refactor(state): use std.RawGet and keyof OpencodeState in annotations

Author: sudo-tee

lua/opencode/state.lua

@@ -1 +1 @@
-return require('opencode.state.init')
+return require('opencode.state.init') --[[@as OpencodeStateModule]]

lua/opencode/state/init.lua

@@ -23,7 +23,7 @@
 ---@class OpencodeToggleDecision
 ---@field action 'open'|'close'|'hide'|'close_hidden'|'restore_hidden'|'migrate'
 
----@class OpencodeState
+---@class OpencodeStateData
 ---@field windows OpencodeWindowState|nil
 ---@field is_opening boolean
 ---@field input_content table
@@ -33,9 +33,10 @@
 ---@field last_output_window_position integer[]|nil
 ---@field last_code_win_before_opencode integer|nil
 ---@field current_code_buf number|nil
+---@field current_code_view table|nil
 ---@field saved_window_options table|nil
----@field display_route any|nil
----@field current_mode string
+---@field display_route string|nil
+---@field current_mode string|nil
 ---@field last_output number
 ---@field last_sent_context OpencodeContext|nil
 ---@field current_context_config OpencodeContextConfig|nil
@@ -55,25 +56,14 @@
 ---@field job_count number
 ---@field user_message_count table<string, number>
 ---@field opencode_server OpencodeServer|nil
----@field api_client OpencodeApiClient
+---@field api_client OpencodeApiClient|nil
 ---@field event_manager EventManager|nil
 ---@field pre_zoom_width integer|nil
 ---@field last_window_width_ratio number|nil
 ---@field required_version string
 ---@field opencode_cli_version string|nil
 ---@field current_cwd string|nil
 ---@field _hidden_buffers OpencodeHiddenBuffers|nil
----@field append fun(key:string, value:any)
----@field remove fun(key:string, idx:number)
----@field subscribe fun(key:string|string[]|nil, cb:fun(key:string, new_val:any, old_val:any))
----@field unsubscribe fun(key:string|nil, cb:fun(key:string, new_val:any, old_val:any))
----@field is_running fun():boolean
----@field session OpencodeSessionStateMutations
----@field jobs OpencodeJobStateMutations
----@field ui OpencodeUiStateMutations
----@field model OpencodeModelStateMutations
----@field renderer OpencodeRendererStateMutations
----@field context OpencodeContextStateMutations
 
 local store = require('opencode.state.store')
 local session = require('opencode.state.session')
@@ -83,6 +73,18 @@ local model = require('opencode.state.model')
 local renderer = require('opencode.state.renderer')
 local context = require('opencode.state.context')
 
