| | |
| | | import serveHandler from "serve-handler" |
| | | import { WebSocketServer } from "ws" |
| | | import { randomUUID } from "crypto" |
| | | import { Mutex } from "async-mutex" |
| | | |
| | | const ORIGIN_NAME = "origin" |
| | | const UPSTREAM_NAME = "upstream" |
| | | const QUARTZ_SOURCE_BRANCH = "v4-alpha" |
| | | const QUARTZ_SOURCE_BRANCH = "v4" |
| | | const cwd = process.cwd() |
| | | const cacheDir = path.join(cwd, ".quartz-cache") |
| | | const cacheFile = "./.quartz-cache/transpiled-build.mjs" |
| | |
| | | }, |
| | | baseDir: { |
| | | string: true, |
| | | default: "", |
| | | describe: "base path to serve your local server on", |
| | | }, |
| | | port: { |
| | |
| | | } |
| | | |
| | | async function stashContentFolder(contentFolder) { |
| | | await fs.promises.rm(contentCacheFolder, { force: true, recursive: true }) |
| | | await fs.promises.cp(contentFolder, contentCacheFolder, { |
| | | force: true, |
| | | recursive: true, |
| | |
| | | |
| | | function gitPull(origin, branch) { |
| | | const flags = ["--no-rebase", "--autostash", "-s", "recursive", "-X", "ours", "--no-edit"] |
| | | spawnSync("git", ["pull", ...flags, origin, branch], { stdio: "inherit" }) |
| | | const out = spawnSync("git", ["pull", ...flags, origin, branch], { stdio: "inherit" }) |
| | | if (out.stderr) { |
| | | throw new Error(`Error while pulling updates: ${out.stderr}`) |
| | | } |
| | | } |
| | | |
| | | yargs(hideBin(process.argv)) |
| | |
| | | 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: "copy", label: "Copy an existing folder", hint: "overwrites `content`" }, |
| | | { |
| | | value: "symlink", |
| | | label: "Symlink an existing folder", |
| | |
| | | |
| | | async function rmContentFolder() { |
| | | const contentStat = await fs.promises.lstat(contentFolder) |
| | | if (contentStat) { |
| | | if (contentStat.isSymbolicLink()) { |
| | | await fs.promises.unlink(contentFolder) |
| | | } else { |
| | | await rimraf(contentFolder) |
| | | } |
| | | if (contentStat.isSymbolicLink()) { |
| | | await fs.promises.unlink(contentFolder) |
| | | } else { |
| | | await rimraf(contentFolder) |
| | | } |
| | | } |
| | | |
| | |
| | | |
| | | await rmContentFolder() |
| | | if (setupStrategy === "copy") { |
| | | await fs.promises.cp(originalFolder, contentFolder, { recursive: true }) |
| | | await fs.promises.cp(originalFolder, contentFolder, { |
| | | recursive: true, |
| | | preserveTimestamps: true, |
| | | }) |
| | | } else if (setupStrategy === "symlink") { |
| | | await fs.promises.symlink(originalFolder, contentFolder, "dir") |
| | | } |
| | |
| | | const contentFolder = path.join(cwd, argv.directory) |
| | | console.log(chalk.bgGreen.black(`\n 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`, |
| | | ) |
| | | await stashContentFolder(contentFolder) |
| | | console.log( |
| | | "Pulling updates... you may need to resolve some `git` conflicts if you've made changes to components or plugins.", |
| | | ) |
| | | execSync( |
| | | `git remote show upstream || git remote add upstream https://github.com/jackyzha0/quartz.git`, |
| | | ) |
| | | gitPull(UPSTREAM_NAME, QUARTZ_SOURCE_BRANCH) |
| | | await popContentFolder(contentFolder) |
| | | console.log("Ensuring dependencies are up to date") |
| | | spawnSync("npm", ["i"], { stdio: "inherit" }) |
| | | console.log(chalk.green("Done!")) |
| | | }) |
| | | .command( |
| | | "restore", |
| | | "Try to restore your content folder from the cache", |
| | | CommonArgv, |
| | | async (argv) => { |
| | | const contentFolder = path.join(cwd, argv.directory) |
| | | await popContentFolder(contentFolder) |
| | | }, |
| | | ) |
| | | .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 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")) |
| | | |
| | | // stash symlink file |
| | | await stashContentFolder(contentFolder) |
| | | |
| | | // follow symlink and copy content |
| | | await fs.promises.cp(linkTarg, contentFolder, { |
| | | recursive: true, |
| | | preserveTimestamps: true, |
| | | }) |
| | | } |
| | | |
| | | const currentTimestamp = new Date().toLocaleString("en-US", { |
| | | dateStyle: "medium", |
| | | timeStyle: "short", |
| | | }) |
| | | spawnSync("git", ["commit", "-am", `Quartz sync: ${currentTimestamp}`], { stdio: "inherit" }) |
| | | spawnSync("git", ["add", "."], { stdio: "inherit" }) |
| | | spawnSync("git", ["commit", "-m", `Quartz sync: ${currentTimestamp}`], { stdio: "inherit" }) |
| | | |
| | | if (contentStat.isSymbolicLink()) { |
| | | // put symlink back |
| | | await popContentFolder(contentFolder) |
| | | } |
| | | } |
| | | |
| | | await stashContentFolder(contentFolder) |
| | |
| | | ], |
| | | }) |
| | | |
| | | const buildMutex = new Mutex() |
| | | const timeoutIds = new Set() |
| | | const build = async (clientRefresh) => { |
| | | await buildMutex.acquire() |
| | | const result = await ctx.rebuild().catch((err) => { |
| | | console.error(`${chalk.red("Couldn't parse Quartz configuration:")} ${fp}`) |
| | | console.log(`Reason: ${chalk.grey(err)}`) |
| | |
| | | const { default: buildQuartz } = await import(cacheFile + `?update=${randomUUID()}`) |
| | | await buildQuartz(argv, clientRefresh) |
| | | clientRefresh() |
| | | buildMutex.release() |
| | | } |
| | | |
| | | const rebuild = (clientRefresh) => { |
| | | timeoutIds.forEach((id) => clearTimeout(id)) |
| | | timeoutIds.clear() |
| | | timeoutIds.add(setTimeout(() => build(clientRefresh), 250)) |
| | | } |
| | | |
| | |
| | | wss.on("connection", (ws) => connections.push(ws)) |
| | | const clientRefresh = () => connections.forEach((conn) => conn.send("rebuild")) |
| | | |
| | | if (argv.baseDir !== "" && !argv.baseDir.startsWith("/")) { |
| | | argv.baseDir = "/" + argv.baseDir |
| | | } |
| | | |
| | | await build(clientRefresh) |
| | | const server = http.createServer(async (req, res) => { |
| | | if (argv.baseDir && !req.url?.startsWith(argv.baseDir)) { |
| | | console.log( |
| | | chalk.red( |
| | | `[404] ${req.url} (warning: link outside of site, this is likely a Quartz bug)`, |
| | | ), |
| | | ) |
| | | res.writeHead(404) |
| | | res.end() |
| | | return |
| | | } |
| | | |
| | | // strip baseDir prefix |
| | | req.url = req.url?.slice(argv.baseDir.length) |
| | | |
| | | const serve = async () => { |
| | | await serveHandler(req, res, { |
| | | public: argv.output, |
| | |
| | | const status = res.statusCode |
| | | const statusString = |
| | | status >= 200 && status < 300 ? chalk.green(`[${status}]`) : chalk.red(`[${status}]`) |
| | | console.log(statusString + chalk.grey(` ${req.url}`)) |
| | | console.log(statusString + chalk.grey(` ${argv.baseDir}${req.url}`)) |
| | | } |
| | | |
| | | const redirect = (newFp) => { |
| | | newFp = argv.baseDir + newFp |
| | | res.writeHead(302, { |
| | | Location: newFp, |
| | | }) |
| | | console.log(chalk.yellow("[302]") + chalk.grey(` ${req.url} -> ${newFp}`)) |
| | | console.log(chalk.yellow("[302]") + chalk.grey(` ${argv.baseDir}${req.url} -> ${newFp}`)) |
| | | res.end() |
| | | } |
| | | |
| | |
| | | return serve() |
| | | }) |
| | | server.listen(argv.port) |
| | | console.log(chalk.cyan(`Started a Quartz server listening at http://localhost:${argv.port}`)) |
| | | console.log( |
| | | chalk.cyan( |
| | | `Started a Quartz server listening at http://localhost:${argv.port}${argv.baseDir}`, |
| | | ), |
| | | ) |
| | | console.log("hint: exit with ctrl+c") |
| | | chokidar |
| | | .watch(["**/*.ts", "**/*.tsx", "**/*.scss", "package.json"], { |