feat: add tool type check for formatters

Author: sudo-tee

lua/opencode/ui/formatter/tools/apply_patch.lua

@@ -20,6 +20,10 @@ end
 ---@param output Output
 ---@param part OpencodeMessagePart
 function M.format(output, part)
+  if part.tool ~= 'apply_patch' then
+    return
+  end
+
   local metadata = part.state and part.state.metadata or {}
   for _, file in ipairs(metadata.files or {}) do
     local utils = require('opencode.ui.formatter.utils')

lua/opencode/ui/formatter/tools/glob.lua

@@ -3,6 +3,10 @@ local M = {}
 ---@param output Output
 ---@param part OpencodeMessagePart
 function M.format(output, part)
+  if part.tool ~= 'glob' then
+    return
+  end
+
   local input = part.state and part.state.input or {}
   local metadata = part.state and part.state.metadata or {}
 

lua/opencode/ui/formatter/tools/grep.lua

@@ -19,6 +19,10 @@ end
 ---@param output Output
 ---@param part OpencodeMessagePart
 function M.format(output, part)
+  if part.tool ~= 'grep' then
+    return
+  end
+
   local metadata = part.state and part.state.metadata or {}
   local input = part.state and part.state.input or nil
 

lua/opencode/ui/formatter/tools/list.lua

@@ -3,6 +3,9 @@ local M = {}
 ---@param output Output
 ---@param part OpencodeMessagePart
 function M.format(output, part)
+  if part.tool ~= 'list' then
+    return
+  end
   local input = part.state and part.state.input or {}
   local metadata = part.state and part.state.metadata or {}
   local tool_output = part.state and part.state.output or ''

lua/opencode/ui/formatter/tools/question.lua

@@ -3,6 +3,10 @@ local M = {}
 ---@param output Output
 ---@param part OpencodeMessagePart
 function M.format(output, part)
+  if part.tool ~= 'question' then
+    return
+  end
+
   local input = part.state and part.state.input or {}
   local metadata = part.state and part.state.metadata or {}
 

lua/opencode/ui/formatter/tools/task.lua

@@ -25,6 +25,10 @@ end
 ---@param part OpencodeMessagePart
 ---@param get_child_parts? fun(session_id: string): OpencodeMessagePart[]?
 function M.format(output, part, get_child_parts)
+  if part.tool ~= 'task' then
+    return
+  end
+
   local input = part.state and part.state.input or {}
   local metadata = part.state and part.state.metadata or {}
   local tool_output = part.state and part.state.output or ''

lua/opencode/ui/formatter/tools/todowrite.lua

@@ -3,11 +3,20 @@ local M = {}
 ---@param output Output
 ---@param part OpencodeMessagePart
 function M.format(output, part)
+  if part.tool ~= 'todowrite' then
+    return
+  end
   local utils = require('opencode.ui.formatter.utils')
   local config = require('opencode.config')
 
   local icons = require('opencode.ui.icons')
-  utils.format_action(output, icons.get('plan'), 'plan', (part.state and part.state.title or ''), utils.get_duration_text(part))
+  utils.format_action(
+    output,
+    icons.get('plan'),
+    'plan',
+    (part.state and part.state.title or ''),
+    utils.get_duration_text(part)
+  )
   if not config.ui.output.tools.show_output then
     return
   end

lua/opencode/ui/formatter/tools/webfetch.lua

@@ -4,8 +4,18 @@ local utils = require('opencode.ui.formatter.utils')
 ---@param output Output
 ---@param part OpencodeMessagePart
 function M.format(output, part)
+  if part.tool ~= 'webfetch' then
+    return
+  end
+
   local icons = require('opencode.ui.icons')
-  utils.format_action(output, icons.get('web'), 'fetch', part.state and part.state.input and part.state.input.url, utils.get_duration_text(part))
+  utils.format_action(
+    output,
+    icons.get('web'),
+    'fetch',
+    part.state and part.state.input and part.state.input.url,
+    utils.get_duration_text(part)
+  )
 end
 
 ---@param _ OpencodeMessagePart