From 2d7dd26882a20c1834eb019ff427e6f3418d0ee4 Mon Sep 17 00:00:00 2001
From: LittleSheep <littlesheep.code@hotmail.com>
Date: Sun, 3 Aug 2025 22:11:31 +0800
Subject: [PATCH] :sparkles: Post with image / media

---
 .../Client/src/components/AttachmentItem.vue  |   2 +
 .../Client/src/components/PostEditor.vue      | 166 +++++++++++++++---
 DysonNetwork.Sphere/Client/vite.config.ts     |   4 +
 3 files changed, 151 insertions(+), 21 deletions(-)

diff --git a/DysonNetwork.Sphere/Client/src/components/AttachmentItem.vue b/DysonNetwork.Sphere/Client/src/components/AttachmentItem.vue
index cd70071..341a17e 100644
--- a/DysonNetwork.Sphere/Client/src/components/AttachmentItem.vue
+++ b/DysonNetwork.Sphere/Client/src/components/AttachmentItem.vue
@@ -4,6 +4,8 @@
       <img src="/image-broken.jpg" class="w-32 h-32 rounded-md" />
     </template>
   </n-image>
+  <audio v-else-if="itemType == 'audio'" :src="remoteSource" controls />
+  <video v-else-if="itemType == 'video'" :src="remoteSource" controls />
 </template>
 
 <script lang="ts" setup>
diff --git a/DysonNetwork.Sphere/Client/src/components/PostEditor.vue b/DysonNetwork.Sphere/Client/src/components/PostEditor.vue
index d1da17d..d5706d3 100644
--- a/DysonNetwork.Sphere/Client/src/components/PostEditor.vue
+++ b/DysonNetwork.Sphere/Client/src/components/PostEditor.vue
@@ -1,30 +1,58 @@
 <template>
-  <div class="flex flex-col gap-3">
-    <pub-select v-model:value="publisher" />
-    <n-input
-      type="textarea"
-      placeholder="What's happended?!"
-      v-model:value="content"
-      @keydown.meta.enter.exact="submit"
-      @keydown.ctrl.enter.exact="submit"
-    />
-    <div class="flex justify-between">
-      <div class="flex gap-2"></div>
-      <n-button type="primary" icon-placement="right" :loading="submitting" @click="submit">
-        Post
-        <template #icon>
-          <n-icon><send-round /></n-icon>
-        </template>
-      </n-button>
+  <n-upload
+    abstract
+    with-credentials
+    @remove="handleRemove"
+    :create-thumbnail-url="createThumbnailUrl"
+    :custom-request="customRequest"
+    v-model:file-list="fileList"
+  >
+    <div class="flex flex-col gap-3">
+      <pub-select v-model:value="publisher" />
+      <n-input
+        type="textarea"
+        placeholder="What's happended?!"
+        v-model:value="content"
+        @keydown.meta.enter.exact="submit"
+        @keydown.ctrl.enter.exact="submit"
+      />
+      <n-upload-file-list />
+      <div class="flex justify-between">
+        <div class="flex gap-2">
+          <n-upload-trigger #="{ handleClick }" abstract>
+            <n-button @click="handleClick">
+              <n-icon><upload-round /></n-icon>
+            </n-button>
+          </n-upload-trigger>
+        </div>
+        <n-button type="primary" icon-placement="right" :loading="submitting" @click="submit">
+          Post
+          <template #icon>
+            <n-icon><send-round /></n-icon>
+          </template>
+        </n-button>
+      </div>
     </div>
-  </div>
+  </n-upload>
 </template>
 
 <script setup lang="ts">
-import { NInput, NButton, NIcon } from 'naive-ui'
+import {
+  NInput,
+  NButton,
+  NIcon,
+  NUpload,
+  NUploadFileList,
+  NUploadTrigger,
+  useMessage,
+  type UploadSettledFileInfo,
+  type UploadCustomRequestOptions,
+  create,
+  type UploadFileInfo,
+} from 'naive-ui'
+import { SendRound, UploadRound } from '@vicons/material'
 import { ref } from 'vue'
-
-import { SendRound } from '@vicons/material'
+import * as tus from 'tus-js-client'
 
 import PubSelect from './PubSelect.vue'
 
