ass/metrics.js

const fs = require('fs-extra');
const path = require('path');
const { s3enabled } = require('./config.json');
const { formatBytes } = require('./utils');
const { bucketSize } = require('./storage');

const TLog = require('@tycrek/log');
const log = new TLog({ level: 'debug', timestamp: { enabled: false } });

module.exports = () => {
	const data = fs.readJsonSync(path.join(__dirname, 'data.json'));
	const { users } = fs.readJsonSync(path.join(__dirname, 'auth.json'));
	Object.keys(users).forEach((token) => users[token].count = 0);

	let totalSize = 0;
	let oldSize = 0;
	Object.values(data).forEach(({ token, size }) => {
		try {
			totalSize += size;
			if (token === undefined) oldSize += size; // skipcq: JS-0127
			else {
				if (!users[token].size) users[token].size = 0;
				users[token].size += size;
				users[token].count++;
			}
		} catch (ex) {
			// Silently handle missing tokens from dev environment -tycrek
		}
	});

	// Get AWS size
	bucketSize()
		.then((s3size) => {
			log.info('---- Usage metrics ----')
				.blank()
				.info('Users', Object.keys(users).length)
				.info('Files', Object.keys(data).length)
				.info('S3 size', s3enabled ? s3size : '--')
				.blank()
				.info('Total size', formatBytes(totalSize))
				.info('Old files', formatBytes(oldSize))
				.blank();

			Object.values(users).forEach(({ username, count, size }) => log.info(`- ${username}`, formatBytes(size), `${count} files`));
			process.exit(0);
		})
		.catch(log.c.error);
}

if (require.main === module) module.exports();