elements.py/cli.py

import os
import json
import subprocess
import sys
from src.help import help


def clear_console():
    # Clear console based on the operating system
    if os.name == "nt":  # For Windows
        os.system("cls")
    else:  # For Unix/Linux/Mac
        os.system("clear")


def check_python_installation():
    try:
        python_version = subprocess.check_output(
            ["python", "--version"], stderr=subprocess.STDOUT, universal_newlines=True
        )
        print(f"Python is already installed ({python_version})")
        return True
    except FileNotFoundError:
        return False


def install_python():
    print("Python is not installed. Please download and install it from:")
    print("https://www.python.org/downloads/")
    input("Press Enter to continue after installing Python...")
    sys.exit(1)


def check_rpc_config():
    if os.path.exists("rpc_config.json"):
        return True
    return False


def get_rpc_config():
    rpc_host = input("Enter the RPC host: ")
    rpc_port = input("Enter the RPC port: ")
    rpc_user = input("Enter the RPC user: ")
    rpc_password = input("Enter the RPC password: ")

    rpc_config = {
        "rpc_host": rpc_host,
        "rpc_port": rpc_port,
        "rpc_user": rpc_user,
        "rpc_password": rpc_password,
    }

    with open("rpc_config.json", "w") as config_file:
        json.dump(rpc_config, config_file)


def run_script(script_name):
    script_path = os.path.join("src", f"{script_name}.py")

    if os.path.exists(script_path):
        # print(f"Running {script_name}...")
        os.system(f"python {script_path}")
    else:
        print(f"Error: {script_name} not found!")


def main():
    setup_completed = False

    while not check_python_installation():
        install_python()

    subprocess.run(["pip", "install", "-r", "requirements.txt"])

    while not check_rpc_config():
        clear_console()
        print("RPC config file not found.")
        user_input = input(
            "Do you have credentials for your Elements node RPC (yes/no)? "
        ).lower()

        if user_input in ["no", "n"]:
            print(
                "Please follow the instructions to set up an Elements node: [https://docs.liquid.net/docs/building-on-liquid]"
            )
            input("Press Enter to continue after setting up the Elements node...")
        elif user_input in ["yes", "y"]:
            get_rpc_config()
        else:
            print("Invalid input. Please enter 'y' or 'n'.")

    setup_completed = True

    print("You are ready to proceed.")
    clear_console()

    print("\nElements Python Command Line Interface")
    help()

    while True:
        user_input = input("Enter a command: ")

        if user_input.lower() == "exit":
            break
        if user_input.lower() == "help":
            help()
        if setup_completed:
            run_script(user_input)
        else:
            print("Error: Setup not completed. Use 'help' to see available commands.")


if __name__ == "__main__":
    main()