From 8fd75ffbfda30edd5a134a1fbf9b81ac3cebb2ff Mon Sep 17 00:00:00 2001
From: Jacky Zhao <j.zhao2k19@gmail.com>
Date: Mon, 24 Jul 2023 00:42:00 +0000
Subject: [PATCH] support attachments folder

---
 quartz/plugins/index.ts |  101 ++++++++++++++++++++++++++++++++++++++++++++------
 1 files changed, 89 insertions(+), 12 deletions(-)

diff --git a/quartz/plugins/index.ts b/quartz/plugins/index.ts
index 4d0b600..a6231db 100644
--- a/quartz/plugins/index.ts
+++ b/quartz/plugins/index.ts
@@ -1,5 +1,81 @@
-import { StaticResources } from '../resources'
-import { PluginTypes } from './types'
+import { GlobalConfiguration } from "../cfg"
+import { QuartzComponent } from "../components/types"
+import { StaticResources } from "../resources"
+import { joinStyles } from "../theme"
+import { EmitCallback, PluginTypes } from "./types"
+import styles from "../styles/base.scss"
+import { FilePath, ServerSlug } from "../path"
+
+export type ComponentResources = {
+  css: string[]
+  beforeDOMLoaded: string[]
+  afterDOMLoaded: string[]
+}
+
+export function getComponentResources(plugins: PluginTypes): ComponentResources {
+  const allComponents: Set<QuartzComponent> = new Set()
+  for (const emitter of plugins.emitters) {
+    const components = emitter.getQuartzComponents()
+    for (const component of components) {
+      allComponents.add(component)
+    }
+  }
+
+  const componentResources = {
+    css: new Set<string>(),
+    beforeDOMLoaded: new Set<string>(),
+    afterDOMLoaded: new Set<string>(),
+  }
+
+  for (const component of allComponents) {
+    const { css, beforeDOMLoaded, afterDOMLoaded } = component
+    if (css) {
+      componentResources.css.add(css)
+    }
+    if (beforeDOMLoaded) {
+      componentResources.beforeDOMLoaded.add(beforeDOMLoaded)
+    }
+    if (afterDOMLoaded) {
+      componentResources.afterDOMLoaded.add(afterDOMLoaded)
+    }
+  }
+
+  return {
+    css: [...componentResources.css],
+    beforeDOMLoaded: [...componentResources.beforeDOMLoaded],
+    afterDOMLoaded: [...componentResources.afterDOMLoaded],
+  }
+}
+
+function joinScripts(scripts: string[]): string {
+  // wrap with iife to prevent scope collision
+  return scripts.map((script) => `(function () {${script}})();`).join("\n")
+}
+
+export async function emitComponentResources(
+  cfg: GlobalConfiguration,
+  res: ComponentResources,
+  emit: EmitCallback,
+): Promise<FilePath[]> {
+  const fps = await Promise.all([
+    emit({
+      slug: "index" as ServerSlug,
+      ext: ".css",
+      content: joinStyles(cfg.theme, styles, ...res.css),
+    }),
+    emit({
+      slug: "prescript" as ServerSlug,
+      ext: ".js",
+      content: joinScripts(res.beforeDOMLoaded),
+    }),
+    emit({
+      slug: "postscript" as ServerSlug,
+      ext: ".js",
+      content: joinScripts(res.afterDOMLoaded),
+    }),
+  ])
+  return fps
+}
 
 export function getStaticResourcesFromPlugins(plugins: PluginTypes) {
   const staticResources: StaticResources = {
@@ -7,27 +83,28 @@
     js: [],
   }
 
-  for (const plugin of plugins.transformers) {
-    const res = plugin.externalResources
+  for (const transformer of plugins.transformers) {
+    const res = transformer.externalResources ? transformer.externalResources() : {}
     if (res?.js) {
-      staticResources.js = staticResources.js.concat(res.js)
+      staticResources.js.push(...res.js)
     }
     if (res?.css) {
-      staticResources.css = staticResources.css.concat(res.css)
+      staticResources.css.push(...res.css)
     }
   }
 
   return staticResources
 }
 
-export * from './transformers'
-export * from './filters'
-export * from './emitters'
+export * from "./transformers"
+export * from "./filters"
+export * from "./emitters"
 
-declare module 'vfile' {
+declare module "vfile" {
   // inserted in processors.ts
   interface DataMap {
-    slug: string
-    filePath: string
+    slug: ServerSlug
+    allSlugs: ServerSlug[]
+    filePath: FilePath
   }
 }

--
Gitblit v1.10.0