#!/bin/python3
import jarvis
import git
from pathlib import Path
from multiprocessing import Value, Process, freeze_support
from importlib import reload as ireload
from time import sleep


def run():
    while True:
        ireload(jarvis)
        jarvis.run()


def restart():
    global jarvis_process
    Path(get_pid_file()).unlink()
    jarvis_process.terminate()
    jarvis_process = Process(target=run, name="jarvis")
    jarvis_process.start()


def update():
    repo = git.Repo(".")
    dirty = repo.is_dirty()
    if dirty:
        print("    Local system has uncommitted changes.")
    current_hash = repo.head.object.hexsha
    origin = repo.remotes.origin
    origin.fetch()
    if current_hash != origin.refs["main"].object.hexsha:
        if dirty:
            return 2
        origin.pull()
        return 0
    return 1


def get_pid_file():
    return f"jarvis.{get_pid()}.pid"


def get_pid():
    global jarvis_process
    return jarvis_process.pid


def cli():
    pfile = Path(get_pid_file())
    while not pfile.exists():
        sleep(0.2)
    print(
        """
  All systems online.

Command List:
    (R)eload
    (U)pdate
    (Q)uit
    """
    )
    while True:
        cmd = input("> ")
        if cmd.lower() in ["q", "quit", "e", "exit"]:
            print("  Shutting down core systems...")
            pfile.unlink()
            break
        if cmd.lower() in ["u", "update"]:
            print("  Updating core systems...")
            status = update()
            if status == 0:
                restart()
                pfile = Path(get_pid_file())
                while not pfile.exists():
                    sleep(0.2)
                print("    Core systems successfully updated.")
            elif status == 1:
                print("    No core updates available.")
            elif status == 2:
                print("    Core updates available, but not applied.")
        if cmd.lower() in ["r", "reload"]:
            print("  Reloading core systems...")
            restart()
            pfile = Path(get_pid_file())
            while not pfile.exists():
                sleep(0.2)
            print("  All systems reloaded.")


if __name__ == "__main__":
    freeze_support()
    pid_file = Value("i", 0)
    jarvis_process = Process(target=run, name="jarvis")
    print("J.A.R.V.I.S. initializing....")
    print("  Updating core systems...")
    status = update()
    if status == 0:
        print("    Core systems successfully updated")
    elif status == 1:
        print("    No core updates available.")
    elif status == 2:
        print("    Core updates available, but not applied.")
    print("  Starting core systems...")
    jarvis_process.start()
    cli()
    if jarvis_process.is_alive():
        jarvis_process.terminate()
    jarvis_process.join(30)
    print("All systems shut down.")