123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282
import { basename, dirname, extname, join } from "./deps/path.js";import { existsSync } from "./deps/fs.js";import { Directory, Page } from "./filesystem.js";import { concurrent } from "./utils.js";
export default class Source {  root = new Directory({ path: "/" });
  data = new Map();  pages = new Map();  staticFiles = new Map();  assets = new Set();  ignored = new Set();
  constructor(site) {    this.site = site;  }
  /**   * Returns the Directory instance of a path   */  getDirectory(path, create = false) {    let dir = this.root;
    path.split("/").forEach((name) => {      if (!name || !dir) {        return;      }
      if (create && !dir.dirs.has(name)) {        dir.createDirectory(name);      }
      dir = dir.dirs.get(name);    });
    return dir;  }
  /**   * Check whether a file is included in the static files   */  isStatic(file) {    for (const entry of this.staticFiles) {      const [from] = entry;
      if (file.startsWith(from)) {        return entry;      }    }
    return false;  }
  /**   * Load a directory recursively   */  async loadDirectory(directory = this.root) {    const path = this.site.src(directory.src.path);
    return concurrent(      Deno.readDir(path),      (entry) => this.#loadEntry(directory, entry),    );  }
  /**   * Reload some files   */  async loadFile(file) {    const entry = {      name: basename(file),      isFile: true,      isDirectory: false,      isSymlink: false,    };
    //Is a file inside _data folder    if (file.match(/\/_data\//)) {      const dir = file.split("/_data/", 2).shift();      const directory = this.getDirectory(dir, true);      return this.#loadDataFolderEntry(        join(directory.src.path, "_data"),        entry,        directory.data,      );    }
    const directory = this.getDirectory(dirname(file), true);    await this.#loadEntry(directory, entry);  }
  /**   * Load an entry from a directory   */  async #loadEntry(directory, entry) {    if (entry.isSymlink || entry.name.startsWith(".")) {      return;    }
    const path = join(directory.src.path, entry.name);
    if (this.staticFiles.has(path) || this.ignored.has(path)) {      return;    }
    if (entry.isDirectory && entry.name === "_data") {      directory.data = await this.#loadDataFolder(path);      return;    }
    if (entry.isFile && entry.name.match(/^_data\.\w+$/)) {      directory.data = await this.#loadData(path);      return;    }
    if (entry.name.startsWith("_")) {      return;    }
    if (entry.isFile) {      const page = await this.#loadPage(path);
      if (page) {        directory.setPage(entry.name, page);      } else {        directory.unsetPage(entry.name);      }      return;    }
    if (entry.isDirectory) {      const subDirectory = directory.createDirectory(entry.name);      await this.loadDirectory(subDirectory);      return;    }  }
  /**   * Create and returns a Page   */  async #loadPage(path) {    let ext, load;
    for (const [key, value] of this.pages) {      if (path.endsWith(key)) {        ext = key;        load = value;        break;      }    }
    if (!load) {      return;    }
    const fullPath = this.site.src(path);
    if (!existsSync(fullPath)) {      return;    }
    const info = await Deno.stat(fullPath);    const src = {      path: path.slice(0, -ext.length),      lastModified: info.mtime,      created: info.birthtime,      ext,    };
    const page = new Page(src);    page.data = await load(fullPath);
    const subext = extname(page.src.path);
    if (subext && !this.assets.has(ext)) {      page.dest.path = page.src.path.slice(0, -subext.length);      page.dest.ext = subext;    } else {      page.dest.path = page.src.path;      page.dest.ext = this.assets.has(ext) ? ext : ".html";    }
    if (!page.data.date) {      page.data.date = getDate(page.src, page.dest);    } else if (!(page.data.date instanceof Date)) {      throw new Error(        'Invalid date. Use "yyyy-mm-dd" or "yyy-mm-dd hh:mm:ss" formats',      );    }
    return page;  }
  /**   * Load a _data.* file and return the content   */  async #loadData(path) {    for (const [ext, loader] of this.data) {      if (path.endsWith(ext)) {        return loader(this.site.src(path));      }    }  }
  /**   * Load a _data folder and return the content of all files   */  async #loadDataFolder(path) {    const data = {};
    for (const entry of Deno.readDirSync(this.site.src(path))) {      await this.#loadDataFolderEntry(path, entry, data);    }
    return data;  }
  /**   * Load a data file inside a _data folder   */  async #loadDataFolderEntry(path, entry, data) {    if (      entry.isSymlink || entry.name.startsWith(".") ||      entry.name.startsWith("_")    ) {      return;    }
    if (entry.isFile) {      const name = basename(entry.name, extname(entry.name));      data[name] = Object.assign(        data[name] || {},        await this.#loadData(join(path, entry.name)),      );      return;    }
    if (entry.isDirectory) {      data[entry.name] = await this.#loadDataFolder(join(path, entry.name));    }  }}
function getDate(src, dest) {  const fileName = basename(src.path);  const dayInPath = fileName.match(/^(\d{4})-(\d{2})-(\d{2})_/);
  if (dayInPath) {    const [found, year, month, day] = dayInPath;    dest.path = dest.path.replace(found, "");    return new Date(parseInt(year), parseInt(month) - 1, parseInt(day));  }
  const timeInPath = fileName.match(    /^(\d{4})-(\d{2})-(\d{2})-(\d{2})-(\d{2})-(\d{2})_/,  );
  if (timeInPath) {    const [found, year, month, day, hour, minute, second] = timeInPath;    dest.path = dest.path.replace(found, "");    return new Date(      parseInt(year),      parseInt(month) - 1,      parseInt(day),      parseInt(hour),      parseInt(minute),      parseInt(second),    );  }
  const orderInPath = fileName.match(/^(\d+)_/);
  if (orderInPath) {    const [found, timestamp] = orderInPath;    dest.path = dest.path.replace(found, "");    return new Date(parseInt(timestamp));  }
  return src.created || src.lastModified;}