install.py

#!/usr/bin/python
# -*- coding: utf-8 -*-

# Imports.
import sys # System stuff.
import os # Operating System functions.
from colorama import Fore

# Distro package manager commands
class Distros:
    @staticmethod
    def debian():
        return [
            'sudo apt update',
            'sudo apt-get install -y python3 python3-pip python-dev',
        ]
    @staticmethod
    def arch():
        return [
            'sudo pacman -Syu',
            'sudo pacman -S python python-pip',
        ]
    @staticmethod
    def void():
        return Distros.debian()
    @staticmethod
    def opensuse():
        return [
            'sudo zypper in python310',
        ]
        # Do I need to add more here?

def handle_distros():
    # Distros
    distros = {
        'Debian': [
            'Parrot',
            'Ubuntu',
            'Kali',
            'Mint',
        ],
        'Arch': [
            'Garuda',
            'Manjaro',
        ],
        'Void' : [],
        'openSUSE': [
            'openSUSE Tumbleweed',
            'openSUSE Leap',
        ]
    }

    # auto detect
    distro = os.popen('cat /etc/issue').read()
    auto = f'{Fore.RED}- Unsupported disto'
    for k, v in distros.items():
        if k.lower() in distro.lower():
            auto = f'{Fore.YELLOW}- {k}'
            break
        else:
            txt = ''
            for sub in v:
                if sub.lower() in distro.lower():
                    txt = f'{Fore.YELLOW}- {k} ({sub})'
                    break
            if txt != '':
                auto = txt
                break

    message = ''
    for key, val in distros.items():
        message += f'\n\t- {Fore.GREEN}{key}'
        for sub in val:
            message += f'\n\t    - {Fore.YELLOW}{sub}'
    
    print(f"""
        {Fore.GREEN}- auto
            {auto}{message}
        {Fore.WHITE}""")

    ans = ''
    while not ans in [k.lower() for k, _ in distros.items()] + ['auto']: 
        # user input to select a distro
        ans=input(f"{Fore.YELLOW}>_ {Fore.CYAN}What is your operating system?: {Fore.WHITE}").lower()

    if 'auto' in ans:
        ans = auto.split('- ')[1].lower()

    if 'debian' in ans:
        return Distros.debian()
    elif 'arch' in ans:
        return Distros.arch()
    elif 'void' in ans:
        return Distros.void()
    elif 'opensuse' in ans:
        return Distros.opensuse()
    
    print(f'\n\t{Fore.RED}Unsuported distro: {Fore.WHITE}{ans}\n')
    exit(1)

def check_shell_config(location):
    # check if the cryptex alias is in the given file
    with open(location, 'rt') as f:
        check = 'alias cryptex' in f.read()
    return check
    
def handle_shell():
    shell = os.readlink(f'/proc/{os.getppid()}/exe')

    # Display a unsuported shell message if the shell isn't suported
    message = f'\n\t\t{Fore.RED}Unsuported shell'
    supported = ['bash', 'zsh', 'fish']
    for s in supported:
        if s in shell:
            message = ''
            break

    # Auto generate the list of shels
    shells = ''
    for s in supported:
        shells += f'\n\t{Fore.GREEN}- {s}'
    print(f"""
        {Fore.GREEN}- auto
            {Fore.YELLOW}- {shell}{message}{shells}
    {Fore.WHITE}""")

    # Check for input
    ans = ''
    options = ['auto'] + supported
    while ans not in options:
        ans=input(f"{Fore.YELLOW}>_ {Fore.CYAN}What is your shell?: {Fore.WHITE}").lower()

    # If the input was not 'auto' then set the shell to what the user entered
    if 'auto' not in ans:
        shell = ans

    # set the path to the config based on what shell you are setting it up for
    user = os.environ['HOME']
    path = ''
    if 'bash' in shell:
        path = f'{user}/.bashrc'
    elif 'zsh' in shell:
        path = f'{user}/.zshrc'
    elif 'fish' in shell:
        path = f'{user}/.config/fish/config.fish'
    else:
        print(f'\n\t{Fore.RED}Unsuported shell: {Fore.WHITE}{shell}\n')
        exit(1)

    # Check if the cyrptex alias already exists in the given shell
    if check_shell_config(path):
        print(f'\n\t{Fore.BLUE}Alias already exists in config: {Fore.WHITE}{path}\n')
        return ''

    command = 'echo \'alias cryptex="python3 ~/.Cryptex/src/main.py"\'' 
    return f'{command} >> {path}'
    
def main():
    # Check for updates
    from src import Update
    Update()

    # List over commands to run
    commands = []

    commands += handle_distros()

    # Cryptex related commands
    commands += [
        'pip install -r requirements.txt',
        'rm -rf ~/.Cryptex',
        'mkdir ~/.Cryptex',
        'cp -r . ~/.Cryptex',
    ]

    # Shell related commands
    commands += [handle_shell()]

    # Run the commands
    for c in commands:
        if len(c) <= 0: continue
        print(f'\n\t{Fore.GREEN}RUNNING: {Fore.WHITE}{c}\n')
        os.system(c)

    # End message
    print(f"""
    {Fore.GREEN}Installation finished.
    Restart the terminal and type {Fore.YELLOW}cryptex {Fore.GREEN}to run the program{Fore.WHITE}
    """)
    exit()

if __name__ == '__main__':
    try:
        main()
    except KeyboardInterrupt:
        print(f'\n{Fore.YELLOW}You interrupted the program.{Fore.WHITE}')
        try:
            sys.exit(0)
        except SystemExit:
            os._exit(0)