|
@@ -1,13 +1,19 @@
|
|
|
import { create } from "zustand";
|
|
|
import { persist } from "zustand/middleware";
|
|
|
import { FETCH_COMMIT_URL, FETCH_TAG_URL } from "../constant";
|
|
|
+import { requestUsage } from "../requests";
|
|
|
|
|
|
export interface UpdateStore {
|
|
|
lastUpdate: number;
|
|
|
remoteVersion: string;
|
|
|
|
|
|
+ used?: number;
|
|
|
+ subscription?: number;
|
|
|
+ lastUpdateUsage: number;
|
|
|
+
|
|
|
version: string;
|
|
|
- getLatestVersion: (force: boolean) => Promise<string>;
|
|
|
+ getLatestVersion: (force?: boolean) => Promise<void>;
|
|
|
+ updateUsage: (force?: boolean) => Promise<void>;
|
|
|
}
|
|
|
|
|
|
export const UPDATE_KEY = "chat-update";
|
|
@@ -26,22 +32,23 @@ function queryMeta(key: string, defaultValue?: string): string {
|
|
|
return ret;
|
|
|
}
|
|
|
|
|
|
+const ONE_MINUTE = 60 * 1000;
|
|
|
+
|
|
|
export const useUpdateStore = create<UpdateStore>()(
|
|
|
persist(
|
|
|
(set, get) => ({
|
|
|
lastUpdate: 0,
|
|
|
remoteVersion: "",
|
|
|
|
|
|
+ lastUpdateUsage: 0,
|
|
|
+
|
|
|
version: "unknown",
|
|
|
|
|
|
async getLatestVersion(force = false) {
|
|
|
- set(() => ({ version: queryMeta("version") }));
|
|
|
+ set(() => ({ version: queryMeta("version") ?? "unknown" }));
|
|
|
|
|
|
- const overTenMins = Date.now() - get().lastUpdate > 10 * 60 * 1000;
|
|
|
- const shouldFetch = force || overTenMins;
|
|
|
- if (!shouldFetch) {
|
|
|
- return get().version ?? "unknown";
|
|
|
- }
|
|
|
+ const overTenMins = Date.now() - get().lastUpdate > 10 * ONE_MINUTE;
|
|
|
+ if (!force && !overTenMins) return;
|
|
|
|
|
|
try {
|
|
|
// const data = await (await fetch(FETCH_TAG_URL)).json();
|
|
@@ -53,10 +60,19 @@ export const useUpdateStore = create<UpdateStore>()(
|
|
|
remoteVersion: remoteId,
|
|
|
}));
|
|
|
console.log("[Got Upstream] ", remoteId);
|
|
|
- return remoteId;
|
|
|
} catch (error) {
|
|
|
console.error("[Fetch Upstream Commit Id]", error);
|
|
|
- return get().version ?? "";
|
|
|
+ }
|
|
|
+ },
|
|
|
+
|
|
|
+ async updateUsage(force = false) {
|
|
|
+ const overOneMinute = Date.now() - get().lastUpdateUsage >= ONE_MINUTE;
|
|
|
+ if (!overOneMinute && !force) return;
|
|
|
+
|
|
|
+ const usage = await requestUsage();
|
|
|
+
|
|
|
+ if (usage) {
|
|
|
+ set(() => usage);
|
|
|
}
|
|
|
},
|
|
|
}),
|