webstudio/packages/cli/src/cli.ts

import { exit, argv } from "node:process";
import { hideBin } from "yargs/helpers";
import { GLOBAL_CONFIG_FILE } from "./config";
import { createFileIfNotExists } from "./fs-utils";
import { link, linkOptions } from "./commands/link";
import { sync, syncOptions } from "./commands/sync";
import { build, buildOptions } from "./commands/build";
import { initFlow } from "./commands/init-flow";
import makeCLI from "yargs";
import packageJson from "../package.json" assert { type: "json" };
import type { CommonYargsArgv } from "./commands/yargs-types";

export const main = async () => {
  try {
    await createFileIfNotExists(GLOBAL_CONFIG_FILE, "{}");

    const cmd: CommonYargsArgv = makeCLI(hideBin(argv))
      .strict()
      .fail(function (msg, err, yargs) {
        if (err) {
          throw err; // preserve stack
        }

        console.error(msg);

        console.error(yargs.help());

        process.exit(1);
      })
      .wrap(null)
      .option("v", {
        describe: "Show version number",
        type: "boolean",
      })
      .option("h", {
        describe: "Show all commands",
        alias: "help",
        type: "boolean",
      })
      .scriptName("webstudio")
      .usage(
        `Webstudio CLI (${packageJson.version}) allows you to setup, sync, build and preview your project.`
      );

    cmd.version(packageJson.version).alias("v", "version");

    cmd.command(
      ["build"],
      "Build the project",
      (yargs: CommonYargsArgv) => {
        return buildOptions(yargs).demandOption(
          "template",
          "Please specify a template to use for the build"
        );
      },
      build
    );
    cmd.command(["link"], "Link the project with the cloud", linkOptions, link);
    cmd.command(["sync"], "Sync your project", syncOptions, sync);
    cmd.command(["$0", "init"], "Setup the project", buildOptions, initFlow);

    await cmd.parse();
  } catch (error) {
    console.error(error);
    exit(1);
  }
};