memos/web/src/store/v1/user.ts

import axios from "axios";
import { create } from "zustand";
import * as api from "@/helpers/api";
import { UpdateUserResponse, User } from "@/types/proto/api/v2/user_service_pb";

interface UserV1Store {
  userMapByUsername: Record<string, User>;
  getOrFetchUserByUsername: (username: string) => Promise<User>;
  getUserByUsername: (username: string) => User;
  updateUser: (user: Partial<User>, updateMask: string[]) => Promise<User>;
}

// Request cache is used to prevent multiple requests.
const requestCache = new Map<string, Promise<any>>();

const useUserV1Store = create<UserV1Store>()((set, get) => ({
  userMapByUsername: {},
  getOrFetchUserByUsername: async (username: string) => {
    const userMap = get().userMapByUsername;
    if (userMap[username]) {
      return userMap[username] as User;
    }
    if (requestCache.has(username)) {
      return await requestCache.get(username);
    }

    const promise = api.getUserByUsername(username);
    requestCache.set(username, promise);
    const {
      data: { user: user },
    } = await promise;
    if (!user) {
      throw new Error("User not found");
    }
    requestCache.delete(username);
    userMap[username] = user;
    set(userMap);
    return user;
  },
  getUserByUsername: (username: string) => {
    const userMap = get().userMapByUsername;
    return userMap[username] as User;
  },
  updateUser: async (user: Partial<User>, updateMask: string[]) => {
    const {
      data: { user: updatedUser },
    } = await axios.post<UpdateUserResponse>(`/api/v2/users/${user.username}`, {
      user,
      updateMask,
    });
    if (!updatedUser) {
      throw new Error("User not found");
    }
    const userMap = get().userMapByUsername;
    userMap[updatedUser.username] = updatedUser;
    set(userMap);
    return updatedUser;
  },
}));

export default useUserV1Store;