CyTube/bin/admin.js

#!/usr/bin/env node

const Config = require('../lib/config');
Config.load('config.yaml');

if (!Config.get('service-socket.enabled')){
    console.error('The Service Socket is not enabled.');
    process.exit(1);
}

const net = require('net');
const path = require('path');
const readline = require('node:readline/promises');

const socketPath = path.resolve(__dirname, '..', Config.get('service-socket.socket'));
const rl = readline.createInterface({
    input: process.stdin,
    output: process.stdout
});

async function doCommand(params) {
    return new Promise((resolve, reject) => {
        const client = net.createConnection(socketPath);

        client.on('connect', () => {
            client.write(JSON.stringify(params) + '\n');
        });

        client.on('data', data => {
            client.end();
            resolve(JSON.parse(data));
        });

        client.on('error', error => {
            reject(error);
        });
    });
}

let commands = [
    {
        command: 'ban-channel',
        handler: async args => {
            if (args.length !== 3) {
                console.log('Usage: ban-channel <name> <externalReason> <internalReason>');
                process.exit(1);
            }

            let [name, externalReason, internalReason] = args;
            let answer = await rl.question(`Ban ${name} with external reason "${externalReason}" and internal reason "${internalReason}"? `);

            if (!/^[yY]$/.test(answer)) {
                console.log('Aborted.');
                process.exit(1);
            }

            let res = await doCommand({
                command: 'ban-channel',
                name,
                externalReason,
                internalReason
            });

            switch (res.status) {
                case 'error':
                    console.log('Error:', res.error);
                    process.exit(1);
                    break;
                case 'success':
                    console.log('Ban succeeded.');
                    process.exit(0);
                    break;
                default:
                    console.log(`Unknown result: ${res.status}`);
                    process.exit(1);
                    break;
            }
        }
    },
    {
        command: 'unban-channel',
        handler: async args => {
            if (args.length !== 1) {
                console.log('Usage: unban-channel <name>');
                process.exit(1);
            }

            let [name] = args;
            let answer = await rl.question(`Unban ${name}? `);

            if (!/^[yY]$/.test(answer)) {
                console.log('Aborted.');
                process.exit(1);
            }

            let res = await doCommand({
                command: 'unban-channel',
                name
            });

            switch (res.status) {
                case 'error':
                    console.log('Error:', res.error);
                    process.exit(1);
                    break;
                case 'success':
                    console.log('Unban succeeded.');
                    process.exit(0);
                    break;
                default:
                    console.log(`Unknown result: ${res.status}`);
                    process.exit(1);
                    break;
            }
        }
    },
    {
        command: 'show-banned-channel',
        handler: async args => {
            if (args.length !== 1) {
                console.log('Usage: show-banned-channel <name>');
                process.exit(1);
            }

            let [name] = args;

            let res = await doCommand({
                command: 'show-banned-channel',
                name
            });

            switch (res.status) {
                case 'error':
                    console.log('Error:', res.error);
                    process.exit(1);
                    break;
                case 'success':
                    if (res.ban != null) {
                            console.log(`Channel: ${name}`);
                        console.log(`Ban issued: ${res.ban.createdAt}`);
                        console.log(`Banned by: ${res.ban.bannedBy}`);
                        console.log(`External reason:\n${res.ban.externalReason}`);
                        console.log(`Internal reason:\n${res.ban.internalReason}`);
                    } else {
                        console.log(`Channel ${name} is not banned.`);
                    }
                    process.exit(0);
                    break;
                default:
                    console.log(`Unknown result: ${res.status}`);
                    process.exit(1);
                    break;
            }
        }
    }
];

let found = false;
commands.forEach(cmd => {
    if (cmd.command === process.argv[2]) {
        found = true;
        cmd.handler(process.argv.slice(3)).then(() => {
            process.exit(0);
        }).catch(error => {
            console.log('Error in command:', error.stack);
        });
    }
});

if (!found) {
    console.log('Available commands:');
    commands.forEach(cmd => {
        console.log(`  * ${cmd.command}`);
    });
    process.exit(1);
}