craftbeerpi4-pione/cbpi/cli.py

import argparse
import datetime
import logging
import subprocess
import sys
import re
import requests
import yaml
from cbpi.utils.utils import load_config
from zipfile import ZipFile
from cbpi.craftbeerpi import CraftBeerPi
import os
import platform
import pathlib
import pkgutil
import shutil
import yaml
import click
from subprocess import call
import zipfile
from colorama import Fore, Back, Style
from importlib import import_module
import importlib
from jinja2 import Template
from importlib_metadata import metadata, version
from tabulate import tabulate
from PyInquirer import prompt, print_json

def create_config_file():
    if os.path.exists(os.path.join(".", 'config', "config.yaml")) is False:
        srcfile = os.path.join(os.path.dirname(__file__), "config", "config.yaml")
        destfile = os.path.join(".", 'config')
        shutil.copy(srcfile, destfile)
        
    if os.path.exists(os.path.join(".", 'config', "actor.json")) is False:
        srcfile = os.path.join(os.path.dirname(__file__), "config", "actor.json")
        destfile = os.path.join(".", 'config')
        shutil.copy(srcfile, destfile)

    if os.path.exists(os.path.join(".", 'config', "sensor.json")) is False:
        srcfile = os.path.join(os.path.dirname(__file__), "config", "sensor.json")
        destfile = os.path.join(".", 'config')
        shutil.copy(srcfile, destfile)

    if os.path.exists(os.path.join(".", 'config', "kettle.json")) is False:
        srcfile = os.path.join(os.path.dirname(__file__), "config", "kettle.json")
        destfile = os.path.join(".", 'config')
        shutil.copy(srcfile, destfile)

    if os.path.exists(os.path.join(".", 'config', "fermenter_data.json")) is False:
        srcfile = os.path.join(os.path.dirname(__file__), "config", "fermenter_data.json")
        destfile = os.path.join(".", 'config')
        shutil.copy(srcfile, destfile)

    if os.path.exists(os.path.join(".", 'config', "step_data.json")) is False:
        srcfile = os.path.join(os.path.dirname(__file__), "config", "step_data.json")
        destfile = os.path.join(".", 'config')
        shutil.copy(srcfile, destfile)

    if os.path.exists(os.path.join(".", 'config', "config.json")) is False:
        srcfile = os.path.join(os.path.dirname(__file__), "config", "config.json")
        destfile = os.path.join(".", 'config')
        shutil.copy(srcfile, destfile)

    if os.path.exists(os.path.join(".", 'config', "dashboard", "cbpi_dashboard_1.json")) is False:
        srcfile = os.path.join(os.path.dirname(__file__), "config", "dashboard", "cbpi_dashboard_1.json")
        destfile = os.path.join(".", "config", "dashboard")
        shutil.copy(srcfile, destfile)

    if os.path.exists(os.path.join(".", 'config', "carftbeerpi.service")) is False:
        srcfile = os.path.join(os.path.dirname(__file__), "config", "craftbeerpi.service")
        destfile = os.path.join(".", 'config')
        shutil.copy(srcfile, destfile)

    if os.path.exists(os.path.join(".", 'config', "chromium.desktop")) is False:
        srcfile = os.path.join(os.path.dirname(__file__), "config", "chromium.desktop")
        destfile = os.path.join(".", 'config')
        shutil.copy(srcfile, destfile)

    print("Config Folder created")


def create_home_folder_structure():
    pathlib.Path(os.path.join(".", 'logs/sensors')).mkdir(parents=True, exist_ok=True)
    pathlib.Path(os.path.join(".", 'config')).mkdir(parents=True, exist_ok=True)
    pathlib.Path(os.path.join(".", 'config/dashboard')).mkdir(parents=True, exist_ok=True)
    pathlib.Path(os.path.join(".", 'config/dashboard/widgets')).mkdir(parents=True, exist_ok=True)
    pathlib.Path(os.path.join(".", 'config/recipes')).mkdir(parents=True, exist_ok=True)
    pathlib.Path(os.path.join(".", 'config/fermenterrecipes')).mkdir(parents=True, exist_ok=True)
    pathlib.Path(os.path.join(".", 'config/upload')).mkdir(parents=True, exist_ok=True) 
    print("Folder created")


