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