66 lines
1.7 KiB
TypeScript
66 lines
1.7 KiB
TypeScript
import { cookies } from 'next/headers';
|
|
import { notFound } from 'next/navigation';
|
|
|
|
import { auth } from '@/app/(auth)/auth';
|
|
import { Chat } from '@/components/chat';
|
|
import { getChatById, getMessagesByChatId } from '@/lib/db/queries';
|
|
import { convertToUIMessages } from '@/lib/utils';
|
|
import { DataStreamHandler } from '@/components/data-stream-handler';
|
|
import { DEFAULT_CHAT_MODEL } from '@/lib/ai/models';
|
|
|
|
export default async function Page(props: { params: Promise<{ id: string }> }) {
|
|
const params = await props.params;
|
|
const { id } = params;
|
|
const chat = await getChatById({ id });
|
|
|
|
if (!chat) {
|
|
notFound();
|
|
}
|
|
|
|
const session = await auth();
|
|
|
|
if (chat.visibility === 'private') {
|
|
if (!session || !session.user) {
|
|
return notFound();
|
|
}
|
|
|
|
if (session.user.id !== chat.userId) {
|
|
return notFound();
|
|
}
|
|
}
|
|
|
|
const messagesFromDb = await getMessagesByChatId({
|
|
id,
|
|
});
|
|
|
|
const cookieStore = await cookies();
|
|
const chatModelFromCookie = cookieStore.get('chat-model');
|
|
|
|
if (!chatModelFromCookie) {
|
|
return (
|
|
<>
|
|
<Chat
|
|
id={chat.id}
|
|
initialMessages={convertToUIMessages(messagesFromDb)}
|
|
selectedChatModel={DEFAULT_CHAT_MODEL}
|
|
selectedVisibilityType={chat.visibility}
|
|
isReadonly={session?.user?.id !== chat.userId}
|
|
/>
|
|
<DataStreamHandler id={id} />
|
|
</>
|
|
);
|
|
}
|
|
|
|
return (
|
|
<>
|
|
<Chat
|
|
id={chat.id}
|
|
initialMessages={convertToUIMessages(messagesFromDb)}
|
|
selectedChatModel={chatModelFromCookie.value}
|
|
selectedVisibilityType={chat.visibility}
|
|
isReadonly={session?.user?.id !== chat.userId}
|
|
/>
|
|
<DataStreamHandler id={id} />
|
|
</>
|
|
);
|
|
}
|