def setup_one_wire():
    print("Setting up 1Wire")
    with open('/boot/config.txt', 'w') as f:
        f.write("dtoverlay=w1-gpio,gpiopin=4,pullup=on")
    print("/boot/config.txt created")

def list_one_wire():
    print("List 1Wire")
    call(["modprobe", "w1-gpio"])
    call(["modprobe", "w1-therm"])
    try:
        for dirname in os.listdir('/sys/bus/w1/devices'):
            if (dirname.startswith("28") or dirname.startswith("10")):
                print(dirname)
    except Exception as e:
        print(e)

def copy_splash():
    srcfile = os.path.join(".", "config", "splash.png")
    destfile = os.path.join(".", 'config')
    shutil.copy(srcfile, destfile)
    print("Splash Srceen created")


def clear_db():
    import os.path
    if os.path.exists(os.path.join(".", "craftbeerpi.db")) is True:
        os.remove(os.path.join(".", "craftbeerpi.db"))
        print("database cleared")

def recursive_chown(path, owner, group):
    for dirpath, dirnames, filenames in os.walk(path):
        shutil.chown(dirpath, owner, group)
        for filename in filenames:
            shutil.chown(os.path.join(dirpath, filename), owner, group)

def check_for_setup():
    if os.path.exists(os.path.join(".", "config", "config.yaml")) is False:
        print("***************************************************")
        print("CraftBeerPi Config File not found: %s" % os.path.join(".", "config", "config.yaml"))
        print("Please run 'cbpi setup' before starting the server ")
        print("***************************************************")
        return False
    if os.path.exists(os.path.join(".", "config", "upload")) is False:
        print("***************************************************")
        print("CraftBeerPi upload folder not found: %s" % os.path.join(".", "config/upload"))
        print("Please run 'cbpi setup' before starting the server ")
        print("***************************************************")
        return False
#    if os.path.exists(os.path.join(".", "config", "fermenterrecipes")) is False:
#        print("***************************************************")
#        print("CraftBeerPi fermenterrecipes folder not found: %s" % os.path.join(".", "config/fermenterrecipes"))
#        print("Please run 'cbpi setup' before starting the server ")
#        print("***************************************************")
#        return False
    backupfile = os.path.join(".", "restored_config.zip")
    if os.path.exists(os.path.join(backupfile)) is True:
        print("***************************************************")
        print("Found backup of config. Starting restore")
        required_content=['dashboard/', 'recipes/', 'upload/', 'config.json', 'config.yaml']
        zip=zipfile.ZipFile(backupfile)
        zip_content_list = zip.namelist()
        zip_content = True
        print("Checking content of zip file")
        for content in required_content:
            try:
                check = zip_content_list.index(content)
            except:
                zip_content = False

        if zip_content == True:
            print("Found correct content. Starting Restore process")
            output_path = pathlib.Path(os.path.join(".", 'config'))
            system = platform.system()
            print(system)
            if system != "Windows":
                owner = output_path.owner()
                group = output_path.group()
            print("Removing old config folder")
            shutil.rmtree(output_path, ignore_errors=True) 
            print("Extracting zip file to config folder")
            zip.extractall(output_path)
            zip.close()
            if system != "Windows":
                print("Changing owner and group of config folder recursively to {}:{}".format(owner,group))
                recursive_chown(output_path, owner, group)
            print("Removing backup file")
            os.remove(backupfile)
        else:
            print("Wrong Content in zip file. No restore possible")
            print("Removing zip file")
            os.remove(backupfile)
        print("***************************************************")

        return True 
    else:
        return True

