Adam Brangenberg
2023-07-31 cbae88fc4e9b98764cfccca2e85f265c4b894573
quartz/plugins/emitters/assets.ts
@@ -1,36 +1,55 @@
import { globbyStream } from "globby"
import {
  FilePath, slugifyFilePath,
} from "../../path"
import { FilePath, slugifyFilePath } from "../../path"
import { QuartzEmitterPlugin } from "../types"
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
    },
  }
}