File page

perguth · perguth · commit 6049738efa3c · 2025-01-29T16:31:37.000Z
diff --git a/islands/Decrypt.tsx b/islands/Decrypt.tsx
@@ -1,7 +1,11 @@
 // islands/Decrypt.tsx
 import { useEffect, useState } from 'preact/hooks'
 
-export default function Decrypt() {
+interface DecryptProps {
+  fileExists: boolean | undefined
+}
+
+export default function Decrypt({ fileExists }: DecryptProps) {
   const [status, setStatus] = useState('')
   const [error, setError] = useState('')
   const [downloaded, setDownloaded] = useState(false)
@@ -15,6 +19,10 @@ export default function Decrypt() {
           return
         }
 
+        if (fileExists === false) {
+          throw new Error('File not found or expired')
+        }
+
         // Extract data from hash
         const key = hash.slice(0, 64)
         const iv = hash.slice(64, 96)
@@ -23,8 +31,8 @@ export default function Decrypt() {
 
         setStatus('Decrypting file...')
 
-        // Fetch encrypted file using key as filename
-        const response = await fetch(`/${key}.enc`)
+        // Get file from backend using the current URL path
+        const response = await fetch(`/${key}`)
         if (!response.ok) {
           throw new Error(
             response.status === 404
@@ -93,7 +101,7 @@ export default function Decrypt() {
       const key = hash.slice(0, 64)
       const iv = hash.slice(64, 96)
 
-      const response = await fetch(`/${key}.enc`, {
+      const response = await fetch(`/${key}`, {
         method: 'DELETE',
         headers: {
           'Authorization': `Bearer ${key}${iv}`,
@@ -102,7 +110,7 @@ export default function Decrypt() {
 
       if (!response.ok) throw new Error('Delete failed')
       setStatus('✓ File deleted')
-      setDownloaded(false) // Hide delete button after successful deletion
+      setDownloaded(false)
     } catch (_err) {
       setError('Failed to delete file')
     }
diff --git a/routes/d/[hash].tsx b/routes/d/[hash].tsx
@@ -1,8 +1,69 @@
 // routes/d/[hash].tsx
 import { Head } from '$fresh/runtime.ts'
-import Decrypt from '@/islands/Decrypt.tsx'
+import { Handlers } from '$fresh/server.ts'
+import { recordDownload } from '../../utils/stats.ts'
+import Decrypt from '../../islands/Decrypt.tsx'
 
-export default function DecryptPage() {
+interface FileData {
+  content: Uint8Array
+  size: number
+  created: string
+  deletionKey: string
+}
+
+const kv = await Deno.openKv()
+
+async function getFileData(key: string): Promise<FileData | null> {
+  const res = await kv.get<FileData>(['files', key])
+  return res.value
+}
+
+// This handler will serve the file content
+export const handler: Handlers = {
+  async GET(req, ctx) {
+    const { hash } = ctx.params
+    const key = hash.slice(0, 64) // First 64 chars are the key
+
+    // Check if file exists
+    const fileData = await getFileData(key)
+    if (!fileData) {
+      // Fall back to UI to show error if file not found
+      return ctx.render()
+    }
+
+    // If direct file request, serve file content
+    const url = new URL(req.url)
+    if (url.searchParams.has('download')) {
+      await recordDownload(fileData.size)
+
+      const stream = new ReadableStream({
+        start(controller) {
+          controller.enqueue(fileData.content)
+          controller.close()
+        },
+      })
+
+      return new Response(stream, {
+        headers: {
+          'content-type': 'application/octet-stream',
+          'cache-control': 'no-store',
+          'content-length': fileData.size.toString(),
+        },
+      })
+    }
+
+    // Otherwise render the UI
+    return ctx.render({
+      fileExists: true,
+      created: fileData.created,
+      size: fileData.size,
+    })
+  },
+}
+
+export default function DecryptPage(
+  props: { data: { fileExists?: boolean; created?: string; size?: number } },
+) {
   return (
     <>
       <Head>
@@ -12,10 +73,20 @@ export default function DecryptPage() {
         <div class='max-w-md w-full space-y-8'>
           <div>
             <h1 class='mt-6 text-center text-3xl font-extrabold text-gray-900'>
-              Decrypting your file
+              {props.data?.fileExists ? 'Decrypt your file' : 'File not found'}
             </h1>
+            {props.data?.fileExists && props.data?.created && (
+              <p class='mt-2 text-center text-sm text-gray-600'>
+                Uploaded {new Date(props.data.created).toLocaleString()}
+              </p>
+            )}
+            {props.data?.fileExists && props.data?.size && (
+              <p class='mt-2 text-center text-sm text-gray-600'>
+                Size: {(props.data.size / 1024).toFixed(1)} KB
+              </p>
+            )}
           </div>
-          <Decrypt />
+          <Decrypt fileExists={props.data?.fileExists} />
         </div>
       </div>
     </>
