From c538c151c7462ad0395ff2c15c5e11e89e362aa8 Mon Sep 17 00:00:00 2001
From: Striven <sg.striven@cutecat.club>
Date: Sat, 04 Apr 2026 19:47:16 +0000
Subject: [PATCH] Initial commit
---
quartz/cli/handlers.js | 130 +++++++++++++++++++++++++++++--------------
1 files changed, 87 insertions(+), 43 deletions(-)
diff --git a/quartz/cli/handlers.js b/quartz/cli/handlers.js
index 6b23d80..eb340d0 100644
--- a/quartz/cli/handlers.js
+++ b/quartz/cli/handlers.js
@@ -1,11 +1,11 @@
import { promises } from "fs"
import path from "path"
import esbuild from "esbuild"
-import chalk from "chalk"
+import { styleText } from "util"
import { sassPlugin } from "esbuild-sass-plugin"
import fs from "fs"
import { intro, outro, select, text } from "@clack/prompts"
-import { rimraf } from "rimraf"
+import { rm } from "fs/promises"
import chokidar from "chokidar"
import prettyBytes from "pretty-bytes"
import { execSync, spawnSync } from "child_process"
@@ -34,13 +34,22 @@
} from "./constants.js"
/**
+ * Resolve content directory path
+ * @param contentPath path to resolve
+ */
+function resolveContentPath(contentPath) {
+ if (path.isAbsolute(contentPath)) return path.relative(cwd, contentPath)
+ return path.join(cwd, contentPath)
+}
+
+/**
* Handles `npx quartz create`
* @param {*} argv arguments for `create`
*/
export async function handleCreate(argv) {
console.log()
- intro(chalk.bgGreen.black(` Quartz v${version} `))
- const contentFolder = path.join(cwd, argv.directory)
+ intro(styleText(["bgGreen", "black"], ` Quartz v${version} `))
+ const contentFolder = resolveContentPath(argv.directory)
let setupStrategy = argv.strategy?.toLowerCase()
let linkResolutionStrategy = argv.links?.toLowerCase()
const sourceDirectory = argv.source
@@ -52,12 +61,16 @@
// Error handling
if (!sourceDirectory) {
outro(
- chalk.red(
- `Setup strategies (arg '${chalk.yellow(
+ styleText(
+ "red",
+ `Setup strategies (arg '${styleText(
+ "yellow",
`-${CreateArgv.strategy.alias[0]}`,
- )}') other than '${chalk.yellow(
+ )}') other than '${styleText(
+ "yellow",
"new",
- )}' require content folder argument ('${chalk.yellow(
+ )}' require content folder argument ('${styleText(
+ "yellow",
`-${CreateArgv.source.alias[0]}`,
)}') to be set`,
),
@@ -66,19 +79,23 @@
} else {
if (!fs.existsSync(sourceDirectory)) {
outro(
- chalk.red(
- `Input directory to copy/symlink 'content' from not found ('${chalk.yellow(
+ styleText(
+ "red",
+ `Input directory to copy/symlink 'content' from not found ('${styleText(
+ "yellow",
sourceDirectory,
- )}', invalid argument "${chalk.yellow(`-${CreateArgv.source.alias[0]}`)})`,
+ )}', invalid argument "${styleText("yellow", `-${CreateArgv.source.alias[0]}`)})`,
),
)
process.exit(1)
} else if (!fs.lstatSync(sourceDirectory).isDirectory()) {
outro(
- chalk.red(
- `Source directory to copy/symlink 'content' from is not a directory (found file at '${chalk.yellow(
+ styleText(
+ "red",
+ `Source directory to copy/symlink 'content' from is not a directory (found file at '${styleText(
+ "yellow",
sourceDirectory,
- )}', invalid argument ${chalk.yellow(`-${CreateArgv.source.alias[0]}`)}")`,
+ )}', invalid argument ${styleText("yellow", `-${CreateArgv.source.alias[0]}`)}")`,
),
)
process.exit(1)
@@ -110,7 +127,7 @@
if (contentStat.isSymbolicLink()) {
await fs.promises.unlink(contentFolder)
} else {
- await rimraf(contentFolder)
+ await rm(contentFolder, { recursive: true, force: true })
}
}
@@ -216,7 +233,11 @@
* @param {*} argv arguments for `build`
*/
export async function handleBuild(argv) {
- console.log(chalk.bgGreen.black(`\n Quartz v${version} \n`))
+ if (argv.serve) {
+ argv.watch = true
+ }
+
+ console.log(`\n${styleText(["bgGreen", "black"], ` Quartz v${version} `)} \n`)
const ctx = await esbuild.context({
entryPoints: [fp],
outfile: cacheFile,
@@ -291,13 +312,13 @@
}
if (cleanupBuild) {
- console.log(chalk.yellow("Detected a source code change, doing a hard rebuild..."))
+ console.log(styleText("yellow", "Detected a source code change, doing a hard rebuild..."))
await cleanupBuild()
}
const result = await ctx.rebuild().catch((err) => {
- console.error(`${chalk.red("Couldn't parse Quartz configuration:")} ${fp}`)
- console.log(`Reason: ${chalk.grey(err)}`)
+ console.error(`${styleText("red", "Couldn't parse Quartz configuration:")} ${fp}`)
+ console.log(`Reason: ${styleText("gray", err)}`)
process.exit(1)
})
release()
@@ -322,9 +343,10 @@
clientRefresh()
}
+ let clientRefresh = () => {}
if (argv.serve) {
const connections = []
- const clientRefresh = () => connections.forEach((conn) => conn.send("rebuild"))
+ clientRefresh = () => connections.forEach((conn) => conn.send("rebuild"))
if (argv.baseDir !== "" && !argv.baseDir.startsWith("/")) {
argv.baseDir = "/" + argv.baseDir
@@ -334,7 +356,8 @@
const server = http.createServer(async (req, res) => {
if (argv.baseDir && !req.url?.startsWith(argv.baseDir)) {
console.log(
- chalk.red(
+ styleText(
+ "red",
`[404] ${req.url} (warning: link outside of site, this is likely a Quartz bug)`,
),
)
@@ -369,8 +392,10 @@
})
const status = res.statusCode
const statusString =
- status >= 200 && status < 300 ? chalk.green(`[${status}]`) : chalk.red(`[${status}]`)
- console.log(statusString + chalk.grey(` ${argv.baseDir}${req.url}`))
+ status >= 200 && status < 300
+ ? styleText("green", `[${status}]`)
+ : styleText("red", `[${status}]`)
+ console.log(statusString + styleText("gray", ` ${argv.baseDir}${req.url}`))
release()
}
@@ -379,7 +404,10 @@
res.writeHead(302, {
Location: newFp,
})
- console.log(chalk.yellow("[302]") + chalk.grey(` ${argv.baseDir}${req.url} -> ${newFp}`))
+ console.log(
+ styleText("yellow", "[302]") +
+ styleText("gray", ` ${argv.baseDir}${req.url} -> ${newFp}`),
+ )
res.end()
}
@@ -424,24 +452,37 @@
return serve()
})
+
server.listen(argv.port)
const wss = new WebSocketServer({ port: argv.wsPort })
wss.on("connection", (ws) => connections.push(ws))
console.log(
- chalk.cyan(
+ styleText(
+ "cyan",
`Started a Quartz server listening at http://localhost:${argv.port}${argv.baseDir}`,
),
)
- console.log("hint: exit with ctrl+c")
- const paths = await globby(["**/*.ts", "**/*.tsx", "**/*.scss", "package.json"])
+ } else {
+ await build(clientRefresh)
+ ctx.dispose()
+ }
+
+ if (argv.watch) {
+ const paths = await globby([
+ "**/*.ts",
+ "quartz/cli/*.js",
+ "quartz/static/**/*",
+ "**/*.tsx",
+ "**/*.scss",
+ "package.json",
+ ])
chokidar
.watch(paths, { ignoreInitial: true })
.on("add", () => build(clientRefresh))
.on("change", () => build(clientRefresh))
.on("unlink", () => build(clientRefresh))
- } else {
- await build(() => {})
- ctx.dispose()
+
+ console.log(styleText("gray", "hint: exit with ctrl+c"))
}
}
@@ -450,8 +491,8 @@
* @param {*} argv arguments for `update`
*/
export async function handleUpdate(argv) {
- const contentFolder = path.join(cwd, argv.directory)
- console.log(chalk.bgGreen.black(`\n Quartz v${version} \n`))
+ const contentFolder = resolveContentPath(argv.directory)
+ console.log(`\n${styleText(["bgGreen", "black"], ` Quartz v${version} `)} \n`)
console.log("Backing up your content")
execSync(
`git remote show upstream || git remote add upstream https://github.com/jackyzha0/quartz.git`,
@@ -464,7 +505,7 @@
try {
gitPull(UPSTREAM_NAME, QUARTZ_SOURCE_BRANCH)
} catch {
- console.log(chalk.red("An error occurred above while pulling updates."))
+ console.log(styleText("red", "An error occurred above while pulling updates."))
await popContentFolder(contentFolder)
return
}
@@ -491,9 +532,9 @@
const res = spawnSync("npm", ["i"], opts)
if (res.status === 0) {
- console.log(chalk.green("Done!"))
+ console.log(styleText("green", "Done!"))
} else {
- console.log(chalk.red("An error occurred above while installing dependencies."))
+ console.log(styleText("red", "An error occurred above while installing dependencies."))
}
}
@@ -502,7 +543,7 @@
* @param {*} argv arguments for `restore`
*/
export async function handleRestore(argv) {
- const contentFolder = path.join(cwd, argv.directory)
+ const contentFolder = resolveContentPath(argv.directory)
await popContentFolder(contentFolder)
}
@@ -511,15 +552,15 @@
* @param {*} argv arguments for `sync`
*/
export async function handleSync(argv) {
- const contentFolder = path.join(cwd, argv.directory)
- console.log(chalk.bgGreen.black(`\n Quartz v${version} \n`))
+ const contentFolder = resolveContentPath(argv.directory)
+ console.log(`\n${styleText(["bgGreen", "black"], ` Quartz v${version} `)}\n`)
console.log("Backing up your content")
if (argv.commit) {
const contentStat = await fs.promises.lstat(contentFolder)
if (contentStat.isSymbolicLink()) {
const linkTarg = await fs.promises.readlink(contentFolder)
- console.log(chalk.yellow("Detected symlink, trying to dereference before committing"))
+ console.log(styleText("yellow", "Detected symlink, trying to dereference before committing"))
// stash symlink file
await stashContentFolder(contentFolder)
@@ -554,7 +595,7 @@
try {
gitPull(ORIGIN_NAME, QUARTZ_SOURCE_BRANCH)
} catch {
- console.log(chalk.red("An error occurred above while pulling updates."))
+ console.log(styleText("red", "An error occurred above while pulling updates."))
await popContentFolder(contentFolder)
return
}
@@ -563,14 +604,17 @@
await popContentFolder(contentFolder)
if (argv.push) {
console.log("Pushing your changes")
- const res = spawnSync("git", ["push", "-uf", ORIGIN_NAME, QUARTZ_SOURCE_BRANCH], {
+ const currentBranch = execSync("git rev-parse --abbrev-ref HEAD").toString().trim()
+ const res = spawnSync("git", ["push", "-uf", ORIGIN_NAME, currentBranch], {
stdio: "inherit",
})
if (res.status !== 0) {
- console.log(chalk.red(`An error occurred above while pushing to remote ${ORIGIN_NAME}.`))
+ console.log(
+ styleText("red", `An error occurred above while pushing to remote ${ORIGIN_NAME}.`),
+ )
return
}
}
- console.log(chalk.green("Done!"))
+ console.log(styleText("green", "Done!"))
}
--
Gitblit v1.10.0