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

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'
};

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
	};
});