124 lines
2.7 KiB
TypeScript
124 lines
2.7 KiB
TypeScript
'use server'
|
|
|
|
import { revalidatePath } from 'next/cache'
|
|
import { redirect } from 'next/navigation'
|
|
import { kv } from '@vercel/kv'
|
|
|
|
import { auth } from '@/auth'
|
|
import { type Chat } from '@/lib/types'
|
|
|
|
export async function getChats(userId?: string | null) {
|
|
if (!userId) {
|
|
return []
|
|
}
|
|
|
|
try {
|
|
const pipeline = kv.pipeline()
|
|
const chats: string[] = await kv.zrange(`user:chat:${userId}`, 0, -1, {
|
|
rev: true
|
|
})
|
|
|
|
for (const chat of chats) {
|
|
pipeline.hgetall(chat)
|
|
}
|
|
|
|
const results = await pipeline.exec()
|
|
|
|
return results as Chat[]
|
|
} catch (error) {
|
|
return []
|
|
}
|
|
}
|
|
|
|
export async function getChat(id: string, userId: string) {
|
|
const chat = await kv.hgetall<Chat>(`chat:${id}`)
|
|
|
|
if (!chat || (userId && chat.userId !== userId)) {
|
|
return null
|
|
}
|
|
|
|
return chat
|
|
}
|
|
|
|
export async function removeChat({ id, path }: { id: string; path: string }) {
|
|
const session = await auth()
|
|
|
|
if (!session) {
|
|
return {
|
|
error: 'Unauthorized'
|
|
}
|
|
}
|
|
|
|
const uid = await kv.hget<string>(`chat:${id}`, 'userId')
|
|
|
|
// Convert both IDs to strings before comparing
|
|
// This is necessary because the session user ID is a string, while the chat user ID is a number
|
|
// The strict inequality check (!==) in JavaScript checks both value and type, so it would fail if the types are different
|
|
// By converting both IDs to strings, we ensure that the comparison is done between two strings, and it will work correctly even if the IDs are originally of different types
|
|
if (String(uid) !== String(session?.user?.id)) {
|
|
return {
|
|
error: 'Unauthorized'
|
|
}
|
|
}
|
|
|
|
await kv.del(`chat:${id}`)
|
|
await kv.zrem(`user:chat:${session.user.id}`, `chat:${id}`)
|
|
|
|
revalidatePath('/')
|
|
return revalidatePath(path)
|
|
}
|
|
|
|
export async function clearChats() {
|
|
const session = await auth()
|
|
|
|
if (!session?.user?.id) {
|
|
return {
|
|
error: 'Unauthorized'
|
|
}
|
|
}
|
|
|
|
const chats: string[] = await kv.zrange(`user:chat:${session.user.id}`, 0, -1)
|
|
if (!chats.length) {
|
|
return redirect('/')
|
|
}
|
|
const pipeline = kv.pipeline()
|
|
|
|
for (const chat of chats) {
|
|
pipeline.del(chat)
|
|
pipeline.zrem(`user:chat:${session.user.id}`, chat)
|
|
}
|
|
|
|
await pipeline.exec()
|
|
|
|
revalidatePath('/')
|
|
return redirect('/')
|
|
}
|
|
|
|
export async function getSharedChat(id: string) {
|
|
const chat = await kv.hgetall<Chat>(`chat:${id}`)
|
|
|
|
if (!chat || !chat.sharePath) {
|
|
return null
|
|
}
|
|
|
|
return chat
|
|
}
|
|
|
|
export async function shareChat(chat: Chat) {
|
|
const session = await auth()
|
|
|
|
if (!session?.user?.id || session.user.id !== chat.userId) {
|
|
return {
|
|
error: 'Unauthorized'
|
|
}
|
|
}
|
|
|
|
const payload = {
|
|
...chat,
|
|
sharePath: `/share/${chat.id}`
|
|
}
|
|
|
|
await kv.hmset(`chat:${chat.id}`, payload)
|
|
|
|
return payload
|
|
}
|