Implement basic upload and listing

2 files changed, 81 insertions, 15 deletions

diff --git a/backend/config.py b/backend/config.py
index 6cfab69..3f70c39 100644
--- a/backend/config.py
+++ b/backend/config.py
@@ -1 +1,5 @@
 homepage = "https://stumpf.es/"
+fileroot = "/media/disk1/files.stumpf.es/" 
+global_limit = 0
+user_subdirs = True
+database = "cargohold.db3"
diff --git a/backend/main.py b/backend/main.py
index ff703be..716ca13 100644
--- a/backend/main.py
+++ b/backend/main.py
@@ -1,33 +1,95 @@
-import HTTP
 import json
 import config
+import cgi
+import os
+import sqlite3
+import mimetypes
 
-def playout(filename):
-	return ["", [('Content-Type','text/html'), ("X-Accel-Redirect", filename)], None]
+def playout(filename, content = "text/html"):
+	return ["", [('Content-Type', content if content else "application/binary"), ("X-Accel-Redirect", filename)], None]
 
 def home():
 	return ["", [('Content-Type','text/html'), ("Location", config.homepage)], "302 Home"]
 
+def target_filename_internal(session, file):
+	target = ""
+	if not file:
+		file = ""
+	if config.user_subdirs:
+		target = session["user"] + "/"
+	return target + session["path"] + "/" + file
+	
+
+def target_filename(session, file):
+	return config.fileroot + target_filename_internal(session, file)
+
+def resolve_alias(alias):
+	session = None
+	db = sqlite3.connect(config.database, check_same_thread = False)
+	cursor = db.cursor()
+	cursor.execute("SELECT user, real, access, storage FROM aliases WHERE alias=:alias", {"alias": alias})
+	data = cursor.fetchone()
+	if data:
+		session = {"user": data[0], "path": data[1], "access": data[2], "limit": data[3]}
+	else:
+		print("Unknown alias " + alias)
+	db.close()
+	return session
+
+def listing(session):
+	target = target_filename(session, None)
+	files = os.listdir(target)
+	return [json.dumps(files), [('Content-Type','application/json')], "200 OK"]
+
+def upload(session, post):
+	if post["file"].filename:
+		target = target_filename(session, os.path.basename(post["file"].filename))
+		try:
+			open(target, 'wb').write(post["file"].file.read())
+			print("Uploaded " + target)
+			return ["", [('Content-Type','text/html')], "200 OK"]
+		except PermissionError as e:
+			print("Failed to store " + target + ": " + e.msg)
+			return ["", [('Content-Type','text/html')], "500 Error"]
+	else:
+		print("Upload failed")
+		return ["", [('Content-Type','text/html')], "500 Error"]
+
 def route(path, env, session, post):
-	if path[0] == "8cabc1fce52bcb565ae203267ce7e73f69a9272e":
-		return playout("interface/listing.htm")
-	if path[0] == "test":
-		return playout("interface/listing.htm")
+	# Get mapped user/path/limits
+	session = resolve_alias(path[0])
+
+	if not session:
+		return home()
+
+	print(json.dumps(session))
+
+	if len(path) > 1 and path[1] == "upload":
+		return upload(session, post)
+
+	if len(path) > 1 and path[1] == "listing":
+		return listing(session)
+
+	if len(path) > 1 and path[1] == "file":
+		print("/data/" + target_filename_internal(session, path[2]))
+		return playout("/data/" + target_filename_internal(session, path[2]), mimetypes.guess_type(path[2])[0])
 
-	# Default path
-	return home()
+	return playout("/interface/listing.htm")
 
 def handle_request(env, response):
 	path = env.get('PATH_INFO', '').lstrip('/').split('/')
-	post = {}
+	post = None
 	headers = []
+	session = None
 
 	# Read POST data
 	try:
-		content_length = int(env.get('CONTENT_LENGTH', '0'))
-		post_raw = env["wsgi.input"].read(content_length).decode('utf-8')
-		if env.get('CONTENT_TYPE', '') == "multipart/form-data":
-			post = HTTP.formdata(post_raw)
+		if env.get('CONTENT_TYPE', '').startswith("multipart/form-data"):
+			post = cgi.FieldStorage(environ=env, fp=env['wsgi.input'], keep_blank_values=True)
+		else:
+			content_length = int(env.get('CONTENT_LENGTH', '0'))
+			post_raw = env["wsgi.input"].read(content_length).decode('utf-8')
+			post = post_raw
 	except ValueError as e:
 		post = None
 
@@ -35,4 +97,4 @@ def handle_request(env, response):
 	data, addtl_headers, code = route(path, env, session, post)
 	headers.extend(addtl_headers)
 	response(code if code else "200 OK", headers)
-	return [bytes(data)]
+	return [bytes(data, "utf-8")]