mirror of
https://iceshrimp.dev/iceshrimp/iceshrimp
synced 2024-12-21 01:58:10 +09:00
61 lines
1.0 KiB
TypeScript
61 lines
1.0 KiB
TypeScript
import DriveFile from '../../../models/drive-file';
|
|
import define from '../define';
|
|
import fetchMeta from '../../../misc/fetch-meta';
|
|
|
|
export const meta = {
|
|
desc: {
|
|
'ja-JP': 'ドライブの情報を取得します。',
|
|
'en-US': 'Get drive information.'
|
|
},
|
|
|
|
tags: ['drive', 'account'],
|
|
|
|
requireCredential: true,
|
|
|
|
kind: 'drive-read',
|
|
|
|
res: {
|
|
type: 'object',
|
|
properties: {
|
|
capacity: {
|
|
type: 'number'
|
|
},
|
|
usage: {
|
|
type: 'number'
|
|
}
|
|
}
|
|
}
|
|
};
|
|
|
|
export default define(meta, async (ps, user) => {
|
|
const instance = await fetchMeta();
|
|
|
|
// Calculate drive usage
|
|
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;
|
|
}
|
|
return 0;
|
|
});
|
|
|
|
return {
|
|
capacity: 1024 * 1024 * instance.localDriveCapacityMb,
|
|
usage: usage
|
|
};
|
|
});
|