123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332
import {  join,  dirname,  resolve,  extname,  basename,} from "./deps/path.js";import {  ensureDir,  emptyDir,  exists,  copy,} from "./deps/fs.js";import { gray } from "./deps/colors.js";import Source from "./source.js";import Searcher from "./searcher.js";
export default class Site {  searcher = new Searcher(this);  engines = new Map();  before = new Map();  after = new Map();  filters = new Map();
  constructor(options) {    options.src = resolve(options.src);    options.dest = resolve(options.dest);    this.options = options;
    this.source = new Source(options.src);  }
  /**   * Use a plugin   */  use(plugin) {    plugin(this);    return this;  }
  /**   * Register a data loader for some extensions   */  data(extensions, loader) {    extensions.forEach((extension) => this.source.data.set(extension, loader));    return this;  }
  /**   * Register a page/assets loader for some extensions   */  load(extensions, loader, asset = false) {    extensions.forEach((extension) => this.source.pages.set(extension, loader));
    if (asset) {      extensions.forEach((extension) => this.source.assets.add(extension));    }    return this;  }
  /**   * Register a transformer executed before render some extensions   */  beforeRender(extensions, transformer) {    extensions.forEach((extension) => {      const transformers = this.before.get(extension) || [];      transformers.push(transformer);      this.before.set(extension, transformers);    });    return this;  }
  /**   * Register a transformer executed after render some extensions   */  afterRender(extensions, transformer) {    extensions.forEach((extension) => {      const transformers = this.after.get(extension) || [];      transformers.push(transformer);      this.after.set(extension, transformers);    });    return this;  }
  /**   * Register template engine used for some extensions   */  engine(extensions, engine) {    extensions.forEach((extension) => this.engines.set(extension, engine));    this.load(extensions, engine.load.bind(engine));
    for (const [name, filter] of this.filters) {      engine.addFilter(name, filter);    }
    return this;  }
  /**   * Register a template filter   */  filter(name, filter) {    this.filters.set(name, filter);
    for (const engine of this.engines.values()) {      engine.addFilter(name, filter);    }
    return this;  }
  /**   * Copy static files/folders without processing   */  copy(from, to = from) {    this.source.staticFiles.set(join("/", from), join("/", to));    return this;  }
  /**   * Build the entire site   */  async build() {    await emptyDir(this.options.dest);
    for (const [from, to] of this.source.staticFiles) {      await this.#copyStatic(from, to);    }
    await this.source.loadDirectory();    await this.#buildPages();  }
  /**   * Rebuild some files that might be changed   */  async update(files) {    const directories = [];
    for (const file of files) {      // file inside a _data folder      if (file.match(/\/_data\//)) {        directories.push(join("/", file.split("/_data/").shift()));        await this.source.loadFile(file);        continue;      }
      // file inside a _includes folder      if (file.match(/\/_includes\//)) {        directories.push(join("/", file.split("/_includes/").shift()));        continue;      }
      // _data.* file      if (file.match(/\/_data.\w+$/)) {        directories.push(dirname(file));        await this.source.loadFile(file);        continue;      }
      //Static file      const entry = this.source.isStatic(file);      if (entry) {        const [from, to] = entry;
        await this.#copyStatic(file, join(to, file.slice(from.length)));        continue;      }
      //Default      await this.source.loadFile(file);    }
    const filter = (page) =>      files.has(page.src.path + page.src.ext) ||      directories.some((path) => page.src.path.startsWith(path));
    return this.#buildPages(filter);  }
  /**   * Return the site pages   */  *getPages(filter = null, directory = "/", recursive = true) {    const from = this.source.getDirectory(directory);
    for (const [page, dir] of from.getPages(recursive)) {      if (page.data.draft && !this.options.dev) {        continue;      }
      if (!filter || filter(page)) {        yield [page, dir];      }    }  }
  /**   * Copy a static file   */  async #copyStatic(from, to) {    const pathFrom = join(this.options.src, from);    const pathTo = join(this.options.dest, to);
    if (await exists(pathFrom)) {      await ensureDir(dirname(pathTo));      console.log(`🔥 ${from}`);      return copy(pathFrom, pathTo, { overwrite: true });    }  }
  /**   * Build the pages   */  async #buildPages(filter) {    for (const entry of this.getPages(filter)) {      const [page, dir] = entry;      const transformers = this.before.get(page.src.ext);
      if (transformers) {        for (const transform of transformers) {          await transform(page, dir);        }      }
      this.#urlPage(page);    }
    for (const entry of this.getPages(filter)) {      const [page, dir] = entry;
      await this.#renderPage(page, dir);
      if (!page.rendered) {        continue;      }
      const transformers = this.after.get(page.dest.ext);
      if (transformers) {        for (const transform of transformers) {          await transform(page, dir);        }      }
      await this.#savePage(page);    }  }
  /**   * Generate the url and dest info of a page   */  #urlPage(page) {    const { dest } = page;
    if (page.data.permalink) {      const ext = extname(page.data.permalink);      dest.ext = ext || ".html";      dest.path = ext        ? page.data.permalink.slice(0, -ext.length)        : page.data.permalink;    }
    if (dest.ext === ".html" && basename(dest.path) !== "index") {      dest.path = join(dest.path, "index");    }
    page.url = (dest.ext === ".html" && dest.path.endsWith("/index"))      ? dest.path.slice(0, -5)      : dest.path + dest.ext;  }
  /**   * Render a page   */  async #renderPage(page) {    const engine = this.#getEngine(page.src.ext);
    let content = page.data.content;    let pageData = page.fullData;    let layout = pageData.layout;
    if (engine) {      pageData.explorer = this.explorer;      content = await engine.render(content, pageData);    }
    while (layout) {      const engine = this.#getEngine(layout);      const path = join(engine.includes, layout);      const layoutData = await engine.load(path);      pageData = {        ...layoutData,        ...pageData,        content,        search: this.searcher,      };
      content = await engine.render(layoutData.content, pageData);      layout = layoutData.layout;    }
    page.rendered = content;  }
  /**   * Save a page   */  async #savePage(page) {    page.dest.saved = true;    const dest = page.dest.path + page.dest.ext;    const src = page.src.path + page.src.ext;
    console.log(`🔥 ${dest} ${gray(src)}`);
    const filename = join(this.options.dest, dest);    await ensureDir(dirname(filename));    return Deno.writeTextFile(filename, page.rendered);  }
  /**   * Get the engine used by a path or extension   */  #getEngine(path) {    for (const [ext, engine] of this.engines) {      if (path.endsWith(ext)) {        return engine;      }    }  }}