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/bootstrap-cli.mjs |  410 ++++++++++++++++++++++++++++++++++++++++++++++-----------
 1 files changed, 327 insertions(+), 83 deletions(-)

diff --git a/quartz/bootstrap-cli.mjs b/quartz/bootstrap-cli.mjs
index e6bb3f0..f25484b 100755
--- a/quartz/bootstrap-cli.mjs
+++ b/quartz/bootstrap-cli.mjs
@@ -1,113 +1,357 @@
 #!/usr/bin/env node
-import { promises, readFileSync } from 'fs'
-import yargs from 'yargs'
-import path from 'path'
-import { hideBin } from 'yargs/helpers'
-import esbuild from 'esbuild'
-import chalk from 'chalk'
-import { sassPlugin } from 'esbuild-sass-plugin'
+import { promises, readFileSync } from "fs"
+import yargs from "yargs"
+import path from "path"
+import { hideBin } from "yargs/helpers"
+import esbuild from "esbuild"
+import chalk from "chalk"
+import { sassPlugin } from "esbuild-sass-plugin"
+import fs from "fs"
+import { intro, isCancel, outro, select, text } from "@clack/prompts"
+import { rimraf } from "rimraf"
+import prettyBytes from "pretty-bytes"
+import { spawnSync } from "child_process"
 
+const UPSTREAM_NAME = "upstream"
+const QUARTZ_SOURCE_BRANCH = "v4-alpha"
+const cwd = process.cwd()
+const cacheDir = path.join(cwd, ".quartz-cache")
 const cacheFile = "./.quartz-cache/transpiled-build.mjs"
 const fp = "./quartz/build.ts"
 const { version } = JSON.parse(readFileSync("./package.json").toString())
+const contentCacheFolder = path.join(cacheDir, "content-cache")
 
