Custom listdir

SuperZombi · SuperZombi · commit 8e8cff11e3e0 · 2025-01-11T22:11:37.000+02:00
diff --git a/src/HTTPServer.py b/src/HTTPServer.py
@@ -96,7 +96,7 @@ def do_GET(self):
 			if SETTINGS.get("root") == "index":
 				self.path = URL.path + 'index.html'
 			else:
-				return self.copyfile(self.list_directory(path), self.wfile)
+				return self.copyfile(list_directory(path, URL.path), self.wfile)
 		else:
 			if not os.path.exists(path):
 				self.path = URL.path + '.html'
diff --git a/src/utils.py b/src/utils.py
@@ -2,6 +2,8 @@
 import socket
 import json
 import tkinter as tk
+import io
+import urllib
 
 
 def resource_path(relative_path):
@@ -31,6 +33,54 @@ def get_host(host=""):
 		return "127.0.0.1"
 	return host
 
+def get_files_and_folders(path):
+	files = []
+	folders = []
+	for item in os.listdir(path):
+		full_path = os.path.join(path, item)
+		if os.path.isdir(full_path):
+			folders.append(item)
+		elif os.path.isfile(full_path):
+			files.append(item)
+	return files, folders
+
+def list_directory(path, urlpath):
+	files, folders = get_files_and_folders(path)
+
+	folders_html = ""
+	for folder in folders:
+		folders_html += f"<li>📁<a href='{folder}/'>{folder}/</a></li>"
+
+	files_html = ""
+	for file in files:
+		folders_html += f"<li>🗎<a href='{file}'>{file}</a></li>"
+
+	nothing = "<p>¯\\_(ツ)_/¯</p>"
+	html = f"""
+	<!DOCTYPE html>
+	<html>
+	<head>
+		<title>Directory listing</title>
+		<meta http-equiv='Content-Type' content='text/html; charset=utf-8'/>
+		<meta name='viewport' content='width=device-width'>
+	</head>
+	<body>
+		<h2>{urllib.parse.unquote(urlpath)}</h2>
+		<hr>
+		<ul>
+			{"<li><a href='../'>↩</a></li>" if urlpath != "/" else ""}
+			{nothing if (not folders_html and not files_html) else ""}
+			{folders_html}
+			{files_html}
+		</ul>
+	</body>
+	</html>
+	"""
+	f = io.BytesIO()
+	f.write(bytes(html, 'utf-8'))
+	f.seek(0)
+	return f
+
 class EntryWithPlaceholder(tk.Entry):
 	def __init__(self, master, placeholder, color='grey'):
 		super().__init__(master)
