script.js

"use strict";

//#region REQUIRES EXTERNAL MODULES
const electron = require("electron");
const {
    remote
} = require("electron");
const {
    dialog
} = require("electron").remote;
const fs = require("fs");
const exec = require("child_process").exec;
const path = require('path');
const os = require('os');
const ping = require('ping');
const libAppSettings = require("lib-app-settings");
var piWifi = require('pi-wifi');
//#endregion REQUIRES EXTERNAL MODULES

//#region GLOBAL DECLARATIONS
const settingsFile = ".settings";
const appSettings = new libAppSettings(settingsFile);
const netInterface = "wlan0";
const wpaFilename = "/etc/wpa_supplicant/wpa_supplicant.conf";
var ssidConnectRequest = "";
var saveSSIDs = [];
var quitApp = false;
var plcIP = "100.100.100.50";
var plcCheckTimeSet = 10;
var plcCheckTimeAcc = 0;
var plcChecking = false;
let appDir = path.dirname(require.main.filename);

// Page component references.
var lblWANIP = document.getElementById("lblWANIP");
var lblLANIP = document.getElementById("lblLANIP");
var lblPLCIP = document.getElementById("lblPLCIP");
var txtSSID = document.getElementById("txtSSID");
var lstSSIDs = document.getElementById("lstSSIDs");
//#endregion GLOBAL DECLARATIONS

//#region ONSCREEN KEYBOARD
var lowerKeys = ["\`", "1", "2", "3", "4", "5", "6", "7", "8", "9", "0", "-", "=", 
    "q", "w", "e", "r", "t", "y", "u", "i", "o", "p", "[", "]", "\\", 
    " ", "a", "s", "d", "f", "g", "h", "j", "k", "l", ";", "\'", " ", 
    " ", " ", "z", "x", "c", "v", "b", "n", "m", ",", ".", "/", " "];
var shiftKeys = ["~", "!", "@", "#", "$", "%", "^", "&", "*", "(", ")", "_", "+", 
    "Q", "W", "E", "R", "T", "Y", "U", "I", "O", "P", "{", "}", "|", 
    " ", "A", "S", "D", "F", "G", "H", "J", "K", "L", ":", "\"", " ", 
    " ", " ", "Z", "X", "C", "V", "B", "N", "M", "<", ">", "?", " "];
var loadedKeys = lowerKeys;  // Initialize the keyboard to lowercase keys.

function loadKeys(keys){
    let row = 1;
    let keysPerRow = 13;
    let i = 1;
    removeChildren(`row${row}`);
    for (let key of keys){
        let keyEl = addItemtoDiv(`row${row}`, key, "btnKey");
        if (key == " "){  // Empty key
            keyEl.classList.add("background");
        }
        else{
            // Add the click event handler
            keyEl.addEventListener("click", ()=>{
                txtKeyboardEntry.value += key;
            });
        }
        i += 1;
        if (i > keysPerRow && row < 4){
            i=1;
            row += 1;
            removeChildren(`row${row}`);
        }
    }
}

//#endregion ONSCREEN KEYBOARD

//#region LAN FUNCTIONS
function plcCheckConn(){
    plcChecking = true;
    console.log("ping " + plcIP);
    if (lblPLCIP.classList.contains("invalid")){
        lblPLCIP.innerText = "Checking...";
    }
    ping.sys.probe(plcIP, (isAlive) => {
        lblPLCIP.innerText = plcIP;
        isAlive ? lblPLCIP.classList.remove("invalid") : lblPLCIP.classList.add("invalid");
        plcChecking = false;
        plcCheckTimeAcc = plcCheckTimeSet;
    });
}

