feat: #2 add access control by

access code

app/store/access.ts

@@ -0,0 +1,30 @@
+import { create } from "zustand";
+import { persist } from "zustand/middleware";
+import { queryMeta } from "../utils";
+
+export interface AccessControlStore {
+  accessCode: string;
+
+  updateCode: (_: string) => void;
+  enabledAccessControl: () => boolean;
+}
+
+export const ACCESS_KEY = "access-control";
+
+export const useAccessStore = create<AccessControlStore>()(
+  persist(
+    (set, get) => ({
+      accessCode: "",
+      enabledAccessControl() {
+        return queryMeta("access") === "enabled";
+      },
+      updateCode(code: string) {
+        set((state) => ({ accessCode: code }));
+      },
+    }),
+    {
+      name: ACCESS_KEY,
+      version: 1,
+    }
+  )
+);

app/store/app.ts

@@ -308,6 +308,7 @@ export const useChatStore = create<ChatStore>()(
           onMessage(content, done) {
             if (done) {
               botMessage.streaming = false;
+              botMessage.content = content;
               get().onNewMessage(botMessage);
             } else {
               botMessage.content = content;

app/store/index.ts

@@ -1,2 +1,3 @@
 export * from "./app";
 export * from "./update";
+export * from "./access";