import type { Argv } from "yargs"

import { cmd } from "./cmd"

import { bootstrap } from "../bootstrap"

import { UI } from "../ui"

import { Executor } from "../../molecule/executor"

import type { Molecule } from "../../molecule/types"

import { BashTool } from "../../tool/bash"

import { WriteTool } from "../../tool/write"

import { EditTool } from "../../tool/edit"

import { ReadTool } from "../../tool/read"

import { GrepTool } from "../../tool/grep"

import { GlobTool } from "../../tool/glob"

import { ListTool } from "../../tool/ls"

export const MoleculeCommand = cmd({

command: "molecule",

describe: "Execute molecules - atomic, auditable work units",

builder: (yargs: Argv) => {

return yargs.command(

"run <spec-file>",

"Execute a molecule from spec file",

(yargs) => {

return yargs

.positional("spec-file", {

describe: "Path to molecule spec file (JSON or TypeScript)",

type: "string",

demandOption: true,

})

.option("dry-run", {

describe: "Validate without executing",

type: "boolean",

default: false,

})

},

async (args) => {

const specFile = args["spec-file"] as string

await bootstrap(process.cwd(), async () => {

const file = Bun.file(specFile)

if (!(await file.exists())) {

UI.error(`Spec file not found: ${specFile}`)

process.exit(1)

}

const content = await file.text()

let spec: Molecule.Spec

if (specFile.endsWith(".json")) {

spec = JSON.parse(content)

} else if (specFile.endsWith(".ts") || specFile.endsWith(".js")) {

const module = await import(specFile)

spec = module.default || module.spec

} else {

UI.error("Spec file must be .json, .ts, or .js")

process.exit(1)

}

if (args["dry-run"]) {

UI.println(UI.Style.TEXT_SUCCESS_BOLD + "✓ Spec is valid")

UI.println(UI.Style.TEXT_NORMAL + " ID: " + spec.id)

UI.println(UI.Style.TEXT_NORMAL + " Description: " + spec.description)

UI.println(UI.Style.TEXT_NORMAL + " Actions: " + spec.actions.length)

UI.println(UI.Style.TEXT_NORMAL + " Oracles: " + spec.oracles.length)

return

}

const toolRegistry = new Map()

toolRegistry.set("bash", BashTool)

toolRegistry.set("write", WriteTool)

toolRegistry.set("edit", EditTool)

toolRegistry.set("read", ReadTool)

toolRegistry.set("grep", GrepTool)

toolRegistry.set("glob", GlobTool)

toolRegistry.set("list", ListTool)

const ctx: Executor.Context = {

sessionID: "molecule-" + Date.now(),

messageID: "msg-" + Date.now(),

agent: "build",

toolRegistry,

}

UI.println(UI.Style.TEXT_INFO_BOLD + "▸ Executing molecule: " + spec.id)

UI.println(UI.Style.TEXT_DIM + " " + spec.description)

const startTime = Date.now()

const result = await Executor.execute(spec, ctx)

const duration = Date.now() - startTime

if (result.success) {

UI.println(UI.Style.TEXT_SUCCESS_BOLD + "✓ Success" + UI.Style.TEXT_DIM + ` (${duration}ms)`)

UI.println(UI.Style.TEXT_NORMAL + " Input hash: " + result.attestation.inputHash.slice(0, 16) + "...")

UI.println(UI.Style.TEXT_NORMAL + " Output hash: " + result.attestation.outputHash.slice(0, 16) + "...")

if (result.attestation.oracleResults.length > 0) {

UI.println(UI.Style.TEXT_INFO_BOLD + " Oracles:")

for (const oracle of result.attestation.oracleResults) {

const status = oracle.passed ? "✓" : "✗"

const color = oracle.passed ? UI.Style.TEXT_SUCCESS : UI.Style.TEXT_DANGER

UI.println(color + " " + status + " " + oracle.oracle.check)

}

}

} else {

UI.println(UI.Style.TEXT_DANGER_BOLD + "✗ Failed" + UI.Style.TEXT_DIM + ` (${duration}ms)`)

if (result.errors) {

for (const error of result.errors) {

UI.println(UI.Style.TEXT_DANGER + " " + error.message)

}

}

process.exit(1)

}

})

},

)

},

handler: () => {},

})