refactor(state): add mutation type annotations for state modules

Author: sudo-tee

lua/opencode/state/context.lua

@@ -1,5 +1,11 @@
+
 local store = require('opencode.state.store')
 
+---@class OpencodeContextStateMutations
+---@field set_current_context_config fun(config: OpencodeContextConfig|nil)
+---@field set_context_updated_at fun(timestamp: number|nil)
+---@field set_current_cwd fun(cwd: string|nil)
+
 local M = {}
 
 ---@param config OpencodeContextConfig|nil

lua/opencode/state/init.lua

@@ -23,64 +23,6 @@
 ---@class OpencodeToggleDecision
 ---@field action 'open'|'close'|'hide'|'close_hidden'|'restore_hidden'|'migrate'
 
----@class OpencodeSessionStateMutations
----@field set_active fun(session: Session|nil, opts?: OpencodeProtectedStateSetOptions)
----@field clear_active fun(opts?: OpencodeProtectedStateSetOptions)
----@field set_restore_points fun(points: RestorePoint[], opts?: OpencodeProtectedStateSetOptions)
----@field reset_restore_points fun(opts?: OpencodeProtectedStateSetOptions)
----@field set_last_sent_context fun(context: OpencodeContext|nil)
----@field set_user_message_count fun(count: table<string, number>)
-
----@class OpencodeJobStateMutations
----@field increment_count fun(delta?: integer, opts?: OpencodeProtectedStateSetOptions)
----@field decrement_count fun(delta?: integer, opts?: OpencodeProtectedStateSetOptions)
----@field set_count fun(count: integer, opts?: OpencodeProtectedStateSetOptions)
----@field set_server fun(server: OpencodeServer|nil, opts?: OpencodeProtectedStateSetOptions)
----@field clear_server fun(opts?: OpencodeProtectedStateSetOptions)
----@field set_api_client fun(client: OpencodeApiClient|nil)
----@field set_event_manager fun(manager: EventManager|nil)
----@field set_opencode_cli_version fun(version: string|nil)
-
----@class OpencodeUiStateMutations
----@field set_windows fun(windows: OpencodeWindowState|nil)
----@field clear_windows fun()
----@field set_opening fun(is_opening: boolean)
----@field set_panel_focused fun(is_focused: boolean)
----@field set_last_focused_window fun(win_type: 'input'|'output'|nil)
----@field set_display_route fun(route: any)
----@field clear_display_route fun()
----@field set_last_code_window fun(win_id: integer|nil)
----@field set_current_code_buf fun(bufnr: integer|nil)
----@field set_last_window_width_ratio fun(ratio: number|nil)
----@field clear_last_window_width_ratio fun()
----@field set_input_content fun(lines: table)
----@field set_saved_window_options fun(opts: table|nil)
----@field set_pre_zoom_width fun(width: integer|nil)
-
----@class OpencodeModelStateMutations
----@field set_mode fun(mode: string|nil)
----@field clear_mode fun()
----@field set_model fun(model: string|nil)
----@field clear_model fun()
----@field set_model_info fun(info: table|nil)
----@field set_variant fun(variant: string|nil)
----@field clear_variant fun()
----@field set_mode_model_map fun(mode_map: table<string, string>)
----@field set_mode_model_override fun(mode: string, model: string)
-
----@class OpencodeRendererStateMutations
----@field set_messages fun(messages: OpencodeMessage[]|nil)
----@field set_current_message fun(message: OpencodeMessage|nil)
----@field set_last_user_message fun(message: OpencodeMessage|nil)
----@field set_pending_permissions fun(permissions: OpencodePermission[])
----@field set_cost fun(cost: number)
----@field set_tokens_count fun(count: number)
-
----@class OpencodeContextStateMutations
----@field set_current_context_config fun(config: OpencodeContextConfig|nil)
----@field set_context_updated_at fun(timestamp: number|nil)
----@field set_current_cwd fun(cwd: string|nil)
-
 ---@class OpencodeState
 ---@field windows OpencodeWindowState|nil
 ---@field is_opening boolean

lua/opencode/state/jobs.lua

@@ -1,5 +1,15 @@
 local store = require('opencode.state.store')
 
+---@class OpencodeJobStateMutations
+---@field increment_count fun(delta?: integer, opts?: OpencodeProtectedStateSetOptions)
+---@field decrement_count fun(delta?: integer, opts?: OpencodeProtectedStateSetOptions)
+---@field set_count fun(count: integer, opts?: OpencodeProtectedStateSetOptions)
+---@field set_server fun(server: OpencodeServer|nil, opts?: OpencodeProtectedStateSetOptions)
+---@field clear_server fun(opts?: OpencodeProtectedStateSetOptions)
+---@field set_api_client fun(client: OpencodeApiClient|nil)
+---@field set_event_manager fun(manager: EventManager|nil)
+---@field set_opencode_cli_version fun(version: string|nil)
+
 local M = {}
 
 ---@param delta integer|nil

lua/opencode/state/renderer.lua

@@ -1,5 +1,14 @@
+
 local store = require('opencode.state.store')
 
+---@class OpencodeRendererStateMutations
+---@field set_messages fun(messages: OpencodeMessage[]|nil)
+---@field set_current_message fun(message: OpencodeMessage|nil)
+---@field set_last_user_message fun(message: OpencodeMessage|nil)
+---@field set_pending_permissions fun(permissions: OpencodePermission[])
+---@field set_cost fun(cost: number)
+---@field set_tokens_count fun(count: number)
+
 local M = {}
 
 ---@param messages OpencodeMessage[]|nil

lua/opencode/state/session.lua

@@ -1,5 +1,13 @@
 local store = require('opencode.state.store')
 
+---@class OpencodeSessionStateMutations
+---@field set_active fun(session: Session|nil, opts?: OpencodeProtectedStateSetOptions)
+---@field clear_active fun(opts?: OpencodeProtectedStateSetOptions)
+---@field set_restore_points fun(points: RestorePoint[], opts?: OpencodeProtectedStateSetOptions)
+---@field reset_restore_points fun(opts?: OpencodeProtectedStateSetOptions)
+---@field set_last_sent_context fun(context: OpencodeContext|nil)
+---@field set_user_message_count fun(count: table<string, number>)
+
 local M = {}
 
 ---@param session Session|nil

lua/opencode/state/ui.lua

@@ -1,5 +1,21 @@
 local store = require('opencode.state.store')
 
+---@class OpencodeUiStateMutations
+---@field set_windows fun(windows: OpencodeWindowState|nil)
+---@field clear_windows fun()
+---@field set_opening fun(is_opening: boolean)
+---@field set_panel_focused fun(is_focused: boolean)
+---@field set_last_focused_window fun(win_type: 'input'|'output'|nil)
+---@field set_display_route fun(route: any)
+---@field clear_display_route fun()
+---@field set_last_code_window fun(win_id: integer|nil)
+---@field set_current_code_buf fun(bufnr: integer|nil)
+---@field set_last_window_width_ratio fun(ratio: number|nil)
+---@field clear_last_window_width_ratio fun()
+---@field set_input_content fun(lines: table)
+---@field set_saved_window_options fun(opts: table|nil)
+---@field set_pre_zoom_width fun(width: integer|nil)
+
 local M = {}
 
 local _state = store.state()