barrelsby/src/fileTree.ts

Automatic TypeScript barrels for your entire code base

import fs from 'fs';
import path from 'path';

import { Logger } from './options/logger';
import { convertPathSeparator } from './utilities';
import { Directory } from './interfaces/directory.interface';

/** Build directory information recursively. */
export function buildTree(directory: string, barrelName: string, logger: Logger): Directory {
  logger.debug(`Building directory tree for ${convertPathSeparator(directory)}`);
  const names = fs.readdirSync(directory);
  const result: Directory = {
    directories: [],
    files: [],
    name: path.basename(directory),
    path: convertPathSeparator(directory),
  };
  names.forEach((name: string) => {
    const fullPath = path.join(directory, name);
    if (fs.statSync(fullPath).isDirectory()) {
      result.directories.push(buildTree(fullPath, barrelName, logger));
    } else {
      const convertedPath = convertPathSeparator(fullPath);
      const file = {
        name,
        path: convertedPath,
      };
      result.files.push(file);
      if (file.name === barrelName) {
        logger.debug(`Found existing barrel @ ${convertedPath}`);
        result.barrel = file;
      }
    }
  });
  return result;
}

/** Walk an entire directory tree recursively. */
export function walkTree(directory: Directory, callback: (directory: Directory) => void) {
  callback(directory);
  directory.directories.forEach(childDirectory => walkTree(childDirectory, callback));
}