feat(state): add server port setter and session helpers

Author: sudo-tee

lua/opencode/context.lua

@@ -116,12 +116,10 @@ end
 -- Delegate global state management to ChatContext
 function M.add_selection(selection)
   ChatContext.add_selection(selection)
-  state.context.set_context_updated_at(vim.uv.now())
 end
 
 function M.remove_selection(selection)
   ChatContext.remove_selection(selection)
-  state.context.set_context_updated_at(vim.uv.now())
 end
 
 function M.clear_selections()
@@ -216,13 +214,11 @@ function M.add_file(file)
 
   file = vim.fn.fnamemodify(file, ':p')
   ChatContext.add_file(file)
-  state.context.set_context_updated_at(vim.uv.now())
 end
 
 function M.remove_file(file)
   file = vim.fn.fnamemodify(file, ':p')
   ChatContext.remove_file(file)
-  state.context.set_context_updated_at(vim.uv.now())
 end
 
 function M.clear_files()
@@ -231,12 +227,10 @@ end
 
 function M.add_subagent(subagent)
   ChatContext.add_subagent(subagent)
-  state.context.set_context_updated_at(vim.uv.now())
 end
 
 function M.remove_subagent(subagent)
   ChatContext.remove_subagent(subagent)
-  state.context.set_context_updated_at(vim.uv.now())
 end
 
 function M.clear_subagents()
@@ -249,7 +243,6 @@ end
 
 function M.load()
   ChatContext.load()
-  state.context.set_context_updated_at(vim.uv.now())
 end
 
 -- Context creation with delta logic (delegates to ChatContext)

lua/opencode/image_handler.lua

@@ -150,7 +150,6 @@ function M.paste_image_from_clipboard()
 
   if success then
     context.add_file(image_path)
-    state.context.set_context_updated_at(os.time())
     vim.notify('Image saved and added to context: ' .. vim.fn.fnamemodify(image_path, ':t'), vim.log.levels.INFO)
     return true
   end

lua/opencode/state/jobs.lua

@@ -31,6 +31,13 @@ function M.clear_server()
   return store.set('opencode_server', nil)
 end
 
+---@param port integer
+function M.set_server_port(port)
+  store.mutate('opencode_server', function(server)
+    server.port = port
+  end)
+end
+
 ---@param client OpencodeApiClient|nil
 function M.set_api_client(client)
   return store.set('api_client', client)

lua/opencode/state/session.lua

@@ -41,4 +41,22 @@ function M.set_user_message_count(count)
   return store.set('user_message_count', count)
 end
 
+---Increment/decrement the message count for a session, clamped to >= 0
+---@param session_id string
+---@param delta integer
+function M.increment_user_message_count(session_id, delta)
+  store.mutate('user_message_count', function(counts)
+    local new_value = (counts[session_id] or 0) + delta
+    counts[session_id] = new_value >= 0 and new_value or 0
+  end)
+end
+
+---Update active_session without emitting a change event, used when a silent
+---in-place update is needed (e.g. session metadata refresh that must not
+---trigger a re-render)
+---@param session Session
+function M.update_silently(session)
+  store.set_raw('active_session', session)
+end
+
 return M

tests/minimal/init.lua

@@ -34,6 +34,8 @@ vim.opt.termguicolors = true
 
 require('opencode')
 
-if vim.treesitter and vim.treesitter.start then
-  vim.treesitter.start = function() end
+-- Safely stub out treesitter start during tests. Accessing `vim.treesitter`
+local ok, ts = pcall(require, 'vim.treesitter')
+if ok and ts and ts.start then
+  ts.start = function() end
 end