+---@class OpencodeStateModule
+---@field store OpencodeStateStore<OpencodeStateData>
+---@field session OpencodeSessionStateMutations
+---@field jobs OpencodeJobStateMutations
+---@field ui OpencodeUiStateMutations
+---@field model OpencodeModelStateMutations
+---@field renderer OpencodeRendererStateMutations
+---@field context OpencodeContextStateMutations
+---@field is_running fun():boolean
+
+---@alias OpencodeState OpencodeStateModule & OpencodeStateData
+---@type OpencodeState
 local M = {
   store = store,
   session = session,

lua/opencode/state/store.lua

@@ -4,7 +4,7 @@
 
 local M = {}
 
----@type OpencodeState
+---@type OpencodeStateData
 local _state = {
   windows = nil,
   is_opening = false,
@@ -49,7 +49,7 @@ local _state = {
 
 local _listeners = {}
 
----@param key string
+---@param key keyof OpencodeStateData
 ---@param opts? OpencodeProtectedStateSetOptions
 local function error_on_raw_write(key, opts)
   if opts and opts.silent then
@@ -63,18 +63,18 @@ function M.state()
   return _state
 end
 
----@generic K extends keyof OpencodeState
+---@generic K extends keyof OpencodeStateData
 ---@param key K
----@return OpencodeState[K]
+---@return std.RawGet<OpencodeStateData, K>
 function M.get(key)
   return _state[key]
 end
 
----@generic K extends keyof OpencodeState
+---@generic K extends keyof OpencodeStateData
 ---@param key K
----@param value OpencodeState[K]
+---@param value std.RawGet<OpencodeStateData, K>
 ---@param opts? OpencodeProtectedStateSetOptions
----@return OpencodeState[K]
+---@return std.RawGet<OpencodeStateData, K>
 function M.set(key, value, opts)
   local old = _state[key]
   opts = opts or { source = 'helper' }
@@ -91,30 +91,30 @@ function M.set(key, value, opts)
   return value
 end
 
----@generic K extends keyof OpencodeState
+---@generic K extends keyof OpencodeStateData
 ---@param key K
----@param value OpencodeState[K]
+---@param value std.RawGet<OpencodeStateData, K>
 ---@param opts? OpencodeProtectedStateSetOptions
----@return OpencodeState[K]
+---@return std.RawGet<OpencodeStateData, K>
 function M.set_raw(key, value, opts)
   local next_opts = vim.tbl_extend('force', { source = 'raw' }, opts or {})
   return M.set(key, value, next_opts)
 end
 
----@generic K extends keyof OpencodeState
+---@generic K extends keyof OpencodeStateData
 ---@param key K
----@param updater fun(current: OpencodeState[K]): OpencodeState[K]
+---@param updater fun(current: std.RawGet<OpencodeStateData, K>): std.RawGet<OpencodeStateData, K>
 ---@param opts? OpencodeProtectedStateSetOptions
----@return OpencodeState[K]
+---@return std.RawGet<OpencodeStateData, K>
 function M.update(key, updater, opts)
   local next_value = updater(_state[key])
   M.set(key, next_value, opts)
   return next_value
 end
 
----@generic K extends keyof OpencodeState
+---@generic K extends keyof OpencodeStateData
 ---@param key K|K[]|nil
----@param cb fun(key:K, new_val:OpencodeState[K], old_val:OpencodeState[K])
+---@param cb fun(key:K, new_val:std.RawGet<OpencodeStateData, K>, old_val:std.RawGet<OpencodeStateData, K>)
 function M.subscribe(key, cb)
   if type(key) == 'table' then
     for _, current_key in ipairs(key) do
@@ -137,9 +137,9 @@ function M.subscribe(key, cb)
   table.insert(_listeners[key], cb)
 end
 
----@generic K extends keyof OpencodeState
+---@generic K extends keyof OpencodeStateData
 ---@param key K|nil
----@param cb fun(key:K, new_val:OpencodeState[K], old_val:OpencodeState[K])
+---@param cb fun(key:K, new_val:std.RawGet<OpencodeStateData, K>, old_val:std.RawGet<OpencodeStateData, K>)
 function M.unsubscribe(key, cb)
   key = key or '*'
   local list = _listeners[key]
@@ -154,10 +154,10 @@ function M.unsubscribe(key, cb)
   end
 end
 
----@generic K extends keyof OpencodeState
+---@generic K extends keyof OpencodeStateData
 ---@param key K
----@param new_val OpencodeState[K]
----@param old_val OpencodeState[K]
+---@param new_val std.RawGet<OpencodeStateData, K>
+---@param old_val std.RawGet<OpencodeStateData, K>
 function M.emit(key, new_val, old_val)
   vim.schedule(function()
     if _listeners[key] then
@@ -177,9 +177,9 @@ function M.emit(key, new_val, old_val)
   end)
 end
 
----@generic K extends keyof OpencodeState
+---@generic K extends keyof OpencodeStateData
 ---@param key K
----@param value OpencodeState[K] extends any[] and OpencodeState[K][integer] or never
+---@param value std.RawGet<OpencodeStateData, K> extends any[] and std.RawGet<OpencodeStateData, K>[integer] or never
 function M.append(key, value)
   if type(value) ~= 'table' then
     error('Value must be a table to append')
@@ -196,7 +196,7 @@ function M.append(key, value)
   M.emit(key, _state[key], old)
 end
 
----@generic K extends keyof OpencodeState
+---@generic K extends keyof OpencodeStateData
 ---@param key K
 ---@param idx integer
 function M.remove(key, idx)