o#!/bin/sh
show_menu(){
    normal=`echo "\033[m"`
    menu=`echo "\033[36m"` #blue
    number=`echo "\033[33m"` #yellow
    bgred=`echo "\033[41m"`
    fgred=`echo "\033[31m"`
    printf "\n${menu}********* WAGO Provisioning Tool ***********${normal}\n"
    printf "${menu} ${number} 1)${menu} Install Docker ${normal}\n"
    printf "${menu} ${number} 2)${menu} Disable PLC Runtime ${normal}\n"
    printf "${menu} ${number} 3)${menu} Disable OPC-UA & IO-Check Services${normal}\n"
    printf "${menu} ${number} 4)${menu} Install KBUS MQTT Daemon (obsolete: Use API in Menu 1,f)${normal}\n"
    printf "${menu} ${number} 5)${menu} Install PFC200 Containers ${normal}\n"
    printf "${menu} ${number} 6)${menu} Install CC100 Containers ${normal}\n"
    printf "${menu} ${number} 7)${menu} Install DataPlotterApp 2.4 ${normal}\n"
    printf "${menu} ${number} 8)${menu} PFC200 FW Updates${normal}\n"
    printf "${menu} ${number} 9)${menu} Restart PLC ${normal}\n"
    printf "${menu}*********************************************${normal}\n"
    printf "Please enter a menu option and enter or ${fgred}x to exit. ${normal}"
    read opt
}

option_picked(){
    msgcolor=`echo "\033[01;31m"` # bold red
    normal=`echo "\033[00;00m"` # normal white
    message=${@:-"${normal}Error: No message passed"}
    printf "${msgcolor}${message}${normal}\n"
}

show_pfc200_container_menu(){
    normal=`echo "\033[m"`
    menu=`echo "\033[36m"` #blue
    number=`echo "\033[33m"` #yellow
    bgred=`echo "\033[41m"`
    fgred=`echo "\033[31m"`
    printf "\n${menu}********* Docker Containers ***********${normal}\n"
    printf "${menu} ${number} a)${menu} Install Node-RED latest${normal}\n"
    printf "${menu} ${number} b)${menu} Install Mosquitto ${normal}\n"
    printf "${menu} ${number} c)${menu} Install KBUS Modbus ${normal}\n"
    printf "${menu} ${number} d)${menu} Install Grafana ${normal}\n"
    printf "${menu} ${number} e)${menu} Install InfluxDB ${normal}\n"
    printf "${menu} ${number} f)${menu} Install KBUS Daemon Container ${normal}\n"
    printf "${menu} ${number} g)${menu} Install Python3 PyModbus Container. ${normal}\n"
    printf "${menu} ${number} z)${menu} Main Menu ${normal}\n"
    printf "${menu}*********************************************${normal}\n"
    printf "Please enter a menu option and enter or ${fgred}x to exit. ${normal}"
    read opt
}
show_cc100_container_menu(){
    normal=`echo "\033[m"`
    menu=`echo "\033[36m"` #blue
    number=`echo "\033[33m"` #yellow
    bgred=`echo "\033[41m"`
    fgred=`echo "\033[31m"`
    printf "\n${menu}********* Docker Containers ***********${normal}\n"
    printf "${menu} ${number} p)${menu} Install Node-RED latest for CC100${normal}\n"
    printf "${menu} ${number} r)${menu} Install InfluxDB ${normal}\n"
    printf "${menu} ${number} s)${menu} Install Mosquitto 2.0 ${normal}\n"
    printf "${menu} ${number} z)${menu} Main Menu ${normal}\n"
    printf "${menu}*********************************************${normal}\n"
    printf "Please enter a menu option and enter or ${fgred}x to exit. ${normal}"
    read opt
}