@@ -33,6 +61,8 @@ const emits = defineEmits(['posted'])
 const publisher = ref<string | undefined>()
 const content = ref('')
 
+const fileList = ref<UploadFileInfo[]>([])
+
 const submitting = ref(false)
 
 async function submit() {
@@ -44,11 +74,105 @@ async function submit() {
     },
     body: JSON.stringify({
       content: content.value,
+      attachments: fileList.value
+        .filter((e) => e.url != null)
+        .map((e) => e.url!.split('/').reverse()[0]),
     }),
   })
 
   submitting.value = false
   content.value = ''
+  fileList.value = []
   emits('posted')
 }
+
+const messageDisplay = useMessage()
+
+function customRequest({
+  file,
+  headers,
+  withCredentials,
+  onFinish,
+  onError,
+  onProgress,
+}: UploadCustomRequestOptions) {
+  const requestHeaders: Record<string, string> = {}
+  const upload = new tus.Upload(file.file, {
+    endpoint: '/cgi/drive/tus',
+    retryDelays: [0, 3000, 5000, 10000, 20000],
+    removeFingerprintOnSuccess: false,
+    uploadDataDuringCreation: false,
+    metadata: {
+      filename: file.name,
+      'content-type': file.type ?? 'application/octet-stream',
+    },
+    headers: {
+      'X-DirectUpload': 'true',
+      ...requestHeaders,
+      ...headers,
+    },
+    onShouldRetry: () => false,
+    onError: function (error) {
+      if (error instanceof tus.DetailedError) {
+        const failedBody = error.originalResponse?.getBody()
+        if (failedBody != null)
+          messageDisplay.error(`Upload failed: ${failedBody}`, {
+            duration: 10000,
+            closable: true,
+          })
+      }
+      console.error('[DRIVE] Upload failed:', error)
+      onError()
+    },
+    onProgress: function (bytesUploaded, bytesTotal) {
+      onProgress({ percent: (bytesUploaded / bytesTotal) * 100 })
+    },
+    onSuccess: function (payload) {
+      const rawInfo = payload.lastResponse.getHeader('x-fileinfo')
+      const jsonInfo = JSON.parse(rawInfo as string)
+      console.log('[DRIVE] Upload successful: ', jsonInfo)
+      file.url = `/cgi/drive/files/${jsonInfo.id}`
+      file.type = jsonInfo.mime_type
+      onFinish()
+    },
+    onBeforeRequest: function (req) {
+      const xhr = req.getUnderlyingObject()
+      xhr.withCredentials = withCredentials
+    },
+  })
+  upload.findPreviousUploads().then(function (previousUploads) {
+    if (previousUploads.length) {
+      upload.resumeFromPreviousUpload(previousUploads[0])
+    }
+    upload.start()
+  })
+}
+
+function createThumbnailUrl(
+  _file: File | null,
+  fileInfo: UploadSettledFileInfo,
+): string | undefined {
+  if (!fileInfo) return undefined
+  return fileInfo.url ?? undefined
+}
+
+function handleRemove(data: { file: UploadFileInfo; fileList: UploadFileInfo[] }) {
+  if (data.file.url == null) return true
+  const messageReactive = messageDisplay.loading('Deleting files...', {
+    duration: 0,
+  })
+  return new Promise((resolve) => {
+    fetch(`/cgi/drive/files/${data.file.url!.split('/').reverse()[0]}`, { method: 'DELETE' })
+      .then(() => {
+        messageReactive.destroy()
+        messageDisplay.success('File has been deleted')
+        resolve(true)
+      })
+      .catch((err) => {
+        messageReactive.destroy()
+        messageDisplay.error('Unable to delete this file: ' + err)
+        resolve(false)
+      })
+  })
+}
 </script>
diff --git a/DysonNetwork.Sphere/Client/vite.config.ts b/DysonNetwork.Sphere/Client/vite.config.ts
index 387d074..72e9eec 100644
--- a/DysonNetwork.Sphere/Client/vite.config.ts
+++ b/DysonNetwork.Sphere/Client/vite.config.ts
@@ -19,6 +19,10 @@ export default defineConfig({
   },
   server: {
     proxy: {
+      '/api/tus': {
+        target: 'http://localhost:5090',
+        changeOrigin: true,
+      },
       '/api': {
         target: 'http://localhost:5071',
         changeOrigin: true,