Jacky Zhao
2023-08-17 0998bc355e6425e6b2bdf3d2da7124aa7b63b2a2
quartz/bootstrap-cli.mjs
@@ -355,6 +355,7 @@
      ],
    })
    const timeoutIds = new Set()
    const build = async (clientRefresh) => {
      const result = await ctx.rebuild().catch((err) => {
        console.error(`${chalk.red("Couldn't parse Quartz configuration:")} ${fp}`)
@@ -380,6 +381,11 @@
      clientRefresh()
    }
    const rebuild = (clientRefresh) => {
      timeoutIds.forEach((id) => clearTimeout(id))
      timeoutIds.add(setTimeout(() => build(clientRefresh), 250))
    }
    if (argv.serve) {
      const wss = new WebSocketServer({ port: 3001 })
      const connections = []
@@ -388,7 +394,7 @@
      await build(clientRefresh)
      const server = http.createServer(async (req, res) => {
        const serve = async (fp) => {
        const serve = async () => {
          await serveHandler(req, res, {
            public: argv.output,
            directoryListing: false,
@@ -400,11 +406,11 @@
        }
        const redirect = (newFp) => {
          res.writeHead(301, {
          res.writeHead(302, {
            Location: newFp,
          })
          console.log(chalk.yellow("[301]") + chalk.grey(` ${req.url} -> ${newFp}`))
          return res.end()
          console.log(chalk.yellow("[302]") + chalk.grey(` ${req.url} -> ${newFp}`))
          res.end()
        }
        let fp = req.url?.split("?")[0] ?? "/"
@@ -415,7 +421,8 @@
          // does /trailing/index.html exist? if so, serve it
          const indexFp = path.posix.join(fp, "index.html")
          if (fs.existsSync(path.posix.join(argv.output, indexFp))) {
            return serve(indexFp)
            req.url = fp
            return serve()
          }
          // does /trailing.html exist? if so, redirect to /trailing
@@ -424,7 +431,7 @@
            base += ".html"
          }
          if (fs.existsSync(path.posix.join(argv.output, base))) {
            return redirect(base)
            return redirect(fp.slice(0, -1))
          }
        } else {
          // /regular
@@ -434,7 +441,8 @@
            base += ".html"
          }
          if (fs.existsSync(path.posix.join(argv.output, base))) {
            return serve(base)
            req.url = fp
            return serve()
          }
          // does /regular/index.html exist? if so, redirect to /regular/
@@ -444,7 +452,7 @@
          }
        }
        return serve(fp)
        return serve()
      })
      server.listen(argv.port)
      console.log(chalk.cyan(`Started a Quartz server listening at http://localhost:${argv.port}`))
@@ -455,7 +463,7 @@
        })
        .on("all", async () => {
          console.log(chalk.yellow("Detected a source code change, doing a hard rebuild..."))
          await build(clientRefresh)
          rebuild(clientRefresh)
        })
    } else {
      await build(() => {})