From d79011e39abe8dc8b6bc73d601bcdfd8cfcb68b4 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E4=B8=89=E5=92=B2=E6=99=BA=E5=AD=90?= <sxzz@sxzz.moe>
Date: Thu, 24 Nov 2022 01:17:54 +0800
Subject: [PATCH] feat: support attachment uploading

---
 components/common/CommonNotFound.vue     |  2 +-
 components/publish/PublishAttachment.vue | 24 +++++++
 components/publish/PublishWidget.vue     | 90 +++++++++++++++++++++---
 package.json                             |  1 +
 pnpm-lock.yaml                           |  6 ++
 shims.d.ts                               |  1 +
 6 files changed, 115 insertions(+), 9 deletions(-)
 create mode 100644 components/publish/PublishAttachment.vue
 create mode 100644 shims.d.ts

diff --git a/components/common/CommonNotFound.vue b/components/common/CommonNotFound.vue
index d0f0b696..1cf80863 100644
--- a/components/common/CommonNotFound.vue
+++ b/components/common/CommonNotFound.vue
@@ -2,7 +2,7 @@
   <div flex="~ col" items-center>
     <div i-ri:forbid-line text-10 mt10 mb2 />
     <div text-lg>
-      <slot>Not found</slot>
+      <slot>404 Not Found</slot>
     </div>
   </div>
 </template>
diff --git a/components/publish/PublishAttachment.vue b/components/publish/PublishAttachment.vue
new file mode 100644
index 00000000..1e82c07a
--- /dev/null
+++ b/components/publish/PublishAttachment.vue
@@ -0,0 +1,24 @@
+<script setup lang="ts">
+import type { Attachment } from 'masto'
+
+withDefaults(defineProps<{
+  attachment: Attachment
+  alt?: string
+  removable?: boolean
+}>(), {
+  removable: true,
+})
+
+defineEmits<{
+  (evt: 'remove'): void
+}>()
+</script>
+
+<template>
+  <div relative group>
+    <status-attachment :attachment="attachment" w-full />
+    <div absolute right-2 top-2 hover:bg="gray/40" transition-100 p-1 rounded-5 cursor-pointer op-0 group-hover:op-100>
+      <div v-if="removable" i-ri:close-line text-3 @click="$emit('remove')" />
+    </div>
+  </div>
+</template>
diff --git a/components/publish/PublishWidget.vue b/components/publish/PublishWidget.vue
index 0cded01c..b6e4bdf7 100644
--- a/components/publish/PublishWidget.vue
+++ b/components/publish/PublishWidget.vue
@@ -1,5 +1,5 @@
 <script setup lang="ts">
-import type { CreateStatusParamsWithStatus } from 'masto'
+import type { Attachment, CreateStatusParams, CreateStatusParamsWithStatus } from 'masto'
 
 const {
   draftKey,
@@ -19,13 +19,65 @@ function getDefaultStatus(): CreateStatusParamsWithStatus {
     inReplyToId,
   }
 }
