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