76 lines
2.4 KiB
TypeScript
76 lines
2.4 KiB
TypeScript
import type { Command } from "./commands"
|
|
import type { Command as SDKCommand } from "@opencode-ai/sdk"
|
|
import { showAlertDialog, showPromptDialog } from "../stores/alerts"
|
|
import { activeSessionId, executeCustomCommand } from "../stores/sessions"
|
|
import { getLogger } from "./logger"
|
|
|
|
const log = getLogger("actions")
|
|
|
|
export function commandRequiresArguments(template?: string): boolean {
|
|
if (!template) return false
|
|
return /\$(?:\d+|ARGUMENTS)/.test(template)
|
|
}
|
|
|
|
export async function promptForCommandArguments(command: SDKCommand): Promise<string | null> {
|
|
if (!commandRequiresArguments(command.template)) {
|
|
return ""
|
|
}
|
|
|
|
try {
|
|
return await showPromptDialog(`Arguments for /${command.name}`, {
|
|
title: "Custom command",
|
|
variant: "info",
|
|
inputLabel: "Arguments",
|
|
inputPlaceholder: "e.g. foo bar",
|
|
inputDefaultValue: "",
|
|
confirmLabel: "Run",
|
|
cancelLabel: "Cancel",
|
|
})
|
|
} catch (error) {
|
|
log.error("Failed to prompt for command arguments", error)
|
|
showAlertDialog("Failed to open arguments prompt.", {
|
|
title: "Command arguments",
|
|
variant: "error",
|
|
})
|
|
return null
|
|
}
|
|
}
|
|
|
|
function formatCommandLabel(name: string): string {
|
|
if (!name) return ""
|
|
return name.charAt(0).toUpperCase() + name.slice(1)
|
|
}
|
|
|
|
export function buildCustomCommandEntries(instanceId: string, commands: SDKCommand[]): Command[] {
|
|
return commands.map((cmd) => ({
|
|
id: `custom:${instanceId}:${cmd.name}`,
|
|
label: formatCommandLabel(cmd.name),
|
|
description: cmd.description ?? "Custom command",
|
|
category: "Custom Commands",
|
|
keywords: [cmd.name, ...(cmd.description ? cmd.description.split(/\s+/).filter(Boolean) : [])],
|
|
action: async () => {
|
|
const sessionId = activeSessionId().get(instanceId)
|
|
if (!sessionId || sessionId === "info") {
|
|
showAlertDialog("Select a session before running a custom command.", {
|
|
title: "Session required",
|
|
variant: "warning",
|
|
})
|
|
return
|
|
}
|
|
try {
|
|
const args = await promptForCommandArguments(cmd)
|
|
if (args === null) {
|
|
return
|
|
}
|
|
await executeCustomCommand(instanceId, sessionId, cmd.name, args)
|
|
} catch (error) {
|
|
log.error("Failed to run custom command", error)
|
|
showAlertDialog("Failed to run custom command. Check the console for details.", {
|
|
title: "Command failed",
|
|
variant: "error",
|
|
})
|
|
}
|
|
},
|
|
}))
|
|
}
|