from http.server import BaseHTTPRequestHandler, HTTPServer
import pdb
import pywinauto
import json
import subprocess
import zlib
import tkinter
import tkinter.ttk
import os
#ProcessChildSendString
def ProcessChildSendString(lProcess,lString):
    lByteString = zlib.compress(lString.encode("utf-8"))
    #Вернуть потенциальные \n
    lByteString = lByteString.replace(b'\n',b'{{n}}')
    #Отправить сообщение в дочерний процесс
    lProcess.stdin.write(lByteString+bytes('\n',"utf-8"))
    #print(str(lByteString+bytes('\n',"utf-8")))
    lProcess.stdin.flush()
    #Вернуть результат
    return

#ProcessChildReadWaitString
def ProcessChildReadWaitString(lProcess):
    #Ожидаем ответ от процесса
    lResult = lProcess.stdout.readline()
    #Вернуть потенциальные \n
    lResult = lResult.replace(b'{{n}}',b'\n')
    #print("check")
    #print(str(lResult))
    lResult = zlib.decompress(lResult[0:-1])
    lResult = lResult.decode("utf-8")
    #Вернуть результат
    return lResult

#ProcessChildSendObject
def ProcessChildSendObject(inProcess,inObject):
    #Выполнить отправку сконвертированного объекта в JSON
    ProcessChildSendString(inProcess,json.dumps(inObject))
    return
#ProcessChildReadWaitObject
def ProcessChildReadWaitObject(inProcess):
    #Выполнить получение и разбор объекта
    lResult=json.loads(ProcessChildReadWaitString(inProcess));
    return lResult;

#ProcessChildSendReadWaitString
def ProcessChildSendReadWaitString(lProcess,lString):
    ProcessChildSendString(lProcess,lString)
    #Вернуть результат
    return ProcessChildReadWaitString(lProcess)
#ProcessChildSendReadWaitObject
def ProcessChildSendReadWaitObject(inProcess,inObject):
    ProcessChildSendObject(inProcess,inObject)
    #Вернуть результат
    return ProcessChildReadWaitString(inProcess)

#pywinauto
def checkBitnessIs64():
    lResult=True

    return lResult;


def GetControl(inControlSpecificationArray):
    #Выполнить идентификацию объектов, если передан массив
    lResultList=[];
    if len(inControlSpecificationArray) > 0:
        #Выполнить подключение к объекту
        lRPAApplication = pywinauto.Application()
        #Проверка разрядности
        
        lRPAApplication.connect(**inControlSpecificationArray[0])
        lTempObject=lRPAApplication.window(**inControlSpecificationArray[0])

        #Циклическое прохождение к недрам объекта
        for lWindowSpecification in inControlSpecificationArray[1:]:
            lTempObject=lTempObject.window(**lWindowSpecification)
        #Получить инфо объект
        lTempObjectInfo = lTempObject.wrapper_object().element_info
        #Добавить информацию об обнаруженом объекте
        lResultList.append({"name":lTempObjectInfo.name, "rich_text":lTempObjectInfo.rich_text, "process_id":lTempObjectInfo.process_id,"handle":lTempObjectInfo.handle});
    return lResultList

def GetChildControls(inControlSpecificationArray):
    #Выполнить идентификацию объектов, если передан массив
    lResultList=[];
    if len(inControlSpecificationArray) > 0:
        #Выполнить подключение к объекту
        lRPAApplication = pywinauto.Application()
        lRPAApplication.connect(**inControlSpecificationArray[0])
        lTempObject=lRPAApplication.window(**inControlSpecificationArray[0])

        #Циклическое прохождение к недрам объекта
        for lWindowSpecification in inControlSpecificationArray[1:]:
            lTempObject=lTempObject.window(**lWindowSpecification)
        #Получить инфо объект
        lTempObjectInfo = lTempObject.wrapper_object().element_info
        #Добавить информацию об обнаруженом объекте
        lResultList.append({"name":lTempObjectInfo.name, "rich_text":lTempObjectInfo.rich_text, "process_id":lTempObjectInfo.process_id,"handle":lTempObjectInfo.handle});
    return lResultList
