From d5583d0d6da7325ab0fce4abb9f9ae5efd7737ac Mon Sep 17 00:00:00 2001
From: TZGyn <ca538901d@mozmail.com>
Date: Sat, 9 Dec 2023 16:07:10 +0800
Subject: [PATCH] able to change shortener project

---
 .../src/routes/(app)/links/+page.server.ts    |  5 ++++-
 frontend/src/routes/(app)/links/+page.svelte  | 22 +++++++++++++++++++
 .../(app)/projects/[uuid]/+page.server.ts     |  4 +++-
 .../routes/(app)/projects/[uuid]/+page.svelte | 22 +++++++++++++++++++
 .../src/routes/api/shortener/[id]/+server.ts  |  6 ++++-
 5 files changed, 56 insertions(+), 3 deletions(-)

diff --git a/frontend/src/routes/(app)/links/+page.server.ts b/frontend/src/routes/(app)/links/+page.server.ts
index c793421..7e65b17 100644
--- a/frontend/src/routes/(app)/links/+page.server.ts
+++ b/frontend/src/routes/(app)/links/+page.server.ts
@@ -7,10 +7,13 @@ export const load = (async (event) => {
 	const shorteners = await db.query.shortener.findMany({
 		with: {
 			visitor: true,
+			project: true,
 		},
 		where: (shortener, { eq, and, isNull }) =>
 			and(eq(shortener.userId, user.id), isNull(shortener.projectId)),
 	})
 
-	return { shorteners }
+	const projects = await db.query.project.findMany()
+
+	return { shorteners, projects }
 }) satisfies PageServerLoad
diff --git a/frontend/src/routes/(app)/links/+page.svelte b/frontend/src/routes/(app)/links/+page.svelte
index 3cce6cc..f4a1838 100644
--- a/frontend/src/routes/(app)/links/+page.svelte
+++ b/frontend/src/routes/(app)/links/+page.svelte
@@ -5,6 +5,7 @@
 	import * as Dialog from '$lib/components/ui/dialog'
 	import * as Card from '$lib/components/ui/card'
 	import * as DropdownMenu from '$lib/components/ui/dropdown-menu'
+	import * as Select from '$lib/components/ui/select'
 	import { Input } from '$lib/components/ui/input'
 	import { Label } from '$lib/components/ui/label'
 	import {
@@ -43,12 +44,14 @@
 	let editDialogOpen = false
 	let editShortenerCode = ''
 	let editShortenerLink = ''
+	let editShortenerCategory: any = undefined
 	let isEditLoading = false
 
 	const openEditDialog = (code: string, link: string) => {
 		editDialogOpen = true
 		editShortenerCode = code
 		editShortenerLink = link
+		editShortenerCategory = undefined
 	}
 
 	const editShortener = async (code: string, link: string) => {
@@ -57,6 +60,9 @@
 			method: 'put',
 			body: JSON.stringify({
 				link,
+				projectId: editShortenerCategory
+					? editShortenerCategory.value
+					: undefined,
 			}),
 		})
 		await invalidateAll()
@@ -184,6 +190,22 @@
 					bind:value={editShortenerLink}
 					class="col-span-3" />
 			</div>
+			<div class="grid grid-cols-4 items-center gap-4">
+				<Label class="text-right">Category</Label>
+				<Select.Root
+					bind:selected={editShortenerCategory}
+					multiple={false}>
+					<Select.Trigger class="col-span-3">
+						<Select.Value placeholder="Select a Category" />
+					</Select.Trigger>
+					<Select.Content>
+						{#each data.projects as project}
+							<Select.Item value={project.id}
+								>{project.name}</Select.Item>
+						{/each}
+					</Select.Content>
+				</Select.Root>
+			</div>
 		</div>
 		<Dialog.Footer>
 			<Button
diff --git a/frontend/src/routes/(app)/projects/[uuid]/+page.server.ts b/frontend/src/routes/(app)/projects/[uuid]/+page.server.ts
index 99a5cc6..0761246 100644
--- a/frontend/src/routes/(app)/projects/[uuid]/+page.server.ts
+++ b/frontend/src/routes/(app)/projects/[uuid]/+page.server.ts
@@ -27,7 +27,9 @@ export const load = (async (event) => {
 				),
 		})
 
-		return { project, shorteners }
+		const projects = await db.query.project.findMany()
+
+		return { project, shorteners, projects }
 	} catch (error) {
 		throw redirect(303, '/projects')
 	}
diff --git a/frontend/src/routes/(app)/projects/[uuid]/+page.svelte b/frontend/src/routes/(app)/projects/[uuid]/+page.svelte
index 3cce6cc..f4a1838 100644
--- a/frontend/src/routes/(app)/projects/[uuid]/+page.svelte
+++ b/frontend/src/routes/(app)/projects/[uuid]/+page.svelte
@@ -5,6 +5,7 @@
 	import * as Dialog from '$lib/components/ui/dialog'
 	import * as Card from '$lib/components/ui/card'
 	import * as DropdownMenu from '$lib/components/ui/dropdown-menu'
+	import * as Select from '$lib/components/ui/select'
 	import { Input } from '$lib/components/ui/input'
 	import { Label } from '$lib/components/ui/label'
 	import {
@@ -43,12 +44,14 @@
 	let editDialogOpen = false
 	let editShortenerCode = ''
 	let editShortenerLink = ''
+	let editShortenerCategory: any = undefined
 	let isEditLoading = false
 
 	const openEditDialog = (code: string, link: string) => {
 		editDialogOpen = true
 		editShortenerCode = code
 		editShortenerLink = link
+		editShortenerCategory = undefined
 	}
 
 	const editShortener = async (code: string, link: string) => {
@@ -57,6 +60,9 @@
 			method: 'put',
 			body: JSON.stringify({
 				link,
+				projectId: editShortenerCategory
+					? editShortenerCategory.value
+					: undefined,
 			}),
 		})
 		await invalidateAll()
@@ -184,6 +190,22 @@
 					bind:value={editShortenerLink}
 					class="col-span-3" />
 			</div>
+			<div class="grid grid-cols-4 items-center gap-4">
+				<Label class="text-right">Category</Label>
+				<Select.Root
+					bind:selected={editShortenerCategory}
+					multiple={false}>
+					<Select.Trigger class="col-span-3">
+						<Select.Value placeholder="Select a Category" />
+					</Select.Trigger>
+					<Select.Content>
+						{#each data.projects as project}
+							<Select.Item value={project.id}
+								>{project.name}</Select.Item>
+						{/each}
+					</Select.Content>
+				</Select.Root>
+			</div>
 		</div>
 		<Dialog.Footer>
 			<Button
diff --git a/frontend/src/routes/api/shortener/[id]/+server.ts b/frontend/src/routes/api/shortener/[id]/+server.ts
index 51bc6fc..4a3fb31 100644
--- a/frontend/src/routes/api/shortener/[id]/+server.ts
+++ b/frontend/src/routes/api/shortener/[id]/+server.ts
@@ -31,6 +31,7 @@ export const GET: RequestHandler = async (event) => {
 
 const updateShortenerSchema = z.object({
 	link: z.string().url(),
+	projectId: z.number().nullable(),
 })
 
 export const PUT: RequestHandler = async (event) => {
@@ -52,7 +53,10 @@ export const PUT: RequestHandler = async (event) => {
 
 	await db
 		.update(shortenerSchema)
-		.set({ link: updateShortener.data.link })
+		.set({
+			link: updateShortener.data.link,
+			projectId: updateShortener.data.projectId ?? undefined,
+		})
 		.where(
 			and(
 				eq(shortenerSchema.code, shortenerId),