function getIPAddresses(){
    var eth0 = false;
    var wlan0 = false;

    var ifaces = os.networkInterfaces();
    Object.keys(ifaces).forEach(function (ifname) {
        var alias = 0;
        ifaces[ifname].forEach(function (iface) {
            if ('IPv4' !== iface.family || iface.internal !== false) {
            // skip over internal (i.e. 127.0.0.1) and non-ipv4 addresses
            return;
            }
            if (alias >= 1) {
            // this single interface has multiple ipv4 addresses
                console.log(ifname + ':' + alias, iface.address);
            } else {
            // this interface has only one ipv4 adress
                console.log(ifname, iface.address);
            }
            if (ifname == "eth0"){
                lblLANIP.innerText = iface.address;
                lblLANIP.classList.remove("invalid");
                eth0 = true;
            }
            if (ifname == "wlan0"){
                lblWANIP.innerText = iface.address;
                lblWANIP.classList.remove("invalid");
                wlan0 = true;
            }
            ++alias;
        });
    });
    if (!eth0) {
        lblLANIP.innerText = "UNPLUGGED";
        lblLANIP.classList.add("invalid");
    }
    if (!wlan0) {
        lblWANIP.innerText = "UNPLUGGED";
        lblWANIP.classList.add("invalid");
    }
}
//#endregion LAN FUNCTIONS

//#region SYSTEM FUNCTIONS
function reboot(){
    exec("sudo reboot", (error, stdout, stderr) => {
        if (error) {
            console.log(`error: ${error.message}`);
            return;
        }
        if (stderr) {
            console.log(`stderr: ${stderr}`);
            return;
        }
        console.log(`stdout: ${stdout}`);
        quitApp = true;
    });
}
//#endregion SYSTEM FUNCTIONS

//#region DOM HELPER FUNCTIONS
function addItemtoDiv(divById, itemInnerText, classAdd) {
    var newItem = document.createElement("div");
    newItem.innerText = itemInnerText;
    newItem.classList.add(classAdd);
    return document.getElementById(divById).appendChild(newItem);
}


function removeChildren(elId) {
    let elChildren = document.getElementById(elId);
    while (elChildren.childElementCount > 0) {
        elChildren.removeChild(elChildren.lastChild);
    }
}

function showPage(page){
    divHomePage.classList.add("hide");
    divSettingsPage.classList.add("hide");
    divKeyboardPage.classList.add("hide");
    divMessagePage.classList.add("hide");
    if (page=="Home") divHomePage.classList.remove("hide");
    else if (page=="Settings") divSettingsPage.classList.remove("hide");
    else if (page=="Keyboard") divKeyboardPage.classList.remove("hide");
    else if (page=="Message") divMessagePage.classList.remove("hide");
}
//#endregion DOM HELPER FUNCTIONS

//#region NOTIFICATION WINDOWS
function showWarningMessageBox(message) {
    lblMessage.innerText = message;
    showPage("Message");
}

function showOKMessageBox(message) {
    lblMessage.innerText = message;
    showPage("Message");
}
//#endregion NOTIFICATION WINDOWS

//#region SAVED SSID FUNCTIONS
function addSSID(ssid, index){
    var newItem = document.createElement("div");
    newItem.innerText = ssid;
    newItem.classList.add("btn");
    newItem.classList.add("btnTight");
    newItem.setAttribute("data-netIndex", index.toString());
    lstConnectSSIDs.appendChild(newItem).addEventListener("click", (e)=>{
        selectSSID(e.target);
    });
}


function loadSSIDs(){
    piWifi.listNetworks(function(err, networksArray) {
        if (err) {
          return console.log(err.message);
        }
        saveSSIDs = networksArray;
        console.log(saveSSIDs);
        let index = 0;
        for (let network of networksArray){
            addSSID(network.ssid, index);
            index++;
        }
        addPSKfromWPA();
    });
}

function selectSSID(el){
    console.log(el.innerText);
    lblSSID.innerText = el.innerText;
    let netId = parseInt(el.getAttribute("data-netIndex"));
    ssidConnect(el.innerText, saveSSIDs[netId].psk);
}

function getConnectedSSID(){
    piWifi.status(netInterface, (err, status)=>{
        if (!err){
            lblSSID.innerText = status.ssid;
        }
    });
}

function addPSKfromWPA(){
    // ConnectToId is no worky.  This work-around pulls the password from the wpa_supplicant.conf file and makes
    // a call to Connect instead.  Not super secure because of the direct access to the password, but whatevs.
    let wpa_data = fs.readFileSync(wpaFilename).toString('utf-8');
    let networks = wpa_data.split("network=");
    for (let i=1; i<networks.length; i++){
        let psk = networks[i].split('psk="')[1].split('"')[0];
        console.log({index: i, PSK: psk});
        saveSSIDs[i-1].psk = psk;
    }
}
//#endregion SAVED SSID FUNCTIONS

