import type { Argv } from "yargs"
import { App } from "../../app/app"
import { Bus } from "../../bus"
import { Provider } from "../../provider/provider"
import { Session } from "../../session"
import { Share } from "../../share/share"
import { Message } from "../../session/message"
import { UI } from "../ui"
import { VERSION } from "../version"

const COLOR = [
  UI.Style.TEXT_SUCCESS_BOLD,
  UI.Style.TEXT_INFO_BOLD,
  UI.Style.TEXT_HIGHLIGHT_BOLD,
  UI.Style.TEXT_WARNING_BOLD,
]

const TOOL: Record<string, [string, string]> = {
  opencode_todowrite: ["Todo", UI.Style.TEXT_WARNING_BOLD],
  opencode_todoread: ["Todo", UI.Style.TEXT_WARNING_BOLD],
  opencode_bash: ["Bash", UI.Style.TEXT_DANGER_BOLD],
  opencode_edit: ["Edit", UI.Style.TEXT_SUCCESS_BOLD],
  opencode_glob: ["Glob", UI.Style.TEXT_INFO_BOLD],
  opencode_grep: ["Grep", UI.Style.TEXT_INFO_BOLD],
  opencode_list: ["List", UI.Style.TEXT_INFO_BOLD],
  opencode_read: ["Read", UI.Style.TEXT_HIGHLIGHT_BOLD],
  opencode_write: ["Write", UI.Style.TEXT_SUCCESS_BOLD],
}

export const RunCommand = {
  command: "run [message..]",
  describe: "Run OpenCode with a message",
  builder: (yargs: Argv) => {
    return yargs
      .positional("message", {
        describe: "Message to send",
        type: "string",
        array: true,
        default: [],
      })
      .option("session", {
        describe: "Session ID to continue",
        type: "string",
      })
  },
  handler: async (args: {
    message: string[]
    session?: string
    printLogs?: boolean
  }) => {
    const message = args.message.join(" ")
    await App.provide(
      {
        cwd: process.cwd(),
        version: VERSION,
      },
      async () => {
        await Share.init()
        const session = args.session
          ? await Session.get(args.session)
          : await Session.create()

        UI.println(UI.Style.TEXT_HIGHLIGHT_BOLD + "◍  OpenCode", VERSION)
        UI.empty()
        UI.println(UI.Style.TEXT_NORMAL_BOLD + "> ", message)
        UI.empty()
        UI.println(
          UI.Style.TEXT_INFO_BOLD +
            "~  https://dev.opencode.ai/s/" +
            session.id.slice(-8),
        )
        UI.empty()

        function printEvent(color: string, type: string, title: string) {
          UI.println(
            color + `|`,
            UI.Style.TEXT_NORMAL +
              UI.Style.TEXT_DIM +
              ` ${type.padEnd(7, " ")}`,
            "",
            UI.Style.TEXT_NORMAL + title,
          )
        }

        Bus.subscribe(Message.Event.PartUpdated, async (evt) => {
          if (evt.properties.sessionID !== session.id) return
          const part = evt.properties.part
          const message = await Session.getMessage(
            evt.properties.sessionID,
            evt.properties.messageID,
          )

          if (
            part.type === "tool-invocation" &&
            part.toolInvocation.state === "result"
          ) {
            const metadata =
              message.metadata.tool[part.toolInvocation.toolCallId]
            const [tool, color] = TOOL[part.toolInvocation.toolName] ?? [
              part.toolInvocation.toolName,
              UI.Style.TEXT_INFO_BOLD,
            ]
            printEvent(color, tool, metadata.title)
          }

          if (part.type === "text") {
            if (part.text.includes("\n")) {
              UI.empty()
              UI.println(part.text)
              UI.empty()
              return
            }
            printEvent(UI.Style.TEXT_NORMAL_BOLD, "Text", part.text)
          }
        })

        const { providerID, modelID } = await Provider.defaultModel()
        await Session.chat({
          sessionID: session.id,
          providerID,
          modelID,
          parts: [
            {
              type: "text",
              text: message,
            },
          ],
        })
        UI.empty()
      },
    )
  },
}