import { cookies } from "next/headers";
import { redirect } from "next/navigation";
import { Suspense } from "react";
import { Chat } from "@/components/chat";
import { DataStreamHandler } from "@/components/data-stream-handler";
import { LovartHome } from "@/components/home/lovart-home";
import { generateUUID } from "@/lib/utils";
import { auth } from "../(auth)/auth";
import {
  getDefaultAiModel,
  getEnabledAiModelsWithProvider,
  getUserById,
  getSiteSettings,
  getSystemSetting,
} from "@/lib/db/queries";
import { isAiModelAgentAuto } from "@/lib/db/ai-model-flags";

export default function Page() {
  return (
    <Suspense fallback={<div className="flex h-dvh" />}>
      <NewChatPage />
    </Suspense>
  );
}

async function NewChatPage() {
  const id = generateUUID();

  const [session, cookieStore, siteSettings, homeVariantSetting, surpriseCategoriesSetting] =
    await Promise.all([
      auth(),
      cookies(),
      getSiteSettings(),
      getSystemSetting({ key: "home_variant" }).catch(() => undefined),
      getSystemSetting({ key: "home_surprise_categories" }).catch(() => undefined),
    ]);
  
  const homeVariant = homeVariantSetting?.value === "lovart" ? "lovart" : "classic";
  const surpriseCategories = Array.isArray(surpriseCategoriesSetting?.value)
    ? surpriseCategoriesSetting?.value
        .map((item: unknown) => String(item || "").trim())
        .filter(Boolean)
    : ["全部", "精选"];

  if (!session?.user && homeVariant !== "lovart") {
    redirect("/login");
  }

  if (homeVariant === "lovart") {
    return (
      <LovartHome
        user={
          session?.user
            ? {
                id: session.user.id,
                email: session.user.email,
                name: session.user.name,
                type: session.user.type,
              }
            : null
        }
        isAuthenticated={Boolean(session?.user)}
        siteName={siteSettings.siteName || "AI Chatbot"}
        siteLogo={siteSettings.siteLogo || siteSettings.siteLogoDark || ""}
        surpriseCategories={surpriseCategories}
        heroPromoText={siteSettings.lovartPromoText}
        heroPromoActionText={siteSettings.lovartPromoActionText}
        heroTitleText={siteSettings.lovartHeroTitle}
        heroSubtitleText={siteSettings.lovartHeroSubtitle}
      />
    );
  }
  
  const modelIdFromCookie = cookieStore.get("chat-model");

  // 如果没有保存的模型 ID，从数据库获取默认模型
  let initialModelId = modelIdFromCookie?.value;
  
  if (!initialModelId) {
    // 尝试获取默认模型
    const defaultModel = await getDefaultAiModel();

    const currentUser = session?.user?.id
      ? await getUserById({ id: session.user.id }).catch(() => undefined)
      : undefined;
    const allowHiddenModels =
      session?.user?.type === "admin" || Boolean(currentUser?.isHighVip);

    if (
      defaultModel &&
      !isAiModelAgentAuto(defaultModel) &&
      (allowHiddenModels || !defaultModel.isHidden)
    ) {
      initialModelId = defaultModel.id;
    } else {
      // 如果没有设置默认模型，获取第一个启用的模型
      const enabledModels = await getEnabledAiModelsWithProvider(false, true);
      const visibleModels = enabledModels.filter(
        ({ model }) =>
          !isAiModelAgentAuto(model) &&
          (allowHiddenModels || !model.isHidden)
      );
      if (visibleModels.length > 0) {
        initialModelId = visibleModels[0].model.id;
      }
    }
  }

  return (
    <>
      <Chat
        autoResume={false}
        id={id}
        initialChatModel={initialModelId || ""}
        initialMessages={[]}
        initialVisibilityType="private"
        isReadonly={false}
        key={id}
        userType={session?.user?.type}
      />
      <DataStreamHandler />
    </>
  );
}