giuliomoro@15: #!/usr/bin/python
giuliomoro@15: 
giuliomoro@15: # Detect the Python version to switch code between 2.x and 3.x
giuliomoro@15: # http://stackoverflow.com/questions/9079036/detect-python-version-at-runtime
giuliomoro@15: import sys
giuliomoro@15: 
giuliomoro@15: from os import walk
giuliomoro@15: from os import path
giuliomoro@15: from os import listdir
giuliomoro@15: import inspect
giuliomoro@15: import os
giuliomoro@15: import pickle
giuliomoro@15: import datetime
giuliomoro@15: 
giuliomoro@15: if sys.version_info[0] == 2:
giuliomoro@15:     # Version 2.x
giuliomoro@15:     import BaseHTTPServer
giuliomoro@15:     import urllib2
giuliomoro@15:     import urlparse
giuliomoro@15: elif sys.version_info[0] == 3:
giuliomoro@15:     # Version 3.x
giuliomoro@15:     from http.server import BaseHTTPRequestHandler, HTTPServer
giuliomoro@15:     import urllib as urllib2
giuliomoro@15: 
giuliomoro@15: # Go to right folder. 
giuliomoro@15: scriptdir = os.path.dirname(os.path.abspath(inspect.getfile(inspect.currentframe()))) # script directory
giuliomoro@15: os.chdir(scriptdir) # does this work?
giuliomoro@15: 
giuliomoro@15: PSEUDO_PATH = '../tests/'
giuliomoro@15: pseudo_files = []
giuliomoro@15: for filename in listdir(PSEUDO_PATH):
giuliomoro@15:     print(filename)
giuliomoro@15:     if filename.endswith('.xml'):
giuliomoro@15:         pseudo_files.append(filename)
giuliomoro@15: 
giuliomoro@15: curSaveIndex = 0;
giuliomoro@15: curFileName = 'test-0.xml'
giuliomoro@15: while(path.isfile('../saves/'+curFileName)):
giuliomoro@15:     curSaveIndex += 1;
giuliomoro@15:     curFileName = 'test-'+str(curSaveIndex)+'.xml'
giuliomoro@15: 
giuliomoro@15: if len(pseudo_files) > 0:
giuliomoro@15:     pseudo_index = curSaveIndex % len(pseudo_files)
giuliomoro@15: else:
giuliomoro@15:     pseudo_index = 0
giuliomoro@15: 
giuliomoro@15: print('URL: http://localhost:8000/index.html')
giuliomoro@15: 
giuliomoro@15: def send404(s):
giuliomoro@15:     s.send_response(404)
giuliomoro@15:     s.send_header("Content-type", "text/html")
giuliomoro@15:     s.end_headers()
giuliomoro@15: 	
giuliomoro@15: def processFile(s):
giuliomoro@15:     if sys.version_info[0] == 2:
giuliomoro@15:         s.path = s.path.rsplit('?')
giuliomoro@15:         s.path = s.path[0]
giuliomoro@15:         s.path = s.path[1:len(s.path)]
giuliomoro@15:         st = s.path.rsplit(',')
giuliomoro@15:         lenSt = len(st)
giuliomoro@15:         fmt = st[lenSt-1].rsplit('.')
giuliomoro@15:         fpath = "../"+urllib2.unquote(s.path)
giuliomoro@15:         size = path.getsize(fpath)
giuliomoro@15:         fileDump = open(fpath)
giuliomoro@15:         s.send_response(200)
giuliomoro@15: 
giuliomoro@15:         if (fmt[1] == 'html'):
giuliomoro@15:             s.send_header("Content-type", 'text/html')
giuliomoro@15:         elif (fmt[1] == 'css'):
giuliomoro@15:             s.send_header("Content-type", 'text/css')
giuliomoro@15:         elif (fmt[1] == 'js'):
giuliomoro@15:             s.send_header("Content-type", 'application/javascript')
giuliomoro@15:         else:
giuliomoro@15:             s.send_header("Content-type", 'application/octet-stream')
giuliomoro@15:         s.send_header("Content-Length", size)
giuliomoro@15:         s.end_headers()
giuliomoro@15:         s.wfile.write(fileDump.read())
giuliomoro@15:         fileDump.close()
giuliomoro@15:     elif sys.version_info[0] == 3:
giuliomoro@15:         s.path = s.path.rsplit('?')
giuliomoro@15:         s.path = s.path[0]
giuliomoro@15:         s.path = s.path[1:len(s.path)]
giuliomoro@15:         st = s.path.rsplit(',')
giuliomoro@15:         lenSt = len(st)
giuliomoro@15:         fmt = st[lenSt-1].rsplit('.')
giuliomoro@15:         fpath = "../"+urllib2.parse.unquote(s.path)
giuliomoro@15:         s.send_response(200)
giuliomoro@15:         if (fmt[1] == 'html'):
giuliomoro@15:             s.send_header("Content-type", 'text/html')
giuliomoro@15:             fileDump = open(fpath, encoding='utf-8')
giuliomoro@15:             fileBytes = bytes(fileDump.read(), "utf-8")
giuliomoro@15:             fileDump.close()
giuliomoro@15:         elif (fmt[1] == 'css'):
giuliomoro@15:             s.send_header("Content-type", 'text/css')
giuliomoro@15:             fileDump = open(fpath, encoding='utf-8')
giuliomoro@15:             fileBytes = bytes(fileDump.read(), "utf-8")
giuliomoro@15:             fileDump.close()
giuliomoro@15:         elif (fmt[1] == 'js'):
giuliomoro@15:             s.send_header("Content-type", 'application/javascript')
giuliomoro@15:             fileDump = open(fpath, encoding='utf-8')
giuliomoro@15:             fileBytes = bytes(fileDump.read(), "utf-8")
giuliomoro@15:             fileDump.close()
giuliomoro@15:         else:
giuliomoro@15:             s.send_header("Content-type", 'application/octet-stream')
giuliomoro@15:             fileDump = open(fpath, 'rb')
giuliomoro@15:             fileBytes = fileDump.read()
giuliomoro@15:             fileDump.close()
giuliomoro@15:         s.send_header("Content-Length", len(fileBytes))
giuliomoro@15:         s.end_headers()
giuliomoro@15:         s.wfile.write(fileBytes)
giuliomoro@15: 
giuliomoro@15: def keygen(s):
giuliomoro@15: 	reply = ""
giuliomoro@15: 	options = s.path.rsplit('?')
giuliomoro@15: 	options = options[1].rsplit('=')
giuliomoro@15: 	key = options[1]
giuliomoro@15: 	print("Registered key "+key)
giuliomoro@15: 	if os.path.isfile("saves/save-"+key+".xml"):
giuliomoro@15: 		reply = "<response><state>NO</state><key>"+key+"</key></response>"
giuliomoro@15: 	else:
giuliomoro@15: 		reply = "<response><state>OK</state><key>"+key+"</key></response>"
giuliomoro@15: 	s.send_response(200)
giuliomoro@15: 	s.send_header("Content-type", "application/xml")
giuliomoro@15: 	s.end_headers()
giuliomoro@15: 	s.wfile.write(reply)
giuliomoro@15: 	file = open("../saves/save-"+key+".xml",'w')
giuliomoro@15: 	file.write("<waetresult key="+key+"/>")
giuliomoro@15: 	file.close();
giuliomoro@15: 
giuliomoro@15: def saveFile(self):
giuliomoro@15:     global curFileName
giuliomoro@15:     global curSaveIndex
giuliomoro@15:     options = self.path.rsplit('?')
giuliomoro@15:     options = options[1].rsplit('=')
giuliomoro@15:     key = options[1]
giuliomoro@15:     varLen = int(self.headers['Content-Length'])
giuliomoro@15:     postVars = self.rfile.read(varLen)
giuliomoro@15:     print("Saving file key "+key)
giuliomoro@15:     file = open('../saves/save-'+key+'.xml','w')
giuliomoro@15:     file.write(postVars)
giuliomoro@15:     file.close()
giuliomoro@15:     try:
giuliomoro@15:         wbytes = os.path.getsize('../saves/save-'+key+'.xml')
giuliomoro@15:     except OSError:
giuliomoro@15:         self.send_response(200)
giuliomoro@15:         self.send_header("Content-type", "text/xml")
giuliomoro@15:         self.end_headers()
giuliomoro@15:         self.wfile.write('<response state="error"><message>Could not open file</message></response>')
giuliomoro@15:     self.send_response(200)
giuliomoro@15:     self.send_header("Content-type", "text/xml")
giuliomoro@15:     self.end_headers()
giuliomoro@15:     self.wfile.write('<response state="OK"><message>OK</message><file bytes="'+str(wbytes)+'">"saves/'+curFileName+'"</file></response>')
giuliomoro@15:     curSaveIndex += 1
giuliomoro@15:     curFileName = 'test-'+str(curSaveIndex)+'.xml'
giuliomoro@15: 
giuliomoro@15: def http_do_HEAD(s):
giuliomoro@15:     s.send_response(200)
giuliomoro@15:     s.send_header("Content-type", "text/html")
giuliomoro@15:     s.end_headers()
giuliomoro@15: 
giuliomoro@15: def http_do_GET(request):
giuliomoro@15:     if(request.client_address[0] == "127.0.0.1"):
giuliomoro@15:         if (request.path == "/favicon.ico"):
giuliomoro@15:             send404(request)
giuliomoro@15:         elif (request.path.split('?',1)[0] == "/php/keygen.php"):
giuliomoro@15:             keygen(request);
giuliomoro@15:         else:
giuliomoro@15:             request.path = request.path.split('?',1)[0]
giuliomoro@15:             if (request.path == '/'):
giuliomoro@15:                 request.path = '/index.html'
giuliomoro@15:             elif (request.path == '/pseudo.xml'):
giuliomoro@15:                 request.path = '/'+PSEUDO_PATH + pseudo_files[pseudo_index]
giuliomoro@15:                 print(request.path)
giuliomoro@15:                 pseudo_index += 1
giuliomoro@15:                 pseudo_index %= len(pseudo_files)
giuliomoro@15:             processFile(request)
giuliomoro@15:     else:
giuliomoro@15:         send404(request)
giuliomoro@15: 
giuliomoro@15: def http_do_POST(request):
giuliomoro@15:     if(request.client_address[0] == "127.0.0.1"):
giuliomoro@15:         if (request.path.rsplit('?',1)[0] == "/save" or request.path.rsplit('?',1)[0] == "/php/save.php"):
giuliomoro@15:             saveFile(request)
giuliomoro@15:         else:
giuliomoro@15:             send404(request)
giuliomoro@15: 
giuliomoro@15: if sys.version_info[0] == 2:
giuliomoro@15:     class MyHandler(BaseHTTPServer.BaseHTTPRequestHandler):
giuliomoro@15:         def do_HEAD(s):
giuliomoro@15:             http_do_HEAD(s)
giuliomoro@15:         def do_GET(request):
giuliomoro@15:             http_do_GET(request)
giuliomoro@15:         def do_POST(request):
giuliomoro@15:             http_do_POST(request)
giuliomoro@15:     def run(server_class=BaseHTTPServer.HTTPServer,handler_class=MyHandler):
giuliomoro@15:         server_address = ('', 8000)
giuliomoro@15:         httpd = server_class(server_address, handler_class)
giuliomoro@15:         httpd.serve_forever()
giuliomoro@15:     run()
giuliomoro@15: elif sys.version_info[0] == 3:
giuliomoro@15:     class MyHandler(BaseHTTPRequestHandler):
giuliomoro@15:         def do_HEAD(s):
giuliomoro@15:             send404(s)
giuliomoro@15:         def do_GET(request):
giuliomoro@15:             http_do_GET(request)
giuliomoro@15:         def do_POST(request):
giuliomoro@15:             http_do_POST(request)
giuliomoro@15:     def run(server_class=HTTPServer,handler_class=MyHandler):
giuliomoro@15:         server_address = ('', 8000)
giuliomoro@15:         httpd = server_class(server_address, handler_class)
giuliomoro@15:         httpd.serve_forever()
giuliomoro@15:     run()