1 files changed, 24 insertions, 0 deletions
diff --git a/lua/mason-core/ui/state.lua b/lua/mason-core/ui/state.lua
new file mode 100644
index 00000000..9d7bcdda
--- /dev/null
+++ b/lua/mason-core/ui/state.lua
@@ -0,0 +1,24 @@
+local M = {}
+
+---@generic T : table
+---@param initial_state T
+---@param subscriber fun(state: T)
+function M.create_state_container(initial_state, subscriber)
+    -- we do deepcopy to make sure instances of state containers doesn't mutate the initial state
+    local state = vim.deepcopy(initial_state)
+    local has_unsubscribed = false
+
+    ---@param mutate_fn fun(current_state: table)
+    return function(mutate_fn)
+        mutate_fn(state)
+        if not has_unsubscribed then
+            subscriber(state)
+        end
+    end, function()
+        return state
+    end, function(val)
+        has_unsubscribed = val
+    end
+end
+
+return M