From 40c7c325e7315aa9955016d01cdc9ae32c1e8366 Mon Sep 17 00:00:00 2001
From: ranomier <ranomier@fragomat.net>
Date: Fri, 13 Dec 2024 20:02:45 +0100
Subject: [PATCH] added argparse and upload parameter

---
 mserve | 42 +++++++++++++++++++++++++++++++-----------
 1 file changed, 31 insertions(+), 11 deletions(-)

diff --git a/mserve b/mserve
index 7bbb7bf..563f193 100755
--- a/mserve
+++ b/mserve
@@ -2,19 +2,39 @@
 # docstring=serve files with http.
 
 import subprocess as sp
+import argparse
 
+CFG = {
+    "base": [
+        "miniserve",
+        "--auth=guest:Super9000!",
+        "--color-scheme-dark=monokai",
+        "--qrcode",
+        "--dirs-first",
+        "--hide-version-footer",
+        "--show-wget-footer",
+        "--title=Rano's quick share",
+    ],
+    "folder": "./"
+}
 
-def run():
-    sp.run(["miniserve",
-           "--auth=guest:Super9000!",
-           "--color-scheme-dark=monokai",
-           "--qrcode",
-           "--dirs-first",
-           "--hide-version-footer",
-           "--show-wget-footer",
-           "--title=Rano's quick share"
-            ])
+def main():
+    ap = argparse.ArgumentParser()
+    ap.add_argument("-u", "--upload", help="enable upload", action="store_true")
+    args = ap.parse_args()
+
+    cmd = CFG["base"]
+
+    if args.upload:
+        cmd.extend(["--upload-files=./", "--overwrite-files"])
+
+    cmd.append(CFG["folder"])
+
+    try:
+        sp.run(cmd)
+    except KeyboardInterrupt:
+        print("\ninterrupted by user")
 
 if __name__ == "__main__":
-    run()
+    main()
 # implement a switch for easy --upload-files=DIR