def plugins_list():
    result = []
    print("")
    print(Fore.LIGHTYELLOW_EX,"List of active plugins", Style.RESET_ALL)
    print("")
    discovered_plugins = {
        name: importlib.import_module(name)
        for finder, name, ispkg
        in pkgutil.iter_modules()
        if name.startswith('cbpi') and len(name) > 4
    }
    for key, module in discovered_plugins.items():
        try:
            meta = metadata(key)
            result.append(dict(Name=meta["Name"], Version=meta["Version"], Author=meta["Author"], Homepage=meta["Home-page"], Summary=meta["Summary"]))
                        
        except Exception as e:
            print(e)
    print(Fore.LIGHTGREEN_EX, tabulate(result, headers="keys"), Style.RESET_ALL)
    


def plugin_create():

    
    

    print("Plugin Creation")
    print("")

    questions = [
        {
            'type': 'input',
            'name': 'name',
            'message': 'Plugin Name:',
        }
    ]

    answers = prompt(questions)

    name = "cbpi4_" + answers["name"]
    if os.path.exists(os.path.join(".", name)) is True:
        print("Cant create Plugin. Folder {} already exists ".format(name))
        return

    url = 'https://github.com/Manuel83/craftbeerpi4-plugin-template/archive/main.zip'
    r = requests.get(url)
    with open('temp.zip', 'wb') as f:
        f.write(r.content)

    with ZipFile('temp.zip', 'r') as repo_zip:
        repo_zip.extractall()

    os.rename("./craftbeerpi4-plugin-template-main", os.path.join(".", name))
    os.rename(os.path.join(".", name, "src"), os.path.join(".", name, name))

    import jinja2

    templateLoader = jinja2.FileSystemLoader(searchpath=os.path.join(".", name))
    templateEnv = jinja2.Environment(loader=templateLoader)
    TEMPLATE_FILE = "setup.py"
    template = templateEnv.get_template(TEMPLATE_FILE)
    outputText = template.render(name=name)

    with open(os.path.join(".", name, "setup.py"), "w") as fh:
        fh.write(outputText)

    TEMPLATE_FILE = "MANIFEST.in"
    template = templateEnv.get_template(TEMPLATE_FILE)
    outputText = template.render(name=name)
    with open(os.path.join(".", name, "MANIFEST.in"), "w") as fh:
        fh.write(outputText)

    TEMPLATE_FILE = os.path.join("/", name, "config.yaml")
    template = templateEnv.get_template(TEMPLATE_FILE)
    outputText = template.render(name=name)

    with open(os.path.join(".", name, name, "config.yaml"), "w") as fh:
        fh.write(outputText)

        
    print("")
    print("")
    print("Plugin {}{}{} created! ".format(Fore.LIGHTGREEN_EX, name, Style.RESET_ALL) )
    print("")
    print("Developer Documentation: https://openbrewing.gitbook.io/craftbeerpi4_support/readme/development")
    print("")
    print("Happy developing! Cheers")
    print("")
    print("")
    


@click.group()
def main():
    print("---------------------")
    print("Welcome to CBPi")
    print("---------------------")
    level = logging.INFO
    logging.basicConfig(level=level, format='%(asctime)s - %(levelname)s - %(name)s - %(message)s')
    pass


@click.command()
def setup():
    '''Create Config folder'''
    print("Setting up CraftBeerPi")
    create_home_folder_structure()
    create_config_file()


@click.command()
@click.option('--list', is_flag=True, help="List all 1Wire Devices")
@click.option('--setup', is_flag=True, help="Setup 1Wire on Raspberry Pi")
def onewire(list, setup):
    '''Setup 1wire on Raspberry Pi'''
    if setup is True:
        setup_one_wire()
    if list is True:
        list_one_wire()



@click.command()
def start():
    '''Lets go brewing'''
    if check_for_setup() is False:
        return
    print("Starting up CraftBeerPi ...")
    cbpi = CraftBeerPi()
    cbpi.start()


@click.command()
def plugins():
    '''List active plugins'''
    plugins_list()
    return




