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

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,
    }
  )
);