@mtayfur/opencode-prompt-enhancer 0.0.5 → 0.0.7

package/README.md

@@ -1,11 +1,20 @@
 # opencode-prompt-enhancer
 
-OpenCode TUI plugin that rewrites rough prompt drafts into stronger prompts with workspace context.
+OpenCode TUI plugin that rewrites rough prompt drafts into clearer, stronger prompts.
 
-## Package
+## What it does
 
-- npm: `opencode-prompt-enhancer`
-- GitHub: `mtayfur/opencode-prompt-enhancer`
+- Rewrites rough prompt drafts into clearer, stronger prompts.
+- Uses lightweight workspace context.
+- Keeps the original intent and language, and does not read file contents.
+
+## Context used
+
+The enhancer uses:
+
+- the current working directory
+- recent user prompts in the current session
+- files changed in the current session
 
 ## Install
 
@@ -14,20 +23,20 @@ Add the package to OpenCode's `tui.json` plugin list:
 ```jsonc
 {
   "plugin": [
-    "opencode-prompt-enhancer@latest"
+    "@mtayfur/opencode-prompt-enhancer@latest"
   ]
 }
 ```
 
-## Use
+OpenCode `>=1.3.14` is required.
 
-After restarting OpenCode, press `Ctrl+E` or run `/enhance`.
-
-The plugin opens a dialog, rewrites your draft with workspace context, and writes the enhanced prompt back into the current input.
-
-## Context
+## Use
 
