Hierarchy.ts

import path from 'path';
import { listDirectories } from 'utils/hierarchy';
import { GitHubApi } from 'utils/apis';
import { Algorithm, Category, File } from 'models';
import { Author } from 'models/File';
import { algorithmsDir } from 'config/paths';
import { execute, pull } from 'utils/misc';

type CommitAuthors = {
  [sha: string]: Author,
}

export class Hierarchy {
  private categories!: Category[];
  readonly path: string = algorithmsDir;

  constructor() {
    this.refresh();
    this.update().catch(console.error);
  }

  refresh() {
    this.categories = listDirectories(this.path)
      .map(categoryName => new Category(path.resolve(this.path, categoryName), categoryName));

    const files: File[] = [];
    this.categories.forEach(category => category.algorithms.forEach(algorithm => files.push(...algorithm.files)));
    this.cacheCommitAuthors().then(commitAuthors => this.cacheContributors(files, commitAuthors));
  }

  async update(commit?: string) {
    await pull(this.path, 'algorithms', commit);
    this.refresh();
  };

  async cacheCommitAuthors(page = 1, commitAuthors: CommitAuthors = {}): Promise<CommitAuthors> {
    const per_page = 100;
    const {data} = await GitHubApi.listCommits('algorithm-visualizer', 'algorithms', {per_page, page});
    const commits: any[] = data;
    for (const {sha, author} of commits) {
      if (!author) continue;
      const {login, avatar_url} = author;
      commitAuthors[sha] = {login, avatar_url};
    }
    if (commits.length < per_page) {
      return commitAuthors;
    } else {
      return this.cacheCommitAuthors(page + 1, commitAuthors);
    }
  }

  async cacheContributors(files: File[], commitAuthors: CommitAuthors) {
    for (const file of files) {
      const stdout = await execute(`git --no-pager log --follow --no-merges --format="%H" -- "${path.relative(this.path, file.path)}"`, {
        cwd: this.path,
      });
      const output = stdout.toString().replace(/\n$/, '');
      const shas = output.split('\n').reverse();
      const contributors: Author[] = [];
      for (const sha of shas) {
        const author = commitAuthors[sha];
        if (author && !contributors.find(contributor => contributor.login === author.login)) {
          contributors.push(author);
        }
      }
      file.contributors = contributors;
    }
  }

  find(categoryKey: string, algorithmKey: string) {
    const category = this.categories.find(category => category.key === categoryKey);
    if (!category) return;
    const algorithm = category.algorithms.find(algorithm => algorithm.key === algorithmKey);
    if (!algorithm) return;

    const categoryName = category.name;
    const algorithmName = algorithm.name;
    const files = algorithm.files;
    const description = algorithm.description;

    return {categoryKey, categoryName, algorithmKey, algorithmName, files, description};
  }

  iterate(callback: (category: Category, algorithm: Algorithm) => void) {
    this.categories.forEach(category => category.algorithms.forEach(algorithm => callback(category, algorithm)));
  }

  toJSON() {
    const {categories} = this;
    return {categories};
  }
}