123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180
import { posix } from "../deps/path.ts";import { emptyDir, ensureDir } from "../deps/fs.ts";import { concurrent } from "./utils/concurrent.ts";import { sha1 } from "./utils/digest.ts";import { log } from "./utils/log.ts";import binaryLoader from "./loaders/binary.ts";
import type { Page, StaticFile } from "./file.ts";
export interface Options {  dest: string;}
/** Generic interface for Writer */export interface Writer {  savePages(pages: Page[]): Promise<Page[]>;  copyFiles(files: StaticFile[]): Promise<StaticFile[]>;  clear(): Promise<void>;  removeFiles(files: string[]): Promise<void>;}
/** * Class to write the generated pages and static files * in the dest folder. */export class FSWriter implements Writer {  dest: string;
  #outputs = new Map<string, [number, string, string]>();  #saveCount = 0;
  constructor(options: Options) {    this.dest = options.dest;  }
  /**   * Save the pages in the dest folder   * Returns an array of pages that have been saved   */  async savePages(pages: Page[]): Promise<Page[]> {    const savedPages: Page[] = [];    ++this.#saveCount;
    await concurrent(      pages,      async (page) => {        if (await this.savePage(page)) {          savedPages.push(page);        }      },    );
    return savedPages;  }
  /**   * Save a page in the dest folder   * Returns a boolean indicating if the page has saved   */  async savePage(page: Page): Promise<boolean> {    const { sourcePath, outputPath, content } = page;    // Ignore empty pages    if (!content) {      log.warn(        `[Lume] <cyan>Skipped page</cyan> ${page.data.url} (file content is empty)`,      );      return false;    }
    const filename = posix.join(this.dest, outputPath);    const id = filename.toLowerCase();    const hash = await sha1(content);    const previous = this.#outputs.get(id);    this.#outputs.set(id, [this.#saveCount, sourcePath, hash]);
    if (previous) {      const [previousCount, previousSourcePath, previousHash] = previous;
      if (previousCount === this.#saveCount) {        throw new Error(          `The pages ${sourcePath} and ${previousSourcePath} have the same output path "${outputPath}". Use distinct 'url' values to resolve the conflict.`,        );      }
      // The page content didn't change      if (previousHash === hash) {        return false;      }    }
    log.info(`🔥 ${page.data.url} <- <gray>${sourcePath}</gray>`);
    await ensureDir(posix.dirname(filename));
    page.content instanceof Uint8Array      ? await Deno.writeFile(filename, page.content)      : await Deno.writeTextFile(filename, page.content as string);
    return true;  }
  /**   * Copy the static files in the dest folder   */  async copyFiles(files: StaticFile[]): Promise<StaticFile[]> {    const copyFiles: StaticFile[] = [];
    await concurrent(      files,      async (file) => {        if (await this.copyFile(file)) {          copyFiles.push(file);        }      },    );
    return copyFiles;  }
  /**   * Copy a static file in the dest folder   * Returns a boolean indicating if the file has saved   */  async copyFile(file: StaticFile): Promise<boolean> {    const { entry } = file;
    if (entry.flags.has("saved")) {      return false;    }
    entry.flags.add("saved");    const pathTo = posix.join(this.dest, file.outputPath);
    try {      await ensureDir(posix.dirname(pathTo));
      if (entry.flags.has("remote")) {        await Deno.writeFile(          pathTo,          (await entry.getContent(binaryLoader)).content as Uint8Array,        );      } else {        // Copy file https://github.com/denoland/deno/issues/19425        Deno.writeFileSync(pathTo, Deno.readFileSync(entry.src));      }      log.info(        `🔥 ${file.outputPath} <- <gray>${          entry.flags.has("remote") ? entry.src : entry.path        }</gray>`,      );      return true;    } catch {      // Ignored    }
    return false;  }
  /** Empty the dest folder */  async clear() {    await emptyDir(this.dest);    this.#outputs.clear();  }
  async removeFiles(files: string[]) {    await concurrent(      files,      async (file) => {        try {          const outputPath = posix.join(this.dest, file);          this.#outputs.delete(outputPath.toLowerCase());          await Deno.remove(outputPath);        } catch {          // Ignored        }      },    );  }}