forked from github/plane
42 lines
1.2 KiB
TypeScript
42 lines
1.2 KiB
TypeScript
import React, { createContext, ReactElement } from "react";
|
|
|
|
import useSWR, { KeyedMutator } from "swr";
|
|
|
|
// services
|
|
import userService from "services/user.service";
|
|
// constants
|
|
import { CURRENT_USER } from "constants/fetch-keys";
|
|
// types
|
|
import type { IUser } from "types";
|
|
|
|
interface IUserContextProps {
|
|
user?: IUser;
|
|
isUserLoading: boolean;
|
|
mutateUser: KeyedMutator<IUser>;
|
|
assignedIssuesLength?: number;
|
|
workspaceInvitesLength?: number;
|
|
}
|
|
|
|
export const UserContext = createContext<IUserContextProps>({} as IUserContextProps);
|
|
|
|
export const UserProvider = ({ children }: { children: ReactElement }) => {
|
|
// API to fetch user information
|
|
const { data, error, mutate } = useSWR<IUser>(CURRENT_USER, () => userService.currentUser(), {
|
|
shouldRetryOnError: false,
|
|
});
|
|
|
|
return (
|
|
<UserContext.Provider
|
|
value={{
|
|
user: error ? undefined : data?.user,
|
|
isUserLoading: Boolean(data?.user === undefined && error === undefined),
|
|
mutateUser: mutate,
|
|
assignedIssuesLength: data?.assigned_issues ?? 0,
|
|
workspaceInvitesLength: data?.workspace_invites ?? 0,
|
|
}}
|
|
>
|
|
{children}
|
|
</UserContext.Provider>
|
|
);
|
|
};
|