//#region SCANNED SSID FUNCTIONS
function scanForSSIDs(){
    // Scan for SSIDs and add to the list.
    removeChildren("lstAddSSIDs");
    piWifi.scan((err, networks)=>{
        if (err) {
            console.log("Scan failed..." + err.message);
        }
        else{
            let listed = [];
            for (let network of networks){
                if (!listed.includes(network.ssid) && network.ssid != undefined){
                    let ssidEl = addItemtoDiv("lstAddSSIDs", network.ssid, "btn");
                    ssidEl.classList.add("btnTight");
                    ssidEl.addEventListener("click", ()=>{
                        getPassword(network.ssid);
                    });
                    listed.push(network.ssid);
                }
            }
        }
    });
}

function getPassword(ssid){
    ssidConnectRequest = ssid;
    showPage("Keyboard");
    txtKeyboardEntry.value = "";
}

function ssidConnect(ssid, password){
    var networkDetails = {
        ssid: ssid,
    }
    // Add the password if one was supplied.
    if (password != ""){
        networkDetails.password = password
    }
    piWifi.connectTo(networkDetails, (err)=>{
        if (!err){
            lblSSID.value = ssid;
            showPage("Home");
        }
        else {
            showWarningMessageBox(`Connection failed.\nReboot if error persists.\n${err}\n${err.message}`);
        }
    });
}
//#endregion SCANNED SSID FUNCTIONS

//#region INITIALIZATION
async function init() {
    await wait(1000);
    // Initialize the page components
    lblLANIP.innerText = "UNPLUGGED";
    lblLANIP.classList.add("invalid");
    lblWANIP.innerText = "UNPLUGGED";
    lblWANIP.classList.add("invalid");
    lblPLCIP.innerText = plcIP;
    lblPLCIP.classList.add("invalid");

    // Load the SSID list.
    loadSSIDs();

    // Get the ssid of the current connection.
    getConnectedSSID();

    // Fill the page components
    getIPAddresses();

    // Load the keyboard
    loadKeys(loadedKeys);

    // Run the connected status loop.
    loop();
}

async function loop() {
    while (!quitApp) {
        await wait(1000);
        if (!plcChecking){
            plcCheckTimeAcc -= 1;
            if (plcCheckTimeAcc <= 0){
                plcCheckConn();
                getIPAddresses();
            }
            
        }
    }
}
//#endregion INITIALIZATION

//#region EVENT HANDLERS
function wait(time_ms) {
    return new Promise(function (resolve, reject) {
        setTimeout(() => {
            resolve();
        }, time_ms);
    });
}

function ssid_clicked(e){
    selectSSID(e.target);
}

btnSettings.addEventListener("click", ()=>{
    showPage("Settings");
});

btnHome.addEventListener("click", ()=>{
    showPage("Home");
});

btnKeyCancel.addEventListener("click", ()=>{
    showPage("Settings");
});

btnReboot.addEventListener("click", () => {
    reboot();
});

btnScanWifi.addEventListener("click", ()=>{
    scanForSSIDs();
});


//#region KEYBOARD EVENTS
btnKeyShift.addEventListener("click", ()=>{
    loadedKeys = loadedKeys == lowerKeys ? shiftKeys : lowerKeys;
    loadKeys(loadedKeys);
});

btnKeySpace.addEventListener("click", ()=>{
    txtKeyboardEntry.value += " ";
});

btnKeyBackSp.addEventListener("click", ()=>{
    let entryText = txtKeyboardEntry.value;
    if (entryText.length > 0){
        txtKeyboardEntry.value = entryText.slice(0,-1);
    }
});

btnKeyEnter.addEventListener("click", ()=>{
    ssidConnect(ssidConnectRequest, txtKeyboardEntry.value);
});
//#endregion KEYBOARD EVENTS

btnMsgOk.addEventListener("click", ()=>{
    showPage("Home");
});

//#endregion EVENT HANDLERS

init();