buildLocalBake.ts

#! /usr/bin/env node

import yargs from "yargs"
import { hideBin } from "yargs/helpers"
import { BakeStep, BakeStepConfig, bakeSteps, SiteBaker } from "./SiteBaker.js"
import fs from "fs-extra"
import { normalize } from "path"
import * as db from "../db/db.js"
import { DEFAULT_LOCAL_BAKE_DIR } from "../site/SiteConstants.js"

const bakeDomainToFolder = async (
    baseUrl = "http://localhost:3000/",
    dir = DEFAULT_LOCAL_BAKE_DIR,
    bakeSteps?: BakeStepConfig
) => {
    dir = normalize(dir)
    await fs.mkdirp(dir)
    const baker = new SiteBaker(dir, baseUrl, bakeSteps)
    console.log(`Baking site locally with baseUrl '${baseUrl}' to dir '${dir}'`)

    await db.knexReadonlyTransaction(
        (trx) => baker.bakeAll(trx),
        db.TransactionCloseMode.Close
    )
}

void yargs(hideBin(process.argv))
    .command<{ baseUrl: string; dir: string; steps?: string[] }>(
        "$0 [baseUrl] [dir]",
        "Bake the site to a local folder",
        (yargs) => {
            yargs
                .positional("baseUrl", {
                    type: "string",
                    default: "http://localhost:3000/",
                    describe: "Base URL of the site",
                })
                .positional("dir", {
                    type: "string",
                    default: "localBake",
                    describe: "Directory to save the baked site",
                })
                .option("steps", {
                    type: "array",
                    choices: bakeSteps,
                    description: "Steps to perform during the baking process",
                })
        },
        async ({ baseUrl, dir, steps }) => {
            const bakeSteps = steps ? new Set(steps as BakeStep[]) : undefined
            await bakeDomainToFolder(baseUrl, dir, bakeSteps)
            process.exit(0)
        }
    )
    .help()
    .alias("help", "h")
    .strict().argv