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

Author: sudo-tee

lua/opencode/state/store.lua

@@ -49,7 +49,7 @@ local _state = {
 
 local _listeners = {}
 
----@param key string
+---@param key keyof OpencodeState
 ---@param opts? OpencodeProtectedStateSetOptions
 local function error_on_raw_write(key, opts)
   if opts and opts.silent then
@@ -65,16 +65,16 @@ end
 
 ---@generic K extends keyof OpencodeState
 ---@param key K
----@return OpencodeState[K]
+---@return std.RawGet<OpencodeState, K>
 function M.get(key)
   return _state[key]
 end
 
 ---@generic K extends keyof OpencodeState
 ---@param key K
----@param value OpencodeState[K]
+---@param value std.RawGet<OpencodeState, K>
 ---@param opts? OpencodeProtectedStateSetOptions
----@return OpencodeState[K]
+---@return std.RawGet<OpencodeState, K>
 function M.set(key, value, opts)
   local old = _state[key]
   opts = opts or { source = 'helper' }
@@ -93,19 +93,19 @@ end
 
 ---@generic K extends keyof OpencodeState
 ---@param key K
----@param value OpencodeState[K]
+---@param value std.RawGet<OpencodeState, K>
 ---@param opts? OpencodeProtectedStateSetOptions
----@return OpencodeState[K]
+---@return std.RawGet<OpencodeState, 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
 ---@param key K
----@param updater fun(current: OpencodeState[K]): OpencodeState[K]
+---@param updater fun(current: std.RawGet<OpencodeState, K>): std.RawGet<OpencodeState, K>
 ---@param opts? OpencodeProtectedStateSetOptions
----@return OpencodeState[K]
+---@return std.RawGet<OpencodeState, K>
 function M.update(key, updater, opts)
   local next_value = updater(_state[key])
   M.set(key, next_value, opts)
@@ -114,7 +114,7 @@ end
 
 ---@generic K extends keyof OpencodeState
 ---@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<OpencodeState, K>, old_val:std.RawGet<OpencodeState, K>)
 function M.subscribe(key, cb)
   if type(key) == 'table' then
     for _, current_key in ipairs(key) do
@@ -139,7 +139,7 @@ end
 
 ---@generic K extends keyof OpencodeState
 ---@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<OpencodeState, K>, old_val:std.RawGet<OpencodeState, K>)
 function M.unsubscribe(key, cb)
   key = key or '*'
   local list = _listeners[key]
@@ -156,8 +156,8 @@ end
 
 ---@generic K extends keyof OpencodeState
 ---@param key K
----@param new_val OpencodeState[K]
----@param old_val OpencodeState[K]
+---@param new_val std.RawGet<OpencodeState, K>
+---@param old_val std.RawGet<OpencodeState, K>
 function M.emit(key, new_val, old_val)
   vim.schedule(function()
     if _listeners[key] then
@@ -179,7 +179,7 @@ end
 
 ---@generic K extends keyof OpencodeState
 ---@param key K
----@param value OpencodeState[K] extends any[] and OpencodeState[K][integer] or never
+---@param value std.RawGet<OpencodeState, K> extends any[] and std.RawGet<OpencodeState, K>[integer] or never
 function M.append(key, value)
   if type(value) ~= 'table' then
     error('Value must be a table to append')