From 161b5710cdaa5695a803982fc7988d25550c369e Mon Sep 17 00:00:00 2001
From: www <bofh@fet.at>
Date: Mon, 28 Dec 2020 00:50:17 +0000
Subject: [PATCH] internal files

---
 internfiles/__init__.py | 69 +++++++++++++++++++++++++++++++++++++++++
 internfiles/pth.py      | 38 +++++++++++++++++++++++
 test2.py                |  2 ++
 3 files changed, 109 insertions(+)
 create mode 100644 internfiles/__init__.py
 create mode 100644 internfiles/pth.py

diff --git a/internfiles/__init__.py b/internfiles/__init__.py
new file mode 100644
index 0000000..8d7c864
--- /dev/null
+++ b/internfiles/__init__.py
@@ -0,0 +1,69 @@
+from flask import Blueprint, make_response, render_template
+import os
+from slugify import slugify
+import re
+from flask import redirect, url_for, send_from_directory
+from .pth import pth
+
+
+bp = Blueprint("internfiles", __name__, url_prefix="/internfiles")
+
+
+class Folder:
+    def __init__(self, path="", p=""):
+        self.path = pth(path)
+        self.p = p
+        self.files = []
+        self.dirs = []
+        for it in os.scandir(path):
+            if it.is_file():
+                self.files.append(pth(it.name))
+            if it.is_dir():
+                self.dirs.append(pth(it.name))
+
+    @property
+    def bookmarks(self):
+        f = self.p.split("/")
+        # return [i for i in range(len(f)-1) ]
+        b = ["/".join(f[0:i]) for i in range(len(f))]
+        return b
+
+    def __repr__(self):
+        return f"Folder({self.path})"
+
+
+def load_file_and_folder(path: str = ""):
+    f, directory = (None, None)
+    path = pth(path)
+    datapath = pth("/mnt/save/daten")
+    filepath = datapath + path
+
+    if not str(path) == "" and not (filepath in datapath):
+        return None, None
+
+    if os.path.isfile(filepath):
+        f = filepath
+
+    for d in [path, path - 1, pth("")]:
+        if os.path.isdir(datapath + d):
+            directory = datapath + d
+            break
+    return f, Folder(directory, d)
+
+
+@bp.route("/")
+@bp.route("/<path:path>")
+def web(path=""):
+    f, d = load_file_and_folder(path)
+    if not f and not d.p == path:
+        return redirect(url_for("internfiles.web", path=d.p), code=302)
+    text = None
+    if f and f.endswith(".txt"):
+        text = os.path.abspath(f)
+
+        with open(os.path.abspath(f), "r") as fh:
+            text = fh.read()
+            text = re.sub("\\n\s*\\n", "<br>", text)
+    elif f:
+        return send_from_directory(str(f - 1), str(f[-1]))
+    return render_template("internfiles.html", file=f, dir=d, text=text, path=pth(d.p))
diff --git a/internfiles/pth.py b/internfiles/pth.py
new file mode 100644
index 0000000..84f5f8c
--- /dev/null
+++ b/internfiles/pth.py
@@ -0,0 +1,38 @@
+import os
+from slugify import slugify
+
+
+class pth(str):
+    def __new__(cls, value, *args, **kw):
+        if not value:
+            value = ""
+        value = value.rstrip("/")
+        return str.__new__(cls, value, *args, **kw)
+
+    def __add__(self, add):
+        if isinstance(add, str) or isinstance(add, pth):
+            return pth(os.path.join(str(self), str(add)))  # .rstrip("/")
+        else:
+            return super().__add__(add)
+
+    def __sub__(self, sub):
+        if isinstance(sub, int):
+            return pth("/".join(self.split("/")[:-sub]))
+        else:
+            return super().__sub__(sub)
+
+    def __contains__(self, path):
+        directory = os.path.join(os.path.realpath(str(self)), "")
+        file = os.path.realpath(str(path))
+        return os.path.commonprefix([file, directory]) == directory
+
+    def __getitem__(self, key):
+        return str(self).split("/")[key]
+
+    def slugify(self):
+        s = self.split("/")
+        s = s[:-1]
+        s2 = s[-1]
+        if "." in s[-1]:
+            s2 = ".".join([slugify(p) for p in s2])
+        return "/".join([slugify(p) for p in s] + [s2])
diff --git a/test2.py b/test2.py
index 31f509f..0dd8c2f 100644
--- a/test2.py
+++ b/test2.py
@@ -17,6 +17,7 @@ from solrfet2020 import SolrFet2020
 from slugify import slugify
 
 import wiki
+import internfiles
 
 #logging.basicConfig(level=logging.INFO)
 URL_HOSTNAME='bot.2020.fet.at'
@@ -65,6 +66,7 @@ def auth_user(user_id, token):
     return "development"
 
 app.register_blueprint(wiki.bp)
+app.register_blueprint(internfiles.bp)
 
 def download_file(url):
     local_filename = url.split('/')[-1]