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/emitters/assets.ts |   73 +++++++++++++++++++++++-------------
 1 files changed, 47 insertions(+), 26 deletions(-)

diff --git a/quartz/plugins/emitters/assets.ts b/quartz/plugins/emitters/assets.ts
index 51c8d84..d231d1d 100644
--- a/quartz/plugins/emitters/assets.ts
+++ b/quartz/plugins/emitters/assets.ts
@@ -4,31 +4,52 @@
 import path from "path"
 import fs from "fs"
 
-export const Assets: QuartzEmitterPlugin = () => ({
-  name: "Assets",
-  getQuartzComponents() {
-    return []
-  },
-  async emit({ argv }, _content, _resources, _emit): Promise<FilePath[]> {
-    // glob all non MD/MDX/HTML files in content folder and copy it over
-    const assetsPath = path.join(argv.output, "assets")
+interface Options {
+  attachmentsFolder: string | null
+}
 
-    const fps: FilePath[] = []
-    for await (const rawFp of globbyStream("**", {
-      ignore: ["**/*.md"],
-      cwd: argv.directory,
-    })) {
-      const fp = rawFp as FilePath
-      const ext = path.extname(fp)
-      const src = path.join(argv.directory, fp) as FilePath
-      const name = (slugifyFilePath(fp as FilePath) + ext) as FilePath
-      const dest = path.join(assetsPath, name) as FilePath
-      const dir = path.dirname(dest) as FilePath
-      await fs.promises.mkdir(dir, { recursive: true }) // ensure dir exists
-      await fs.promises.copyFile(src, dest)
-      fps.push(path.join("assets", fp) as FilePath)
-    }
+const defaultOptions: Options = {
+  attachmentsFolder: null,
+}
 
-    return fps
-  },
-})
+export const Assets: QuartzEmitterPlugin<Options> = (userOpts?: Options) => {
+  const { attachmentsFolder } = { ...defaultOptions, ...userOpts }
+
+  return {
+    name: "Assets",
+    getQuartzComponents() {
+      return []
+    },
+    async emit({ argv }, _content, _resources, _emit): Promise<FilePath[]> {
+      // glob all non MD/MDX/HTML files in content folder and copy it over
+      const assetsPath = path.join(argv.output, "assets")
+
+      const fps: FilePath[] = []
+      for await (const rawFp of globbyStream("**", {
+        ignore: ["**/*.md"],
+        cwd: argv.directory,
+      })) {
+        const fp = rawFp as FilePath
+        const ext = path.extname(fp)
+        const src = path.join(argv.directory, fp) as FilePath
+        let name = (slugifyFilePath(fp as FilePath) + ext) as FilePath
+
+        if (attachmentsFolder) {
+          const segments = name.split("/")
+          if (segments.at(-2) === attachmentsFolder) {
+            segments.splice(-2, 1)
+            name = segments.join("/") as FilePath
+          }
+        }
+
+        const dest = path.join(assetsPath, name) as FilePath
+        const dir = path.dirname(dest) as FilePath
+        await fs.promises.mkdir(dir, { recursive: true }) // ensure dir exists
+        await fs.promises.copyFile(src, dest)
+        fps.push(path.join("assets", fp) as FilePath)
+      }
+
+      return fps
+    },
+  }
+}

--
Gitblit v1.10.0