102 lines
2.6 KiB
TypeScript
102 lines
2.6 KiB
TypeScript
import { create } from "zustand";
|
|
import { persist } from "zustand/middleware";
|
|
import { DEFAULT_API_HOST, DEFAULT_MODELS, StoreKey } from "../constant";
|
|
import { getHeaders } from "../client/api";
|
|
import { BOT_HELLO } from "./chat";
|
|
import { getClientConfig } from "../config/client";
|
|
|
|
export interface AccessControlStore {
|
|
accessCode: string;
|
|
token: string;
|
|
|
|
needCode: boolean;
|
|
hideUserApiKey: boolean;
|
|
hideBalanceQuery: boolean;
|
|
disableGPT4: boolean;
|
|
|
|
openaiUrl: string;
|
|
|
|
updateToken: (_: string) => void;
|
|
updateCode: (_: string) => void;
|
|
updateOpenAiUrl: (_: string) => void;
|
|
enabledAccessControl: () => boolean;
|
|
isAuthorized: () => boolean;
|
|
fetch: () => void;
|
|
}
|
|
|
|
let fetchState = 0; // 0 not fetch, 1 fetching, 2 done
|
|
|
|
const DEFAULT_OPENAI_URL =
|
|
getClientConfig()?.buildMode === "export" ? DEFAULT_API_HOST : "/api/openai/";
|
|
console.log("[API] default openai url", DEFAULT_OPENAI_URL);
|
|
|
|
export const useAccessStore = create<AccessControlStore>()(
|
|
persist(
|
|
(set, get) => ({
|
|
token: "",
|
|
accessCode: "",
|
|
needCode: true,
|
|
hideUserApiKey: false,
|
|
hideBalanceQuery: false,
|
|
disableGPT4: false,
|
|
|
|
openaiUrl: DEFAULT_OPENAI_URL,
|
|
|
|
enabledAccessControl() {
|
|
get().fetch();
|
|
|
|
return get().needCode;
|
|
},
|
|
updateCode(code: string) {
|
|
set(() => ({ accessCode: code }));
|
|
},
|
|
updateToken(token: string) {
|
|
set(() => ({ token }));
|
|
},
|
|
updateOpenAiUrl(url: string) {
|
|
set(() => ({ openaiUrl: url }));
|
|
},
|
|
isAuthorized() {
|
|
get().fetch();
|
|
|
|
// has token or has code or disabled access control
|
|
return (
|
|
!!get().token || !!get().accessCode || !get().enabledAccessControl()
|
|
);
|
|
},
|
|
fetch() {
|
|
if (fetchState > 0 || getClientConfig()?.buildMode === "export") return;
|
|
fetchState = 1;
|
|
fetch("/api/config", {
|
|
method: "post",
|
|
body: null,
|
|
headers: {
|
|
...getHeaders(),
|
|
},
|
|
})
|
|
.then((res) => res.json())
|
|
.then((res: DangerConfig) => {
|
|
console.log("[Config] got config from server", res);
|
|
set(() => ({ ...res }));
|
|
|
|
if (res.disableGPT4) {
|
|
DEFAULT_MODELS.forEach(
|
|
(m: any) => (m.available = !m.name.startsWith("gpt-4")),
|
|
);
|
|
}
|
|
})
|
|
.catch(() => {
|
|
console.error("[Config] failed to fetch config");
|
|
})
|
|
.finally(() => {
|
|
fetchState = 2;
|
|
});
|
|
},
|
|
}),
|
|
{
|
|
name: StoreKey.Access,
|
|
version: 1,
|
|
},
|
|
),
|
|
);
|