show_pfc200_fw_menu(){
    normal=`echo "\033[m"`
    menu=`echo "\033[36m"` #blue
    number=`echo "\033[33m"` #yellow
    bgred=`echo "\033[41m"`
    fgred=`echo "\033[31m"`
    printf "\n${menu}********* PFC200 Firwmare Update ***********${normal}\n"
    printf "\n${menu}Please put PLC Run switch to stop before proceeding. ${normal}\n"
    printf "\n${menu}Update takes 15-20min. ${normal}\n"
    printf "${menu} ${number} w)${menu} Remove all Docker images & containers (Warning!) ${normal}\n"    
    printf "${menu} ${number} t)${menu} FW25 ${normal}\n"
    printf "${menu} ${number} u)${menu} FW26 ${normal}\n"
    printf "${menu} ${number} v)${menu} FW27 ${normal}\n"
    printf "${menu} ${number} z)${menu} Main Menu ${normal}\n"
    printf "${menu}*********************************************${normal}\n"
    printf "Please enter a menu option and enter or ${fgred}x to exit. ${normal}"
    read opt
}
clear
show_menu
while [ $opt != '' ]
    do
    if [ $opt = '' ]; then
      exit;
    else
      case $opt in
        1) clear;
            option_picked "Option 1 Picked - Install Docker";
            wget https://github.com/WAGO/docker-ipk/releases/download/v1.0.4-beta/docker_20.10.5_armhf.ipk;
            opkg install docker_20.10.5_armhf.ipk; 
            rm docker_20.10.5_armhf.ipk;
            printf "Docker v20.10.5 Installed.";
            show_menu;
        ;;
        2) clear;
            option_picked "Option 2 Picked - Disable Runtime";
            /etc/config-tools/config_runtime runtime-version=0;
            printf "Stopping Runtime";
            show_menu;
        ;;
        3) clear;
            option_picked "Option 3 Picked - Disable OPC-UA & IO Check Services";
            /etc/config-tools/config_opcua --set=\"state\":\"disable\";
            /etc/config-tools/config_opcua -r 
            printf "OPC-UA Disabled";
            /etc/config-tools/config_iocheckport state="disabled";
            printf "IO Check Disabled";
            show_menu;
        ;;
        4) clear;
            option_picked "Option 4 Picked - Install KBUS Daemon";
            wget https://github.com/jessejamescox/kbus-daemon-installer/archive/refs/heads/main.zip 
            unzip main.zip 
            sh kbus-daemon-installer-main/installer.sh
            printf "KBUS Daemon Installed";
            show_menu;
        ;;
        5) clear; # Docker sub-menu
            option_picked "Option 5 Picked - Install PFC200 Containers";
            printf "Select Container";
            show_pfc200_container_menu;
        ;;
        6) clear; # Docker sub-menu
            option_picked "Option 5 Picked - Install CC100 Containers";
            printf "Select Container";
            show_cc100_container_menu;
        ;;
        7) clear;
            option_picked "Option 6 Picked - Install DataPlotterApp";
            wget https://github.com/braunku/pfc-provisioning-tool/raw/main/install-dataplotter_2.4_armhf.ipk;
            opkg install install-dataplotter_2.4_armhf.ipk; 
            rm install-dataplotter_2.4_armhf.ipk;
            printf "DataPlotterApp 2.4 Installed.  Accessible at http://<pfc-ip-address>/dataplotter/dataplotter.html";
            show_menu;
        ;; 
        z) clear; # FW Update
            show_menu;
        ;;
        8) clear; # Return to main menu
            show_pfc200_fw_menu;
        ;;
        9) clear;
            option_picked "Option 9 Picked - Rebooting";
            reboot now;
            printf "PLC will restart";
            show_menu;
        ;;
        a) clear;
            option_picked "Option a Picked - Install Node-RED";
            docker volume create --name node_red_user_data;
            docker run --restart unless-stopped -d --name node-red --network=host --security-opt seccomp:unconfined -v node_red_user_data:/data nodered/node-red:latest-minimal;
            printf "Node-RED Installed";
            show_pfc200_container_menu;
        ;;
        b) clear;
            option_picked "Option b Picked - Install Mosquitto Broker 1.5";
            docker run -d --restart unless-stopped --name mosquitto --network=host eclipse-mosquitto:1.5;
            printf "Mosquitto Broker 1.5 Installed";
            show_pfc200_container_menu;
        ;;
        c) clear;
            option_picked "Option c Picked - Install KBUS Modbus Coupler";
            docker run -d --init --restart unless-stopped --privileged -p 502:502 --name=pfc-modbus-server -v /var/run/dbus/system_bus_socket:/var/run/dbus/system_bus_socket wagoautomation/pfc-modbus-server; 
            printf "KBUS Modbus Coupler Installed";
            show_pfc200_container_menu;
        ;;
        d) clear;
            option_picked "Option d Picked - Install Grafana";
            docker volume create grafana-storage;
            docker run -d --restart unless-stopped --network=host --name=grafana -v grafana-storage:/var/lib/grafana grafana/grafana;
            printf "Grafana Installed";
            show_pfc200_container_menu;
        ;;
        e) clear;
            option_picked "Option e Picked - Install InfluxDB";
            docker run -d --restart unless-stopped --name=influxdb --network=host -v influx-storage:/etc/influxdb/ arm32v7/influxdb;
            printf "InfluxDB Installed";
            show_pfc200_container_menu;
        ;;
        f) clear;
            option_picked "Option f Picked - Install KBUS Daemon Container";
            docker run -d --init --restart unless-stopped --privileged --network=host --name=kbus -v kbusapidata:/etc/kbus-api -v /var/run/dbus/system_bus_socket:/var/run/dbus/system_bus_socket jessejamescox/pfc-kbus-api;
            printf "KBUS Daemon Installed";
            show_pfc200_container_menu;
        ;;
        g) clear;
            option_picked "Option g Picked - Install Python3 PyModbus Container";
            mkdir dockerbuild-python;
            cd dockerbuild-python;
            wget https://github.com/braunku/pfc-provisioning-tool/raw/main/Dockerfile;
            docker build -t "python3-pymodbus:1.0" .;
             docker run -d --restart unless-stopped --network=host --name=python3-pymodbus python3-pymodbus:1.0;
            printf "Python3 PyModbus Installed";
            show_pfc200_container_menu;
        ;;
        p) clear;
            option_picked "Option p Picked - Install CC100 Node-RED";
            docker run -d --restart unless-stopped --name wago-node-red --privileged=true --security-opt seccomp:unconfined --user=root --network=host -v node_red_user_data:/data nodered/node-red:latest-minimal;
            printf "Node-RED CC100 Installed";
            show_cc100_container_menu;
        ;;
        r) clear;
            option_picked "Option r Picked - Install InfluxDB";
            docker run -d --restart unless-stopped --name=influxdb --network=host -v influx-storage:/etc/influxdb/ arm32v7/influxdb;
            printf "InfluxDB Installed";
            show_cc100_container_menu;
        ;;
        s) clear;
            option_picked "Option s Picked - Install Mosquitto 2.0";
            wget https://github.com/braunku/pfc-provisioning-tool/raw/main/mosquitto.conf;
            docker run -d --restart unless-stopped --name mosquitto2 -p 1883:1883 -p 8883:8883 -p 9001:9001 -v $(pwd)/mosquitto.conf:/mosquitto/config/mosquitto.conf eclipse-mosquitto:latest
            printf "Mosquitto 2.0 Installed";
            show_cc100_container_menu;
        ;;   
        t) clear;
            option_picked "Option u Picked - Update PFC200 FW25";
            docker rm /fw
            read -sp "Enter root password: " PASSWORD
            echo
            docker run -d --restart on-failure:1 --name=fw  --env PASSWORD="$PASSWORD" --env SERVICE_NAME=fw --env MAX_UPDATE_RETRYS=5 --env CERT_NAME=ca.crt --env FILEPATH=/etc/docker/ --env FILE=daemon.json wagoautomation/fw-update-pfc200:04.03.03-25
            printf "Please wait 20 minutes.  Leave PLC alone.";
            sleep 10;
            /etc/config-tools/fwupdate status;
            exit;
        ;;   
        u) clear;
            option_picked "Option u Picked - Update PFC200 FW26";
            docker rm /fw
            read -sp "Enter root password: " PASSWORD
            echo
            docker run -d --restart on-failure:1 --name=fw  --env PASSWORD="$PASSWORD" --env SERVICE_NAME=fw --env MAX_UPDATE_RETRYS=5 --env CERT_NAME=ca.crt --env FILEPATH=/etc/docker/ --env FILE=daemon.json wagoautomation/fw-update-pfc200:04.04.03-26
            printf "Please wait 20 minutes.  Leave PLC alone.";
            sleep 10;
            /etc/config-tools/fwupdate status;
            exit;
        ;;   
        v) clear;
            option_picked "Option u Picked - Update PFC200 FW27";
            docker rm /fw
            read -sp "Enter root password: " PASSWORD
            echo
            docker run -d --restart on-failure:1 --name=fw  --env PASSWORD="$PASSWORD" --env SERVICE_NAME=fw --env MAX_UPDATE_RETRYS=5 --env CERT_NAME=ca.crt --env FILEPATH=/etc/docker/ --env FILE=daemon.json wagoron/fw-update-pfc:04.05.10-27
            printf "Please wait 20 minutes.  Leave PLC alone.";
            sleep 10;
            /etc/config-tools/fwupdate status;
            exit;
        ;;  
        w) clear;
            option_picked "Option w picked - Remove all docker images & containers";
            docker stop $(docker ps -aq) && docker rm $(docker ps -aq) && docker rmi $(docker images -q);
            exit;
        ;;  
        x) clear;
            chmod +x menu.sh;
            printf "Type ./menu.sh to re-open this tool";
            printf "\n";
            exit;
        ;;
        \n)exit;
        ;;
        *)clear;
            option_picked "Pick an option from the menu";
            show_menu;
        ;;
      esac
    fi
done