ChatGPT-Next-Web/app/store/access.ts

import { create } from "zustand";
import { persist } from "zustand/middleware";
import { queryMeta } from "../utils";

export interface AccessControlStore {
  accessCode: string;
  token: string;

  updateToken: (_: string) => void;
  updateCode: (_: string) => void;
  enabledAccessControl: () => boolean;
}

export const ACCESS_KEY = "access-control";

export const useAccessStore = create<AccessControlStore>()(
  persist(
    (set, get) => ({
      token: process.env.NEXT_PUBLIC_OPENAI_API_KEY ?? "",
      accessCode: process.env.NEXT_PUBLIC_CODE ?? "",
      enabledAccessControl() {
        return queryMeta("access") === "enabled";
      },
      updateCode(code: string) {
        set((state) => ({ accessCode: code }));
      },
      updateToken(token: string) {
        set((state) => ({ token }));
      },
    }),
    {
      name: ACCESS_KEY,
      version: 1,
    },
  ),
);