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

/**
 * Module dependencies
 */
import DriveFile from '../../../models/drive-file';

/**
 * Get drive information
 *
 * @param {any} params
 * @param {any} user
 * @return {Promise<any>}
 */
module.exports = (params, user) => new Promise(async (res, rej) => {
	// Calculate drive usage
	const usage = ((await DriveFile
		.aggregate([
			{ $match: { 'metadata.userId': user._id } },
			{
				$project: {
					length: true
				}
			},
			{
				$group: {
					_id: null,
					usage: { $sum: '$length' }
				}
			}
		]))[0] || {
			usage: 0
		}).usage;

	res({
		capacity: user.driveCapacity,
		usage: usage
	});
});