// general imports
const $ = require('jquery');
const encryption = require('./2_encryption');
const wordList = require('./3_wordlist');
const pinInput = require('./4_input_pin');
const xkcdPassword = require('xkcd-password');

// setup vars
const host = '127.0.0.1';
let peerId;
let passphrase;
let connectedPeer;
let connectedPeers = []; // TODO: Save new peers in array

// setup generator
const generator = new xkcdPassword();
generator.initWithWordList(wordList);

// setup encryption
(async () => {
    peerId = await generator.generate().then(words => words.join('-'));
    encryption.setup();
    if (localStorage.getItem('database') === 'success' && await encryption.check()) {
        pinInput.init(async pin => {
            try {
                await encryption.decryptPrivate(await encryption.getPrivate(), pin);
                chat()
            } catch (e) {
                // TODO: 3 passphrase tries
                console.error('Passphrase is wrong!');
                pinInput.failure();
            }
        });
    } else {
        pinInput.init(pin => {
            console.log('[LOG] No existing keys found! Generating...');
            pinInput.generate();
            passphrase = pin;
            (async () => await encryption.generate(peerId, passphrase).then(() => chat()))()
        });
    }
})();

/**
 * Initializes chat functions
 */
function chat() {
    // hide pin input and display chat
    $('#enter_pin').hide();
    $('#chat').fadeIn();

    // start the peer
    const peer = new Peer(peerId, {host: host, port: 8080, path: '/api', debug: 0});

    // Peer events
    peer.on('open', id => console.log('[LOG] Your ID is', id));
    peer.on('error', err => console.error(err));
    peer.on('connection', conn => {
        connectedPeer = conn;
        console.log('[LOG] Connected with', connectedPeer.peer);
        connectedPeer.on('open', async () => transferKey(await encryption.getPublic()));
        connectedPeer.on('data', async message => {
            console.log('[LOG] Received new message!');
            await receivedMessage(message);
        })
    });

    /**
     * Connects to a peer via his id
     * @param id
     * @returns {Promise<void>}
     */
    async function connect(id) {
        const connectionId = (await generator.generate()).join('-');
        console.log('[LOG] Connecting to', id);
        console.log('[LOG] Your connection ID is', connectionId);
        connectedPeer = peer.connect(id, {label: connectionId, reliable: true});
        console.log('[LOG] Connected with', connectedPeer.peer);
        connectedPeer.on('open', async () => transferKey(await encryption.getPublic()));
        connectedPeer.on('data', async message => {
            console.log('[LOG] Received new message!');
            await receivedMessage(message);
        })
    }

    /**
     * Sends a message to the peer with which you're currently connected
     * @param message
     * @returns {Promise<void>}
     */
    async function sendMessage(message) {
        console.log(`[LOG] Sending message '${message}' to ${connectedPeer.peer}`);
        connectedPeer.send({
            type: 'text',
            data: await encryption.encrypt(message, await encryption.get(connectedPeer.peer))
        });
        await receivedMessage(message, true);
    }

    /**
     * Transfers the (public) key to the currently connected peer
     * @param key
     */
    function transferKey(key) {
        console.log(`[LOG] Transferring key to ${connectedPeer.peer}`);
        connectedPeer.send({type: 'key', data: key});
    }

    /**
     * Renders and processes the incoming messages
     * @param message
     * @param self
     */
    async function receivedMessage(message, self = false) {
        if (self) {
            $('#messages').append(`<span style="color: green">${message}</span><br>`);
        } else {
            if (message.type === 'text') {
                await encryption.decrypt(message.data, await encryption.get(connectedPeer.peer), await encryption.getPrivate(), passphrase)
                    .then(plaintext => $('#messages').append(`${plaintext}<br>`));
            } else if (message.type === 'key') {
                await encryption.store(connectedPeer.peer, message.data)
            }
        }
    }

    /**
     * Events after load
     */
    $(document).ready(() => {
        $('#add_peer_id').on('click', async () => await connect($('#peer_id').val()));
        $('#send_message').on('click', async () => await sendMessage($('#message').val()));

        $('[toggle-contact-modal]').on('click', () => $('#add_contact_modal').toggleClass('is-active'))
    });
}

//encryption.test(); // TESTING IF ENCRYPTION WORKS