def GetWindowList(inWindowSpecificationArray = []):
    #Выполнить идентификацию объектов, если передан массив
    lResultList={};
    if len(inWindowSpecificationArray) > 0:
        #Выполнить подключение к объекту
        lRPAApplication = pywinauto.Application()
        lTempObject=lRPAApplication.connect(**inWindowSpecificationArray[0])

        #Циклическое прохождение к недрам объекта
        for lWindowSpecification in inWindowSpecificationArray[1:]:
            lTempObject=lTempObject.window(**lWindowSpecification)
            lTempObjectInfo = lTempObject.wrapper_object().element_info
            lResultList2.append({"name":lTempObjectInfo.name, "rich_text":lTempObjectInfo.rich_text, "process_id":lTempObjectInfo.process_id,"handle":lTempObjectInfo.handle});
        
        #Получить список объектов
        lResultList=pywinauto.findwindows.find_elements()
    else:
        #Получить список объектов
        lResultList=pywinauto.findwindows.find_elements()    
    lResultList2=[]
    for lI in lResultList:
        lResultList2.append({"name":lI.name, "rich_text":lI.rich_text, "process_id":lI.process_id,"handle":lI.handle});
    return lResultList2

# HTTPRequestHandler class
class testHTTPServer_RequestHandler(BaseHTTPRequestHandler):
    # GET
    def do_GET(self):
        if self.path == "/":
            # Send response status code
            self.send_response(200)

            #pdb.set_trace()
         
            # Send headers
            self.send_header('Content-type','text/html')
            self.end_headers()
         
            # Send message back to client
            f = open("Index.xhtml", "r") 
            message = f.read()
            f.close()
            # Write content as utf-8 data
            self.wfile.write(bytes(message, "utf8"))

        if self.path == '/3rdParty/Semantic-UI-CSS-master/semantic.min.css':
            # Send response status code
            self.send_response(200)

            #pdb.set_trace()
         
            # Send headers
            self.send_header('Content-type','text/css')
            self.end_headers()
         
            # Send message back to client
            f = open("3rdParty\Semantic-UI-CSS-master\semantic.min.css", "r") 
            message = f.read()
            f.close()
            # Write content as utf-8 data
            self.wfile.write(bytes(message, "utf8"))
        if self.path == '/3rdParty/Semantic-UI-CSS-master/semantic.min.js':
            # Send response status code
            self.send_response(200)

            #pdb.set_trace()
             
            # Send headers
            self.send_header('Content-type','application/javascript')
            self.end_headers()
         
            # Send message back to client
            f = open("3rdParty\Semantic-UI-CSS-master\semantic.min.js", "r") 
            message = f.read()
            f.close()
            # Write content as utf-8 data
            self.wfile.write(bytes(message, "utf8"))
        if self.path == '/3rdParty/jQuery/jquery-3.1.1.min.js':
            # Send response status code
            self.send_response(200)
            # Send headers
            self.send_header('Content-type','application/javascript')
            self.end_headers()
            # Send message back to client
            f = open("3rdParty\jQuery\jquery-3.1.1.min.js", "r") 
            message = f.read()
            f.close()
            # Write content as utf-8 data
            self.wfile.write(bytes(message, "utf8"))
        if self.path == '/3rdParty/Google/LatoItalic.css':
            # Send response status code
            self.send_response(200)
            # Send headers
            self.send_header('Content-type','text/css')
            self.end_headers()
            # Send message back to client
            f = open("3rdParty\Google\LatoItalic.css", "r") 
            message = f.read()
            f.close()
            # Write content as utf-8 data
            self.wfile.write(bytes(message, "utf8"))
        if self.path == '/3rdParty/Semantic-UI-CSS-master/themes/default/assets/fonts/icons.woff2':
            # Send response status code
            self.send_response(200)
            # Send headers
            self.send_header('Content-type','font/woff2')
            self.end_headers()
            # Send message back to client
            f = open("3rdParty\Semantic-UI-CSS-master\themes\default\assets\fonts\icons.woff2", "r") 
            message = f.read()
            f.close()
            # Write content as utf-8 data
            self.wfile.write(bytes(message, "utf8"))
        #Action ObjectInspector GetObjectList
        if self.path == '/ObjectDetector/JSONGetWindowList':

            #ReadRequest
            #lInputByteArray=self.rfile.read()
            #print(str(len(os.stat(self.rfile).st_size)))
            
            # Send response status code
            self.send_response(200)
      
            # Send headers
            self.send_header('Content-type','application/json')
            self.end_headers()
         
            # Send message back to client
            #{'functionName':'', 'argsArray':[]}
            lRequestObject={'functionName':'ElementGetChildElementList','argsArray':[]}
            #Отправить запрос в дочерний процесс, который отвечает за работу с Windows окнами
            ProcessChildSendObject(p,lRequestObject)
            #Получить ответ от дочернего процесса
            lResponseObject=ProcessChildReadWaitObject(p)
            message = json.dumps(lResponseObject)
            
            # Write content as utf-8 data
            self.wfile.write(bytes(message, "utf8"))  
    # POST
    def do_POST(self):
        #Action ObjectInspector GetObjectList
        if self.path == '/ObjectDetector/JSONGetWindowListArgs':
            #ReadRequest
            lInputByteArrayLength = int(self.headers.get('Content-Length'))
            lInputByteArray=self.rfile.read(lInputByteArrayLength)
            #Превращение массива байт в объект
            print(lInputByteArray.decode('utf8'))
            lInputObject=json.loads(lInputByteArray.decode('utf8'))
            
            
            # Send response status code
            self.send_response(200)
      
            # Send headers
            self.send_header('Content-type','application/json')
            self.end_headers()
         
            # Send message back to client
            #{'functionName':'', 'argsArray':[]}
            lRequestObject={'functionName':'ElementGetChildElementList','argsArray':lInputObject}
            
            #Отправить запрос в дочерний процесс, который отвечает за работу с Windows окнами
            ProcessChildSendObject(p,lRequestObject)
            
            #Получить ответ от дочернего процесса
            lResponseObject=ProcessChildReadWaitObject(p)
            print(str(lResponseObject))
            message = json.dumps(lResponseObject)
            
            # Write content as utf-8 data
            self.wfile.write(bytes(message, "utf8"))
        if self.path == '/GUIAction':
            #ReadRequest
            lInputByteArrayLength = int(self.headers.get('Content-Length'))
            lInputByteArray=self.rfile.read(lInputByteArrayLength)
            #Превращение массива байт в объект
            lInputObject=json.loads(lInputByteArray.decode('utf8'))
            # Send response status code
            self.send_response(200)
            # Send headers
            self.send_header('Content-type','application/json')
            self.end_headers()
            # Send message back to client
            #{'functionName':'', 'argsArray':[]}
            lRequestObject=lInputObject
            #Отправить запрос в дочерний процесс, который отвечает за работу с Windows окнами
            ProcessChildSendObject(p,lRequestObject)
            #Получить ответ от дочернего процесса
            lResponseObject=ProcessChildReadWaitObject(p)
            message = json.dumps(lResponseObject)
            # Write content as utf-8 data
            self.wfile.write(bytes(message, "utf8"))  
        return
    
