#!/usr/bin/python
# Server that will accept connections from a Vim channel.
# Run this server and then in Vim you can open the channel:
#  :let handle = connect('localhost:8765', 'json')
#
# Then Vim can send requests to the server:
#  :let response = sendexpr(handle, 'hello!')
#
# And you can control Vim by typing a JSON message here, e.g.:
#   ["ex","echo 'hi there'"]
#
# See ":help channel-demo" in Vim.

import SocketServer
import json
import socket
import sys
import threading

thesocket = None

class ThreadedTCPRequestHandler(SocketServer.BaseRequestHandler):

    def handle(self):
        print "=== socket opened ==="
        global thesocket
        thesocket = self.request
        while True:
            try:
                data = self.request.recv(4096)
            except socket.error:
                print "=== socket error ==="
                break
            except IOError:
                print "=== socket closed ==="
                break
            if data == '':
                print "=== socket closed ==="
                break
            print "received: {}".format(data)
            try:
                decoded = json.loads(data)
            except ValueError:
                print "json decoding failed"
                decoded = [0, '']

            if decoded[1] == 'hello!':
                response = "got it"
            else:
                response = "what?"
            encoded = json.dumps([decoded[0], response])
            print "sending {}".format(encoded)
            self.request.sendall(encoded)
        thesocket = None

class ThreadedTCPServer(SocketServer.ThreadingMixIn, SocketServer.TCPServer):
    pass

if __name__ == "__main__":
    HOST, PORT = "localhost", 8765

    server = ThreadedTCPServer((HOST, PORT), ThreadedTCPRequestHandler)
    ip, port = server.server_address

    # Start a thread with the server -- that thread will then start one
    # more thread for each request
    server_thread = threading.Thread(target=server.serve_forever)

    # Exit the server thread when the main thread terminates
    server_thread.daemon = True
    server_thread.start()
    print "Server loop running in thread: ", server_thread.name

    print "Listening on port {}".format(PORT)
    while True:
        typed = sys.stdin.readline()
        if "quit" in typed:
            print "Goodbye!"
            break
        if thesocket is None:
            print "No socket yet"
        else:
            print "sending {}".format(typed)
            thesocket.sendall(typed)

    server.shutdown()
    server.server_close()