From 062b794005fc669575db93790209549140a21107 Mon Sep 17 00:00:00 2001
From: Alexandr Karpov <alexandr.d.karpov@gmail.com>
Date: Tue, 25 Apr 2023 11:42:06 +0300
Subject: [PATCH] added json, fixed big files preview

---
 akarpov/files/previews/__init__.py            |  2 +
 .../files/previews/application/__init__.py    |  2 +-
 akarpov/files/previews/application/json.py    | 84 +++++++++++++++++++
 akarpov/files/previews/text/common.py         |  2 +
 akarpov/files/previews/text/html.py           |  2 +
 akarpov/files/previews/text/plain.py          |  2 +
 6 files changed, 93 insertions(+), 1 deletion(-)
 create mode 100644 akarpov/files/previews/application/json.py

diff --git a/akarpov/files/previews/__init__.py b/akarpov/files/previews/__init__.py
index 3420969..9adc404 100644
--- a/akarpov/files/previews/__init__.py
+++ b/akarpov/files/previews/__init__.py
@@ -9,6 +9,7 @@
         "docx": application.docx.view,
         "vnd.oasis.opendocument.text": application.odt.view,
         "x-httpd-php": text.common.view,
+        "json": application.json.view,
     },
     "audio": {
         "aac": audio.basic.view,
@@ -77,6 +78,7 @@
         "oga": audio.oga.view,
         "pdf": application.pdf.view,
         "html": text.html.view,
+        "json": application.json.view,
     }
     | source_code
     | fonts_ext
diff --git a/akarpov/files/previews/application/__init__.py b/akarpov/files/previews/application/__init__.py
index ce34ad6..1fa0548 100644
--- a/akarpov/files/previews/application/__init__.py
+++ b/akarpov/files/previews/application/__init__.py
@@ -1 +1 @@
-from . import doc, docx, odt, pdf, zip  # noqa
+from . import doc, docx, json, odt, pdf, zip  # noqa
diff --git a/akarpov/files/previews/application/json.py b/akarpov/files/previews/application/json.py
new file mode 100644
index 0000000..11a4fd1
--- /dev/null
+++ b/akarpov/files/previews/application/json.py
@@ -0,0 +1,84 @@
+from akarpov.files.models import File
+
+
+def view(file: File):
+    static = """
+    <style>
+    pre {outline: 1px solid #ccc; padding: 5px; margin: 5px; }
+    .string { color: green; }
+    .number { color: darkorange; }
+    .boolean { color: blue; }
+    .null { color: magenta; }
+    .key { color: red; }
+    </style>
+    """
+    if file.file_size < 200 * 1024:
+        req = (
+            f"""
+    getJSON('{file.file.url}',
+    """
+            + """
+    function(err, data) {
+      if (err !== null) {
+        console.log('Something went wrong: ' + err);
+      } else {
+        var str = JSON.stringify(data, undefined, 4);
+        output(syntaxHighlight(str));
+      }
+    });
+    """
+        )
+    else:
+        req = """
+        output("file is too large, download to view")
+        """
+
+    content = (
+        """
+    <div id="json" class="col-auto"></div>
+    <script>
+    function output(inp) {
+        document.getElementById("json").appendChild(document.createElement('pre')).innerHTML = inp;
+    }
+
+    function syntaxHighlight(json) {
+        json = json.replace(/&/g, '&amp;').replace(/</g, '&lt;').replace(/>/g, '&gt;');
+        return json.replace(
+/("(\\u[a-zA-Z0-9]{4}|\\[^u]|[^\\"])*"(\\s*:)?|\b(true|false|null)\b|-?\\d+(?:\\.\\d*)?(?:[eE][+\\-]?\\d+)?)/g,
+        function (match) {
+            var cls = 'number';
+            if (/^"/.test(match)) {
+                if (/:$/.test(match)) {
+                    cls = 'key';
+                } else {
+                    cls = 'string';
+                }
+            } else if (/true|false/.test(match)) {
+                cls = 'boolean';
+            } else if (/null/.test(match)) {
+                cls = 'null';
+            }
+            return '<span class="' + cls + '">' + match + '</span>';
+        });
+    }
+    var getJSON = function(url, callback) {
+        var xhr = new XMLHttpRequest();
+        xhr.open('GET', url, true);
+        xhr.responseType = 'json';
+        xhr.onload = function() {
+          var status = xhr.status;
+          if (status === 200) {
+            callback(null, xhr.response);
+          } else {
+            callback(status, xhr.response);
+          }
+        };
+        xhr.send();
+    };
+    """
+        + req
+        + """
+    </script>
+    """
+    )
+    return static, content
diff --git a/akarpov/files/previews/text/common.py b/akarpov/files/previews/text/common.py
index 8b98e23..2071bd5 100644
--- a/akarpov/files/previews/text/common.py
+++ b/akarpov/files/previews/text/common.py
@@ -30,6 +30,8 @@
 
 
 def view(file: File) -> (str, str):
+    if file.file_size > 10 * 1024 * 1024:
+        return "", "file is too large to view"
     extension = file.file.path.split(".")[-1]
     if extension in language_previews:
         extension = language_previews[extension]
diff --git a/akarpov/files/previews/text/html.py b/akarpov/files/previews/text/html.py
index 7db7e21..c1f36ca 100644
--- a/akarpov/files/previews/text/html.py
+++ b/akarpov/files/previews/text/html.py
@@ -4,6 +4,8 @@
 
 
 def view(file: File) -> (str, str):
+    if file.file_size > 10 * 1024 * 1024:
+        return "", "file is too large to view"
     static = f"""
     <meta property="og:title" content="{file.name}" />
 <link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/highlight.js/11.7.0/styles/atom-one-light.min.css">
diff --git a/akarpov/files/previews/text/plain.py b/akarpov/files/previews/text/plain.py
index 03437e9..b8ac6a0 100644
--- a/akarpov/files/previews/text/plain.py
+++ b/akarpov/files/previews/text/plain.py
@@ -6,6 +6,8 @@
 
 
 def view(file: File) -> (str, str):
+    if file.file_size > 10 * 1024 * 1024:
+        return "", "file is too large to view"
     extension = file.file.path.split(".")[-1]
     if hasattr(text, extension):
         return getattr(text, extension).view(file)