index.js

#!/usr/bin/env node
'use strict'

import chalk from 'chalk'
import boxen from 'boxen'
import clear from 'clear'
import inquirer from 'inquirer'
import Enquirer from 'enquirer'
import QuantumCircuit from 'quantum-circuit'
import terminalLink from 'terminal-link'
import terminalImage from 'terminal-image'
import playSound from 'play-sound'
import cliProgress from 'cli-progress'
import lolcatjs from 'lolcatjs'

import fs from 'fs'
import { fileURLToPath } from 'url'
import { dirname } from 'path'
import * as path from 'path'
const __filename = fileURLToPath(import.meta.url);
const __dirname = dirname(__filename);

const data = {
    name: chalk.bold.green('Adam Godel'),
    website: chalk.green(terminalLink('adamgodel.me', 'https://adamgodel.me')),
    labelWebsite: chalk.white('Website:'),
    github: chalk.green(terminalLink('@adam-godel', 'https://github.com/adam-godel')),
    labelGithub: chalk.white('GitHub:'),
    email: chalk.green(terminalLink('agodel@bu.edu', 'mailto:agodel@bu.edu')),
    labelEmail: chalk.white('Email:')
}

const card = boxen(
    [
        `${data.name}`,
        ``,
        `${data.labelWebsite} ${data.website}`,
        `${data.labelGithub} ${data.github}`,
        `${data.labelEmail} ${data.email}`
    ].join('\n'),
    {
        margin: 1,
        padding: 1,
        borderStyle: 'double',
        borderColor: 'green'
    }
);

