123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161
import { posix } from "../deps/path.ts";import { emptyDir, ensureDir } from "../deps/fs.ts";import { concurrent, sha1 } from "./utils.ts";import { Exception } from "./errors.ts";
import type { Page, StaticFile } from "./filesystem.ts";import type Logger from "./logger.ts";
export interface Options {  src: string;  dest: string;  logger: Logger;}
/** * Class to write the generated pages and static files * in the dest folder. */export default class Writer {  src: string;  dest: string;  logger: Logger;  #saveCount = 0;  #outputs = new Map<string, [number, string, string]>();
  constructor(options: Options) {    this.src = options.src;    this.dest = options.dest;    this.logger = options.logger;  }
  /**   * 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> {    // Ignore empty files    if (!page.content) {      return false;    }    const src = page.src.path      ? page.src.path + (page.src.ext || "")      : "(generated)";    const dest = page.dest.path + page.dest.ext;    const id = dest.toLowerCase();    const hash = await sha1(page.content);    const previous = this.#outputs.get(id);    this.#outputs.set(id, [this.#saveCount, src, hash]);
    if (previous) {      const [previousCount, previousPage, previousHash] = previous;
      if (previousCount === this.#saveCount) {        throw new Exception(          "A page will overwrite another page. Use distinct `url` values to resolve the conflict.",          {            page,            previousPage,            destination: dest,          },        );      }
      // The page content didn't change      if (previousHash === hash) {        return false;      }    }
    this.logger.log(`🔥 ${dest.replace(/index\.html?$/, "")} <dim>${src}</dim>`);
    const filename = posix.join(this.dest, dest);    await ensureDir(posix.dirname(filename));
    page.content instanceof Uint8Array      ? await Deno.writeFile(filename, page.content)      : await Deno.writeTextFile(filename, page.content);
    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 { src, dest, saved, removed } = file;
    if (saved || removed) {      return false;    }
    const pathFrom = posix.join(this.src, src);    const pathTo = posix.join(this.dest, dest);
    try {      await ensureDir(posix.dirname(pathTo));      await Deno.copyFile(pathFrom, pathTo);      this.logger.log(`🔥 ${dest} <dim>${src}</dim>`);      file.saved = true;      return true;    } catch (err) {      if (err instanceof Deno.errors.NotFound) {        try {          await Deno.remove(pathTo);          this.logger.log(`❌ <dim>${dest}</dim>`);          file.removed = true;        } catch {          // Ignored        }      }    }
    return false;  }
  /** Empty the dest folder */  async clear() {    await emptyDir(this.dest);    this.#outputs.clear();  }}