modelverse/scripts/prompt.py

import urllib
import urllib2
import threading
import subprocess
import os
import sys

sys.path.append("../interface/HUTN")
sys.path.append("interface/HUTN")
from hutn_compiler.compiler import main as hutn_compile

memory = {}

def send_data(commands, address, username):
    def flush_data(data):
        if data:
            urllib2.urlopen(urllib2.Request(address, urllib.urlencode({"op": "set_input", "data": json.dumps(data), "username": username}))).read()
        return []

    var_list = {}
    data = []
    for p in v:
        if isinstance(p, int):
            if p not in var_list:
                data = flush_data(data)
                val = urllib2.urlopen(urllib2.Request(address, urllib.urlencode({"op": "get_output", "username": username}))).read()
                val = val.split("=", 2)[1].split("&", 1)[0]
                var_list[p] = val
                continue
            else:
                val = var_list[p]
                t = "R"
        else:
            val = p
            t = "V"
        data.append([t, val])
    flush_data(data)

def local_print(string):
    if os.name == "posix":
        # Nicer colour output when using posix (and thus supporting colour)
        string = "\033[92m%s\033[0m" % string
    print(string)

def remote_print(string):
    if os.name == "posix":
        # Nicer colour output when using posix (and thus supporting colour)
        string = "\033[94m%s\033[0m" % string
    print(string)

local_print("Welcome to the local shell!")
local_print("Please specify Modelverse location (default: 127.0.0.1:8001)")

location = raw_input()
if location == "":
    address = "http://127.0.0.1:8001/"

local_print("Username (default: test)")
username = raw_input()
if username == "":
    username = "test"
else:
    username = username.strip()

# If user shouldn't exist yet, we create it
urllib2.urlopen(urllib2.Request(address, urllib.urlencode({"op": "set_input", "element_type": "V", "value": '"%s"' % username, "username": "user_manager"}))).read()

local_print("Switching context to Modelverse: all data is piped.")
local_print("Use command $ to switch to HUTN parsing mode.")
local_print("Use \\ before the value to use a primitive value different from a string, which is already JSON serialized.")
local_print("To quit: execute command 'quit'")

def print_output():
    while 1:
        output = urllib2.urlopen(urllib2.Request(address, urllib.urlencode({"op": "get_output", "username": username}))).read()
        l, r = output.split("&", 1)
        if "value" in l:
            output = l
        else:
            output = r
        _, output = output.split("=", 1)
        remote_print("%s" % str(output))

thrd = threading.Thread(target=print_output)
thrd.daemon = True
thrd.start()

while 1:
    command = raw_input()
    if command == "quit":
        local_print("Received quit: breaking connection to Modelverse immediately!")
        break

    if command.startswith("$"):
        # Invoke the HUTN parser
        tmp_file = "__action.alc"
        with open(tmp_file, 'w') as f:
            f.write("")
        local_print("Entering HUTN coding environment.")
        local_print("There is no nice editor right now, so please just modify the file '__action.alc' in this folder.")
        while 1:
            local_print("When you are done, press <return> to continue.")
            raw_input()
            local_print("File contents:")
            with open(tmp_file, 'r') as f:
                local_print(f.read())
            local_print("Sending file through parser!")

            try:
                commands = hutn_compile(tmp_file, "interface/HUTN/grammars/actionlanguage.g", "CS")
                local_print("Compilation succesfully terminated!")
                local_print("Upload this file? [Y/n]")
                i = raw_input()
                if i == "n\n":
                    continue
                send_data(commands, address)
            except:
                import traceback
                traceback.print_exc()
                local_print("Error during compilation!")
                local_print("Do you want to try again? [Y/n]")
                i = raw_input()
                if i[0] == "n":
                    break

            local_print("HUTN input finished, going back to regular prompt!")
    else:
        # Just send a normal request
        if command.startswith("\\"):
            command = command[1:]
        else:
            command = '"%s"' % command

        urllib2.urlopen(urllib2.Request(address, urllib.urlencode({"op": "set_input", "element_type": "V", "value": command, "username": username}))).read()