docmost/apps/server/src/core/page/page.util.ts

import { MovePageDto } from './dto/move-page.dto';
import { EntityManager } from 'typeorm';

export enum OrderingEntity {
  workspace = 'SPACE',
  space = 'SPACE',
  page = 'PAGE',
}

export type TreeNode = {
  id: string;
  title: string;
  icon?: string;
  children?: TreeNode[];
};

export function orderPageList(arr: string[], payload: MovePageDto): void {
  const { id, after, before } = payload;

  // Removing the item we are moving from the array first.
  const index = arr.indexOf(id);
  if (index > -1) arr.splice(index, 1);

  if (after) {
    const afterIndex = arr.indexOf(after);
    if (afterIndex > -1) {
      arr.splice(afterIndex + 1, 0, id);
    } else {
      // Place the item at the end if the after ID is not found.
      arr.push(id);
    }
  } else if (before) {
    const beforeIndex = arr.indexOf(before);
    if (beforeIndex > -1) {
      arr.splice(beforeIndex, 0, id);
    } else {
      // Place the item at the end if the before ID is not found.
      arr.push(id);
    }
  } else {
    // If neither after nor before is provided, just add the id at the end
    if (!arr.includes(id)) {
      arr.push(id);
    }
  }
}

/**
 * Remove an item from an array and save the entity
 * @param entity - The entity instance (Page or Workspace)
 * @param arrayField - The name of the field which is an array
 * @param itemToRemove - The item to remove from the array
 * @param manager - EntityManager instance
 */
export async function removeFromArrayAndSave<T>(
  entity: T,
  arrayField: string,
  itemToRemove: any,
  manager: EntityManager,
) {
  const array = entity[arrayField];
  const index = array.indexOf(itemToRemove);
  if (index > -1) {
    array.splice(index, 1);
    await manager.save(entity);
  }
}

export function transformPageResult(result: any[]): any[] {
  return result.map((row) => {
    const processedRow = {};
    for (const key in row) {
      const newKey = key.split('_').slice(1).join('_');
      if (newKey === 'childrenIds' && !row[key]) {
        processedRow[newKey] = [];
      } else {
        processedRow[newKey] = row[key];
      }
    }
    return processedRow;
  });
}