55 lines
1.5 KiB
Python
55 lines
1.5 KiB
Python
|
#!/usr/bin/env python
|
||
|
|
|
||
|
|
from BaseHTTPServer import BaseHTTPRequestHandler, HTTPServer
|
||
|
|
import sys
|
||
|
|
import socket
|
||
|
|
import json
|
||
|
|
import struct
|
||
|
|
import re
|
||
|
|
import select
|
||
|
|
|
||
|
|
server_address = ('',8128)
|
||
|
|
timeout = 15
|
||
|
|
|
||
|
|
# Send an encoded message to stdout.
|
||
|
|
def extensionCall(messageContent):
|
||
|
|
sys.stdout.write(struct.pack('@I', len(messageContent)))
|
||
|
|
sys.stdout.write(messageContent)
|
||
|
|
sys.stdout.flush()
|
||
|
|
# wait for extension response
|
||
|
|
i, o, e = select.select( [sys.stdin], [], [], timeout)
|
||
|
|
if (i):
|
||
|
|
rawLength = sys.stdin.read(4)
|
||
|
|
if len(rawLength) == 0:
|
||
|
|
return True
|
||
|
|
messageLength = struct.unpack('@I', rawLength)[0]
|
||
|
|
message = sys.stdin.read(messageLength)
|
||
|
|
return message
|
||
|
|
else:
|
||
|
|
return False
|
||
|
|
|
||
|
|
class PostHTTPRequestHandler(BaseHTTPRequestHandler):
|
||
|
|
def do_GET(self):
|
||
|
|
response = extensionCall('{"command" : "status"}')
|
||
|
|
if response:
|
||
|
|
self.send_response(200)
|
||
|
|
else:
|
||
|
|
self.send_response(500)
|
||
|
|
self.send_header('Content-type', 'application/json')
|
||
|
|
self.end_headers()
|
||
|
|
self.wfile.write(response)
|
||
|
|
|
||
|
|
def do_POST(self):
|
||
|
|
buffer = self.rfile.read(int(self.headers['Content-Length']))
|
||
|
|
response = extensionCall(buffer)
|
||
|
|
if response:
|
||
|
|
self.send_response(200)
|
||
|
|
else:
|
||
|
|
self.send_response(500)
|
||
|
|
self.send_header('Content-type', 'application/json')
|
||
|
|
self.end_headers()
|
||
|
|
self.wfile.write(response)
|
||
|
|
|
||
|
|
httpd = HTTPServer(server_address, PostHTTPRequestHandler)
|
||
|
|
httpd.serve_forever()
|