insomnia/packages/insomnia-app/app/ui/redux/proto-selectors.ts

import { createSelector } from 'reselect';

import type { ProtoDirectory } from '../../models/proto-directory';
import type { ProtoFile } from '../../models/proto-file';
import { selectActiveWorkspace, selectEntitiesLists } from './selectors';

export interface ExpandedProtoDirectory {
  files: ProtoFile[];
  dir: ProtoDirectory | null;
  subDirs: ExpandedProtoDirectory[];
}

const selectAllProtoFiles = createSelector(
  selectEntitiesLists,
  entities => entities.protoFiles,
);

const selectAllProtoDirectories = createSelector(
  selectEntitiesLists,
  entities => entities.protoDirectories,
);

export const selectExpandedActiveProtoDirectories = createSelector(
  selectActiveWorkspace,
  selectAllProtoFiles,
  selectAllProtoDirectories,
  (workspace, allFiles, allDirs): ExpandedProtoDirectory[] => {
    // Get files where the parent is the workspace
    const individualFiles = allFiles.filter(pf => pf.parentId === workspace?._id);
    // Get directories where the parent is the workspace
    const rootDirs = allDirs.filter(pd => pd.parentId === workspace?._id);
    // Expand each directory
    const expandedDirs = rootDirs.map(dir => expandDir(dir, allFiles, allDirs));

    if (individualFiles.length) {
      return [
        {
          files: individualFiles,
          dir: null,
          subDirs: [],
        },
        ...expandedDirs,
      ];
    }

    return expandedDirs;
  },
);

const expandDir = (
  dir: ProtoDirectory,
  allFiles: ProtoFile[],
  allDirs: ProtoDirectory[],
): ExpandedProtoDirectory => {
  const filesInDir = allFiles.filter(pf => pf.parentId === dir._id);
  const subDirs = allDirs.filter(pd => pd.parentId === dir._id);
  // Expand sub directories
  const expandedSubDirs = subDirs.map(subDir => expandDir(subDir, allFiles, allDirs));
  return {
    dir,
    files: filesInDir,
    subDirs: expandedSubDirs,
  };
};