1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586
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);    }  }}