50 lines
1.3 KiB
TypeScript
50 lines
1.3 KiB
TypeScript
|
import { create } from "zustand";
|
||
|
|
import { persist } from "zustand/middleware";
|
||
|
|
import { FETCH_COMMIT_URL } from "../constant";
|
||
|
|
import { getCurrentCommitId } from "../utils";
|
||
|
|
|
||
|
|
export interface UpdateStore {
|
||
|
|
lastUpdate: number;
|
||
|
|
remoteId: string;
|
||
|
|
|
||
|
|
getLatestCommitId: (force: boolean) => Promise<string>;
|
||
|
|
}
|
||
|
|
|
||
|
|
export const UPDATE_KEY = "chat-update";
|
||
|
|
|
||
|
|
export const useUpdateStore = create<UpdateStore>()(
|
||
|
|
persist(
|
||
|
|
(set, get) => ({
|
||
|
|
lastUpdate: 0,
|
||
|
|
remoteId: "",
|
||
|
|
|
||
|
|
async getLatestCommitId(force = false) {
|
||
|
|
const overOneHour = Date.now() - get().lastUpdate > 3600 * 1000;
|
||
|
|
const shouldFetch = force || overOneHour;
|
||
|
|
if (!shouldFetch) {
|
||
|
|
return getCurrentCommitId();
|
||
|
|
}
|
||
|
|
|
||
|
|
try {
|
||
|
|
const data = await (await fetch(FETCH_COMMIT_URL)).json();
|
||
|
|
const sha = data[0].sha as string;
|
||
|
|
const remoteId = sha.substring(0, 7);
|
||
|
|
set(() => ({
|
||
|
|
lastUpdate: Date.now(),
|
||
|
|
remoteId,
|
||
|
|
}));
|
||
|
|
console.log("[Got Upstream] ", remoteId);
|
||
|
|
return remoteId;
|
||
|
|
} catch (error) {
|
||
|
|
console.error("[Fetch Upstream Commit Id]", error);
|
||
|
|
return getCurrentCommitId();
|
||
|
|
}
|
||
|
|
},
|
||
|
|
}),
|
||
|
|
{
|
||
|
|
name: UPDATE_KEY,
|
||
|
|
version: 1,
|
||
|
|
}
|
||
|
|
)
|
||
|
|
);
|