misskey/src/server/api/endpoints/drive/stream.ts

62 lines
1.5 KiB
TypeScript
Raw Normal View History

2018-07-07 12:19:00 +02:00
import $ from 'cafy'; import ID from '../../../../misc/cafy-id';
2018-03-29 13:32:18 +02:00
import DriveFile, { pack } from '../../../../models/drive-file';
2018-06-18 02:54:53 +02:00
import { ILocalUser } from '../../../../models/user';
2016-12-28 23:49:51 +01:00
/**
* Get drive stream
*/
2018-07-05 19:58:29 +02:00
export default (params: any, user: ILocalUser) => new Promise(async (res, rej) => {
2016-12-28 23:49:51 +01:00
// Get 'limit' parameter
2018-07-05 16:36:07 +02:00
const [limit = 10, limitErr] = $.num.optional.range(1, 100).get(params.limit);
2017-03-03 11:33:14 +01:00
if (limitErr) return rej('invalid limit param');
2016-12-28 23:49:51 +01:00
2018-03-29 07:48:47 +02:00
// Get 'sinceId' parameter
2018-07-05 16:36:07 +02:00
const [sinceId, sinceIdErr] = $.type(ID).optional.get(params.sinceId);
2018-03-29 07:48:47 +02:00
if (sinceIdErr) return rej('invalid sinceId param');
2016-12-28 23:49:51 +01:00
2018-03-29 07:48:47 +02:00
// Get 'untilId' parameter
2018-07-05 16:36:07 +02:00
const [untilId, untilIdErr] = $.type(ID).optional.get(params.untilId);
2018-03-29 07:48:47 +02:00
if (untilIdErr) return rej('invalid untilId param');
2016-12-28 23:49:51 +01:00
2018-03-29 07:48:47 +02:00
// Check if both of sinceId and untilId is specified
2017-12-20 18:20:02 +01:00
if (sinceId && untilId) {
2018-03-29 07:48:47 +02:00
return rej('cannot set sinceId and untilId');
2016-12-28 23:49:51 +01:00
}
// Get 'type' parameter
2018-07-05 16:36:07 +02:00
const [type, typeErr] = $.str.optional.match(/^[a-zA-Z\/\-\*]+$/).get(params.type);
2017-03-03 11:33:14 +01:00
if (typeErr) return rej('invalid type param');
2016-12-28 23:49:51 +01:00
// Construct query
const sort = {
2017-01-22 10:13:12 +01:00
_id: -1
2016-12-28 23:49:51 +01:00
};
const query = {
2018-03-29 07:48:47 +02:00
'metadata.userId': user._id
2017-03-03 11:33:14 +01:00
} as any;
if (sinceId) {
2017-01-22 10:13:12 +01:00
sort._id = 1;
2016-12-28 23:49:51 +01:00
query._id = {
2017-03-03 11:33:14 +01:00
$gt: sinceId
2016-12-28 23:49:51 +01:00
};
2017-12-20 18:20:02 +01:00
} else if (untilId) {
2016-12-28 23:49:51 +01:00
query._id = {
2017-12-20 18:20:02 +01:00
$lt: untilId
2016-12-28 23:49:51 +01:00
};
}
2017-03-08 19:50:09 +01:00
if (type) {
2017-11-08 06:43:42 +01:00
query.contentType = new RegExp(`^${type.replace(/\*/g, '.+?')}$`);
2016-12-28 23:49:51 +01:00
}
// Issue query
const files = await DriveFile
.find(query, {
limit: limit,
sort: sort
2017-01-17 03:11:22 +01:00
});
2016-12-28 23:49:51 +01:00
// Serialize
res(await Promise.all(files.map(async file =>
2018-02-02 00:21:30 +01:00
await pack(file))));
2016-12-28 23:49:51 +01:00
});