import { isFuture, isPast, isToday } from "date-fns"; import set from "lodash/set"; import sortBy from "lodash/sortBy"; import { action, computed, observable, makeObservable, runInAction } from "mobx"; import { computedFn } from "mobx-utils"; // helpers import { getDate } from "@/helpers/date-time.helper"; import { orderCycles, shouldFilterCycle } from "@/helpers/cycle.helper"; // services import { CycleService } from "@/services/cycle.service"; import { IssueService } from "@/services/issue"; import { ProjectService } from "@/services/project"; // mobx import { RootStore } from "@/store/root.store"; // types import { ICycle, CycleDateCheckData } from "@plane/types"; export interface ICycleStore { // loaders loader: boolean; // observables fetchedMap: Record; cycleMap: Record; activeCycleIdMap: Record; // computed currentProjectCycleIds: string[] | null; currentProjectCompletedCycleIds: string[] | null; currentProjectUpcomingCycleIds: string[] | null; currentProjectIncompleteCycleIds: string[] | null; currentProjectDraftCycleIds: string[] | null; currentProjectActiveCycleId: string | null; // computed actions getFilteredCycleIds: (projectId: string) => string[] | null; getFilteredCompletedCycleIds: (projectId: string) => string[] | null; getCycleById: (cycleId: string) => ICycle | null; getCycleNameById: (cycleId: string) => string | undefined; getActiveCycleById: (cycleId: string) => ICycle | null; getProjectCycleIds: (projectId: string) => string[] | null; // actions validateDate: (workspaceSlug: string, projectId: string, payload: CycleDateCheckData) => Promise; // fetch fetchWorkspaceCycles: (workspaceSlug: string) => Promise; fetchAllCycles: (workspaceSlug: string, projectId: string) => Promise; fetchActiveCycle: (workspaceSlug: string, projectId: string) => Promise; fetchCycleDetails: (workspaceSlug: string, projectId: string, cycleId: string) => Promise; // crud createCycle: (workspaceSlug: string, projectId: string, data: Partial) => Promise; updateCycleDetails: ( workspaceSlug: string, projectId: string, cycleId: string, data: Partial ) => Promise; deleteCycle: (workspaceSlug: string, projectId: string, cycleId: string) => Promise; // favorites addCycleToFavorites: (workspaceSlug: string, projectId: string, cycleId: string) => Promise; removeCycleFromFavorites: (workspaceSlug: string, projectId: string, cycleId: string) => Promise; } export class CycleStore implements ICycleStore { // observables loader: boolean = false; cycleMap: Record = {}; activeCycleIdMap: Record = {}; //loaders fetchedMap: Record = {}; // root store rootStore; // services projectService; issueService; cycleService; constructor(_rootStore: RootStore) { makeObservable(this, { // observables loader: observable.ref, cycleMap: observable, activeCycleIdMap: observable, fetchedMap: observable, // computed currentProjectCycleIds: computed, currentProjectCompletedCycleIds: computed, currentProjectUpcomingCycleIds: computed, currentProjectIncompleteCycleIds: computed, currentProjectDraftCycleIds: computed, currentProjectActiveCycleId: computed, // actions fetchWorkspaceCycles: action, fetchAllCycles: action, fetchActiveCycle: action, fetchCycleDetails: action, createCycle: action, updateCycleDetails: action, deleteCycle: action, addCycleToFavorites: action, removeCycleFromFavorites: action, }); this.rootStore = _rootStore; this.projectService = new ProjectService(); this.issueService = new IssueService(); this.cycleService = new CycleService(); } // computed /** * returns all cycle ids for a project */ get currentProjectCycleIds() { const projectId = this.rootStore.app.router.projectId; if (!projectId || !this.fetchedMap[projectId]) return null; let allCycles = Object.values(this.cycleMap ?? {}).filter((c) => c?.project_id === projectId); allCycles = sortBy(allCycles, [(c) => c.sort_order]); const allCycleIds = allCycles.map((c) => c.id); return allCycleIds; } /** * returns all completed cycle ids for a project */ get currentProjectCompletedCycleIds() { const projectId = this.rootStore.app.router.projectId; if (!projectId || !this.fetchedMap[projectId]) return null; let completedCycles = Object.values(this.cycleMap ?? {}).filter((c) => { const endDate = getDate(c.end_date); const hasEndDatePassed = endDate && isPast(endDate); const isEndDateToday = endDate && isToday(endDate); return c.project_id === projectId && hasEndDatePassed && !isEndDateToday; }); completedCycles = sortBy(completedCycles, [(c) => c.sort_order]); const completedCycleIds = completedCycles.map((c) => c.id); return completedCycleIds; } /** * returns all upcoming cycle ids for a project */ get currentProjectUpcomingCycleIds() { const projectId = this.rootStore.app.router.projectId; if (!projectId || !this.fetchedMap[projectId]) return null; let upcomingCycles = Object.values(this.cycleMap ?? {}).filter((c) => { const startDate = getDate(c.start_date); const isStartDateUpcoming = startDate && isFuture(startDate); return c.project_id === projectId && isStartDateUpcoming; }); upcomingCycles = sortBy(upcomingCycles, [(c) => c.sort_order]); const upcomingCycleIds = upcomingCycles.map((c) => c.id); return upcomingCycleIds; } /** * returns all incomplete cycle ids for a project */ get currentProjectIncompleteCycleIds() { const projectId = this.rootStore.app.router.projectId; if (!projectId || !this.fetchedMap[projectId]) return null; let incompleteCycles = Object.values(this.cycleMap ?? {}).filter((c) => { const endDate = getDate(c.end_date); const hasEndDatePassed = endDate && isPast(endDate); return c.project_id === projectId && !hasEndDatePassed; }); incompleteCycles = sortBy(incompleteCycles, [(c) => c.sort_order]); const incompleteCycleIds = incompleteCycles.map((c) => c.id); return incompleteCycleIds; } /** * returns all draft cycle ids for a project */ get currentProjectDraftCycleIds() { const projectId = this.rootStore.app.router.projectId; if (!projectId || !this.fetchedMap[projectId]) return null; let draftCycles = Object.values(this.cycleMap ?? {}).filter( (c) => c.project_id === projectId && !c.start_date && !c.end_date ); draftCycles = sortBy(draftCycles, [(c) => c.sort_order]); const draftCycleIds = draftCycles.map((c) => c.id); return draftCycleIds; } /** * returns active cycle id for a project */ get currentProjectActiveCycleId() { const projectId = this.rootStore.app.router.projectId; if (!projectId) return null; const activeCycle = Object.keys(this.activeCycleIdMap ?? {}).find( (cycleId) => this.cycleMap?.[cycleId]?.project_id === projectId ); return activeCycle || null; } /** * @description returns filtered cycle ids based on display filters and filters * @param {TCycleDisplayFilters} displayFilters * @param {TCycleFilters} filters * @returns {string[] | null} */ getFilteredCycleIds = computedFn((projectId: string) => { const filters = this.rootStore.cycleFilter.getFiltersByProjectId(projectId); const searchQuery = this.rootStore.cycleFilter.searchQuery; if (!this.fetchedMap[projectId]) return null; let cycles = Object.values(this.cycleMap ?? {}).filter( (c) => c.project_id === projectId && c.name.toLowerCase().includes(searchQuery.toLowerCase()) && shouldFilterCycle(c, filters ?? {}) ); cycles = orderCycles(cycles); const cycleIds = cycles.map((c) => c.id); return cycleIds; }); /** * @description returns filtered cycle ids based on display filters and filters * @param {TCycleDisplayFilters} displayFilters * @param {TCycleFilters} filters * @returns {string[] | null} */ getFilteredCompletedCycleIds = computedFn((projectId: string) => { const filters = this.rootStore.cycleFilter.getFiltersByProjectId(projectId); const searchQuery = this.rootStore.cycleFilter.searchQuery; if (!this.fetchedMap[projectId]) return null; let cycles = Object.values(this.cycleMap ?? {}).filter( (c) => c.project_id === projectId && c.status.toLowerCase() === "completed" && c.name.toLowerCase().includes(searchQuery.toLowerCase()) && shouldFilterCycle(c, filters ?? {}) ); cycles = sortBy(cycles, [(c) => !c.start_date]); const cycleIds = cycles.map((c) => c.id); return cycleIds; }); /** * @description returns cycle details by cycle id * @param cycleId * @returns */ getCycleById = computedFn((cycleId: string): ICycle | null => this.cycleMap?.[cycleId] ?? null); /** * @description returns cycle name by cycle id * @param cycleId * @returns */ getCycleNameById = computedFn((cycleId: string): string => this.cycleMap?.[cycleId]?.name); /** * @description returns active cycle details by cycle id * @param cycleId * @returns */ getActiveCycleById = computedFn((cycleId: string): ICycle | null => this.activeCycleIdMap?.[cycleId] && this.cycleMap?.[cycleId] ? this.cycleMap?.[cycleId] : null ); /** * @description returns list of cycle ids of the project id passed as argument * @param projectId */ getProjectCycleIds = computedFn((projectId: string): string[] | null => { if (!this.fetchedMap[projectId]) return null; let cycles = Object.values(this.cycleMap ?? {}).filter((c) => c.project_id === projectId); cycles = sortBy(cycles, [(c) => c.sort_order]); const cycleIds = cycles.map((c) => c.id); return cycleIds || null; }); /** * @description validates cycle dates * @param workspaceSlug * @param projectId * @param payload * @returns */ validateDate = async (workspaceSlug: string, projectId: string, payload: CycleDateCheckData) => await this.cycleService.cycleDateCheck(workspaceSlug, projectId, payload); /** * @description fetch all cycles * @param workspaceSlug * @returns ICycle[] */ fetchWorkspaceCycles = async (workspaceSlug: string) => await this.cycleService.getWorkspaceCycles(workspaceSlug).then((response) => { runInAction(() => { response.forEach((cycle) => { set(this.cycleMap, [cycle.id], { ...this.cycleMap[cycle.id], ...cycle }); set(this.fetchedMap, cycle.project_id, true); }); }); return response; }); /** * @description fetches all cycles for a project * @param workspaceSlug * @param projectId * @returns */ fetchAllCycles = async (workspaceSlug: string, projectId: string) => { try { this.loader = true; await this.cycleService.getCyclesWithParams(workspaceSlug, projectId).then((response) => { runInAction(() => { response.forEach((cycle) => { set(this.cycleMap, [cycle.id], cycle); }); set(this.fetchedMap, projectId, true); this.loader = false; }); return response; }); } catch (error) { this.loader = false; return undefined; } }; /** * @description fetches active cycle for a project * @param workspaceSlug * @param projectId * @returns */ fetchActiveCycle = async (workspaceSlug: string, projectId: string) => await this.cycleService.getCyclesWithParams(workspaceSlug, projectId, "current").then((response) => { runInAction(() => { response.forEach((cycle) => { set(this.activeCycleIdMap, [cycle.id], true); set(this.cycleMap, [cycle.id], cycle); }); }); return response; }); /** * @description fetches cycle details * @param workspaceSlug * @param projectId * @param cycleId * @returns */ fetchCycleDetails = async (workspaceSlug: string, projectId: string, cycleId: string) => await this.cycleService.getCycleDetails(workspaceSlug, projectId, cycleId).then((response) => { runInAction(() => { set(this.cycleMap, [response.id], { ...this.cycleMap?.[response.id], ...response }); }); return response; }); /** * @description creates a new cycle * @param workspaceSlug * @param projectId * @param data * @returns */ createCycle = async (workspaceSlug: string, projectId: string, data: Partial) => await this.cycleService.createCycle(workspaceSlug, projectId, data).then((response) => { runInAction(() => { set(this.cycleMap, [response.id], response); }); return response; }); /** * @description updates cycle details * @param workspaceSlug * @param projectId * @param cycleId * @param data * @returns */ updateCycleDetails = async (workspaceSlug: string, projectId: string, cycleId: string, data: Partial) => { try { runInAction(() => { set(this.cycleMap, [cycleId], { ...this.cycleMap?.[cycleId], ...data }); }); const response = await this.cycleService.patchCycle(workspaceSlug, projectId, cycleId, data); this.fetchCycleDetails(workspaceSlug, projectId, cycleId); return response; } catch (error) { console.log("Failed to patch cycle from cycle store"); this.fetchAllCycles(workspaceSlug, projectId); this.fetchActiveCycle(workspaceSlug, projectId); throw error; } }; /** * @description deletes a cycle * @param workspaceSlug * @param projectId * @param cycleId */ deleteCycle = async (workspaceSlug: string, projectId: string, cycleId: string) => await this.cycleService.deleteCycle(workspaceSlug, projectId, cycleId).then(() => { runInAction(() => { delete this.cycleMap[cycleId]; delete this.activeCycleIdMap[cycleId]; }); }); /** * @description adds a cycle to favorites * @param workspaceSlug * @param projectId * @param cycleId * @returns */ addCycleToFavorites = async (workspaceSlug: string, projectId: string, cycleId: string) => { const currentCycle = this.getCycleById(cycleId); try { runInAction(() => { if (currentCycle) set(this.cycleMap, [cycleId, "is_favorite"], true); }); // updating through api. const response = await this.cycleService.addCycleToFavorites(workspaceSlug, projectId, { cycle: cycleId }); return response; } catch (error) { runInAction(() => { if (currentCycle) set(this.cycleMap, [cycleId, "is_favorite"], false); }); throw error; } }; /** * @description removes a cycle from favorites * @param workspaceSlug * @param projectId * @param cycleId * @returns */ removeCycleFromFavorites = async (workspaceSlug: string, projectId: string, cycleId: string) => { const currentCycle = this.getCycleById(cycleId); try { runInAction(() => { if (currentCycle) set(this.cycleMap, [cycleId, "is_favorite"], false); }); const response = await this.cycleService.removeCycleFromFavorites(workspaceSlug, projectId, cycleId); return response; } catch (error) { runInAction(() => { if (currentCycle) set(this.cycleMap, [cycleId, "is_favorite"], true); }); throw error; } }; }