AikidoSec-safe-chain/packages/safe-chain-bun/bin/safe-chain-bun.js

#!/usr/bin/env node

import { setup } from "../src/setup.js";
import { teardown } from "../src/teardown.js";

if (process.argv.length < 3) {
  console.error("No command provided. Please provide a command to execute.");
  console.log();
  writeHelp();
  process.exit(1);
}

const command = process.argv[2];

if (command === "help" || command === "--help" || command === "-h") {
  writeHelp();
  process.exit(0);
}

if (command === "setup") {
  const configFile = process.argv[3];
  setup(configFile);
} else if (command === "teardown") {
  const configFile = process.argv[3];
  teardown(configFile);
} else {
  console.error(`Unknown command: ${command}.`);
  console.log();
  writeHelp();
  process.exit(1);
}

function writeHelp() {
  console.log("Usage: safe-chain-bun <command>");
  console.log();
  console.log("Available commands: setup, teardown, help");
  console.log();
  console.log("- safe-chain-bun setup: Register Safe-Chain-Bun as a security scanner in ~/.bunfig.toml");
  console.log("- safe-chain-bun setup <file>: Register Safe-Chain-Bun as a security scanner in specified bunfig.toml file");
  console.log("- safe-chain-bun teardown: Remove Safe-Chain-Bun scanner from ~/.bunfig.toml");
  console.log("- safe-chain-bun teardown <file>: Remove Safe-Chain-Bun scanner from specified bunfig.toml file");
  console.log();
}