From 130f5b1536b117ea711df13d77fdc69d431befec Mon Sep 17 00:00:00 2001
From: KishinZW <sc-1123@qq.com>
Date: Sat, 3 Aug 2024 20:14:04 +0800
Subject: [PATCH] fully implemented file upload

---
 components/ui/progress/Progress.vue |  39 ++++++++++
 components/ui/progress/index.ts     |   1 +
 pages/content/create.vue            | 112 ++++++++++++++++++----------
 pages/content/index.vue             |   3 +
 server/db/schema.ts                 |   2 +-
 server/trpc/routers/content.ts      |  12 +--
 server/trpc/routers/index.ts        |   2 +
 server/trpc/routers/pool.ts         |   2 +-
 8 files changed, 123 insertions(+), 50 deletions(-)
 create mode 100644 components/ui/progress/Progress.vue
 create mode 100644 components/ui/progress/index.ts

diff --git a/components/ui/progress/Progress.vue b/components/ui/progress/Progress.vue
new file mode 100644
index 0000000..872e73e
--- /dev/null
+++ b/components/ui/progress/Progress.vue
@@ -0,0 +1,39 @@
+<script setup lang="ts">
+import { type HTMLAttributes, computed } from 'vue'
+import {
+  ProgressIndicator,
+  ProgressRoot,
+  type ProgressRootProps,
+} from 'radix-vue'
+import { cn } from '@/lib/utils'
+
+const props = withDefaults(
+  defineProps<ProgressRootProps & { class?: HTMLAttributes['class'] }>(),
+  {
+    modelValue: 0,
+  },
+)
+
+const delegatedProps = computed(() => {
+  const { class: _, ...delegated } = props
+
+  return delegated
+})
+</script>
+
+<template>
+  <ProgressRoot
+    v-bind="delegatedProps"
+    :class="
+      cn(
+        'relative h-2 w-full overflow-hidden rounded-full bg-primary/20',
+        props.class,
+      )
+    "
+  >
+    <ProgressIndicator
+      class="h-full w-full flex-1 bg-primary transition-all"
+      :style="`transform: translateX(-${100 - (props.modelValue ?? 0)}%);`"
+    />
+  </ProgressRoot>
+</template>
diff --git a/components/ui/progress/index.ts b/components/ui/progress/index.ts
new file mode 100644
index 0000000..eace989
--- /dev/null
+++ b/components/ui/progress/index.ts
@@ -0,0 +1 @@
+export { default as Progress } from './Progress.vue'
diff --git a/pages/content/create.vue b/pages/content/create.vue
index 62d798e..44cdfe8 100644
--- a/pages/content/create.vue
+++ b/pages/content/create.vue
@@ -1,12 +1,12 @@
 <template>
-  <Title>登录</Title>
+  <Title>上传内容</Title>
   <div class="w-full lg:grid h-full">
     <div class="flex items-center justify-center">
       <div class="mx-auto grid w-[400px] gap-6">
         <Card>
           <CardHeader>
             <CardTitle class="text-2xl">
-              上传新内容
+              上传内容
             </CardTitle>
             <CardDescription>
               上传图片或视频以显示到食堂显示屏上
@@ -24,21 +24,22 @@
               </div>
               <div class="grid gap-2">
                 <Label for="category">类型</Label>
-                <Popover v-model:open="open">
+                <Popover v-model:open="unfoldCheckbox">
                   <PopoverTrigger as-child>
                     <Button
                       variant="outline"
                       role="combobox"
-                      :aria-expanded="open"
-                      class="w-24"
+                      :aria-expanded="unfoldCheckbox"
+                      class="flex w-full justify-between"
                     >
-                      选择类型
+                      <p>
+                        {{ checkedCategory ? checkedCategory : '选择类型' }}
+                      </p>
                       <ChevronsUpDown class="ml-2 h-4 w-4 shrink-0 opacity-50" />
                     </Button>
                   </PopoverTrigger>
                   <PopoverContent class="w-[200px] p-2">
                     <Command>
-                      <CommandInput class="h-9" placeholder="Search framework..." />
                       <CommandEmpty>请选择类型</CommandEmpty>
                       <CommandList>
                         <CommandGroup>
@@ -47,9 +48,11 @@
                             :key="pool.id"
                             :value="pool.id"
                             @select="(ev: any) => {
