from sultan.api import Sultan
import subprocess
import time
from collections import deque 
import os
from . import consts


class shellRunner:
    def __init__(self):
        pass

    def feed(self, command): # string.split()
        self._process_command(command)

    def get_bash_var(self, var):
        return os.environ.get(var)

    def run_script(self, script):
        try:
            subprocess.check_call(script, shell=True, executable='/bin/bash')
        except subprocess.CalledProcessError as ex:
            print("command failed: ", script)
            return 1
        return 0
    
    def _process_command(self, command):

        if self._is_complex(command):
            return self._process_complex(command)
        return self._process_one_command(command)

    def _process_one_command(self, command):
        command = command.split()
        streaming = False
        output = subprocess.run('which ' + command[0],
                        stdout=subprocess.PIPE,
                        stderr=subprocess.PIPE,
                        shell=True)
        path = output.stdout.decode('utf-8').strip()
        if path and command[0] in consts.DISPLAY:
            try:
                subprocess.check_call(command)
            except subprocess.CalledProcessError:
                print("[pyshell] command failed...")
                return 1
            except KeyboardInterrupt:
                print("keyboard interupted")
            return 0
        if command[0] == 'export':
            if len(command) < 2:
                print("exporting nothing is not allowed")
                return
            split_cmd = command[1].split('=')
            if len(split_cmd) > 1:
                 os.environ[split_cmd[0]] = split_cmd[1]
            idx = 1
        elif '=' in command[0]:
            split_cmd = command[0].split('=')
            if len(split_cmd) < 2:
                print("Error: cannot set = to nothing")
                return 1
            os.environ[split_cmd[0]] = split_cmd[1]
            return 0
        elif 'cd' in command[0]:
            if len(command) < 2:
                print('cannot change dir to nothing')
                return
            os.chdir(command[1])
            return 0
        elif 'pip' in command[0]:
            if 'VIRTUAL_ENV' in os.environ:
                s = Sultan.load(src=os.path.join(os.environ['VIRTUAL_ENV'], 'bin', 'activate'),
                                executable='/bin/bash')
            else:
                s = Sultan.load()
            streaming = True
            bas_cmd = 'pip'
            idx = 1
        else:
            if command[0] in consts.STREAMING:
                streaming = True
            bas_cmd = command[1] if command[0] == 'sudo' else command[0]
            s = Sultan.load(sudo=True if command[0] == 'sudo' else False)
            idx = 2 if command[0] == 'sudo' else 1
            if bas_cmd == 'apt-get': 
                bas_cmd = 'apt__get'

        options = ''
        while idx < len(command):
            options += command[idx] + " "
            idx += 1
        if streaming:
            cmd_string = 'global result; result = s.' + bas_cmd + '(\'' + options + '\')' + '.run(streaming=True)'
            try:
                exec(cmd_string, globals(), locals())      
                complete = False
                while not complete:
                    complete = result.is_complete
                    for line in result.stdout:
                        print(line)
                    for line in result.stderr:
                        print(line)
                    time.sleep(.5)
            except Exception as e:
                print(e)
                print("command failed!")
                return 1
            except KeyboardInterrupt:
                print("keyboard interupted")
        else:
            cmd_string = 'global result; result = s.' + bas_cmd + '(\'' + options + '\')' + '.run()'
            try:
                exec(cmd_string, globals(), locals())
                result.print_stdout()
            except Exception as e:
                print(e)
                print("command failed!")
                return 1


    def _is_complex(self, command):
        return False
    
    # TODO: add complex bash command handling 
    def _process_complex(self, command):
        pass