import set from "lodash/set"; import sortBy from "lodash/sortBy"; import { action, computed, makeObservable, observable, runInAction } from "mobx"; import { computedFn } from "mobx-utils"; // types import { IIssueLabel, IIssueLabelTree } from "@plane/types"; // helpers import { buildTree } from "@/helpers/array.helper"; // services import { IssueLabelService } from "@/services/issue"; // store import { RootStore } from "@/store/root.store"; export interface ILabelStore { //Loaders fetchedMap: Record; //Observable labelMap: Record; // computed projectLabels: IIssueLabel[] | undefined; projectLabelsTree: IIssueLabelTree[] | undefined; workspaceLabels: IIssueLabel[] | undefined; //computed actions getProjectLabels: (projectId: string | null) => IIssueLabel[] | undefined; getLabelById: (labelId: string) => IIssueLabel | null; // fetch actions fetchWorkspaceLabels: (workspaceSlug: string) => Promise; fetchProjectLabels: (workspaceSlug: string, projectId: string) => Promise; // crud actions createLabel: (workspaceSlug: string, projectId: string, data: Partial) => Promise; updateLabel: ( workspaceSlug: string, projectId: string, labelId: string, data: Partial ) => Promise; updateLabelPosition: ( workspaceSlug: string, projectId: string, draggingLabelId: string, droppedParentId: string | null, droppedLabelId: string | undefined, dropAtEndOfList: boolean ) => Promise; deleteLabel: (workspaceSlug: string, projectId: string, labelId: string) => Promise; } export class LabelStore implements ILabelStore { // root store rootStore; // root store labelMap labelMap: Record = {}; //loaders fetchedMap: Record = {}; // services issueLabelService; constructor(_rootStore: RootStore) { makeObservable(this, { labelMap: observable, fetchedMap: observable, // computed projectLabels: computed, projectLabelsTree: computed, fetchProjectLabels: action, createLabel: action, updateLabel: action, updateLabelPosition: action, deleteLabel: action, }); // root store this.rootStore = _rootStore; // services this.issueLabelService = new IssueLabelService(); } /** * Returns the labelMap belongs to a specific workspace */ get workspaceLabels() { const currentWorkspaceDetails = this.rootStore.workspaceRoot.currentWorkspace; const workspaceSlug = this.rootStore.router.workspaceSlug || ""; if (!currentWorkspaceDetails || !this.fetchedMap[workspaceSlug]) return; return sortBy( Object.values(this.labelMap).filter((label) => label.workspace_id === currentWorkspaceDetails.id), "sort_order" ); } /** * Returns the labelMap belonging to the current project */ get projectLabels() { const projectId = this.rootStore.router.projectId; const workspaceSlug = this.rootStore.router.workspaceSlug || ""; if (!projectId || !(this.fetchedMap[projectId] || this.fetchedMap[workspaceSlug])) return; return sortBy( Object.values(this.labelMap).filter((label) => label?.project_id === projectId), "sort_order" ); } /** * Returns the labelMap in a tree format */ get projectLabelsTree() { if (!this.projectLabels) return; return buildTree(this.projectLabels); } getProjectLabels = computedFn((projectId: string | null) => { const workspaceSlug = this.rootStore.router.workspaceSlug || ""; if (!projectId || !(this.fetchedMap[projectId] || this.fetchedMap[workspaceSlug])) return; return sortBy( Object.values(this.labelMap).filter((label) => label?.project_id === projectId), "sort_order" ); }); /** * get label info from the map of labels in the store using label id * @param labelId */ getLabelById = computedFn((labelId: string): IIssueLabel | null => this.labelMap?.[labelId] || null); /** * Fetches all the labelMap belongs to a specific project * @param workspaceSlug * @param projectId * @returns Promise */ fetchProjectLabels = async (workspaceSlug: string, projectId: string) => await this.issueLabelService.getProjectLabels(workspaceSlug, projectId).then((response) => { runInAction(() => { response.forEach((label) => { set(this.labelMap, [label.id], label); }); set(this.fetchedMap, projectId, true); }); return response; }); /** * Fetches all the labelMap belongs to a specific project * @param workspaceSlug * @param projectId * @returns Promise */ fetchWorkspaceLabels = async (workspaceSlug: string) => await this.issueLabelService.getWorkspaceIssueLabels(workspaceSlug).then((response) => { runInAction(() => { response.forEach((label) => { set(this.labelMap, [label.id], label); }); set(this.fetchedMap, workspaceSlug, true); }); return response; }); /** * Creates a new label for a specific project and add it to the store * @param workspaceSlug * @param projectId * @param data * @returns Promise */ createLabel = async (workspaceSlug: string, projectId: string, data: Partial) => await this.issueLabelService.createIssueLabel(workspaceSlug, projectId, data).then((response) => { runInAction(() => { set(this.labelMap, [response.id], response); }); return response; }); /** * Updates a label for a specific project and update it in the store * @param workspaceSlug * @param projectId * @param labelId * @param data * @returns Promise */ updateLabel = async (workspaceSlug: string, projectId: string, labelId: string, data: Partial) => { const originalLabel = this.labelMap[labelId]; try { runInAction(() => { set(this.labelMap, [labelId], { ...originalLabel, ...data }); }); const response = await this.issueLabelService.patchIssueLabel(workspaceSlug, projectId, labelId, data); return response; } catch (error) { console.log("Failed to update label from project store"); runInAction(() => { set(this.labelMap, [labelId], originalLabel); }); throw error; } }; /** * updates the sort order of a label and updates the label information using API. * @param workspaceSlug * @param projectId * @param labelId * @param parentId * @param index * @param isSameParent * @param prevIndex * @returns */ updateLabelPosition = async ( workspaceSlug: string, projectId: string, draggingLabelId: string, droppedParentId: string | null, droppedLabelId: string | undefined, dropAtEndOfList: boolean ) => { const currLabel = this.labelMap?.[draggingLabelId]; const labelTree = this.projectLabelsTree; let currentArray: IIssueLabel[]; if (!currLabel || !labelTree) return; //If its is dropped in the same parent then, there is not specific label on which it is mentioned then keep it's original position if (currLabel.parent === droppedParentId && !droppedLabelId) return; const data: Partial = { parent: droppedParentId }; // find array in which the label is to be added if (!droppedParentId) currentArray = labelTree; else currentArray = labelTree?.find((label) => label.id === droppedParentId)?.children || []; let droppedLabelIndex = currentArray.findIndex((label) => label.id === droppedLabelId); //if the position of droppedLabelId cannot be determined then drop it at the end of the list if (dropAtEndOfList || droppedLabelIndex === -1) droppedLabelIndex = currentArray.length; //if currently adding to a new array, then let backend assign a sort order if (currentArray.length > 0) { let prevSortOrder: number | undefined, nextSortOrder: number | undefined; if (typeof currentArray[droppedLabelIndex - 1] !== "undefined") { prevSortOrder = currentArray[droppedLabelIndex - 1].sort_order; } if (typeof currentArray[droppedLabelIndex] !== "undefined") { nextSortOrder = currentArray[droppedLabelIndex].sort_order; } let sortOrder: number = 65535; //based on the next and previous labelMap calculate current sort order if (prevSortOrder && nextSortOrder) { sortOrder = (prevSortOrder + nextSortOrder) / 2; } else if (nextSortOrder) { sortOrder = nextSortOrder / 2; } else if (prevSortOrder) { sortOrder = prevSortOrder + 10000; } data.sort_order = sortOrder; } return this.updateLabel(workspaceSlug, projectId, draggingLabelId, data); }; /** * Delete the label from the project and remove it from the labelMap object * @param workspaceSlug * @param projectId * @param labelId */ deleteLabel = async (workspaceSlug: string, projectId: string, labelId: string) => { if (!this.labelMap[labelId]) return; await this.issueLabelService.deleteIssueLabel(workspaceSlug, projectId, labelId).then(() => { runInAction(() => { delete this.labelMap[labelId]; }); }); }; }