import dayjs from "dayjs";
import { createContentLoader } from "vitepress/dist/node/index";

interface Page {
  title: string;
  url: string;
  date: string | number;
  update: string | number; // default is update
  frontmatter: Record<string, any>;
  src: string; // markdown content
  desc: string; // description
}

export type { Page };

export default createContentLoader("./content/**/*.md", {
  includeSrc: true,
  render: true,
  transform(raw): Page[] {
    return raw
      .map(({ url, frontmatter, html, src }) => {
        src = parseSrc(src);
        const date = dayjs(frontmatter.date).toDate().getTime();
        const update = frontmatter.update
          ? dayjs(frontmatter.update).toDate().getTime()
          : date;
        return {
          title: frontmatter.title,
          url,
          frontmatter,
          src: src,
          desc: parseDesc(html, frontmatter.summary),
          date: date,
          update: update,
        };
      })
      .sort((a, b) => b.update - a.update);
  },
});

function parseDesc(src: string | undefined, desc: string): string {
  if (desc && desc !== "") {
    return desc;
  }

  if (src) {
    return src;
  }

  return "";
}

// skip --- ---
function parseSrc(src: string | undefined): string {
  if (!src) {
    return "";
  }

  const pairs = src.split("---");
  if (pairs.length < 2) {
    return "";
  }

  return pairs.slice(2).join("---");
}