-                              if (typeof ev.detail.value === 'number')
-                                form.categoryId = ev.detail.value
-                              open = false;
+                              if (typeof ev.detail.value === 'number') {
+                                form.categoryId = ev.detail.value;
+                                checkedCategory = pool.category;
+                              }
+                              unfoldCheckbox = false;
                             }"
                           >
                             {{ pool.category }}
@@ -79,7 +82,7 @@
                 <Label for="lifespan">有效期（天）</Label>
                 <Input
                   id="lifespan"
-                  v-model="form.lifespan"
+                  v-model="lifespan"
                   type="number"
                   max="180"
                   min="0"
@@ -98,11 +101,12 @@
                   </div>
                 </div>
               </div>
-              <Button v-if="!isPending" type="submit" class="w-full" @click="createContent">
-                登录
+              <Progress v-if="isUploading" v-model="progress" />
+              <Button v-if="!isUploading" type="submit" class="w-full" @click="createContent">
+                创建内容
               </Button>
-              <Button v-if="isPending" type="submit" class="w-full" disabled>
-                <Loader2 v-if="isPending" class="w-4 h-4 mr-2 animate-spin" />
+              <Button v-if="isUploading" type="submit" class="w-full" disabled>
+                <Loader2 v-if="isUploading" class="w-4 h-4 mr-2 animate-spin" />
                 请稍候……
               </Button>
             </div>
@@ -121,7 +125,6 @@ import {
   Command,
   CommandEmpty,
   CommandGroup,
-  CommandInput,
   CommandItem,
   CommandList,
 } from '@/components/ui/command';
@@ -130,37 +133,40 @@ import {
   PopoverContent,
   PopoverTrigger,
 } from '@/components/ui/popover';
+import { Progress } from '@/components/ui/progress';
+import { makeId } from '~/server/trpc/utils/shared';
 
 const { $api } = useNuxtApp();
+const userStore = useUserStore();
 
