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

37 lines
698 B
TypeScript
Raw Normal View History

2018-03-29 13:32:18 +02:00
import DriveFile 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 information
*/
2018-06-18 02:54:53 +02:00
module.exports = (params: any, user: ILocalUser) => new Promise(async (res, rej) => {
2017-03-03 20:28:38 +01:00
// Calculate drive usage
2018-06-15 02:56:59 +02:00
const usage = await DriveFile
.aggregate([{
$match: {
'metadata.userId': user._id,
'metadata.deletedAt': { $exists: false }
}
}, {
$project: {
length: true
}
}, {
$group: {
_id: null,
usage: { $sum: '$length' }
}
}])
.then((aggregates: any[]) => {
if (aggregates.length > 0) {
return aggregates[0].usage;
2017-03-03 20:28:38 +01:00
}
2018-06-15 02:56:59 +02:00
return 0;
});
2016-12-28 23:49:51 +01:00
2017-03-03 20:28:38 +01:00
res({
2018-03-29 07:48:47 +02:00
capacity: user.driveCapacity,
2017-03-03 20:28:38 +01:00
usage: usage
2016-12-28 23:49:51 +01:00
});
2017-03-03 20:28:38 +01:00
});