-The enhancer uses the current working directory, branch, recent user prompts, changed files, and active todos. It passes lightweight context only and does not read file contents.
+1. Open OpenCode in a workspace.
+2. Enter a rough prompt in the TUI prompt.
+3. Press `Ctrl+E` or run `/enhance`.
+4. Review the prefilled dialog, edit it if needed, and confirm.
+5. The enhanced prompt replaces the current input.
 
 ## Development
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mtayfur/opencode-prompt-enhancer",
-  "version": "0.0.5",
+  "version": "0.0.7",
   "description": "OpenCode plugin that rewrites rough drafts into stronger prompts.",
   "publishConfig": {
     "access": "public"

package/plugins/enhancer-system-prompt.ts

@@ -0,0 +1,30 @@
+export const ENHANCER_SYSTEM_PROMPT = `You are a prompt editor for OpenCode, an AI coding assistant.
+
+Goal:
+Rewrite the user's draft into the strongest possible next prompt for OpenCode. Preserve the user's intent, scope, priorities, and language. Improve clarity, specificity, and execution readiness without changing the requested outcome.
+
+Capabilities:
+OpenCode can inspect the workspace, edit files, run commands, and verify changes. When the draft does not explicitly ask for discussion or planning only, rewrite it so OpenCode can act directly.
+
+Rewrite rules:
+- Preserve the request form. Questions stay questions; bug-fix requests stay bug-fix requests; review requests stay review requests.
+- Preserve the language of the draft. Do not translate.
+- Preserve inline code, file paths, command strings, error messages, identifiers, and quoted phrases verbatim when they matter.
+- Preserve explicit constraints, file names, commands, acceptance criteria, and user wording.
+- Use workspace context only when it directly clarifies or narrows the task.
+- Resolve vague references like "this", "that bug", or "the plugin" only when context makes them clear.
+- Do not invent details when context is ambiguous.
+- Expand vague verbs into concrete actions when it helps OpenCode act immediately, such as identify the root cause, apply the smallest correct fix, remove dead code, keep scope tight, preserve behavior, follow local conventions, and verify the changed path.
+- Keep the rewrite proportional to the draft. Do not add background, motivation, or steps the draft did not imply.
+- Match the draft's structure and tone. Keep prose as prose and lists as lists.
+- If the draft is already precise, make only minimal cleanup.
+
+Hard constraints:
+- Do not invent requirements, files, APIs, dependencies, bugs, or acceptance criteria.
+- Do not broaden scope with extra features, refactors, tests, or docs unless the draft implies them.
+- Do not ask follow-up questions.
+- Do not mention context, tags, instructions, or the rewriting process.
+- Do not output explanations, markdown fences, or commentary.
+
+Output:
+Return exactly one enhanced user prompt as plain text.`

package/plugins/prompt-enhancer.tsx

@@ -7,54 +7,15 @@ import type {
   TuiPromptInfo,
   TuiPromptRef,
   TuiRouteCurrent,
-  TuiSidebarFileItem,
-  TuiSidebarTodoItem,
   TuiSlotPlugin,
 } from "@opencode-ai/plugin/tui"
+import { ENHANCER_SYSTEM_PROMPT } from "./enhancer-system-prompt"
 
-const MAX_RECENT_MESSAGES = 6
-const MAX_CHANGED_FILES = 30
-const MAX_PROMPT_PREVIEW_LENGTH = 300
-const MAX_TODOS = 10
+const MAX_RECENT_MESSAGES = 3
+const MAX_CHANGED_FILES = 25
+const MAX_PROMPT_PREVIEW_LENGTH = 250
 const DIALOG_TITLE = "Enhance Prompt"
 const TOAST_TITLE = "Prompt enhancer"
-const TEMP_SESSION_TITLE = "Prompt Enhancer"
-const HOME_PROMPT_PLACEHOLDERS = {
-  normal: ["Fix a TODO in the codebase", "What is the tech stack of this project?", "Fix broken tests"],
-  shell: ["ls -la", "git status", "pwd"],
-}
-
-const ENHANCER_SYSTEM_PROMPT = `You rewrite rough user drafts into strong prompts for OpenCode, an AI coding assistant.
-
-Goal:
-Turn the user's draft into the best possible next prompt for OpenCode. Preserve the user's intent, scope, and priorities exactly. Improve clarity, specificity, and execution readiness without changing the requested outcome.
-
-OpenCode context:
-OpenCode can inspect the workspace, read and edit files, run commands, and verify changes. Unless the draft explicitly asks for planning or explanation only, write the prompt so OpenCode can act directly.
-
-Rules:
-- Preserve the original request type.
-- If the draft is a question, return a better question.
-- If the draft is a bug fix, return a better bug-fix request.
-- If the draft is a review, return a better review request.
-- Preserve explicit constraints, file names, commands, error messages, acceptance criteria, and user wording when they matter.
-- Use workspace context only when it clearly helps disambiguate or narrow the task.
-- Reference specific files, paths, branches, recent prompts, changed files, or todos only when they are directly relevant.
-- Resolve vague references like "this", "that bug", or "the plugin" from context when clear.
-- If context does not clearly resolve a reference, do not invent details.
-- Expand vague verbs into concrete actions when helpful, such as: identify root cause, apply the smallest correct fix, remove dead code, keep scope tight, preserve behavior, follow local conventions, and verify the changed path.
-- Prefer wording that helps OpenCode start work immediately.
-- If the draft is already precise, return it with minimal cleanup.
-
-Do not:
-- Do not invent requirements, files, APIs, dependencies, bugs, or acceptance criteria.
-- Do not broaden scope with extra features, refactors, tests, or docs unless the draft implies them.
-- Do not ask follow-up questions inside the rewritten prompt.
-- Do not mention the existence of context, tags, or these instructions.
-- Do not output explanations, markdown fences, or commentary.
-
-Output:
-Return exactly one enhanced user prompt as plain text.`
 
 type ModelRef = {
   providerID: string
@@ -89,41 +50,20 @@ function parseModelString(value: string | undefined): ModelRef | undefined {
   }
 }
 
-function getModelOverride(options: PluginOptions | undefined): ModelRef | undefined {
-  const value = typeof options?.model === "string" ? options.model : undefined
-  return parseModelString(value)
-}
-
 function isSessionRoute(route: TuiRouteCurrent): route is Extract<TuiRouteCurrent, { name: "session" }> {
   return route.name === "session"
 }
 
-function isUserMessage(message: Message): message is Extract<Message, { role: "user" }> {
-  return message.role === "user"
-}
-
-function isVisibleTextPart(part: Part): part is TextPart {
-  return part.type === "text" && !part.ignored
-}
-
 function extractVisibleText(parts: ReadonlyArray<Part>): string {
   return parts
-    .filter(isVisibleTextPart)
+    .filter((part): part is TextPart => part.type === "text" && !part.ignored)
     .map((part) => part.text)
     .join("")
     .trim()
 }
 
-function truncate(value: string, maxLength: number): string {
-  return value.length > maxLength ? `${value.slice(0, maxLength)}...` : value
-}
-
-function formatTodo(todo: TuiSidebarTodoItem): string {
-  return `  [${todo.status || "?"}] ${todo.content || "untitled"}`
-}
-
 function resolveEnhancerModel(api: Api, options: PluginOptions | undefined): ModelRef | undefined {
-  const override = getModelOverride(options)
+  const override = typeof options?.model === "string" ? parseModelString(options.model) : undefined
   if (override) return override
 
   return parseModelString(api.state.config.small_model || api.state.config.model)
@@ -156,25 +96,14 @@ function samePromptTarget(left: PromptTarget | undefined, right: PromptTarget |
   return false
 }
 
-function resolvePromptTarget(route: TuiRouteCurrent, workspaceID?: string): PromptTarget | undefined {
-  if (route.name === "home") return { name: "home", workspaceID }
-  if (isSessionRoute(route)) {
-    return { name: "session", sessionID: route.params.sessionID }
-  }
-  return undefined
-}
-
-function isPromptTargetActive(route: TuiRouteCurrent, target: PromptTarget): boolean {
-  if (target.name === "home") return route.name === "home"
-  return isSessionRoute(route) && route.params.sessionID === target.sessionID
-}
-
-function currentPromptTarget(api: Api, state: PluginState): PromptTarget | undefined {
-  return state.promptTarget ?? resolvePromptTarget(api.route.current)
-}
-
 function isPromptHandleActive(api: Api, state: PluginState, handle: PromptHandle): boolean {
-  if (!isPromptTargetActive(api.route.current, handle.target)) return false
+  const route = api.route.current
+  const target = handle.target
+  if (target.name === "home") {
+    if (route.name !== "home") return false
+  } else if (!isSessionRoute(route) || route.params.sessionID !== target.sessionID) {
+    return false
+  }
   if (api.state.path.directory !== handle.directory) return false
 
   if (handle.ref) {
@@ -255,65 +184,41 @@ async function restorePrompt(
     return true
   }
 
-  return writePrompt(api, state, handle, prompt.input, signal, prompt)
+  return false
 }
 
 function gatherContext(api: Api): string {
   const sections: string[] = []
 
   const dir = api.state.path.directory
-  const branch = api.state.vcs?.branch
-  sections.push(`Working directory: ${dir}${branch ? ` (branch: ${branch})` : ""}`)
+  sections.push(`Working directory: ${dir}`)
 
   const route = api.route.current
   if (isSessionRoute(route)) {
     const sessionID = route.params.sessionID
     const messages = api.state.session.messages(sessionID)
 
-    const userMessages = messages.filter(isUserMessage)
-    const recent = userMessages.slice(-MAX_RECENT_MESSAGES)
+    const userMessages = messages.filter((message): message is Extract<Message, { role: "user" }> => message.role === "user")
+    const recent = userMessages.slice(-MAX_RECENT_MESSAGES).reverse()
     if (recent.length > 0) {
       const prompts: string[] = []
       for (const msg of recent) {
         const text = extractVisibleText(api.state.part(msg.id))
         if (text) {
-          prompts.push(truncate(text, MAX_PROMPT_PREVIEW_LENGTH))
+          prompts.push(text.length > MAX_PROMPT_PREVIEW_LENGTH ? `${text.slice(0, MAX_PROMPT_PREVIEW_LENGTH)}...` : text)
         }
       }
       if (prompts.length > 0) {
-        sections.push(`Recent user prompts in this session (oldest first):\n${prompts.map((p, i) => `${i + 1}. ${p}`).join("\n")}`)
+        sections.push(`Recent user prompts in this session (newest first):\n${prompts.map((p, i) => `${i + 1}. ${p}`).join("\n")}`)
       }
     }
 
     const diff = api.state.session.diff(sessionID)
     if (diff.length > 0) {
-      const files = diff.slice(0, MAX_CHANGED_FILES).map((f: TuiSidebarFileItem) =>
-        `  ${f.file} (+${f.additions} -${f.deletions})`
-      )
-      const label = diff.length > MAX_CHANGED_FILES
-        ? `Files changed in session (showing ${MAX_CHANGED_FILES} of ${diff.length}):`
-        : `Files changed in session:`
-      sections.push(`${label}\n${files.join("\n")}`)
+      const files = diff.slice(0, MAX_CHANGED_FILES).map((f) => `  ${f.file}`)
+      sections.push(`Files changed in session:\n${files.join("\n")}`)
     }
 
-    const todos = api.state.session.todo(sessionID)
-    if (todos.length > 0) {
-      const todoLines = todos.slice(0, MAX_TODOS).map(formatTodo)
-      sections.push(`Active todos:\n${todoLines.join("\n")}`)
-    }
-  }
-
-  return sections.join("\n\n")
-}
-
-function buildUserMessage(input: string, context: string): string {
-  const sections = [
-    "Rewrite the following user draft into a stronger prompt.",
-    `User draft:\n${input}`,
-  ]
-
-  if (context) {
-    sections.splice(1, 0, `Workspace context (use only if directly relevant):\n${context}`)
   }
 
   return sections.join("\n\n")
@@ -328,17 +233,24 @@ async function enhanceWithModel(
   const directory = api.state.path.directory
   const model = resolveEnhancerModel(api, options)
   const context = gatherContext(api)
+  const userMessage = context
+    ? [
+        "Rewrite the following user draft into a stronger prompt.",
+        `Workspace context (use only if directly relevant):\n${context}`,
+        `User draft:\n${input}`,
+      ].join("\n\n")
+    : ["Rewrite the following user draft into a stronger prompt.", `User draft:\n${input}`].join("\n\n")
 
   const created = await api.client.session.create(
     {
       directory,
-      title: TEMP_SESSION_TITLE,
+      title: `Prompt Enhancer ${Math.random().toString(36).slice(2, 8)}`,
     },
     { signal, throwOnError: true },
   )
 
   const tempSessionID = created.data?.id
-  if (!tempSessionID) throw new Error("Prompt enhancer session creation failed.")
+  if (!tempSessionID) throw new Error("Failed to start prompt enhancer.")
 
   try {
     const response = await api.client.session.prompt(
@@ -350,7 +262,7 @@ async function enhanceWithModel(
         parts: [
           {
             type: "text",
-            text: buildUserMessage(input, context),
+            text: userMessage,
           },
         ],
       },
@@ -358,10 +270,10 @@ async function enhanceWithModel(
     )
 
     const parts = response.data?.parts
-    if (!parts) throw new Error("Enhancer model returned no response.")
+    if (!parts) throw new Error("Enhancer returned no response.")
 
     const enhanced = extractVisibleText(parts)
-    if (!enhanced) throw new Error("Enhancer model returned no text.")
+    if (!enhanced) throw new Error("Enhancer returned no text.")
     return enhanced
   } finally {
     try {
@@ -379,15 +291,19 @@ function openEnhanceDialog(
   signal: AbortSignal,
 ): void {
   if (state.enhancing) {
-    api.ui.toast({ variant: "warning", title: TOAST_TITLE, message: "Enhancement already in progress." })
+    api.ui.toast({ variant: "warning", title: TOAST_TITLE, message: "Enhancement in progress." })
     return
   }
 
   if (signal.aborted) return
 
-  const target = currentPromptTarget(api, state)
+  const target = state.promptTarget ?? (api.route.current.name === "home"
+    ? { name: "home" as const }
+    : isSessionRoute(api.route.current)
+      ? { name: "session" as const, sessionID: api.route.current.params.sessionID }
+      : undefined)
   if (!target) {
-    api.ui.toast({ variant: "warning", title: TOAST_TITLE, message: "Prompt enhancement is only available from a prompt view." })
+    api.ui.toast({ variant: "warning", title: TOAST_TITLE, message: "Enhancement only works from a prompt." })
     return
   }
 
@@ -402,19 +318,19 @@ function openEnhanceDialog(
   api.ui.dialog.replace(() => (
     <api.ui.DialogPrompt
       title={DIALOG_TITLE}
-      placeholder="Describe what you want to do..."
+      placeholder="Describe the task..."
       value={initialValue}
       onCancel={() => api.ui.dialog.clear()}
       onConfirm={(value) => {
         const input = value.trim()
         if (!input) {
-          api.ui.toast({ variant: "warning", title: TOAST_TITLE, message: "Prompt is empty." })
+          api.ui.toast({ variant: "warning", title: TOAST_TITLE, message: "Enter a prompt first." })
           api.ui.dialog.clear()
           return
         }
 
         if (!isPromptHandleActive(api, state, handle)) {
-          api.ui.toast({ variant: "warning", title: TOAST_TITLE, message: "Prompt changed while the dialog was open." })
+          api.ui.toast({ variant: "warning", title: TOAST_TITLE, message: "Prompt changed while dialog was open." })
           api.ui.dialog.clear()
           return
         }
@@ -448,7 +364,7 @@ function openEnhanceDialog(
               api.ui.toast({
                 variant: "warning",
                 title: TOAST_TITLE,
-                message: "Enhanced prompt is ready, but the original prompt is no longer active.",
+                message: "Enhanced prompt is ready, but that prompt is no longer active.",
               })
               return
             }
@@ -456,7 +372,7 @@ function openEnhanceDialog(
             api.ui.toast({
               variant: "success",
               title: "Prompt enhanced",
-              message: "Enhanced prompt written to input.",
+              message: "Enhanced prompt added to input.",
               duration: 3000,
             })
           } catch (error) {
@@ -478,10 +394,10 @@ function openEnhanceDialog(
               }
             }
 
-            const baseMessage = error instanceof Error ? error.message : "Model enhancement failed."
+            const baseMessage = error instanceof Error ? error.message : "Prompt enhancement failed."
             const message = restored
               ? baseMessage
-              : `${baseMessage} Original prompt could not be restored because the active prompt changed.`
+              : `${baseMessage} Original prompt could not be restored because the prompt changed.`
             api.ui.toast({ variant: "error", title: TOAST_TITLE, message })
           } finally {
             state.enhancing = false
@@ -503,7 +419,6 @@ const tui: TuiPlugin = async (api, options) => {
             ref={(ref) => bindPromptRef(state, { name: "home", workspaceID: props.workspace_id }, props.ref, ref)}
             workspaceID={props.workspace_id}
             right={<api.ui.Slot name="home_prompt_right" workspace_id={props.workspace_id} />}
-            placeholders={HOME_PROMPT_PLACEHOLDERS}
           />
         )
       },
@@ -528,7 +443,7 @@ const tui: TuiPlugin = async (api, options) => {
     {
       title: DIALOG_TITLE,
       value: "prompt-enhancer.enhance",
-      description: "Enhance prompt with project context (Ctrl+E)",
+      description: "Enhance current prompt (Ctrl+E)",
       category: "Prompt",
       keybind: "ctrl+e",
       suggested: true,