iceshrimp/src/server/file/index.ts

174 lines
3.9 KiB
TypeScript
Raw Normal View History

2016-12-29 07:49:51 +09:00
/**
* File Server
*/
import * as fs from 'fs';
import * as express from 'express';
import * as bodyParser from 'body-parser';
import * as cors from 'cors';
import * as mongodb from 'mongodb';
import * as _gm from 'gm';
import * as stream from 'stream';
2016-12-29 07:49:51 +09:00
2018-03-29 20:32:18 +09:00
import DriveFile, { getGridFSBucket } from '../../models/drive-file';
2016-12-29 07:49:51 +09:00
const gm = _gm.subClass({
imageMagick: true
});
2016-12-29 07:49:51 +09:00
/**
* Init app
*/
const app = express();
app.disable('x-powered-by');
app.locals.cache = true;
app.use(bodyParser.urlencoded({ extended: true }));
app.use(cors());
/**
* Statics
*/
2017-04-14 20:45:37 +09:00
app.use('/assets', express.static(`${__dirname}/assets`, {
2016-12-29 07:49:51 +09:00
maxAge: 1000 * 60 * 60 * 24 * 365 // 一年
}));
app.get('/', (req, res) => {
res.send('yee haw');
});
app.get('/default-avatar.jpg', (req, res) => {
const file = fs.createReadStream(`${__dirname}/assets/avatar.jpg`);
2016-12-29 07:49:51 +09:00
send(file, 'image/jpeg', req, res);
});
app.get('/app-default.jpg', (req, res) => {
const file = fs.createReadStream(`${__dirname}/assets/dummy.png`);
2016-12-29 07:49:51 +09:00
send(file, 'image/png', req, res);
});
interface ISend {
contentType: string;
stream: stream.Readable;
}
2016-12-29 07:49:51 +09:00
function thumbnail(data: stream.Readable, type: string, resize: number): ISend {
2017-11-16 23:14:19 +09:00
const readable: stream.Readable = (() => {
// 動画であれば
if (/^video\/.*$/.test(type)) {
2018-04-12 02:48:06 +09:00
// TODO
// 使わないことになったストリームはしっかり取り壊す
data.destroy();
return fs.createReadStream(`${__dirname}/assets/thumbnail-not-available.png`);
// 画像であれば
2018-04-12 02:48:06 +09:00
// Note: SVGはapplication/xml
} else if (/^image\/.*$/.test(type) || type == 'application/xml') {
// 0フレーム目を送る
try {
return gm(data).selectFrame(0).stream();
// だめだったら
} catch (e) {
// 使わないことになったストリームはしっかり取り壊す
data.destroy();
return fs.createReadStream(`${__dirname}/assets/thumbnail-not-available.png`);
}
// 動画か画像以外
} else {
2017-11-16 23:14:19 +09:00
data.destroy();
return fs.createReadStream(`${__dirname}/assets/not-an-image.png`);
}
2017-11-16 23:14:19 +09:00
})();
let g = gm(readable);
if (resize) {
g = g.resize(resize, resize);
}
const stream = g
.compress('jpeg')
.quality(80)
2017-12-11 02:59:05 +09:00
.interlace('line')
2017-11-16 23:14:19 +09:00
.stream();
return {
contentType: 'image/jpeg',
stream
};
2016-12-29 07:49:51 +09:00
}
const commonReadableHandlerGenerator = (req: express.Request, res: express.Response) => (e: Error): void => {
console.dir(e);
req.destroy();
res.destroy(e);
};
function send(readable: stream.Readable, type: string, req: express.Request, res: express.Response): void {
readable.on('error', commonReadableHandlerGenerator(req, res));
const data = ((): ISend => {
if (req.query.thumbnail !== undefined) {
return thumbnail(readable, type, req.query.size);
}
return {
contentType: type,
stream: readable
};
})();
if (readable !== data.stream) {
data.stream.on('error', commonReadableHandlerGenerator(req, res));
2016-12-29 07:49:51 +09:00
}
if (req.query.download !== undefined) {
res.header('Content-Disposition', 'attachment');
2016-12-29 07:49:51 +09:00
}
res.header('Content-Type', data.contentType);
2017-04-14 20:45:37 +09:00
data.stream.pipe(res);
2016-12-29 07:49:51 +09:00
data.stream.on('end', () => {
res.end();
});
2016-12-29 07:49:51 +09:00
}
2017-11-07 09:18:40 +09:00
async function sendFileById(req: express.Request, res: express.Response): Promise<void> {
2017-02-06 22:04:00 +09:00
// Validate id
if (!mongodb.ObjectID.isValid(req.params.id)) {
res.status(400).send('incorrect id');
return;
}
2017-11-06 16:32:01 +09:00
const fileId = new mongodb.ObjectID(req.params.id);
2017-12-09 23:03:48 +09:00
// Fetch (drive) file
const file = await DriveFile.findOne({ _id: fileId });
2016-12-29 07:49:51 +09:00
2017-11-07 09:18:40 +09:00
// validate name
if (req.params.name !== undefined && req.params.name !== file.filename) {
2017-11-07 09:18:40 +09:00
res.status(404).send('there is no file has given name');
return;
}
2017-01-26 23:11:42 +09:00
if (file == null) {
2017-11-06 15:37:04 +09:00
res.status(404).sendFile(`${__dirname}/assets/dummy.png`);
2016-12-29 07:49:51 +09:00
return;
}
2017-11-06 16:32:01 +09:00
const bucket = await getGridFSBucket();
const readable = bucket.openDownloadStream(fileId);
send(readable, file.contentType, req, res);
2017-11-07 09:14:39 +09:00
}
2017-11-07 09:14:39 +09:00
/**
* Routing
*/
2017-11-07 09:14:39 +09:00
app.get('/:id', sendFileById);
app.get('/:id/:name', sendFileById);
2016-12-29 07:49:51 +09:00
module.exports = app;