const WebSocket = require('ws');
const https = require('https');
const fs = require('fs');
const readLine = require('readline');
const { v4: uuidv4 } = require('uuid');
const { execSync } = require("child_process");

// Generate tls keys if not found.
if (!fs.existsSync('./tlskey.pem'))
{
    console.log("TLS key files not detected. Generating..");
    execSync("openssl req -newkey rsa:2048 -new -nodes -x509 -days 3650 -keyout tlskey.pem -out tlscert.pem -subj \"/C=SA/ST=SA/L=SA/O=SA/CN=SA\"");
    console.log("New tls key files generated.")
}

const server = https.createServer({
    cert: fs.readFileSync('./tlscert.pem'),
    key: fs.readFileSync('./tlskey.pem')
});

const wss = new WebSocket.Server({ server });

wss.on('connection', (ws) => {
    ws.on('message', (msg) => {
        console.log('Received: ', Buffer.from(msg).toString());
    });
});

// start listening for stdin
const rl = readLine.createInterface({
    input: process.stdin,
    output: process.stdout
});

// On ctrl + c we should close SA connection gracefully.
rl.on('SIGINT', () => {
    console.log('SIGINT received...');
    wss.close();
    rl.close();
    server.close()
});

const sendToAll = (msg) => {
    wss.clients.forEach(ws => {
        ws.send(msg);
    });
}

const askForContractId = () => {
    return new Promise(resolve => {
        rl.question('Contract Id? ', (contractId) => {
            resolve(contractId);
        })
    })
}

server.listen(8080, () => {
    console.log(`wss://localhost:${server.address().port}`)
    console.log("Ready to accept inputs.");

    const inputPump = () => {
        rl.question('', async (inp) => {

            if (inp.length > 0) {
                if (wss.clients.size == 0) {
                    console.log('No Sashimano agents connected yet.')
                }
                else {
                    switch (inp) {
                        case 'create':
                            sendToAll(JSON.stringify({
                                id: uuidv4(),
                                type: 'create',
                                owner_pubkey: 'ed7a4b931bdc5dd79b77a8b6ac293d998c123db42bb3ec2613'
                            }));
                            break;
                        case 'destroy':
                            contractId = await askForContractId();
                            sendToAll(JSON.stringify({
                                id: uuidv4(),
                                type: 'destroy',
                                owner_pubkey: 'ed7a4b931bdc5dd79b77a8b6ac293d998c123db42bb3ec2613',
                                contract_id: contractId
                            }))
                            break;
                        case 'start':
                            contractId = await askForContractId();
                            sendToAll(JSON.stringify({
                                id: uuidv4(),
                                type: 'start',
                                owner_pubkey: 'ed7a4b931bdc5dd79b77a8b6ac293d998c123db42bb3ec2613',
                                contract_id: contractId
                            }))
                            break;
                        case 'stop':
                            contractId = await askForContractId();
                            sendToAll(JSON.stringify({
                                id: uuidv4(),
                                type: 'stop',
                                owner_pubkey: 'ed7a4b931bdc5dd79b77a8b6ac293d998c123db42bb3ec2613',
                                contract_id: contractId
                            }))
                            break;

                        default:
                            console.log('Invalid command. Only valid [create, destroy, start and stop]');
                            break;
                    }

                }

            }

            inputPump();
        })
    }
    inputPump();
});