# This script should be loaded inside a python object of Splash.
# To do so, add the following to one of a scene configuration:
#
# "python" : {
#   "type" : "python",
#   "file" : "./httpserver.py"
# }
#
# For this to work,  the configuration file should be in the same
# directory as this script. Otherwise, modify the path accordingly

import splash
import re
from http.server import BaseHTTPRequestHandler, HTTPServer

hostName = "localhost"
hostPort = 10000
reg = re.compile('/(\w*)\??(\w*)?\&?(\w*)?\&?(.*)?')

class SplashServer(BaseHTTPRequestHandler):
    def sendResult(self, result):
        if len(result) != 0:
            strResult = str(result)
            self.send_response(200)
            self.send_header("Content-Length", len(strResult))
            self.send_header("Content-type", "text")
            self.send_header("Access-Control-Allow-Origin", "*")
            self.end_headers()
            self.wfile.write(bytes(strResult, "utf-8"))
        else:
            self.send_response(204)

    def do_GET(self):
        match = reg.findall(self.path)[0]
        command = match[0]

        if match[0] == "getObjectList":
            result = splash.get_object_list()
            self.sendResult(result)
        elif match[0] == "getObjectTypes":
            result = splash.get_object_types()
            self.sendResult(result)
        elif match[0] == "getObjectAttributeDescription":
            obj = match[1]
            attr = match[2]
            result = splash.get_object_attribute_description(obj, attr)
            self.sendResult(result)
        elif match[0] == "getObjectAttribute":
            obj = match[1]
            attr = match[2]
            result = splash.get_object_attribute(obj, attr)
            self.sendResult(result)
        elif match[0] == "getObjectLinks":
            result = splash.get_object_links()
            self.sendResult(result)
        elif match[0] == "setObjectAttribute":
            obj = match[1]
            attr = match[2]
            value = match[3].split('&')
            for i in range(len(value)):
                converted = False
                try:
                    value[i] = int(value[i])
                    converted = True
                except:
                    pass
                
                if not converted:
                    try:
                        value[i] = float(value[i])
                    except:
                        pass

            if splash.set_object_attribute(obj, attr, value):
                self.sendResult("OK")
            else:
                self.sendResult("")

def splash_init():
    global splashServer
    print("Starting HTTP server")
    splashServer = HTTPServer((hostName, hostPort), SplashServer)
    splashServer.timeout = 0.1

def splash_loop():
    global splashServer
    splashServer.handle_request()

def splash_stop():
    print("Shutting down HTTP server")