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 = require('./data');
	const { users } = fs.readJsonSync(path.join(__dirname, 'auth.json'));
	Object.keys(users).forEach((token) => users[token].count = 0);

	let totalSize = 0;
	let oldSize = 0;
	let d;

	data.get()
		.then((D) => d = D.map(([, resource]) => resource))
		.then(() => {
			console.log(d);
			d.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
			return bucketSize();
		})
		.then((s3size) => {
			log.info('---- Usage metrics ----')
				.blank()
				.info('Users', Object.keys(users).length)
				.info('Files', Object.keys(d).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();