-const form = reactive({
+const unfoldCheckbox = ref(false);
+const checkedCategory = ref('');
+const lifespan = ref(0);
+interface Form {
+  name: string;
+  ownerId: number;
+  duration: number;
+  fileType: string;
+  S3FileId: string;
+  lifespan: number;
+  categoryId: number;
+};
+const form: Form = reactive({
   name: '',
   ownerId: 0,
   duration: 0,
   fileType: '',
   S3FileId: '',
   lifespan: 0,
-  state: 'created',
   categoryId: 0,
 });
 
-const img_types = new Set([
-  'jfif',
-  'pjpeg',
-  'jpeg',
-  'pjp',
-  'jpg',
-  'png',
-]);
-const video_types = new Set([
-  'm4v',
-  'mp4',
-]);
+const allowed_types = new Set(['video', 'image']);
 const { data: categoryList } = useQuery({
   queryKey: ['pool.list'],
   queryFn: () => $api.pool.list.query(),
 });
-const { mutate: createMutation, isPending } = useMutation({
+const { mutate: createMutation } = useMutation({
   mutationFn: $api.content.create.mutate,
   onSuccess: () => toast.success('内容创建成功'),
   onError: err => useErrorHandler(err),
@@ -172,11 +178,9 @@ const { files, open: openFileDialog, reset, onChange } = useFileDialog({
 });
 onChange((filelist: FileList | null) => {
   if (filelist) {
-    const extname = filelist[0].name.split('.')[-1];
-    if (img_types.has(extname)) {
-      form.fileType = 'image';
-    } else if (video_types.has(extname)) {
-      form.fileType = 'video';
+    const fileType = filelist[0].type;
+    if (allowed_types.has(fileType.split('/')[0])) {
+      form.fileType = fileType;
     } else {
       reset();
       toast.error('只能上传图片或视频');
@@ -184,11 +188,43 @@ onChange((filelist: FileList | null) => {
   }
 });
 
-function createContent() {
+const progress = ref(0);
+const isUploading = ref(false);
+async function createContent() {
   if (!files.value) {
     toast.error('未选择文件');
     return;
   }
-  form.lifespan = form.lifespan * 86400;
+  form.lifespan = lifespan.value * 86400;
+  if (userStore.userId) {
+    form.ownerId = userStore.userId;
+    form.S3FileId = `${makeId(20)}/user-${userStore.userId}/file-${files.value[0].name}`;
+  } else {
+    navigateTo('/login');
+    return;
+  }
+
+  try {
+    const uploadURL = await $api.s3.getUploadURL.query({ s3FileId: form.S3FileId });
+    const file = files.value[0];
+    if (uploadURL) {
+      isUploading.value = true;
+      await axios.put(uploadURL, file.slice(), {
+        headers: {
+          'Content-Type': file.type,
+        },
+        onUploadProgress: (p) => {
+          progress.value = Math.floor((p.progress ?? 0) * 100);
+        },
+      });
+    }
+  } catch (err: any) {
+    toast.error(`文件上传失败 ${err}`);
+    isUploading.value = false;
+    return;
+  }
+
+  createMutation(form);
+  isUploading.value = false;
 };
 </script>
diff --git a/pages/content/index.vue b/pages/content/index.vue
index de86bf3..ff6333c 100644
--- a/pages/content/index.vue
+++ b/pages/content/index.vue
@@ -25,6 +25,9 @@
                 <TableHead class="w-64">
                   内容名称
                 </TableHead>
+                <TableHead>
+                  创建者
+                </TableHead>
                 <TableHead>
                   创建时间
                 </TableHead>
diff --git a/server/db/schema.ts b/server/db/schema.ts
index 944f66c..d214c2a 100644
--- a/server/db/schema.ts
+++ b/server/db/schema.ts
@@ -61,7 +61,7 @@ export const contents = sqliteTable('contents', {
   S3FileId: text('s3_file_id').notNull(),
   lifespan: integer('lifespan').notNull(), // in seconds
   state: text('state', { enum: ['created', 'approved', 'rejected', 'inuse', 'outdated'] }).notNull().default('created'),
-  categoryId: text('category_id').references(() => pools.id, setNull),
+  categoryId: integer('category_id').references(() => pools.id, setNull),
   createdAt: integer('created_at', { mode: 'timestamp' }).notNull().$defaultFn(() => new Date()),
 });
 
diff --git a/server/trpc/routers/content.ts b/server/trpc/routers/content.ts
index 04ec4f2..6122f26 100644
--- a/server/trpc/routers/content.ts
+++ b/server/trpc/routers/content.ts
@@ -6,18 +6,11 @@ const nameZod = z.string()
 const idZod = z.number();
 const durationZod = z.number()
   .max(90, { message: '内容展示时长不能超过90秒' });
-const fileTypeZod = z.enum(['image', 'video']);
+const fileTypeZod = z.string();
 const S3FileIdZod = z.string();
 const lifespanZod = z.number()
   .max(15_552_000, { message: '内容有效期不能超过180天' });
-const stateZod = z.enum([
-  'created',
-  'approved',
-  'rejected',
-  'inuse',
-  'outdated',
-]);
-const categoryIdZod = z.string();
+const categoryIdZod = z.number();
 
 export const contentRouter = router({
   create: protectedProcedure
@@ -29,7 +22,6 @@ export const contentRouter = router({
       fileType: fileTypeZod,
       S3FileId: S3FileIdZod,
       lifespan: lifespanZod,
-      state: stateZod,
       categoryId: categoryIdZod,
     }))
     .mutation(async ({ ctx, input }) => {
diff --git a/server/trpc/routers/index.ts b/server/trpc/routers/index.ts
index bf6aaa3..18608f5 100644
--- a/server/trpc/routers/index.ts
+++ b/server/trpc/routers/index.ts
@@ -4,6 +4,7 @@ import { deviceRouter } from './device';
 import { programRouter } from './program';
 import { contentRouter } from './content';
 import { poolRouter } from './pool';
+import { s3Router } from './s3';
 
 export const appRouter = router({
   user: userRouter,
@@ -11,6 +12,7 @@ export const appRouter = router({
   program: programRouter,
   content: contentRouter,
   pool: poolRouter,
+  s3: s3Router,
 });
 
 export type AppRouter = typeof appRouter;
diff --git a/server/trpc/routers/pool.ts b/server/trpc/routers/pool.ts
index 053bc54..a86b484 100644
--- a/server/trpc/routers/pool.ts
+++ b/server/trpc/routers/pool.ts
@@ -30,7 +30,7 @@ export const poolRouter = router({
     }),
 
   list: protectedProcedure
-    .use(requireRoles(['admin']))
+    .use(requireRoles(['admin', 'club']))
     .query(async ({ ctx }) => {
       return await ctx.poolController.getList();
     }),