123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283
import {  join,  dirname,  resolve,  extname,  normalize,  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 Explorer from "./explorer.js";
export default class Site {  explorer = new Explorer(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(plugin) {    plugin(this);    return this;  }
  data(extensions, loader) {    extensions.forEach((extension) => this.source.data.set(extension, loader));    return this;  }
  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;  }
  beforeRender(extensions, transformer) {    extensions.forEach((extension) => {      const transformers = this.before.get(extension) || [];      transformers.push(transformer);      this.before.set(extension, transformers);    });    return this;  }
  afterRender(extensions, transformer) {    extensions.forEach((extension) => {      const transformers = this.after.get(extension) || [];      transformers.push(transformer);      this.after.set(extension, transformers);    });    return this;  }
  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;  }
  filter(name, filter) {    this.filters.set(name, filter);
    for (const engine of this.engines.values()) {      engine.addFilter(name, filter);    }
    return this;  }
  copy(from, to = from) {    this.source.staticFiles.set(join("/", from), join("/", to));    return this;  }
  async build() {    await emptyDir(this.options.dest);
    return Promise.all([      this.#copyStaticFiles(this.options.dest),      this.source.load().then(() => this.#buildPages()),    ]);  }
  async update(files) {    for (const entry of this.source.staticFiles) {      const [from, to] = entry;
      for (const file of files) {        if (file.startsWith(from)) {          await this.#copyEntry([file, join(to, file.slice(from.length))]);          files.delete(file);        }      }    }
    if (!files.size) {      return;    }
    await this.source.update(files);    const filter = (page) =>      files.has(page.src.path + page.src.ext) || !page.dest.saved;    return this.#buildPages(filter);  }
  *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];      }    }  }
  async #copyStaticFiles() {    return parallel(      this.source.staticFiles.entries(),      (entry) => this.#copyEntry(entry),    );  }  async #copyEntry(entry) {    const [from, to] = entry;    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 });    }  }
  async #buildPages(filter) {    await parallel(      this.getPages(filter),      async (entry) => {        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);      },    );
    return parallel(      this.getPages(filter),      async (entry) => {        const [page, dir] = entry;
        await this.#renderPage(page, dir);
        if (!page.content) {          return;        }
        const transformers = this.after.get(page.dest.ext);
        if (transformers) {          for (const transform of transformers) {            await transform(page, dir);          }        }
        return this.#savePage(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;  }
  async #renderPage(page) {    const engine = this.#getEngine(page.src.ext);
    let content = page.content;    let pageData = page.data;    let layout = page.data.layout;
    if (engine) {      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 = Object.assign(        {},        layoutData,        pageData,        { content },      );
      content = await engine.render(layoutData.content, pageData);      layout = layoutData.layout;    }
    page.content = content;  }
  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.content);  }
  #getEngine(path) {    for (const [ext, engine] of this.engines) {      if (path.endsWith(ext)) {        return engine;      }    }  }}
function parallel(iterator, callback) {  return Promise.all(    Array.from(iterator).map((entry) => {      try {        return callback(entry);      } catch (err) {        console.error(`Error in: ${entry}:`);        console.error(err);      }    }),  );}