add file/agent mentions to /tui/prompt-append

Author: ewired

packages/opencode/src/cli/cmd/tui/component/prompt/index.tsx

@@ -98,7 +98,33 @@ export function Prompt(props: PromptProps) {
 
   sdk.event.on(TuiEvent.PromptAppend.type, (evt) => {
     if (!input || input.isDestroyed) return
+    const oldLength = input.plainText.length
     input.insertText(evt.properties.text)
+
+    if (evt.properties.parts.length > 0) {
+      for (const part of evt.properties.parts) {
+        const extmarkId =
+          part.source &&
+          input.extmarks.create({
+            start: oldLength + (part.type === "file" ? part.source.text.start : part.source.start),
+            end: oldLength + (part.type === "file" ? part.source.text.end : part.source.end),
+            virtual: true,
+            styleId: part.type === "agent" ? agentStyleId : fileStyleId,
+            typeId: promptPartTypeId,
+          })
+
+        setStore(
+          produce((draft) => {
+            const partIndex = draft.prompt.parts.length
+            draft.prompt.parts.push(part)
+            if (extmarkId !== undefined) {
+              draft.extmarkToPartIndex.set(extmarkId, partIndex)
+            }
+          }),
+        )
+      }
+    }
+
     setTimeout(() => {
       // setTimeout is a workaround and needs to be addressed properly
       if (!input || input.isDestroyed) return

packages/opencode/src/cli/cmd/tui/event.ts

@@ -1,9 +1,21 @@
 import { BusEvent } from "@/bus/bus-event"
 import { Bus } from "@/bus"
+import { MessageV2 } from "@/session/message-v2"
 import z from "zod"
 
 export const TuiEvent = {
-  PromptAppend: BusEvent.define("tui.prompt.append", z.object({ text: z.string() })),
+  PromptAppend: BusEvent.define(
+    "tui.prompt.append",
+    z.object({
+      text: z.string(),
+      parts: z.array(
+        z.union([
+          MessageV2.AgentPart.omit({ id: true, sessionID: true, messageID: true }),
+          MessageV2.FilePart.omit({ id: true, sessionID: true, messageID: true }),
+        ]),
+      ),
+    }),
+  ),
   CommandExecute: BusEvent.define(
     "tui.command.execute",
     z.object({

packages/opencode/src/server/routes/tui.ts

@@ -3,6 +3,7 @@ import { describeRoute, validator, resolver } from "hono-openapi"
 import z from "zod"
 import { Bus } from "../../bus"
 import { Session } from "../../session"
+import { SessionPrompt } from "../../session/prompt"
 import { TuiEvent } from "@/cli/cmd/tui/event"
 import { AsyncQueue } from "../../util/queue"
 import { errors } from "../error"
@@ -95,9 +96,15 @@ export const TuiRoutes = lazy(() =>
           ...errors(400),
         },
       }),
-      validator("json", TuiEvent.PromptAppend.properties),
+      validator("json", z.object({ text: z.string() })),
       async (c) => {
-        await Bus.publish(TuiEvent.PromptAppend, c.req.valid("json"))
+        const { text } = c.req.valid("json")
+        await Bus.publish(TuiEvent.PromptAppend, {
+          text,
+          parts: (await SessionPrompt.resolvePromptParts(text)).filter(
+            (part) => part.type === "agent" || part.type === "file",
+          ),
+        })
         return c.json(true)
       },
     )

packages/sdk/js/src/gen/types.gen.ts

@@ -615,6 +615,24 @@ export type EventTuiPromptAppend = {
   type: "tui.prompt.append"
   properties: {
     text: string
+    parts: Array<
+      | {
+          type: "agent"
+          name: string
+          source?: {
+            value: string
+            start: number
+            end: number
+          }
+        }
+      | {
+          type: "file"
+          mime: string
+          filename?: string
+          url: string
+          source?: FilePartSource
+        }
+    >
   }
 }
 

packages/sdk/js/src/v2/gen/types.gen.ts

@@ -719,6 +719,24 @@ export type EventTuiPromptAppend = {
   type: "tui.prompt.append"
   properties: {
     text: string
+    parts: Array<
+      | {
+          type: "agent"
+          name: string
+          source?: {
+            value: string
+            start: number
+            end: number
+          }
+        }
+      | {
+          type: "file"
+          mime: string
+          filename?: string
+          url: string
+          source?: FilePartSource
+        }
+    >
   }
 }