@picturelink/barrelsby/src/destinations.ts

Automatic TypeScript barrels for your entire code base

import { walkTree } from "./fileTree";
import { Logger } from "./options/logger";
import { LocationOption } from "./options/options";
import { Directory, Location } from "./utilities";

/** Assess which directories in the tree should contain barrels. */
export function getDestinations(
  rootTree: Directory,
  locationOption: LocationOption,
  barrelName: string,
  logger: Logger
): Directory[] {
  let destinations: Directory[];
  switch (locationOption) {
    case "top":
    default:
      destinations = [rootTree];
      break;
    case "below":
      destinations = rootTree.directories;
      break;
    case "all":
      destinations = [];
      walkTree(rootTree, (directory: Directory) => {
        destinations.push(directory);
      });
      break;
    case "replace":
      destinations = [];
      walkTree(rootTree, (directory: Directory) => {
        if (
          directory.files.some(
            (location: Location) => location.name === barrelName
          )
        ) {
          destinations.push(directory);
        }
      });
      break;
    case "branch":
      destinations = [];
      walkTree(rootTree, (directory: Directory) => {
        if (directory.directories.length > 0) {
          destinations.push(directory);
        }
      });
      break;
  }

  // Sort by length. This means barrels will be created deepest first.
  destinations = destinations.sort((a: Directory, b: Directory) => {
    return b.path.length - a.path.length;
  });

  logger("Destinations:");
  destinations.forEach(destination => logger(destination.path));

  return destinations;
}