Initial implementation of #92. Pending tests.

pinnwand/handler/website.py

@@ -1,5 +1,7 @@
 import logging
 import binascii
+import io
+import zipfile
 
 from typing import Any
 from datetime import datetime
@@ -370,6 +372,55 @@ async def get(self, file_id: str) -> None:  # type: ignore
             self.write(binascii.hexlify(file.raw.encode("utf8")))
 
 
+class PasteDownload(Base):
+    """Download an entire paste."""
+    async def get(self, paste_id: str) -> None:  #type: ignore
+        """Get all files from the database and download them as a zipfile."""
+
+        with database.session() as session:
+            paste = (
+                session.query(database.Paste)
+                .filter(database.Paste.slug == paste_id)
+                .first()
+            )
+
+            if not paste:
+                raise tornado.web.HTTPError(404)
+
+            if paste.exp_date < datetime.now():
+                session.delete(paste)
+                session.commit()
+
+                log.warn(
+                    "FileRaw.get: paste was expired, is your cronjob running?"
+                )
+
+                raise tornado.web.HTTPError(404)
+
+            data = io.BytesIO()
+
+            with zipfile.ZipFile(data, "x") as zf:
+                for file in paste.files:
+                    if file.filename:
+                        filename = (
+                            f"{utility.filename_clean(file.filename)}-{file.slug}.txt"
+                        )
+                    else:
+                        filename = f"{file.slug}.txt"
+
+                    zf.writestr(
+                        filename,
+                        file.raw)
+
+            data.seek(0)
+
+            self.set_header("Content-Type", "application/zip")
+            self.set_header(
+                "Content-Disposition", f"attachment; filename={paste.slug}.zip"
+            )
+            self.write(data.read())
+
+
 class FileDownload(Base):
     """Download a file."""
 

pinnwand/http.py

@@ -1,5 +1,6 @@
 import logging
 import secrets
+import zipfile
 
 from typing import Any, List
 
@@ -23,6 +24,8 @@ def make_application() -> tornado.web.Application:
         (r"/([A-Z2-7]+)(?:#.+)?/hex", handler.website.FileHex),
         (r"/download/([A-Z2-7]+)(?:#.+)?", handler.website.FileDownload),
         (r"/([A-Z2-7]+)(?:#.+)?/download", handler.website.FileDownload),
+        (r"/download-archive/([A-Z2-7]+)(?:#.+)?", handler.website.PasteDownload),
+        (r"/([A-Z2-7]+)(?:#.+)?/download-archive", handler.website.PasteDownload),
         (r"/remove/([A-Z2-7]+)", handler.website.Remove),
     ]
 

pinnwand/template/show.html

@@ -12,7 +12,8 @@
             <a href="/remove/{{ paste.removal }}">Remove now</a>.
         {% end %}
 
-        <a href="/repaste/{{ paste.slug }}">Repaste</a> this paste.
+        <a href="/repaste/{{ paste.slug }}">Repaste</a>, or
+        <a href="/download-archive/{{ paste.slug }}">download</a> this paste.
 
         <button class="btn-link" id="toggle-word-wrap">Toggle word wrap</button>.