import { action, observable, runInAction, makeObservable } from "mobx"; // services import { UserService } from "services/user.service"; import { AuthService } from "services/auth.service"; // interfaces import { IUser, IUserSettings } from "@plane/types"; // store import { RootStore } from "../root.store"; import { IUserMembershipStore, UserMembershipStore } from "./user-membership.store"; export interface IUserRootStore { // states currentUserError: any | null; currentUserLoader: boolean; // observables isUserLoggedIn: boolean | null; currentUser: IUser | null; isUserInstanceAdmin: boolean | null; currentUserSettings: IUserSettings | null; dashboardInfo: any; // fetch actions fetchCurrentUser: () => Promise; fetchCurrentUserInstanceAdminStatus: () => Promise; fetchCurrentUserSettings: () => Promise; // crud actions updateUserOnBoard: () => Promise; updateTourCompleted: () => Promise; updateCurrentUser: (data: Partial) => Promise; updateCurrentUserTheme: (theme: string) => Promise; deactivateAccount: () => Promise; signOut: () => Promise; membership: IUserMembershipStore; } export class UserRootStore implements IUserRootStore { // states currentUserError: any | null = null; currentUserLoader: boolean = false; // observables isUserLoggedIn: boolean | null = null; currentUser: IUser | null = null; isUserInstanceAdmin: boolean | null = null; currentUserSettings: IUserSettings | null = null; dashboardInfo: any = null; membership: UserMembershipStore; // root store rootStore; // services userService; authService; constructor(_rootStore: RootStore) { makeObservable(this, { // states currentUserError: observable.ref, currentUserLoader: observable.ref, // observable currentUser: observable, isUserInstanceAdmin: observable.ref, currentUserSettings: observable, dashboardInfo: observable, // action fetchCurrentUser: action, fetchCurrentUserInstanceAdminStatus: action, fetchCurrentUserSettings: action, updateUserOnBoard: action, updateTourCompleted: action, updateCurrentUser: action, updateCurrentUserTheme: action, deactivateAccount: action, signOut: action, }); this.rootStore = _rootStore; this.userService = new UserService(); this.authService = new AuthService(); this.membership = new UserMembershipStore(_rootStore); } /** * Fetches the current user * @returns Promise */ fetchCurrentUser = async () => { try { this.currentUserLoader = true; const response = await this.userService.currentUser(); runInAction(() => { this.isUserLoggedIn = true; this.currentUser = response; this.currentUserError = null; this.currentUserLoader = false; }); return response; } catch (error) { runInAction(() => { this.currentUserLoader = false; this.currentUserError = error; }); throw error; } }; /** * Fetches the current user instance admin status * @returns Promise */ fetchCurrentUserInstanceAdminStatus = async () => await this.userService.currentUserInstanceAdminStatus().then((response) => { runInAction(() => { this.isUserInstanceAdmin = response.is_instance_admin; }); return response.is_instance_admin; }); /** * Fetches the current user settings * @returns Promise */ fetchCurrentUserSettings = async () => await this.userService.currentUserSettings().then((response) => { runInAction(() => { this.currentUserSettings = response; }); return response; }); /** * Updates the user onboarding status * @returns Promise */ updateUserOnBoard = async () => { try { runInAction(() => { this.currentUser = { ...this.currentUser, is_onboarded: true, } as IUser; }); const user = this.currentUser ?? undefined; if (!user) return; await this.userService.updateUserOnBoard(); } catch (error) { this.fetchCurrentUser(); throw error; } }; /** * Updates the user tour completed status * @returns Promise */ updateTourCompleted = async () => { try { if (this.currentUser) { runInAction(() => { this.currentUser = { ...this.currentUser, is_tour_completed: true, } as IUser; }); const response = await this.userService.updateUserTourCompleted(); return response; } } catch (error) { this.fetchCurrentUser(); throw error; } }; /** * Updates the current user * @param data * @returns Promise */ updateCurrentUser = async (data: Partial) => { try { runInAction(() => { this.currentUser = { ...this.currentUser, ...data, } as IUser; }); const response = await this.userService.updateUser(data); runInAction(() => { this.currentUser = response; }); return response; } catch (error) { this.fetchCurrentUser(); throw error; } }; /** * Updates the current user theme * @param theme * @returns Promise */ updateCurrentUserTheme = async (theme: string) => { try { runInAction(() => { this.currentUser = { ...this.currentUser, theme: { ...this.currentUser?.theme, theme, }, } as IUser; }); const response = await this.userService.updateUser({ theme: { ...this.currentUser?.theme, theme }, } as IUser); return response; } catch (error) { throw error; } }; /** * Deactivates the current user * @returns Promise */ deactivateAccount = async () => await this.userService.deactivateAccount().then(() => { runInAction(() => { this.currentUser = null; this.currentUserError = null; this.isUserLoggedIn = false; }); this.membership = new UserMembershipStore(this.rootStore); this.rootStore.eventTracker.resetSession(); this.rootStore.resetOnSignout(); }); /** * Signs out the current user * @returns Promise */ signOut = async () => await this.authService.signOut().then(() => { runInAction(() => { this.currentUser = null; this.isUserLoggedIn = false; }); this.membership = new UserMembershipStore(this.rootStore); this.rootStore.eventTracker.resetSession(); this.rootStore.resetOnSignout(); }); }