2018-04-12 21:06:18 +00:00
|
|
|
import * as Koa from 'koa';
|
|
|
|
import * as send from 'koa-send';
|
2019-03-18 04:20:49 +00:00
|
|
|
import * as rename from 'rename';
|
2019-02-03 09:16:57 +00:00
|
|
|
import { serverLogger } from '..';
|
2019-03-18 06:23:45 +00:00
|
|
|
import { contentDisposition } from '../../misc/content-disposition';
|
2019-04-07 12:50:36 +00:00
|
|
|
import { DriveFiles } from '../../models';
|
|
|
|
import { InternalStorage } from '../../services/drive/internal-storage';
|
2018-05-03 11:03:14 +00:00
|
|
|
|
2018-05-04 08:59:51 +00:00
|
|
|
const assets = `${__dirname}/../../server/file/assets/`;
|
|
|
|
|
2019-01-22 12:42:05 +00:00
|
|
|
const commonReadableHandlerGenerator = (ctx: Koa.BaseContext) => (e: Error): void => {
|
2019-02-03 09:16:57 +00:00
|
|
|
serverLogger.error(e);
|
2018-05-03 11:03:14 +00:00
|
|
|
ctx.status = 500;
|
|
|
|
};
|
2018-04-12 21:06:18 +00:00
|
|
|
|
2019-01-22 12:42:05 +00:00
|
|
|
export default async function(ctx: Koa.BaseContext) {
|
2019-04-07 12:50:36 +00:00
|
|
|
const key = ctx.params.key;
|
2018-04-12 21:06:18 +00:00
|
|
|
|
|
|
|
// Fetch drive file
|
2019-04-07 12:50:36 +00:00
|
|
|
const file = await DriveFiles.createQueryBuilder('file')
|
|
|
|
.where('file.accessKey = :accessKey', { accessKey: key })
|
|
|
|
.orWhere('file.thumbnailAccessKey = :thumbnailAccessKey', { thumbnailAccessKey: key })
|
|
|
|
.orWhere('file.webpublicAccessKey = :webpublicAccessKey', { webpublicAccessKey: key })
|
|
|
|
.getOne();
|
2018-04-12 21:06:18 +00:00
|
|
|
|
|
|
|
if (file == null) {
|
|
|
|
ctx.status = 404;
|
2019-01-22 12:42:05 +00:00
|
|
|
await send(ctx as any, '/dummy.png', { root: assets });
|
2018-04-12 21:06:18 +00:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2019-04-07 12:50:36 +00:00
|
|
|
if (!file.storedInternal) {
|
2018-05-25 11:19:14 +00:00
|
|
|
ctx.status = 204;
|
2018-04-17 11:04:19 +00:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2019-04-07 12:50:36 +00:00
|
|
|
const isThumbnail = file.thumbnailAccessKey === key;
|
|
|
|
const isWebpublic = file.webpublicAccessKey === key;
|
|
|
|
|
|
|
|
if (isThumbnail) {
|
|
|
|
ctx.set('Content-Type', 'image/jpeg');
|
|
|
|
ctx.set('Content-Disposition', contentDisposition('inline', `${rename(file.name, { suffix: '-thumb', extname: '.jpeg' })}`));
|
|
|
|
ctx.body = InternalStorage.read(key);
|
|
|
|
} else if (isWebpublic) {
|
2019-07-05 08:44:23 +00:00
|
|
|
ctx.set('Content-Type', file.type === 'image/apng' ? 'image/png' : file.type);
|
2019-04-07 12:50:36 +00:00
|
|
|
ctx.set('Content-Disposition', contentDisposition('inline', `${rename(file.name, { suffix: '-web' })}`));
|
|
|
|
ctx.body = InternalStorage.read(key);
|
2018-05-03 11:03:14 +00:00
|
|
|
} else {
|
|
|
|
if ('download' in ctx.query) {
|
2019-04-07 12:50:36 +00:00
|
|
|
ctx.set('Content-Disposition', contentDisposition('attachment', `${file.name}`));
|
2018-05-03 11:03:14 +00:00
|
|
|
}
|
2018-04-12 21:06:18 +00:00
|
|
|
|
2019-04-12 16:43:22 +00:00
|
|
|
const readable = InternalStorage.read(file.accessKey!);
|
2019-04-07 12:50:36 +00:00
|
|
|
readable.on('error', commonReadableHandlerGenerator(ctx));
|
|
|
|
ctx.set('Content-Type', file.type);
|
|
|
|
ctx.body = readable;
|
2018-05-03 11:03:14 +00:00
|
|
|
}
|
2018-04-12 21:06:18 +00:00
|
|
|
}
|