home.tsx 3.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140
  1. "use client";
  2. require("../polyfill");
  3. import { useState, useEffect } from "react";
  4. import styles from "./home.module.scss";
  5. import BotIcon from "../icons/bot.svg";
  6. import LoadingIcon from "../icons/three-dots.svg";
  7. import { getCSSVar, useMobileScreen } from "../utils";
  8. import dynamic from "next/dynamic";
  9. import { Path, SlotID } from "../constant";
  10. import { ErrorBoundary } from "./error";
  11. import {
  12. HashRouter as Router,
  13. Routes,
  14. Route,
  15. useLocation,
  16. } from "react-router-dom";
  17. import { SideBar } from "./sidebar";
  18. import { useAppConfig } from "../store/config";
  19. import { useMaskStore } from "../store/mask";
  20. export function Loading(props: { noLogo?: boolean }) {
  21. return (
  22. <div className={styles["loading-content"] + " no-dark"}>
  23. {!props.noLogo && <BotIcon />}
  24. <LoadingIcon />
  25. </div>
  26. );
  27. }
  28. const Settings = dynamic(async () => (await import("./settings")).Settings, {
  29. loading: () => <Loading noLogo />,
  30. });
  31. const Chat = dynamic(async () => (await import("./chat")).Chat, {
  32. loading: () => <Loading noLogo />,
  33. });
  34. const NewChat = dynamic(async () => (await import("./new-chat")).NewChat, {
  35. loading: () => <Loading noLogo />,
  36. });
  37. const MaskPage = dynamic(async () => (await import("./mask")).MaskPage, {
  38. loading: () => <Loading noLogo />,
  39. });
  40. export function useSwitchTheme() {
  41. const config = useAppConfig();
  42. useEffect(() => {
  43. document.body.classList.remove("light");
  44. document.body.classList.remove("dark");
  45. if (config.theme === "dark") {
  46. document.body.classList.add("dark");
  47. } else if (config.theme === "light") {
  48. document.body.classList.add("light");
  49. }
  50. const metaDescriptionDark = document.querySelector(
  51. 'meta[name="theme-color"][media]',
  52. );
  53. const metaDescriptionLight = document.querySelector(
  54. 'meta[name="theme-color"]:not([media])',
  55. );
  56. if (config.theme === "auto") {
  57. metaDescriptionDark?.setAttribute("content", "#151515");
  58. metaDescriptionLight?.setAttribute("content", "#fafafa");
  59. } else {
  60. const themeColor = getCSSVar("--themeColor");
  61. metaDescriptionDark?.setAttribute("content", themeColor);
  62. metaDescriptionLight?.setAttribute("content", themeColor);
  63. }
  64. }, [config.theme]);
  65. }
  66. const useHasHydrated = () => {
  67. const [hasHydrated, setHasHydrated] = useState<boolean>(false);
  68. useEffect(() => {
  69. setHasHydrated(true);
  70. }, []);
  71. return hasHydrated;
  72. };
  73. function Screen() {
  74. const config = useAppConfig();
  75. const location = useLocation();
  76. const isHome = location.pathname === Path.Home;
  77. const isMobileScreen = useMobileScreen();
  78. return (
  79. <div
  80. className={
  81. styles.container +
  82. ` ${
  83. config.tightBorder && !isMobileScreen
  84. ? styles["tight-container"]
  85. : styles.container
  86. }`
  87. }
  88. >
  89. <SideBar className={isHome ? styles["sidebar-show"] : ""} />
  90. <div className={styles["window-content"]} id={SlotID.AppBody}>
  91. <Routes>
  92. <Route path={Path.Home} element={<Chat />} />
  93. <Route path={Path.NewChat} element={<NewChat />} />
  94. <Route path={Path.Masks} element={<MaskPage />} />
  95. <Route path={Path.Chat} element={<Chat />} />
  96. <Route path={Path.Settings} element={<Settings />} />
  97. </Routes>
  98. </div>
  99. </div>
  100. );
  101. }
  102. export function Home() {
  103. useSwitchTheme();
  104. if (!useHasHydrated()) {
  105. return <Loading />;
  106. }
  107. return (
  108. <ErrorBoundary>
  109. <Router>
  110. <Screen />
  111. </Router>
  112. </ErrorBoundary>
  113. );
  114. }