-export const BuildArgv = {
+const CommonArgv = {
+  directory: {
+    string: true,
+    alias: ["d"],
+    default: "content",
+    describe: "directory to look for content files",
+  },
+  verbose: {
+    boolean: true,
+    alias: ["v"],
+    default: false,
+    describe: "print out extra logging information",
+  },
+}
+
+const SyncArgv = {
+  ...CommonArgv,
+  commit: {
+    boolean: true,
+    default: true,
+    describe: "create a git commit for your unsaved changes",
+  },
+  push: {
+    boolean: true,
+    default: true,
+    describe: "push updates to your Quartz fork",
+  },
+  force: {
+    boolean: true,
+    alias: ["f"],
+    default: true,
+    describe: "whether to apply the --force flag to git commands",
+  },
+  pull: {
+    boolean: true,
+    default: true,
+    describe: "pull updates from your Quartz fork",
+  },
+}
+
+const BuildArgv = {
+  ...CommonArgv,
   output: {
     string: true,
-    alias: ['o'],
-    default: 'public',
-    describe: 'output folder for files'
-  },
-  clean: {
-    boolean: true,
-    default: false,
-    describe: 'clean the output folder before building'
+    alias: ["o"],
+    default: "public",
+    describe: "output folder for files",
   },
   serve: {
     boolean: true,
     default: false,
-    describe: 'run a local server to preview your Quartz'
+    describe: "run a local server to live-preview your Quartz",
   },
   port: {
     number: true,
     default: 8080,
-    describe: 'port to serve Quartz on'
+    describe: "port to serve Quartz on",
   },
-  directory: {
-    string: true,
-    alias: ['d'],
-    default: 'content',
-    describe: 'directory to look for content files'
-  },
-  verbose: {
-    boolean: true,
-    alias: ['v'],
-    default: false,
-    describe: 'print out extra logging information'
+}
+
+function escapePath(fp) {
+  return fp
+    .replace(/\\ /g, " ") // unescape spaces
+    .replace(/^".*"$/, "$1")
+    .replace(/^'.*"$/, "$1")
+    .trim()
+}
+
+function exitIfCancel(val) {
+  if (isCancel(val)) {
+    outro(chalk.red("Exiting"))
+    process.exit(0)
+  } else {
+    return val
   }
 }
 
+async function stashContentFolder(contentFolder) {
+  await fs.promises.cp(contentFolder, contentCacheFolder, {
+    force: true,
+    recursive: true,
+    verbatimSymlinks: true,
+    preserveTimestamps: true,
+  })
+  await fs.promises.rm(contentFolder, { force: true, recursive: true })
+}
+
+async function popContentFolder(contentFolder) {
+  await fs.promises.cp(contentCacheFolder, contentFolder, {
+    force: true,
+    recursive: true,
+    verbatimSymlinks: true,
+    preserveTimestamps: true,
+  })
+  await fs.promises.rm(contentCacheFolder, { force: true, recursive: true })
+}
+
 yargs(hideBin(process.argv))
   .scriptName("quartz")
   .version(version)
-  .usage('$0 <cmd> [args]')
-  .command('build', 'Build Quartz into a bundle of static HTML files', BuildArgv, async (argv) => {
-    await esbuild.build({
-      entryPoints: [fp],
-      outfile: path.join("quartz", cacheFile),
-      bundle: true,
-      keepNames: true,
-      platform: "node",
-      format: "esm",
-      jsx: "automatic",
-      jsxImportSource: "preact",
-      packages: "external",
-      plugins: [
-        sassPlugin({
-          type: 'css-text'
-        }),
-        {
-          name: 'inline-script-loader',
-          setup(build) {
-            build.onLoad({ filter: /\.inline\.(ts|js)$/ }, async (args) => {
-              let text = await promises.readFile(args.path, 'utf8')
-              // remove default exports that we manually inserted
-              text = text.replace('export default', '')
-              text = text.replace('export', '')
+  .usage("$0 <cmd> [args]")
+  .command("create", "Initialize Quartz", CommonArgv, async (argv) => {
+    console.log()
+    intro(chalk.bgGreen.black(` Quartz v${version} `))
+    const contentFolder = path.join(cwd, argv.directory)
+    const setupStrategy = exitIfCancel(
+      await select({
+        message: `Choose how to initialize the content in \`${contentFolder}\``,
+        options: [
+          { value: "new", label: "Empty Quartz" },
+          { value: "copy", label: "Replace with an existing folder", hint: "overwrites `content`" },
+          {
+            value: "symlink",
+            label: "Symlink an existing folder",
+            hint: "don't select this unless you know what you are doing!",
+          },
+          { value: "keep", label: "Keep the existing files" },
+        ],
+      }),
+    )
 
-              const transpiled = await esbuild.build({
-                stdin: {
-                  contents: text,
-                  loader: 'ts',
-                  resolveDir: '.',
-                  sourcefile: path.relative(path.resolve('.'), args.path),
-                },
-                write: false,
-                bundle: true,
-                platform: "browser",
-                format: "esm",
-              })
-              const rawMod = transpiled.outputFiles[0].text
-              return {
-                contents: rawMod,
-                loader: 'text',
-              }
-            })
-          }
+    async function rmContentFolder() {
+      const contentStat = await fs.promises.lstat(contentFolder)
+      if (contentStat) {
+        if (contentStat.isSymbolicLink()) {
+          await fs.promises.unlink(contentFolder)
+        } else {
+          await rimraf(contentFolder)
         }
-      ]
-    }).catch(err => {
-      console.error(`${chalk.red("Couldn't parse Quartz configuration:")} ${fp}`)
-      console.log(`Reason: ${chalk.grey(err)}`)
-      console.log("hint: make sure all the required dependencies are installed (run `npm install`)")
-      process.exit(1)
-    })
+      }
+    }
 
-    const { default: init } = await import(cacheFile)
-    init(argv, version)
+    if (setupStrategy === "copy" || setupStrategy === "symlink") {
+      const originalFolder = escapePath(
+        exitIfCancel(
+          await text({
+            message: "Enter the full path to existing content folder",
+            placeholder:
+              "On most terminal emulators, you can drag and drop a folder into the window and it will paste the full path",
+            validate(fp) {
+              const fullPath = escapePath(fp)
+              if (!fs.existsSync(fullPath)) {
+                return "The given path doesn't exist"
+              } else if (!fs.lstatSync(fullPath).isDirectory()) {
+                return "The given path is not a folder"
+              }
+            },
+          }),
+        ),
+      )
+
+      await rmContentFolder()
+      if (setupStrategy === "copy") {
+        await fs.promises.cp(originalFolder, contentFolder, { recursive: true })
+      } else if (setupStrategy === "symlink") {
+        await fs.promises.symlink(originalFolder, contentFolder, "dir")
+      }
+    } else if (setupStrategy === "new") {
+      await rmContentFolder()
+      await fs.promises.mkdir(contentFolder)
+      await fs.promises.writeFile(
+        path.join(contentFolder, "index.md"),
+        `---
+title: Welcome to Quartz
+---
+
+This is a blank Quartz installation.
+See the [documentation](https://quartz.jzhao.xyz) for how to get started.
+`,
+      )
+    }
+
+    // get a prefered link resolution strategy
+    const linkResolutionStrategy = exitIfCancel(
+      await select({
+        message: `Choose how Quartz should resolve links in your content. You can change this later in \`quartz.config.ts\`.`,
+        options: [
+          {
+            value: "absolute",
+            label: "Treat links as absolute path",
+            hint: "for content made for Quartz 3 and Hugo",
+          },
+          {
+            value: "shortest",
+            label: "Treat links as shortest path",
+            hint: "for most Obsidian vaults",
+          },
+          {
+            value: "relative",
+            label: "Treat links as relative paths",
+            hint: "for just normal Markdown files",
+          },
+        ],
+      }),
+    )
+
+    // now, do config changes
+    const configFilePath = path.join(cwd, "quartz.config.ts")
+    let configContent = await fs.promises.readFile(configFilePath, { encoding: "utf-8" })
+    configContent = configContent.replace(
+      /markdownLinkResolution: '(.+)'/,
+      `markdownLinkResolution: '${linkResolutionStrategy}'`,
+    )
+    await fs.promises.writeFile(configFilePath, configContent)
+
+    outro(`You're all set! Not sure what to do next? Try:
+   • Customizing Quartz a bit more by editing \`quartz.config.ts\`
+   • Running \`npx quartz build --serve\` to preview your Quartz locally
+   • Hosting your Quartz online (see: https://quartz.jzhao.xyz/setup/hosting)
+`)
+  })
+  .command("update", "Get the latest Quartz updates", CommonArgv, async (argv) => {
+    const contentFolder = path.join(cwd, argv.directory)
+    console.log(chalk.bgGreen.black(`\n Quartz v${version} \n`))
+    console.log("Backing up your content")
+    await stashContentFolder(contentFolder)
+    console.log(
+      "Pulling updates... you may need to resolve some `git` conflicts if you've made changes to components or plugins.",
+    )
+    spawnSync("git", ["pull", UPSTREAM_NAME, QUARTZ_SOURCE_BRANCH], { stdio: "inherit" })
+    await popContentFolder(contentFolder)
+    console.log(chalk.green("Done!"))
+  })
+  .command("sync", "Sync your Quartz to and from GitHub.", SyncArgv, async (argv) => {
+    const contentFolder = path.join(cwd, argv.directory)
+    console.log(chalk.bgGreen.black(`\n Quartz v${version} \n`))
+    console.log("Backing up your content")
+
+    if (argv.commit) {
+      const currentTimestamp = new Date().toLocaleString("en-US", {
+        dateStyle: "medium",
+        timeStyle: "short",
+      })
+      spawnSync("git", ["commit", "-am", `Quartz sync: ${currentTimestamp}`], { stdio: "inherit" })
+    }
+
+    await stashContentFolder(contentFolder)
+
+    if (argv.pull) {
+      console.log(
+        "Pulling updates from your repository. You may need to resolve some `git` conflicts if you've made changes to components or plugins.",
+      )
+      spawnSync("git", ["pull", "origin", QUARTZ_SOURCE_BRANCH], { stdio: "inherit" })
+    }
+
+    await popContentFolder(contentFolder)
+    if (argv.push) {
+      console.log("Pushing your changes")
+      const args = argv.force
+        ? ["push", "-f", "origin", QUARTZ_SOURCE_BRANCH]
+        : ["push", "origin", QUARTZ_SOURCE_BRANCH]
+      spawnSync("git", args, { stdio: "inherit" })
+    }
+
+    console.log(chalk.green("Done!"))
+  })
+  .command("build", "Build Quartz into a bundle of static HTML files", BuildArgv, async (argv) => {
+    const result = await esbuild
+      .build({
+        entryPoints: [fp],
+        outfile: path.join("quartz", cacheFile),
+        bundle: true,
+        keepNames: true,
+        platform: "node",
+        format: "esm",
+        jsx: "automatic",
+        jsxImportSource: "preact",
+        packages: "external",
+        metafile: true,
+        sourcemap: true,
+        plugins: [
+          sassPlugin({
+            type: "css-text",
+          }),
+          {
+            name: "inline-script-loader",
+            setup(build) {
+              build.onLoad({ filter: /\.inline\.(ts|js)$/ }, async (args) => {
+                let text = await promises.readFile(args.path, "utf8")
+
+                // remove default exports that we manually inserted
+                text = text.replace("export default", "")
+                text = text.replace("export", "")
+
+                const sourcefile = path.relative(path.resolve("."), args.path)
+                const resolveDir = path.dirname(sourcefile)
+                const transpiled = await esbuild.build({
+                  stdin: {
+                    contents: text,
+                    loader: "ts",
+                    resolveDir,
+                    sourcefile,
+                  },
+                  write: false,
+                  bundle: true,
+                  platform: "browser",
+                  format: "esm",
+                })
+                const rawMod = transpiled.outputFiles[0].text
+                return {
+                  contents: rawMod,
+                  loader: "text",
+                }
+              })
+            },
+          },
+        ],
+      })
+      .catch((err) => {
+        console.error(`${chalk.red("Couldn't parse Quartz configuration:")} ${fp}`)
+        console.log(`Reason: ${chalk.grey(err)}`)
+        console.log(
+          "hint: make sure all the required dependencies are installed (run `npm install`)",
+        )
+        process.exit(1)
+      })
+
+    if (argv.verbose) {
+      const outputFileName = "quartz/.quartz-cache/transpiled-build.mjs"
+      const meta = result.metafile.outputs[outputFileName]
+      console.log(
+        `Successfully transpiled ${Object.keys(meta.inputs).length} files (${prettyBytes(
+          meta.bytes,
+        )})`,
+      )
+    }
+
+    const { default: buildQuartz } = await import(cacheFile)
+    buildQuartz(argv, version)
   })
   .showHelpOnFail(false)
   .help()
   .strict()
-  .demandCommand()
-  .argv
+  .demandCommand().argv

--
Gitblit v1.10.0