index.ts

import { Commit } from "@/cli/commit";
import { Branch } from "@/cli/branch";
import { Setup } from "@/cli/setup";
import { Doctor } from "@/cli/doctor";
import { ModelCommand } from "@/cli/model";
import { EffortCommand } from "@/cli/effort";
import { AliasCommand } from "@/cli/alias";
import { Update } from "@/cli/update";
import { type CliCommand, parseArgs, showHelp, showVersion } from "@/cli/parser";
import { Future } from "@/libs/future";
import { absurd } from "@/libs/types";
import { checkUpdate } from "@/cli/update";

import color from "picocolors";

const NOTIFIER_COMMANDS = new Set<CliCommand["type"]>(["generate", "setup", "doctor", "model", "effort", "branch", "alias"]);

const main = () => {
  const args = process.argv.slice(2);

  const action = parseArgs(args).either(
    (err): Future<Error, void> => {
      console.error(color.red(err.message));
      return Future.reject(err);
    },
    (command): Future<Error, void> => {
      if (NOTIFIER_COMMANDS.has(command.type)) checkUpdate();

      switch (command.type) {
        case "generate":
          return Commit.create().chain((c) => c.run());
        case "setup":
          return Setup.create().chain((s) => s.run());
        case "doctor":
          return Doctor.create().run();
        case "model":
          return ModelCommand.create().chain((m) => m.run());
        case "effort":
          return EffortCommand.create().chain((e) => e.run());
        case "branch":
          return Branch.create().chain((b) => b.run());
        case "alias":
          return AliasCommand.create(command.action).chain((a) => a.run());
        case "update":
          return Update.create().run();
        case "version":
          showVersion();
          return Future.resolve(undefined);
        case "help":
          showHelp();
          return Future.resolve(undefined);
        default:
          absurd(command, `Unhandled command type: ${command}`);
      }
    }
  );

  action.fork(
    (_) => process.exit(1),
    () => process.exit(0)
  );
};

main();