def run():
  print('starting server...')
  # Server settings
  # Choose port 8080, for port 80, which is normally used for a http server, you need root access
  server_address = ('127.0.0.1', 8081)
  httpd = HTTPServer(server_address, testHTTPServer_RequestHandler)
  print('running server...')
  httpd.serve_forever() 

#Start childprocess
p = subprocess.Popen(['C:\\Users\\Acer\\AppData\\Local\\Programs\\Python\\Python37\\python.exe','C:\\Abs\\archive\\scopeSrcUL\\OpenRPA\\winGUI.py'],stdin=subprocess.PIPE,stdout=subprocess.PIPE)

#print(ChildProcessReadWaitString(p))
run()

def Constructor():
    lWindow=tkinter.Tk()

    #{'functionName':'', 'argsArray':[]}
    lRequestObject={'functionName':'ElementGetChildElementList','argsArray':[]}
    #Отправить запрос в дочерний процесс, который отвечает за работу с Windows окнами
    ProcessChildSendObject(p,lRequestObject)
    #Получить ответ от дочернего процесса
    lResponseObject=ProcessChildReadWaitObject(p)
    print("Response:"+str(lResponseObject))
    #Создать дерево
    lWidgetTree = tkinter.ttk.Treeview(lWindow)
    lIterator = 0
    #Добавить информацию об объектах в дерево
    for lListItem in lResponseObject['outputObject']:
        lIterator=lIterator+1
        lItemText=lListItem['name']
        if type(lItemText) is str:
            if len(lItemText) == 0:
                lItemText="No text (Len 0)"
        else:
            lItemText="No Text (NoneType)"
        lWidgetTree.insert("", lIterator, "Item"+str(lIterator), text=lItemText)
    lWidgetTree.pack()
    
    return

#Constructor()