@click.command()
def create():
    '''Create New Plugin'''
    plugin_create()

@click.command()
@click.argument('name')
def autostart(name):
    '''Enable or disable autostart'''
    if(name == "status"):
        if os.path.exists(os.path.join("/etc/systemd/system","craftbeerpi.service")) is True:
            print("CraftBeerPi Autostart is {}ON{}".format(Fore.LIGHTGREEN_EX,Style.RESET_ALL))
        else:
            print("CraftBeerPi Autostart is {}OFF{}".format(Fore.RED,Style.RESET_ALL))
    elif(name == "on"):
        print("Add craftbeerpi.service to systemd")
        try:
            if os.path.exists(os.path.join("/etc/systemd/system","craftbeerpi.service")) is False:
                srcfile = os.path.join(".", "config", "craftbeerpi.service")
                destfile = os.path.join("/etc/systemd/system")
                shutil.copy(srcfile, destfile)
                print("Copied craftbeerpi.service to /etc/systemd/system")
                os.system('systemctl enable craftbeerpi.service')
                print('Enabled craftbeerpi service')
                os.system('systemctl start craftbeerpi.service')
                print('Started craftbeerpi.service')
            else:
                print("craftbeerpi.service is already located in /etc/systemd/system")
        except Exception as e:
            print(e)
            return
        return
    elif(name == "off"): 
        print("Remove craftbeerpi.service from systemd")
        try:
            status = os.popen('systemctl list-units --type=service --state=running | grep craftbeerpi.service').read()
            if status.find("craftbeerpi.service") != -1:
                os.system('systemctl stop craftbeerpi.service')
                print('Stopped craftbeerpi service')
                os.system('systemctl disable craftbeerpi.service')
                print('Removed craftbeerpi.service as service')
            else:
                print('craftbeerpi.service service is not running')

            if os.path.exists(os.path.join("/etc/systemd/system","craftbeerpi.service")) is True:
                os.remove(os.path.join("/etc/systemd/system","craftbeerpi.service")) 
                print("Deleted craftbeerpi.service from /etc/systemd/system")
            else:
                print("craftbeerpi.service is not located in /etc/systemd/system")
        except Exception as e:
            print(e)
            return
        return
    
    


@click.command()
@click.argument('name')
def chromium(name):
    '''Enable or disable autostart'''
    if(name == "status"):
        if os.path.exists(os.path.join("/etc/xdg/autostart/","chromium.desktop")) is True:
            print("CraftBeerPi Chromium Desktop is {}ON{}".format(Fore.LIGHTGREEN_EX,Style.RESET_ALL))
        else:
            print("CraftBeerPi Chromium Desktop is {}OFF{}".format(Fore.RED,Style.RESET_ALL))
    elif(name == "on"):
        print("Add chromium.desktop to /etc/xdg/autostart/")
        try:
            if os.path.exists(os.path.join("/etc/xdg/autostart/","chromium.desktop")) is False:
                srcfile = os.path.join(".", "config", "chromium.desktop")
                destfile = os.path.join("/etc/xdg/autostart/")
                shutil.copy(srcfile, destfile)
                print("Copied chromium.desktop to /etc/xdg/autostart/")
            else:
                print("chromium.desktop is already located in /etc/xdg/autostart/")
        except Exception as e:
            print(e)
            return
        return
    elif(name == "off"): 
        print("Remove chromium.desktop from /etc/xdg/autostart/")
        try:
            if os.path.exists(os.path.join("/etc/xdg/autostart/","chromium.desktop")) is True:
                os.remove(os.path.join("/etc/xdg/autostart/","chromium.desktop"))
                print("Deleted chromium.desktop from /etc/xdg/autostart/")
            else:
                print("chromium.desktop is not located in /etc/xdg/autostart/")
        except Exception as e:
            print(e)
            return
        return

main.add_command(setup)
main.add_command(start)
main.add_command(autostart)
main.add_command(chromium)
main.add_command(plugins)
main.add_command(onewire)
main.add_command(create)