data_loader.ts

import { posix } from "../deps/path.ts";
import { isPlainObject } from "./utils/object.ts";

import type { Entry } from "./fs.ts";
import type { RawData } from "./file.ts";
import type Formats from "./formats.ts";

export interface Options {
  /** The registered file formats */
  formats: Formats;
}

/**
 * Class to load data files.
 */
export default class DataLoader {
  /** List of extensions to load data files and the loader used */
  formats: Formats;

  constructor(options: Options) {
    this.formats = options.formats;
  }

  load(entry: Entry): Promise<RawData | undefined> {
    if (entry.type === "directory") {
      return this.#loadDirectory(entry);
    }

    return this.#loadFile(entry);
  }

  /** Load a _data.* file */
  async #loadFile(entry: Entry): Promise<RawData | undefined> {
    const format = this.formats.search(entry.path);

    if (!format?.dataLoader) {
      return;
    }

    return await entry.getContent(format.dataLoader);
  }

  /** Load a _data directory */
  async #loadDirectory(entry: Entry): Promise<RawData> {
    const data: RawData = {};

    for await (const child of entry.children.values()) {
      await this.loadEntry(child, data);
    }

    return data;
  }

  /**
   * Load a data entry inside a _data directory
   * and append the data to the data object
   */
  async loadEntry(entry: Entry, data: RawData) {
    if (entry.name.startsWith(".") || entry.name.startsWith("_")) {
      return;
    }

    if (entry.type === "file") {
      const name = posix.basename(entry.name, posix.extname(entry.name));
      const fileData = await this.#loadFile(entry) || {};

      if (fileData.content && Object.keys(fileData).length === 1) {
        data[name] = fileData.content;
      } else {
        const target = data[name] as Record<string, unknown> | undefined;
        if (isPlainObject(fileData) || target) {
          data[name] = Object.assign(target || {}, fileData);
        } else {
          data[name] = fileData;
        }
      }

      return;
    }

    if (entry.type === "directory") {
      data[entry.name] = await this.#loadDirectory(entry);
    }
  }
}