feat(image_handler): add mention for pasted_image and mention restore

Author: sudo-tee

lua/opencode/image_handler.lua

@@ -4,6 +4,7 @@ local context = require('opencode.context')
 local state = require('opencode.state')
 
 local M = {}
+local cached_temp_dir = nil
 
 --- Check if a file exists and has content
 --- @param path string
@@ -137,11 +138,25 @@ local function try_base64_clipboard(temp_dir, timestamp)
   return success, success and image_path or nil
 end
 
+--- Restore full path of a pasted image by its name
+--- @param name string
+--- @return string?
+function M.restore_img_path(name)
+  if not cached_temp_dir or not name:find('^pasted_image_') then
+    return nil
+  end
+  local path = cached_temp_dir .. '/' .. name
+  return is_valid_file(path) and path or nil
+end
+
 --- Handle clipboard image data by saving it to a file and adding it to context
 --- @return boolean success True if image was successfully handled
 function M.paste_image_from_clipboard()
-  local temp_dir = vim.fn.tempname()
-  vim.fn.mkdir(temp_dir, 'p')
+  if not cached_temp_dir then
+    cached_temp_dir = vim.fn.tempname()
+    vim.fn.mkdir(cached_temp_dir, 'p')
+  end
+  local temp_dir = cached_temp_dir
   local timestamp = os.date('%Y%m%d_%H%M%S')
   local image_path = string.format('%s/pasted_image_%s.png', temp_dir, timestamp)
 
@@ -156,7 +171,12 @@ function M.paste_image_from_clipboard()
   end
 
   if success then
-    context.add_file(image_path)
+    require('opencode.ui.mention').mention(function(mention_cb)
+      local name = vim.fn.fnamemodify(image_path, ':t')
+      mention_cb(name)
+      context.add_file(image_path)
+    end)
+
     vim.notify('Image saved and added to context: ' .. vim.fn.fnamemodify(image_path, ':t'), vim.log.levels.INFO)
     return true
   end

lua/opencode/ui/mention.lua

@@ -143,6 +143,12 @@ function M.restore_mentions(buf)
       require('opencode.context').add_file(name)
       return
     end
+    if name:find('^pasted_image_') then
+      local restored_path = require('opencode.image_handler').restore_img_path(name)
+      if restored_path then
+        require('opencode.context').add_file(restored_path)
+      end
+    end
   end)
 end
 

tests/unit/image_handler_spec.lua

@@ -40,6 +40,7 @@ describe('image_handler', function()
       wsl_distro = 0,
       added_files = {},
       notifications = {},
+      existing_files = {},
     }
 
     vim.fn = setmetatable({
@@ -50,8 +51,13 @@ describe('image_handler', function()
         return mocks.temp_dir
       end,
       mkdir = function() end,
-      getfsize = function(_)
-        return 100 -- Simulating non-empty file
+      getfsize = function(path)
+        for _, f in ipairs(mocks.existing_files) do
+          if f == path then
+            return 100
+          end
+        end
+        return 0
       end,
       has = function(feature)
         if feature == 'win32' then
@@ -124,6 +130,7 @@ describe('image_handler', function()
   it('handles Darwin clipboard with osascript', function()
     mocks.os_name = 'Darwin'
     mocks.executable['osascript'] = 1
+    table.insert(mocks.existing_files, '/tmp/test_dir/pasted_image_20240101_120000.png')
 
     local success = image_handler.paste_image_from_clipboard()
 
@@ -139,6 +146,7 @@ describe('image_handler', function()
     mocks.os_name = 'Linux'
     mocks.executable['wl-paste'] = 1
     mocks.executable['xclip'] = 0
+    table.insert(mocks.existing_files, '/tmp/test_dir/pasted_image_20240101_120000.png')
 
     local success = image_handler.paste_image_from_clipboard()
 
@@ -151,6 +159,7 @@ describe('image_handler', function()
     mocks.os_name = 'Linux'
     mocks.executable['wl-paste'] = 0
     mocks.executable['xclip'] = 1
+    table.insert(mocks.existing_files, '/tmp/test_dir/pasted_image_20240101_120000.png')
 
     local success = image_handler.paste_image_from_clipboard()
 
@@ -162,6 +171,7 @@ describe('image_handler', function()
   it('handles Windows clipboard', function()
     mocks.os_name = 'Windows_NT'
     mocks.executable['powershell.exe'] = 1
+    table.insert(mocks.existing_files, '/tmp/test_dir/pasted_image_20240101_120000.png')
 
     local success = image_handler.paste_image_from_clipboard()
 
@@ -176,6 +186,7 @@ describe('image_handler', function()
     mocks.os_name = 'Linux'
     mocks.wsl_distro = 1
     mocks.executable['powershell.exe'] = 1
+    table.insert(mocks.existing_files, '/tmp/test_dir/pasted_image_20240101_120000.png')
 
     local success = image_handler.paste_image_from_clipboard()
 
@@ -195,6 +206,7 @@ describe('image_handler', function()
     mocks.os_name = 'Darwin'
     mocks.executable['osascript'] = 0 -- Force failure of system tool
     mocks.clipboard_content = 'data:image/png;base64,fakebasedata'
+    table.insert(mocks.existing_files, '/tmp/test_dir/pasted_image_20240101_120000.png')
 
     local success = image_handler.paste_image_from_clipboard()
 
@@ -228,4 +240,38 @@ describe('image_handler', function()
     assert.is_false(success)
     assert.equals(0, #mocks.added_files)
   end)
+
+  it('restores image path when file exists and name is valid', function()
+    mocks.os_name = 'Darwin'
+    mocks.executable['osascript'] = 1
+    -- Initialize cached_temp_dir
+    image_handler.paste_image_from_clipboard()
+
+    local img_name = 'pasted_image_test.png'
+    local expected_path = mocks.temp_dir .. '/' .. img_name
+    table.insert(mocks.existing_files, expected_path)
+
+    local restored_path = image_handler.restore_img_path(img_name)
+    assert.equals(expected_path, restored_path)
+  end)
+
+  it('returns nil when restoring image path with invalid name', function()
+    mocks.os_name = 'Darwin'
+    mocks.executable['osascript'] = 1
+    image_handler.paste_image_from_clipboard()
+
+    local restored_path = image_handler.restore_img_path('not_a_pasted_image.png')
+    assert.is_nil(restored_path)
+  end)
+
+  it('returns nil when restoring image path and file does not exist', function()
+    mocks.os_name = 'Darwin'
+    mocks.executable['osascript'] = 1
+    image_handler.paste_image_from_clipboard()
+
+    local img_name = 'pasted_image_missing.png'
+
+    local restored_path = image_handler.restore_img_path(img_name)
+    assert.is_nil(restored_path)
+  end)
 end)