laoxong/misskey
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fileserver - support DriveFile w/ GridFS

Browse Source
This commit is contained in:
otofune 2017-11-06 15:35:20 +09:00
parent d0dab265f4
commit a5160a1bba
1 changed files with 17 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
src/file/server.ts
View File

@ -9,7 +9,7 @@ import * as cors from 'cors';
import * as mongodb from 'mongodb'; import * as mongodb from 'mongodb';
import * as gm from 'gm'; import * as gm from 'gm';
import File from '../api/models/drive-file'; import DriveFile, { getGridFSBucket } from '../api/models/drive-file';
/** /**
* Init app * Init app
@ -97,17 +97,28 @@ app.get('/:id', async (req, res) => {
return; return;
} }
const file = await File.findOne({ _id: new mongodb.ObjectID(req.params.id) }); const fileId = new mongodb.ObjectID(req.params.id)
const file = await DriveFile.findOne({ _id: fileId });
if (file == null) { if (file == null) {
res.status(404).sendFile(`${__dirname} / assets / dummy.png`); res.status(404).sendFile(`${__dirname} / assets / dummy.png`);
return; return;
} else if (file.data == null) {
res.sendStatus(400);
return;
} }
send(file.data.buffer, file.type, req, res); const bucket = await getGridFSBucket()
const buffer = await ((id): Promise<Buffer> => new Promise((resolve, reject) => {
const chunks = []
const readableStream = bucket.openDownloadStream(id)
readableStream.on('data', chunk => {
chunks.push(chunk);
})
readableStream.on('end', () => {
resolve(Buffer.concat(chunks))
})
}))(fileId)
send(buffer, file.metadata.type, req, res);
}); });
app.get('/:id/:name', async (req, res) => { app.get('/:id/:name', async (req, res) => {