const questions = [
    {
        type: "list",
        name: "action",
        message: "Select an action",
        choices: [
            {
                name: `About Me`,
                value: () => {
                    console.log("My name is Adam Godel and I am studying mathematics and computer science\nat Boston University. While I am interested in a multitude of topics\nwithin my major, my main interests are quantum computing and web\ndevelopment. Feel free to contact me or look at my projects on GitHub!");
                }
            },
            {
                name: `My Signature`,
                value: async () => {
                    fs.readFile(path.resolve(__dirname, './assets/signature.txt'), 'utf8', function (err, data) {
                        if (err) throw err;
                        lolcatjs.options.seed = 612;
                        lolcatjs.fromString(data);
                    });
                    await new Promise(resolve => setTimeout(resolve, 10)); // to keep followup message after fs
                }
            },
            {
                name: `Flip an Entangled Coin`,
                value: async () => {
                    console.log("The entangled coin is made up of two qubits that always share the same state.");
                    await new Promise(resolve => setTimeout(resolve, 2500));
                    console.log("Preparing to generate quantum circuit...");
                    const bar = new cliProgress.Bar({
                        format: ' >> [\u001b[37m{bar}\u001b[0m] {percentage}%',
                        barCompleteChar: '\u2588',
                        barIncompleteChar: '\u2591',
                        barGlue: '\u001b[90m'
                    });
                    bar.start(120, 0);
                    for (let i = 0; i < 6; i++) {
                        await new Promise(resolve => setTimeout(resolve, Math.floor((Math.random() * 500) + 500)));
                        bar.increment(Math.floor((Math.random() * 4) + 20));
                    }
                    process.stdout.write("\n");

                    // image and sound, all for show of course :)
                    console.log(await terminalImage.file(path.resolve(__dirname, './assets/AntManQuantumania.gif'), {height: '60%'}));
                    await new Promise(resolve => setTimeout(resolve, 2000));
                    const player = playSound();
                    player.play(path.resolve(__dirname, './assets/AntManQuantumania.mp3'), function (err) {
                        if (err) throw err;
                    });
                    console.log("Generating quantum circuit...");
                    bar.start(14000, 0);
                    const times = [221, 273, 561, 701, 744, 763, 899, 665, 741, 977, 996, 854, 752, 827, 753, 782, 822, 819, 797, 53];
                    for (let i = 0; i < times.length; i++) {
                        await new Promise(resolve => setTimeout(resolve, times[i]));
                        bar.increment(times[i+1 < times.length ? i+1 : 0]);
                    }
                    process.stdout.write("\n");
                    await new Promise(resolve => setTimeout(resolve, 2000));
                    
                    // actual initialization of the quantum circuit
                    clear();
                    console.log(card);
                    var circuit = new QuantumCircuit(2);
                    circuit.appendGate("h", 0);
                    circuit.appendGate("cx", [0, 1]);
                    circuit.run();
                    console.log("Quantum circuit ready!");
                    circuit.print();

                    const response = await prompt({
                        type: 'input',
                        name: 'count',
                        message: 'How many times would you like to flip the coin? (Max 1000)'
                    });
                    for (let i = 1; i <= (response.count > 1000 ? 1000 : response.count); i++) {
                        circuit.run();
                        console.log("Flip #"+i+": "+circuit.measure(0)+""+circuit.measure(1));
                    }
                }
            },
            {
                name: `Read/Write Values in QRAM`,
                value: async () => {
                    console.log("Quantum Random Access Memory allows you to use a quantum circuit\nto read and write bit information. This quantum circuit has up to\n12 memory locations encoded using up to 26 qubits in total.");
                    const size = await prompt({
                        type: 'input',
                        name: 'count',
                        message: 'What size would you like the QRAM to be? (Max 12)'
                    });
                    if (Number.isNaN(size.count) || size.count <= 0)
                        return;
                    const count = size.count > 12 ? 12 : parseInt(size.count);

                    var circuit = new QuantumCircuit(2*count+2);
                    for (let i = 0; i < count; i++)
                        circuit.appendGate("ccx", [2*count+1, i, count+i]);
                    for (let i = 0; i < count; i++)
                        circuit.appendGate("ccx", [i, count+i, 2*count]);

                    console.log("QRAM for " + count + " addresses using " + (2*count+2) + " qubits")
                    for (let i = 0; i < count; i++)
                        console.log(((i < 10 && count >= 5) ? ' ' : '') + 'q' + i + ' ' + '---'.repeat(count-1-i) + '-⏺-' + '-|-'.repeat(i) + '-' + '---'.repeat(count-1-i) + '-⏺-' + '-|-'.repeat(i) + '---');
                    for (let i = 0; i < count; i++)
                        console.log(((count+i < 10 && count >= 5) ? ' ' : '') + 'q' + (count+i) + ' ' + '-|-'.repeat(count-1-i) + '-⦻-' + '-|-'.repeat(i) + '-' + '-|-'.repeat(count-1-i) + '-⏺-' + '-|-'.repeat(i) + '---');
                    console.log('q' + (2*count) + ' ' + '-|-'.repeat(count) + '-' + '-⦻-'.repeat(count) + '-⦨-');
                    console.log('q' + (2*count+1) + ' ' + '-⏺-'.repeat(count) + '---'.repeat(count) + '--|-');
                    console.log((count >= 5 ? '  ' : ' ') + 'c ' + '---'.repeat(2*count) + '--⏷-');

                    while (true) {
                        const select = await prompt({
                            type: 'list',
                            name: 'rw',
                            message: 'Select an action',
                            choices: ['Read', 'Write', 'Quit']
                        });
                        if (select.rw == 'Quit') 
                            return;
                        const response = await prompt({
                            type: 'input',
                            name: 'address',
                            message: 'Select an address to ' + (select.rw == 'Write' ? 'write to' : 'read from') + ' (' + '0'.repeat(Math.ceil(Math.log2(count)) > 0 ? Math.ceil(Math.log2(count)) : 1) + ' to ' + (count-1).toString(2) + ')',
                        });
                        function validNum(addr) {
                            for (let i = 2; i <= 9; i++)
                                if (addr.toString().includes(i))
                                    return false;
                            return true;
                        }
                        if (isNaN(response.address) || parseInt(response.address, 2) >= count || !validNum(response.address)) {
                            console.log(chalk.red.bold("Error:") + " Enter an address from " + "0".repeat(Math.ceil(Math.log2(count)) > 0 ? Math.ceil(Math.log2(count)) : 1) + " to " + (count-1).toString(2) + " in binary!");
                            continue;
                        }

                        let qInit = [];
                        for (let i = 0; i < count; i++)
                            qInit.push(parseInt(response.address, 2) == i);
                        for (let i = count; i < 2*count; i++)
                            qInit.push(circuit.measure(i));
                        circuit.run(qInit.concat([false, select.rw == 'Write']));
                        if (select.rw == 'Write')
                            console.log("Applied X gate at address " + response.address.padStart(Math.ceil(Math.log2(count)), '0') + "!");
                        else
                            console.log("Address " + response.address.padStart(Math.ceil(Math.log2(count)), '0') + ": " + circuit.measure(2*count));
                    }
                }
            },
            {
                name: `Exit`,
                value: () => {
                    console.log("Have a nice day!\n");
                    process.exit();
                }
            }
        ]
    }
];
const prompt = inquirer.createPromptModule();

const followup = [
    {
        type: "toggle",
        name: "exit",
        message: "Exit?",
        enabled: 'Yes',
        disabled: 'No',
        default: false
    }
];

async function init() {
    clear();
    console.log(card);
    await prompt(questions).then(answer => answer.action());
    await Enquirer.prompt(followup).then(answer => {
        if (!answer.exit) {
            init();
        } else {
            console.log("Have a nice day!\n");
            process.exit();
        }
    });
}

process.removeAllListeners('warning'); // suppress antlr4 warning
init();