-const draft = useLocalStorage<CreateStatusParamsWithStatus>(storageKey, getDefaultStatus())
+let draft = $(useLocalStorage<CreateStatusParamsWithStatus>(storageKey, getDefaultStatus()))
+let attachments = $(useLocalStorage<Attachment[]>(`${storageKey}-attachments`, []))
+const status = $computed(() => {
+  return {
+    ...draft,
+    mediaIds: attachments.map(a => a.id),
+  } as CreateStatusParams
+})
+
+let isUploading = $ref<boolean>(false)
+
+async function handlePaste(evt: ClipboardEvent) {
+  const files = evt.clipboardData?.files
+  if (!files)
+    return
+
+  await uploadAttachments(Array.from(files))
+}
+
+async function pickAttachments() {
+  if (!globalThis.showOpenFilePicker)
+    // TODO: Safari don't support it.
+    return
+
+  const handles = await showOpenFilePicker({
+    multiple: true,
+    // TODO: add more kinds of files: videos, audios
+    types: [{
+      description: 'Images',
+      accept: {
+        'image/*': ['.png', '.gif', '.jpeg', '.jpg', '.webp', '.avif', '.heic'],
+      },
+    }],
+  })
+  const files = await Promise.all(handles.map(handle => handle.getFile()))
+  await uploadAttachments(files)
+}
+
+async function uploadAttachments(files: File[]) {
+  isUploading = true
+  for (const file of files) {
+    const attachment = await masto.mediaAttachments.create({
+      file,
+    })
+    attachments.push(attachment)
+  }
+  isUploading = false
+}
+
+async function removeAttachment(index: number) {
+  attachments.splice(index, 1)
+}
 
 async function publish() {
   try {
     isSending = true
-    await masto.statuses.create(draft.value)
-    draft.value = getDefaultStatus()
+    await masto.statuses.create(status)
+    draft = getDefaultStatus()
+    attachments = []
   }
   finally {
     isSending = false
@@ -33,8 +85,9 @@ async function publish() {
 }
 
 onUnmounted(() => {
-  if (!draft.value.status) {
-    draft.value = undefined
+  if (!draft.status) {
+    // @ts-expect-error draft cannot be undefined
+    draft = undefined
     nextTick(() => {
       localStorage.removeItem(storageKey)
     })
@@ -44,7 +97,7 @@ onUnmounted(() => {
 
 <template>
   <div
-    flex flex-col gap-4
+    flex flex-col gap-3
     :class="isSending ? 'pointer-events-none' : ''"
   >
     <textarea
@@ -52,11 +105,32 @@ onUnmounted(() => {
       :placeholder="placeholder"
       p2 border-rounded w-full h-40
       bg-gray:10 outline-none border="~ base"
+      @paste="handlePaste"
     />
+
+    <div flex="~" gap-2>
+      <button hover:bg-active p2 rounded-5 @click="pickAttachments">
+        <div i-ri:upload-line />
+      </button>
+    </div>
+
+    <div flex="~ col gap-2" max-h-50vh overflow-auto>
+      <publish-attachment
+        v-for="(att, idx) in attachments" :key="att.id"
+        :attachment="att"
+        @remove="removeAttachment(idx)"
+      />
+    </div>
+
+    <div v-if="isUploading" flex gap-2 justify-end items-center>
+      <div op50 i-ri:loader-2-fill animate-spin text-2xl />
+      Uploading...
+    </div>
+
     <div flex justify-end>
       <button
         btn-solid
-        :disabled="!draft.status"
+        :disabled="isUploading || (attachments.length === 0 && !draft.status)"
         @click="publish"
       >
         Publish!
diff --git a/package.json b/package.json
index 5dceba68..c3e42963 100644
--- a/package.json
+++ b/package.json
@@ -20,6 +20,7 @@
     "@pinia/nuxt": "^0.4.3",
     "@types/fs-extra": "^9.0.13",
     "@types/sanitize-html": "^2.6.2",
+    "@types/wicg-file-system-access": "^2020.9.5",
     "@unocss/nuxt": "^0.46.5",
     "@vue-macros/nuxt": "^0.0.2",
     "@vueuse/nuxt": "^9.5.0",
diff --git a/pnpm-lock.yaml b/pnpm-lock.yaml
index 5439211d..9083aff2 100644
--- a/pnpm-lock.yaml
+++ b/pnpm-lock.yaml
@@ -9,6 +9,7 @@ specifiers:
   '@pinia/nuxt': ^0.4.3
   '@types/fs-extra': ^9.0.13
   '@types/sanitize-html': ^2.6.2
+  '@types/wicg-file-system-access': ^2020.9.5
   '@unocss/nuxt': ^0.46.5
   '@vue-macros/nuxt': ^0.0.2
   '@vueuse/nuxt': ^9.5.0
@@ -36,6 +37,7 @@ devDependencies:
   '@pinia/nuxt': 0.4.3_typescript@4.9.3
   '@types/fs-extra': 9.0.13
   '@types/sanitize-html': 2.6.2
+  '@types/wicg-file-system-access': 2020.9.5
   '@unocss/nuxt': 0.46.5
   '@vue-macros/nuxt': 0.0.2_nuxt@3.0.0
   '@vueuse/nuxt': 9.5.0_nuxt@3.0.0
@@ -1297,6 +1299,10 @@ packages:
     resolution: {integrity: sha512-oh8q2Zc32S6gd/j50GowEjKLoOVOwHP/bWVjKJInBwQqdOYMdPrf1oVlelTlyfFK3CKxL1uahMDAr+vy8T7yMQ==}
     dev: true
 
+  /@types/wicg-file-system-access/2020.9.5:
+    resolution: {integrity: sha512-UYK244awtmcUYQfs7FR8710MJcefL2WvkyHMjA8yJzxd1mo0Gfn88sRZ1Bls7hiUhA2w7ne1gpJ9T5g3G0wOyA==}
+    dev: true
+
   /@typescript-eslint/eslint-plugin/5.42.1_a76fwnskzo2n3ynumjbqxmyj7i:
     resolution: {integrity: sha512-LyR6x784JCiJ1j6sH5Y0K6cdExqCCm8DJUTcwG5ThNXJj/G8o5E56u5EdG4SLy+bZAwZBswC+GYn3eGdttBVCg==}
     engines: {node: ^12.22.0 || ^14.17.0 || >=16.0.0}
diff --git a/shims.d.ts b/shims.d.ts
new file mode 100644
index 00000000..d791bcd2
--- /dev/null
+++ b/shims.d.ts
@@ -0,0 +1 @@
+/// <reference types="@types/wicg-file-system-access" />