Add: view function to serve files

backend/apps/files/views.py

@@ -1,3 +1,81 @@
-from django.shortcuts import render
+import os
 
-# Create your views here.
+from django.conf import settings
+from django.shortcuts import get_object_or_404
+from django.http import HttpResponse, FileResponse
+
+from sorl.thumbnail import get_thumbnail
+
+from .models import PostFileModel
+
+
+def serve_content_file(request, file_hash):
+    """
+    View function to serve content files for download or inline viewing
+    """
+
+    print(f"Requesting file: {file_hash}")
+
+    download = request.GET.get("d")
+
+    thumbnail = request.GET.get("t")
+
+    try:
+        obj_file = get_object_or_404(PostFileModel, hash_blake3=file_hash)
+        file_name = obj_file.name.first().filename
+        file_type = obj_file.file_type
+        source_file = obj_file.file
+
+        if thumbnail:
+            if file_type != "image":
+                source_file = obj_file.thumbnail
+
+        if thumbnail == "sx":
+            file_name += ".thumb_64.jpg"
+            thumbnail_file = get_thumbnail(source_file.path, "64", upscale=False)
+            file = open(
+                os.path.abspath(os.path.join(settings.MEDIA_ROOT, thumbnail_file.name)),
+                "rb",
+            )
+        elif thumbnail == "sm":
+            file_name += ".thumb_256.jpg"
+            thumbnail_file = get_thumbnail(source_file.path, "256", upscale=False)
+            file = open(
+                os.path.abspath(os.path.join(settings.MEDIA_ROOT, thumbnail_file.name)),
+                "rb",
+            )
+        elif thumbnail == "md":
+            file_name += ".thumb_748.jpg"
+            thumbnail_file = get_thumbnail(source_file.path, "748", upscale=False)
+            file = open(
+                os.path.abspath(os.path.join(settings.MEDIA_ROOT, thumbnail_file.name)),
+                "rb",
+            )
+        elif thumbnail == "lg":
+            file_name += ".thumb_1024.jpg"
+            thumbnail_file = get_thumbnail(source_file.path, "1024", upscale=False)
+            file = open(
+                os.path.abspath(os.path.join(settings.MEDIA_ROOT, thumbnail_file.name)),
+                "rb",
+            )
+        elif thumbnail == "xl":
+            file_name += ".thumb_2048.jpg"
+            thumbnail_file = get_thumbnail(source_file.path, "2048", upscale=False)
+            file = open(
+                os.path.abspath(os.path.join(settings.MEDIA_ROOT, thumbnail_file.name)),
+                "rb",
+            )
+        else:
+            file = source_file.file
+
+        response = FileResponse(file)
+        if download == "1":
+            response["Content-Disposition"] = f'attachment; filename="{file_name}"'
+        else:
+            response["Content-Disposition"] = f'inline; filename="{file_name}"'
+
+        return response
+
+    except Exception as e:
+        print(f"Error serving file: {e}")
+        return HttpResponse("File not found", status=404)