Renderer.mount: add tests

.emmyrc.json

@@ -0,0 +1,12 @@
+{
+  "$schema": "https://raw.githubusercontent.com/EmmyLuaLs/emmylua-analyzer-rust/refs/heads/main/crates/emmylua_code_analysis/resources/schema.json",
+  "runtime": {
+    "version": "LuaJIT"
+  },
+  "workspace": {
+    "library": [
+      "$VIMRUNTIME",
+      "library/busted"
+    ]
+  }
+}

.github/workflows/ci.yaml

@@ -16,7 +16,7 @@ jobs:
         run:
           nix-shell --pure --run 'true'
 
-      - name: Type-check with lua-language-server
+      - name: Type-check
         run:
           nix-shell --pure --run 'make lint'
 

.gitmodules

@@ -0,0 +1,4 @@
+[submodule "library/busted"]
+	path = library/busted
+	url = https://github.com/LuaCATS/busted
+	branch = main

.luacheckrc

@@ -1,2 +0,0 @@
--- :vim set ft=lua
-globals = { "vim" }

Makefile

@@ -1,8 +1,10 @@
+export VIMRUNTIME := $(shell nvim -u NORC --headless +'echo $$VIMRUNTIME' +'quitall' 2>&1)
+
 all: lint fmt-check test
 
 lint:
 	@echo "## Typechecking"
-	@lua-language-server --check=lua/u/ --checklevel=Error
+	@emmylua_check .
 
 fmt-check:
 	@echo "## Checking code format"

examples/filetree.lua

@@ -10,8 +10,8 @@
 --   change on the underlying filesystem.
 --------------------------------------------------------------------------------
 
---- @alias u.examples.FsDir { kind: 'dir'; path: string; expanded: boolean; children: u.examples.FsNode[] }
+--- @alias u.examples.FsDir { kind: 'dir', path: string, expanded: boolean, children: u.examples.FsNode[] }
---- @alias u.examples.FsFile { kind: 'file'; path: string }
+--- @alias u.examples.FsFile { kind: 'file', path: string }
 --- @alias u.examples.FsNode u.examples.FsDir | u.examples.FsFile
 --- @alias u.examples.ShowOpts { root_path?: string, width?: number, focus_path?: string }
 
@@ -58,7 +58,7 @@ function H.relative(path, base)
 end
 
 --- @param root_path string
---- @return { tree: u.examples.FsDir; path_to_node: table<string, u.examples.FsNode> }
+--- @return { tree: u.examples.FsDir, path_to_node: table<string, u.examples.FsNode> }
 function H.get_tree_inf(root_path)
   logger:info { 'get_tree_inf', root_path }
   --- @type table<string, u.examples.FsNode>
@@ -113,10 +113,10 @@ function H.populate_dir_children(tree, path_to_node)
 end
 
 --- @param opts {
----   bufnr: number;
+---   bufnr: number,
----   prev_winnr: number;
+---   prev_winnr: number,
----   root_path: string;
+---   root_path: string,
----   focus_path?: string;
+---   focus_path?: string
 --- }
 ---
 --- @return { expand: fun(path: string), collapse: fun(path: string) }
@@ -310,7 +310,7 @@ local function _render_in_buffer(opts)
   --
   local renderer = Renderer.new(opts.bufnr)
   tracker.create_effect(function()
-    --- @type { tree: u.examples.FsDir; path_to_node: table<string, u.examples.FsNode> }
+    --- @type { tree: u.examples.FsDir, path_to_node: table<string, u.examples.FsNode> }
     local tree_inf = s_tree_inf:get()
     local tree = tree_inf.tree
 
@@ -407,9 +407,9 @@ end
 --------------------------------------------------------------------------------
 
 --- @type {
----   bufnr: number;
+---   bufnr: number,
----   winnr: number;
+---   winnr: number,
----   controller: { expand: fun(path: string), collapse: fun(path: string) };
+---   controller: { expand: fun(path: string), collapse: fun(path: string) }
 --- } | nil
 local current_inf = nil
 

examples/notify.lua

@@ -23,10 +23,10 @@ vim.api.nvim_create_autocmd('VimResized', {
 })
 
 --- @alias u.examples.Notification {
----   kind: number;
+---   kind: number,
----   id: number;
+---   id: number,
----   text: string;
+---   text: string,
----   timer: uv.uv_timer_t;
+---   timer: uv.uv_timer_t
 --- }
 
 local M = {}

examples/picker.lua

@@ -44,22 +44,22 @@ local function shallow_copy_arr(arr) return vim.iter(arr):totable() end
 --    shortest portion of this function.
 --------------------------------------------------------------------------------
 
---- @alias u.examples.SelectController {
+--- @alias u.examples.SelectController<T> {
----   get_items: fun(): T[];
+---   get_items: fun(): T[],
----   set_items: fun(items: T[]);
+---   set_items: fun(items: T[]),
----   set_filter_text: fun(filter_text: string);
+---   set_filter_text: fun(filter_text: string),
----   get_selected_indices: fun(): number[];
+---   get_selected_indices: fun(): number[],
----   get_selected_items: fun(): T[];
+---   get_selected_items: fun(): T[],
----   set_selected_indices: fun(indicies: number[], ephemeral?: boolean);
+---   set_selected_indices: fun(indicies: number[], ephemeral?: boolean),
----   close: fun();
+---   close: fun()
 --- }
 --- @alias u.examples.SelectOpts<T> {
----   items: `T`[];
+---   items: `T`[],
----   multi?: boolean;
+---   multi?: boolean,
----   format_item?: fun(item: T): u.renderer.Tree;
+---   format_item?: (fun(item: T): u.renderer.Tree),
----   on_finish?: fun(items: T[], indicies: number[]);
+---   on_finish?: (fun(items: T[], indicies: number[])),
----   on_selection_changed?: fun(items: T[], indicies: number[]);
+---   on_selection_changed?: fun(items: T[], indicies: number[]),
----   mappings?: table<string, fun(select: u.examples.SelectController)>;
+---   mappings?: table<string, fun(select: u.examples.SelectController)>
 --- }
 
 --- @generic T
@@ -326,7 +326,7 @@ function M.create_picker(opts) -- {{{
     safe_wrap(function()
       local items = s_items:get()
       local selected_indices = s_selected_indices:get()
-      --- @type { orig_idx: number; item: T }[]
+      --- @type { orig_idx: number, item: T }[]
       local filtered_items = s_filtered_items:get()
       local cursor_index = s_cursor_index:get()
       local indices = shallow_copy_arr(selected_indices)
@@ -477,7 +477,7 @@ function M.create_picker(opts) -- {{{
       local selected_indices = s_selected_indices:get()
       local top_offset = s_top_offset:get()
       local cursor_index = s_cursor_index:get()
-      --- @type { filtered_idx: number; orig_idx: number; item: T; formatted: string }[]
+      --- @type { filtered_idx: number, orig_idx: number, item: T, formatted: string }[]
       local visible_items = s_visible_items:get()
 
       -- The above has to run in the execution context for the signaling to work, but
@@ -791,14 +791,14 @@ function M.buffers() -- {{{
   -- ensure that `cwd` ends with a trailing slash:
   if cwd[#cwd] ~= '/' then cwd = cwd .. '/' end
 
-  --- @type { name: string; changed: number; bufnr: number }[]
+  --- @type { name: string, changed: number, bufnr: number }[]
   local bufs = vim.fn.getbufinfo { buflisted = 1 }
 
   M.create_picker {
     multi = true,
     items = bufs,
 
-    --- @param item { name: string; changed: number; bufnr: number }
+    --- @param item { name: string, changed: number, bufnr: number }
     format_item = function(item)
       local item_name = item.name
       if item_name == '' then item_name = '[No Name]' end

examples/surround.lua

@@ -21,10 +21,10 @@ local surrounds = {
   ['`'] = { left = '`', right = '`' },
 }
 
---- @type { left: string; right: string } | nil
+--- @type { left: string, right: string } | nil
 local CACHED_BOUNDS = nil
 
---- @return { left: string; right: string }|nil
+--- @return { left: string, right: string }|nil
 local function prompt_for_bounds()
   if vim_repeat.is_repeating() then
     -- If we are repeating, we don't want to prompt for bounds, because
@@ -55,7 +55,7 @@ local function prompt_for_bounds()
 end
 
 --- @param range u.Range
---- @param bounds { left: string; right: string }
+--- @param bounds { left: string, right: string }
 local function do_surround(range, bounds)
   local left = bounds.left
   local right = bounds.right

lua/u/buffer.lua

@@ -62,7 +62,7 @@ end
 function Buffer:lines(start, stop) return Range.from_lines(self.bufnr, start, stop) end
 
 --- @param motion string
---- @param opts? { contains_cursor?: boolean; pos?: u.Pos }
+--- @param opts? { contains_cursor?: boolean, pos?: u.Pos }
 function Buffer:motion(motion, opts)
   opts = vim.tbl_extend('force', opts or {}, { bufnr = self.bufnr })
   return Range.from_motion(motion, opts)

lua/u/opkeymap.lua

@@ -5,7 +5,6 @@ local __U__OpKeymapOpFunc_rhs = nil
 
 --- This is the global utility function used for operatorfunc
 --- in opkeymap
---- @type nil|fun(range: u.Range): fun():any|nil
 --- @param ty 'line'|'char'|'block'
 -- selene: allow(unused_variable)
 function _G.__U__OpKeymapOpFunc(ty)

lua/u/range.lua

@@ -112,7 +112,7 @@ function Range.from_lines(bufnr, start_line, stop_line)
 end
 
 --- @param motion string
---- @param opts? { bufnr?: number; contains_cursor?: boolean; pos?: u.Pos, user_defined?: boolean }
+--- @param opts? { bufnr?: number, contains_cursor?: boolean, pos?: u.Pos, user_defined?: boolean }
 --- @return u.Range|nil
 function Range.from_motion(motion, opts)
   -- Options handling:
@@ -513,7 +513,7 @@ function Range:text() return vim.fn.join(self:lines(), '\n') end
 
 --- @param l number
 -- luacheck: ignore
---- @return { line: string; idx0: { start: number; stop: number; }; lnum: number; range: fun():u.Range; text: fun():string }|nil
+--- @return { line: string, idx0: { start: number, stop: number }, lnum: number, range: (fun():u.Range), text: fun():string }|nil
 function Range:line(l)
   if l < 0 then l = self:line_count() + l + 1 end
   if l > self:line_count() then return end

lua/u/renderer.lua

@@ -1,13 +1,25 @@
+---@diagnostic disable: redefined-local
 function _G.URendererOpFuncSwallow() end
 
-local ENABLE_LOG = false
+--- Comma-separated list of regex patterns for debug logging. Examples:
+--- "render,extmark" or ".*" or "get_tags_at,_on_text_changed" or
+--- "^_on_.*,reconcile$"
+local DEBUG = vim.env.DEBUG or ''
 
-local function log(...)
+--- @param name string
-  if not ENABLE_LOG then return end
+local function log(name, ...)
+  if DEBUG == '' then return end
 
-  local f = assert(io.open(vim.fs.joinpath(vim.fn.stdpath 'log', 'u.renderer.log'), 'a+'))
+  for pattern in DEBUG:gmatch '[^,]+' do
-  f:write(os.date() .. '\t' .. vim.inspect { ... } .. '\n')
+    if name:match(vim.trim(pattern)) then
-  f:close()
+      local f = assert(
+        io.open(vim.fs.joinpath((vim.fn.stdpath 'log') --[[@as string]], 'u.renderer.log'), 'a+')
+      )
+      f:write(os.date() .. '\t' .. vim.inspect { name, ... } .. '\n')
+      f:close()
+      return
+    end
+  end
 end
 
 local M = {}
@@ -15,19 +27,59 @@ local H = {}
 
 --- @alias u.renderer.TagEventHandler fun(tag: u.renderer.Tag, mode: string, lhs: string): string
 
---- @alias u.renderer.TagAttributes { [string]?: unknown; imap?: table<string, u.renderer.TagEventHandler>; nmap?: table<string, u.renderer.TagEventHandler>; vmap?: table<string, u.renderer.TagEventHandler>; xmap?: table<string, u.renderer.TagEventHandler>; omap?: table<string, u.renderer.TagEventHandler>, on_change?: fun(text: string): unknown }
+--- @alias u.renderer.TagAttributes { [string]?: unknown, imap?: table<string, u.renderer.TagEventHandler>, nmap?: table<string, u.renderer.TagEventHandler>, vmap?: table<string, u.renderer.TagEventHandler>, xmap?: table<string, u.renderer.TagEventHandler>, omap?: table<string, u.renderer.TagEventHandler>, on_change?: (fun(text: string): unknown), key?: string|number }
+
+--- @generic TProps
+--- @generic TState
+--- @class u.renderer.FnComponentContext<TProps, TState>
+--- @field phase 'mount'|'update'|'unmount'
+--- @field props TProps
+--- @field state? TState
+--- @field children u.renderer.Tree
+--- @field private on_change? fun(): any
+--- @field private prev_rendered_children? u.renderer.Tree
+local FnComponentContext = {}
+M.FnComponentContext = FnComponentContext
+FnComponentContext.__index = FnComponentContext
+
+--- @param props TProps
+--- @param state? TState
+--- @param children u.renderer.Tree
+function FnComponentContext.new(props, state, children)
+  return setmetatable({
+    phase = 'mount',
+    props = props,
+    state = state,
+    children = children,
+  }, FnComponentContext)
+end
+
+--- @param new_state TState
+function FnComponentContext:update(new_state)
+  self.state = new_state
+  if self.on_change then vim.schedule(function() self.on_change() end) end
+end
+
+--- @private
+--- @param new_state TState
+function FnComponentContext:update_immediate(new_state)
+  self.state = new_state
+  if self.on_change then self.on_change() end
+end
+
+--- @alias u.renderer.FnComponent<TProps, TState> fun(ctx: u.renderer.FnComponentContext<TProps, TState>): u.renderer.Tree
 
 --- @class u.renderer.Tag
 --- @field kind 'tag'
---- @field name string
+--- @field name string | u.renderer.FnComponent<any, any>
 --- @field attributes u.renderer.TagAttributes
 --- @field children u.renderer.Tree
+--- @field private ctx? u.renderer.FnComponentContext
 
 --- @alias u.renderer.Node nil | boolean | string | u.renderer.Tag
 --- @alias u.renderer.Tree u.renderer.Node | u.renderer.Node[]
 
 -- luacheck: ignore
---- @type table<string, fun(attributes: u.renderer.TagAttributes, children: u.renderer.Tree): u.renderer.Tag> & fun(name: string, attributes: u.renderer.TagAttributes, children: u.renderer.Tree): u.renderer.Tag>
 M.h = setmetatable({}, {
   __call = function(_, name, attributes, children)
     return {
@@ -42,39 +94,77 @@ M.h = setmetatable({}, {
       return M.h('text', vim.tbl_deep_extend('force', { hl = name }, attributes), children)
     end
   end,
-})
+}) --[[@as table<string, fun(attributes: u.renderer.TagAttributes, children: u.renderer.Tree): u.renderer.Tag> & fun(name: string, attributes: u.renderer.TagAttributes, children: u.renderer.Tree): u.renderer.Tag>]]
 
 -- Renderer {{{
 --- @class u.renderer.RendererExtmark
---- @field id? number
+--- @field id? integer
---- @field start [number, number]
+--- @field start [integer, integer]
---- @field stop [number, number]
+--- @field stop [integer, integer]
 --- @field opts vim.api.keyset.set_extmark
 --- @field tag u.renderer.Tag
 
 --- @class u.renderer.Renderer
---- @field bufnr number
+--- @field bufnr integer
---- @field ns number
+--- @field ns integer
---- @field changedtick number
+--- @field changedtick integer
---- @field old { lines: string[]; extmarks: u.renderer.RendererExtmark[] }
+--- @field text_content { old: { lines: string[], extmarks: u.renderer.RendererExtmark[] }, curr: { lines: string[], extmarks: u.renderer.RendererExtmark[] } }
---- @field curr { lines: string[]; extmarks: u.renderer.RendererExtmark[] }
+--- @field component_tree { old: u.renderer.Tree  }
 local Renderer = {}
 Renderer.__index = Renderer
 M.Renderer = Renderer
 
 --- @private
---- @param x any
+--- @param tree u.renderer.Tree
---- @return boolean
+--- @param visitors {
-function Renderer.is_tag(x) return type(x) == 'table' and x.kind == 'tag' end
+---   nil_?: (fun(): any),
+---   boolean?: (fun(b: boolean): any),
+---   string?: (fun(s: string): any),
+---   array?: (fun(tags: u.renderer.Node[]): any),
+---   tag?: (fun(tag: u.renderer.Tag): any),
+---   component?: (fun(component: u.renderer.FnComponent, tag: u.renderer.Tag): any),
+---   unknown?: fun(tag: any): any
+--- }
+function Renderer.tree_match(tree, visitors) -- {{{
+  local function is_tag(x) return type(x) == 'table' and x.kind == 'tag' end
+  local function is_tag_arr(x) return type(x) == 'table' and not is_tag(x) end
+
+  if tree == nil then
+    return visitors.nil_ and visitors.nil_() or nil
+  elseif type(tree) == 'boolean' then
+    return visitors.boolean and visitors.boolean(tree) or nil
+  elseif type(tree) == 'string' then
+    return visitors.string and visitors.string(tree) or nil
+  elseif is_tag_arr(tree) then
+    return visitors.array and visitors.array(tree --[[@as any]]) or nil
+  elseif is_tag(tree) then
+    local tag = tree --[[@as u.renderer.Tag]]
+    if type(tag.name) == 'function' then
+      return visitors.component and visitors.component(tag.name --[[@as function]], tag) or nil
+    else
+      return visitors.tag and visitors.tag(tree --[[@as any]]) or nil
+    end
+  else
+    return visitors.unknown and visitors.unknown(tree) or error 'unknown value: not a tag'
+  end
+end -- }}}
 
 --- @private
---- @param x any
+--- @param tree u.renderer.Tree
---- @return boolean
+--- @return 'nil' | 'boolean' | 'string' | 'array' | 'tag' | u.renderer.FnComponent | 'unknown'
-function Renderer.is_tag_arr(x)
+function Renderer.tree_kind(tree) -- {{{
-  if type(x) ~= 'table' then return false end
+  return Renderer.tree_match(tree, {
-  return #x == 0 or not Renderer.is_tag(x)
+    nil_ = function() return 'nil' end,
-end
+    boolean = function() return 'boolean' end,
---- @param bufnr number|nil
+    string = function() return 'string' end,
+    array = function() return 'array' end,
+    tag = function() return 'tag' end,
+    component = function(c) return c end,
+    unknown = function() return 'unknown' end,
+  }) --[[@as any]]
+end -- }}}
+
+--- @param bufnr integer|nil
 function Renderer.new(bufnr) -- {{{
   if bufnr == nil or bufnr == 0 then bufnr = vim.api.nvim_get_current_buf() end
 
@@ -86,8 +176,15 @@ function Renderer.new(bufnr) -- {{{
     bufnr = bufnr,
     ns = vim.b[bufnr]._renderer_ns,
     changedtick = 0,
-    old = { lines = {}, extmarks = {} },
+    text_content = {
-    curr = { lines = {}, extmarks = {} },
+      old = { lines = {}, extmarks = {} },
+      curr = { lines = {}, extmarks = {} },
+    },
+    component_tree = {
+      old = nil,
+      curr = nil,
+      ctx_by_node = {},
+    },
   }, Renderer)
 
   vim.api.nvim_create_autocmd({ 'TextChanged', 'TextChangedI', 'TextChangedP' }, {
@@ -99,8 +196,8 @@ function Renderer.new(bufnr) -- {{{
 end -- }}}
 
 --- @param opts {
----   tree: u.renderer.Tree;
+---   tree: u.renderer.Tree,
----   on_tag?: fun(tag: u.renderer.Tag, start0: [number, number], stop0: [number, number]): any;
+---   on_tag?: fun(tag: u.renderer.Tag, start0: [number, number], stop0: [number, number]): any
 --- }
 function Renderer.markup_to_lines(opts) -- {{{
   --- @type string[]
@@ -119,39 +216,31 @@ function Renderer.markup_to_lines(opts) -- {{{
     curr_col1 = 1
   end
 
-  --- @param node u.renderer.Node
+  --- @param node u.renderer.Tree
   local function visit(node) -- {{{
-    if node == nil or type(node) == 'boolean' then return end
+    Renderer.tree_match(node, {
-
+      string = function(s_node)
-    if type(node) == 'string' then
+        local node_lines = vim.split(s_node, '\n')
-      local node_lines = vim.split(node, '\n')
+        for lnum, s in ipairs(node_lines) do
-      for lnum, s in ipairs(node_lines) do
+          if lnum > 1 then put_line() end
-        if lnum > 1 then put_line() end
+          put(s)
-        put(s)
+        end
-      end
+      end,
-    elseif Renderer.is_tag(node) then
+      array = function(ts)
-      local start0 = { curr_line1 - 1, curr_col1 - 1 }
+        for _, child in ipairs(ts) do
-
-      -- visit the children:
-      if Renderer.is_tag_arr(node.children) then
-        for _, child in
-          ipairs(node.children --[[@as u.renderer.Node[] ]])
-        do
-          -- newlines are not controlled by array entries, do NOT output a line here:
           visit(child)
         end
-      else -- luacheck: ignore
+      end,
-        visit(node.children)
+      tag = function(t)
-      end
+        assert(type(t.name) ~= 'function', 'markup_to_lines does not render components')
 
-      local stop0 = { curr_line1 - 1, curr_col1 - 1 }
+        local start0 = { curr_line1 - 1, curr_col1 - 1 }
-      if opts.on_tag then opts.on_tag(node, start0, stop0) end
+        visit(t.children)
-    elseif Renderer.is_tag_arr(node) then
+        local stop0 = { curr_line1 - 1, curr_col1 - 1 }
-      for _, child in ipairs(node) do
+
-        -- newlines are not controlled by array entries, do NOT output a line here:
+        if opts.on_tag then opts.on_tag(t, start0, stop0) end
-        visit(child)
+      end,
-      end
+    })
-    end
   end -- }}}
   visit(opts.tree)
 
@@ -159,12 +248,12 @@ function Renderer.markup_to_lines(opts) -- {{{
 end -- }}}
 
 --- @param opts {
----   tree: u.renderer.Tree;
+---   tree: u.renderer.Tree,
----   format_tag?: fun(tag: u.renderer.Tag): string;
+---   format_tag?: fun(tag: u.renderer.Tag): string
 --- }
 function Renderer.markup_to_string(opts) return table.concat(Renderer.markup_to_lines(opts), '\n') end
 
---- @param bufnr number
+--- @param bufnr integer
 --- @param old_lines string[] | nil
 --- @param new_lines string[]
 function Renderer.patch_lines(bufnr, old_lines, new_lines) -- {{{
@@ -190,8 +279,10 @@ function Renderer.patch_lines(bufnr, old_lines, new_lines) -- {{{
   end
 
   -- Morph the text to the desired state:
-  local line_changes =
+  local line_changes = H.levenshtein {
-    H.levenshtein(old_lines or vim.api.nvim_buf_get_lines(bufnr, 0, -1, false), new_lines)
+    from = old_lines or vim.api.nvim_buf_get_lines(bufnr, 0, -1, false),
+    to = new_lines,
+  }
   for _, line_change in ipairs(line_changes) do
     local lnum0 = line_change.index - 1
 
@@ -200,7 +291,7 @@ function Renderer.patch_lines(bufnr, old_lines, new_lines) -- {{{
     elseif line_change.kind == 'change' then
       -- Compute inter-line diff, and apply:
       local col_changes =
-        H.levenshtein(vim.split(line_change.from, ''), vim.split(line_change.to, ''))
+        H.levenshtein { from = vim.split(line_change.from, ''), to = vim.split(line_change.to, '') }
 
       for _, col_change in ipairs(col_changes) do
         local cnum0 = col_change.index - 1
@@ -226,7 +317,8 @@ end -- }}}
 function Renderer:render(tree) -- {{{
   local changedtick = vim.b[self.bufnr].changedtick
   if changedtick ~= self.changedtick then
-    self.curr = { lines = vim.api.nvim_buf_get_lines(self.bufnr, 0, -1, false) }
+    self.text_content.curr =
+      { extmarks = {}, lines = vim.api.nvim_buf_get_lines(self.bufnr, 0, -1, false) }
     self.changedtick = changedtick
   end
 
@@ -283,8 +375,8 @@ function Renderer:render(tree) -- {{{
     end, -- }}}
   }
 
-  self.old = self.curr
+  self.text_content.old = self.text_content.curr
-  self.curr = { lines = lines, extmarks = extmarks }
+  self.text_content.curr = { lines = lines, extmarks = extmarks }
   self:_reconcile()
   vim.cmd.doautocmd { args = { 'User', 'Renderer:' .. tostring(self.bufnr) .. ':render' } }
 end -- }}}
@@ -294,7 +386,7 @@ function Renderer:_reconcile() -- {{{
   --
   -- Step 1: morph the text to the desired state:
   --
-  Renderer.patch_lines(self.bufnr, self.old.lines, self.curr.lines)
+  Renderer.patch_lines(self.bufnr, self.text_content.old.lines, self.text_content.curr.lines)
   self.changedtick = vim.b[self.bufnr].changedtick
 
   --
@@ -309,7 +401,7 @@ function Renderer:_reconcile() -- {{{
   vim.api.nvim_buf_clear_namespace(self.bufnr, self.ns, 0, -1)
 
   -- Set current extmarks:
-  for _, extmark in ipairs(self.curr.extmarks) do
+  for _, extmark in ipairs(self.text_content.curr.extmarks) do
     extmark.id = vim.api.nvim_buf_set_extmark(
       self.bufnr,
       self.ns,
@@ -330,7 +422,7 @@ function Renderer:_reconcile() -- {{{
     )
   end
 
-  self.old = self.curr
+  self.text_content.old = self.text_content.curr
 end -- }}}
 
 --- @private
@@ -340,9 +432,9 @@ function Renderer:_expr_map_callback(mode, lhs) -- {{{
   -- find the tag with the smallest intersection that contains the cursor:
   local pos0 = vim.api.nvim_win_get_cursor(0)
   pos0[1] = pos0[1] - 1 -- make it actually 0-based
-  log('_expr_map_callback: pos0:', pos0)
+  log('_expr_map_callback', 'pos0:', pos0)
   local pos_infos = self:get_tags_at(pos0)
-  log('_expr_map_callback: pos_infos:', pos_infos)
+  log('_expr_map_callback', 'pos_infos:', pos_infos)
 
   if #pos_infos == 0 then return lhs end
 
@@ -392,7 +484,7 @@ function Renderer:_on_text_changed() -- {{{
       --- @type integer, integer, vim.api.keyset.extmark_details
       local start_row0, start_col0, details = unpack(extmark)
       local end_row0, end_col0 = details.end_row, details.end_col
-      log('_on_text_changed: fetched current extmark for pos_info', {
+      log('_on_text_changed', 'fetched current extmark for pos_info', {
         pos_info = pos_info,
         curr_extmark = {
           start_row0 = start_row0,
@@ -421,7 +513,7 @@ function Renderer:_on_text_changed() -- {{{
           or ''
         end_col0 = last_line:len()
       end
-      log('_on_text_changed: after position correction', {
+      log('_on_text_changed', 'after position correction', {
         curr_extmark = {
           start_row0 = start_row0,
           start_col0 = start_col0,
@@ -439,7 +531,7 @@ function Renderer:_on_text_changed() -- {{{
       local pos2 = { self.bufnr, end_row0 + 1, end_col0 }
       local ok, lines = pcall(vim.fn.getregion, pos1, pos2, { type = 'v' })
       if not ok then
-        log('_on_text_changed: getregion: invalid-pos ', {
+        log('_on_text_changed', 'getregion: invalid-pos ', {
           { pos1, pos2 },
         })
         vim.api.nvim_echo({
@@ -492,7 +584,7 @@ function Renderer:_debug() -- {{{
         ),
       },
       computed = {
-        extmarks = self.curr.extmarks,
+        extmarks = self.text_content.curr.extmarks,
       },
     }
     vim.api.nvim_buf_set_lines(info_bufnr, 0, -1, true, vim.split(vim.inspect(info), '\n'))
@@ -534,9 +626,9 @@ end -- }}}
 --- (outermost).
 ---
 --- @private (private for now)
---- @param pos0 [number; number]
+--- @param pos0 [integer, integer]
 --- @param mode string?
---- @return { extmark: u.renderer.RendererExtmark; tag: u.renderer.Tag; }[]
+--- @return { extmark: u.renderer.RendererExtmark, tag: u.renderer.Tag }[]
 function Renderer:get_tags_at(pos0, mode) -- {{{
   local cursor_line0, cursor_col0 = pos0[1], pos0[2]
   if not mode then mode = vim.api.nvim_get_mode().mode end
@@ -550,7 +642,8 @@ function Renderer:get_tags_at(pos0, mode) -- {{{
     { details = true, overlap = true }
   )
   log(
-    'get_tags_at: context:',
+    'get_tags_at',
+    'context:',
     { pos0 = pos0, mode = mode, raw_overlapping_extmarks = raw_overlapping_extmarks }
   )
 
@@ -560,58 +653,65 @@ function Renderer:get_tags_at(pos0, mode) -- {{{
   --- @type u.renderer.RendererExtmark[]
   local mapped_extmarks = vim
     .iter(raw_overlapping_extmarks)
-    --- @return u.renderer.RendererExtmark
+    :map(
-    :map(function(ext)
+      --- @return u.renderer.RendererExtmark
-      --- @type number, number, number, { end_row?: number; end_col?: number }|nil
+      function(ext)
-      local id, line0, col0, details = unpack(ext)
+        --- @type integer, integer, integer, { end_row?: number, end_col?: number }|nil
-      local start = { line0, col0 }
+        local id, line0, col0, details = unpack(ext)
-      local stop = { line0, col0 }
+        local start = { line0, col0 }
-      if details and details.end_row ~= nil and details.end_col ~= nil then
+        local stop = { line0, col0 }
-        stop = { details.end_row, details.end_col }
+        if details and details.end_row ~= nil and details.end_col ~= nil then
+          stop = {
+            details.end_row --[[@as integer]],
+            details.end_col --[[@as integer]],
+          }
+        end
+        return { id = id, start = start, stop = stop, opts = details }
       end
-      return { id = id, start = start, stop = stop, opts = details }
+    )
-    end)
     :totable()
 
   local intersecting_extmarks = vim
     .iter(mapped_extmarks)
-    --- @param ext u.renderer.RendererExtmark
+    :filter(
-    :filter(function(ext)
+      --- @param ext u.renderer.RendererExtmark
-      if ext.stop[1] ~= nil and ext.stop[2] ~= nil then
+      function(ext)
-        -- If we've "ciw" and "collapsed" an extmark onto the cursor,
+        if ext.stop[1] ~= nil and ext.stop[2] ~= nil then
-        -- the cursor pos will equal the exmark's start AND end. In this
+          -- If we've "ciw" and "collapsed" an extmark onto the cursor,
-        -- case, we want to include the extmark.
+          -- the cursor pos will equal the exmark's start AND end. In this
-        if
+          -- case, we want to include the extmark.
-          cursor_line0 == ext.start[1]
+          if
-          and cursor_col0 == ext.start[2]
+            cursor_line0 == ext.start[1]
-          and cursor_line0 == ext.stop[1]
+            and cursor_col0 == ext.start[2]
-          and cursor_col0 == ext.stop[2]
+            and cursor_line0 == ext.stop[1]
-        then
+            and cursor_col0 == ext.stop[2]
+          then
+            return true
+          end
+
+          return
+            -- START: line check
+            cursor_line0 >= ext.start[1]
+              -- START: column check
+              and (cursor_line0 ~= ext.start[1] or cursor_col0 >= ext.start[2])
+              -- STOP: line check
+              and cursor_line0 <= ext.stop[1]
+              -- STOP: column check
+              and (
+                cursor_line0 ~= ext.stop[1]
+                or (
+                  mode == 'i'
+                    -- In insert mode, the cursor is "thin", so <= to compensate:
+                    and cursor_col0 <= ext.stop[2]
+                  -- In normal mode, the cursor is "wide", so < to compensate:
+                  or cursor_col0 < ext.stop[2]
+                )
+              )
+        else
           return true
         end
-
-        return
-          -- START: line check
-          cursor_line0 >= ext.start[1]
-            -- START: column check
-            and (cursor_line0 ~= ext.start[1] or cursor_col0 >= ext.start[2])
-            -- STOP: line check
-            and cursor_line0 <= ext.stop[1]
-            -- STOP: column check
-            and (
-              cursor_line0 ~= ext.stop[1]
-              or (
-                mode == 'i'
-                  -- In insert mode, the cursor is "thin", so <= to compensate:
-                  and cursor_col0 <= ext.stop[2]
-                -- In normal mode, the cursor is "wide", so < to compensate:
-                or cursor_col0 < ext.stop[2]
-              )
-            )
-      else
-        return true
       end
-    end)
+    )
     :totable()
 
   -- Sort the tags into smallest (inner) to largest (outer):
@@ -637,18 +737,20 @@ function Renderer:get_tags_at(pos0, mode) -- {{{
   )
 
   -- When we set the extmarks in the step above, we captured the IDs of the
-  -- created extmarks in self.curr.extmarks, which also has which tag each
+  -- created extmarks in self.text_content.curr.extmarks, which also has which tag each
   -- extmark is associated with. Cross-reference with that list to get a list
   -- of tags that we need to fire events for:
-  --- @type { extmark: u.renderer.RendererExtmark; tag: u.renderer.Tag }[]
+  --- @type { extmark: u.renderer.RendererExtmark, tag: u.renderer.Tag }[]
   local matching_tags = vim
     .iter(intersecting_extmarks)
-    --- @param ext u.renderer.RendererExtmark
+    :map(
-    :map(function(ext)
+      --- @param ext u.renderer.RendererExtmark
-      for _, extmark_cache in ipairs(self.curr.extmarks) do
+      function(ext)
-        if extmark_cache.id == ext.id then return { extmark = ext, tag = extmark_cache.tag } end
+        for _, extmark_cache in ipairs(self.text_content.curr.extmarks) do
+          if extmark_cache.id == ext.id then return { extmark = ext, tag = extmark_cache.tag } end
+        end
       end
-    end)
+    )
     :totable()
 
   return matching_tags
@@ -656,9 +758,9 @@ end -- }}}
 
 --- @private
 --- @param tag_or_id string | u.renderer.Tag
---- @return { start: [number, number]; stop: [number, number] } | nil
+--- @return { start: [number, number], stop: [number, number] } | nil
 function Renderer:get_tag_bounds(tag_or_id) -- {{{
-  for _, x in ipairs(self.curr.extmarks) do
+  for _, x in ipairs(self.text_content.curr.extmarks) do
     local pos = { start = x.start, stop = x.stop }
     local does_tag_match = type(tag_or_id) == 'string' and x.tag.attributes.id == tag_or_id
       or x.tag == tag_or_id
@@ -666,6 +768,189 @@ function Renderer:get_tag_bounds(tag_or_id) -- {{{
   end
 end -- }}}
 
+--- @param tree u.renderer.Tree
+function Renderer:mount(tree) -- {{{
+  local function rerender()
+    local H2 = {}
+
+    --- @param tree u.renderer.Tree
+    function H2.unmount(tree)
+      log('Render.mount.umount', { tree = tree })
+      --- @param tree u.renderer.Tree
+      local function visit(tree)
+        Renderer.tree_match(tree, {
+          array = function(tags)
+            log('Render.mount.umount.array', { tags = tags })
+            for _, tag in ipairs(tags) do
+              visit(tag)
+            end
+          end,
+          tag = function(tag)
+            log('Render.mount.umount.tag', { tag = tag })
+            visit(tag.children)
+          end,
+          component = function(Component, tag)
+            log('Render.mount.umount.component', { Component = Component, tag = tag })
+            --- @type u.renderer.FnComponentContext
+            local ctx = assert(tag.ctx, 'could not find context for node')
+
+            -- depth-first:
+            visit(ctx.children)
+
+            -- now unmount current:
+            ctx.phase = 'unmount'
+            Component(ctx)
+            ctx.on_change = nil
+          end,
+        })
+      end
+
+      visit(tree)
+    end
+
+    --- @param old_tree u.renderer.Tree
+    --- @param new_tree u.renderer.Tree
+    --- @return u.renderer.Tree
+    function H2.visit_tree(old_tree, new_tree)
+      local old_tree_kind = Renderer.tree_kind(old_tree)
+      local new_tree_kind = Renderer.tree_kind(new_tree)
+      log('Renderer.mount.visit_tree', {
+        old_tree_kind = old_tree_kind,
+        old_tree = old_tree,
+        new_tree_kind = new_tree_kind,
+        new_tree = new_tree,
+      })
+
+      local new_tree_rendered = Renderer.tree_match(new_tree, {
+        string = function(s) return s end,
+        boolean = function(b) return b end,
+        nil_ = function() return nil end,
+
+        array = function(new_arr)
+          return H2.visit_array(old_tree --[[@as any]], new_arr)
+        end,
+        tag = function(new_tag)
+          log('Renderer.mount.visit_tree.tag', { new_tag = new_tag })
+          local old_children = old_tree_kind == new_tree_kind and old_tree.children or nil
+          return M.h(
+            new_tag.name,
+            new_tag.attributes,
+            H2.visit_tree(old_children --[[@as any]], new_tag.children --[[@as any]])
+          )
+        end,
+
+        component = function(NewC, new_tag)
+          --- @type { tag: u.renderer.Tag, ctx?: u.renderer.FnComponentContext } | nil
+          local old_component_info = Renderer.tree_match(
+            old_tree,
+            { component = function(_, t) return { tag = t, ctx = t.ctx } end }
+          )
+          local old_tag = old_component_info and old_component_info.tag or nil
+          local ctx = old_component_info and old_component_info.ctx or nil
+
+          log('Renderer.mount.visit_tree.component', {
+            NewC = NewC,
+            new_tag = new_tag,
+            old_tag = old_tag,
+          })
+
+          if not ctx then
+            --- @type u.renderer.FnComponentContext
+            ctx = FnComponentContext.new(new_tag.attributes, nil, new_tag.children)
+          else
+            ctx.phase = 'update'
+          end
+          ctx.props = new_tag.attributes
+          ctx.children = new_tag.children
+          ctx.on_change = rerender
+
+          new_tag.ctx = ctx
+          local NewC_rendered_children = NewC(ctx)
+          local result = H2.visit_tree(ctx.prev_rendered_children, NewC_rendered_children)
+          ctx.prev_rendered_children = NewC_rendered_children
+          return result
+        end,
+      })
+
+      if old_tree_kind ~= new_tree_kind then H2.unmount(old_tree) end
+
+      return new_tree_rendered
+    end
+
+    --- @param old_arr u.renderer.Node[]
+    --- @param new_arr u.renderer.Node[]
+    --- @return u.renderer.Node[]
+    function H2.visit_array(old_arr, new_arr)
+      -- We are going to hijack levenshtein in order to compute the
+      -- difference between elements/components. In this model, we need to
+      -- "update" all the nodes, so no nodes are equal. We will rely on
+      -- levenshtein to find the "shortest path" to conforming old => new via
+      -- the cost.of_change function. That will provide the meat of modeling
+      -- what effort it will take to morph one element into the new form.
+      -- What levenshtein gives us for free in this model is also informing
+      -- us what needs to be added (i.e., "mounted"), what needs to be
+      -- deleted ("unmounted") and what needs to be changed ("updated").
+      local changes = H.levenshtein {
+        from = old_arr or {},
+        to = new_arr or {},
+        are_equal = function() return false end,
+        cost = {
+          of_change = function(tree1, tree2, tree1_idx, tree2_idx)
+            --- @return string
+            local function verbose_tree_kind(tree, idx)
+              return Renderer.tree_match(tree, {
+                nil_ = function() return 'nil' end,
+                string = function() return 'string' end,
+                boolean = function() return 'boolean' end,
+                array = function() return 'array' end,
+                tag = function(tag)
+                  return ('tag-%s-%s'):format(tag.name, tostring(tag.attributes.key or idx))
+                end,
+                component = function(C, tag)
+                  return ('component-%s-%s'):format(C, tostring(tag.attributes.key or idx))
+                end,
+              })
+            end
+            local tree1_inf = verbose_tree_kind(tree1, tree1_idx)
+            local tree2_inf = verbose_tree_kind(tree2, tree2_idx)
+            return tree1_inf == tree2_inf and 1 or 2
+          end,
+        },
+      }
+
+      --- @type u.renderer.Node[]
+      local resulting_nodes = {}
+      for _, change in ipairs(changes) do
+        local resulting_node
+        if change.kind == 'add' then
+          resulting_node = H2.visit_tree(nil, change.item)
+        elseif change.kind == 'delete' then
+          H2.visit_tree(change.item, nil)
+        elseif change.kind == 'change' then
+          resulting_node = H2.visit_tree(change.from, change.to)
+        end
+        if resulting_node then table.insert(resulting_nodes, 1, resulting_node) end
+      end
+
+      log('Renderer.mount.visit_array', {
+        old_arr = old_arr,
+        new_arr = new_arr,
+        changes = changes,
+        resulting_nodes = resulting_nodes,
+      })
+
+      return resulting_nodes
+    end
+
+    local renderable_tree = H2.visit_tree(self.component_tree.old, tree)
+    self.component_tree.old = tree
+    self:render(renderable_tree)
+  end
+
+  -- Kick off initial render:
+  rerender()
+end -- }}}
+
 -- }}}
 
 -- TreeBuilder {{{
@@ -733,14 +1018,21 @@ function TreeBuilder:tree() return self.nodes end
 
 -- Levenshtein utility {{{
 -- luacheck: ignore
---- @alias u.renderer.LevenshteinChange<T> ({ kind: 'add'; item: T; index: number; } | { kind: 'delete'; item: T; index: number; } | { kind: 'change'; from: T; to: T; index: number; })
+--- @alias u.renderer.LevenshteinChange<T> ({ kind: 'add', item: T, index: integer } | { kind: 'delete', item: T, index: integer } | { kind: 'change', from: T, to: T, index: integer })
 --- @private
 --- @generic T
---- @param x `T`[]
+--- @param opts {
---- @param y T[]
+---   from: `T`[],
---- @param cost? { of_delete?: fun(x: T): number; of_add?: fun(x: T): number; of_change?: fun(x: T, y: T): number; }
+---   to: T[],
+---   are_equal?: (fun(x: T, y: T, xidx: integer, yidx: integer): boolean),
+---   cost?: {
+---     of_delete?: (fun(x: T, idx: integer): number),
+---     of_add?: (fun(x: T, idx: integer): number),
+---     of_change?: (fun(x: T, y: T, xidx: integer, yidx: integer): number)
+---   }
+--- }
 --- @return u.renderer.LevenshteinChange<T>[] The changes, from last (greatest index) to first (smallest index).
-function H.levenshtein(x, y, cost)
+function H.levenshtein(opts)
   -- At the moment, this whole `cost` plumbing is not used. Deletes have the
   -- same cost as Adds or Changes. I can imagine a future, however, where
   -- fudging with the costs of operations produces a more optimized change-set
@@ -750,12 +1042,13 @@ function H.levenshtein(x, y, cost)
   -- in here even though it's not actively used. Hopefully having this
   -- callback-based plumbing does not cause too much of a performance hit to
   -- the renderer.
-  cost = cost or {}
+  if not opts.are_equal then opts.are_equal = function(x, y) return x == y end end
-  local cost_of_delete_f = cost.of_delete or function() return 1 end
+  if not opts.cost then opts.cost = {} end
-  local cost_of_add_f = cost.of_add or function() return 1 end
+  if not opts.cost.of_add then opts.cost.of_add = function() return 1 end end
-  local cost_of_change_f = cost.of_change or function() return 1 end
+  if not opts.cost.of_change then opts.cost.of_change = function() return 1 end end
+  if not opts.cost.of_delete then opts.cost.of_delete = function() return 1 end end
 
-  local m, n = #x, #y
+  local m, n = #opts.from, #opts.to
   -- Initialize the distance matrix
   local dp = {}
   for i = 0, m do
@@ -776,12 +1069,12 @@ function H.levenshtein(x, y, cost)
   -- Compute the Levenshtein distance dynamically
   for i = 1, m do
     for j = 1, n do
-      if x[i] == y[j] then
+      if opts.are_equal(opts.from[i], opts.to[j], i, j) then
         dp[i][j] = dp[i - 1][j - 1] -- no cost if items are the same
       else
-        local cost_delete = dp[i - 1][j] + cost_of_delete_f(x[i])
+        local cost_delete = dp[i - 1][j] + opts.cost.of_delete(opts.from[i], i)
-        local cost_add = dp[i][j - 1] + cost_of_add_f(y[j])
+        local cost_add = dp[i][j - 1] + opts.cost.of_add(opts.to[j], j)
-        local cost_change = dp[i - 1][j - 1] + cost_of_change_f(x[i], y[j])
+        local cost_change = dp[i - 1][j - 1] + opts.cost.of_change(opts.from[i], opts.to[j], i, j)
         dp[i][j] = math.min(cost_delete, cost_add, cost_change)
       end
     end
@@ -806,9 +1099,9 @@ function H.levenshtein(x, y, cost)
 
     if is_first_min(cost_of_change, cost_of_add, cost_of_delete) then
       -- potential change
-      if x[i] ~= y[j] then
+      if not opts.are_equal(opts.from[i], opts.to[j]) then
         --- @type u.renderer.LevenshteinChange
-        local change = { kind = 'change', from = x[i], index = i, to = y[j] }
+        local change = { kind = 'change', from = opts.from[i], index = i, to = opts.to[j] }
         table.insert(changes, change)
       end
       i = i - 1
@@ -816,13 +1109,13 @@ function H.levenshtein(x, y, cost)
     elseif is_first_min(cost_of_add, cost_of_change, cost_of_delete) then
       -- addition
       --- @type u.renderer.LevenshteinChange
-      local change = { kind = 'add', item = y[j], index = i + 1 }
+      local change = { kind = 'add', item = opts.to[j], index = i + 1 }
       table.insert(changes, change)
       j = j - 1
     elseif is_first_min(cost_of_delete, cost_of_change, cost_of_add) then
       -- deletion
       --- @type u.renderer.LevenshteinChange
-      local change = { kind = 'delete', item = x[i], index = i }
+      local change = { kind = 'delete', item = opts.from[i], index = i }
       table.insert(changes, change)
       i = i - 1
     else

shell.nix

@@ -1,10 +1,10 @@
 {
   pkgs ?
     import
-      # nixos-25.05 (neovim@0.11.2):
+      # nixos-unstable (neovim@0.11.4):
       (fetchTarball {
-        url = "https://github.com/nixos/nixpkgs/archive/1c1c9b3f5ec0421eaa0f22746295466ee6a8d48f.tar.gz";
+        url = "https://github.com/nixos/nixpkgs/archive/0b4defa2584313f3b781240b29d61f6f9f7e0df3.tar.gz";
-        sha256 = "0szvq1swpzyjmyyw929ngxy1khdnd9ba96qds2bm6l6kg4iq3cq0";
+        sha256 = "0p3rrd8wwlk0iwgzm7frkw1k98ywrh0avi7fqjjk87i68n3inxrs";
       })
       { },
 }:
@@ -12,7 +12,7 @@ pkgs.mkShell {
   packages = [
     pkgs.git
     pkgs.gnumake
-    pkgs.lua-language-server
+    pkgs.emmylua-check
     pkgs.lua51Packages.busted
     pkgs.lua51Packages.luacov
     pkgs.lua51Packages.luarocks

spec/renderer_spec.lua

@@ -294,4 +294,99 @@ describe('Renderer', function()
       assert.are.same(bounds, { start = { 0, 9 }, stop = { 0, 19 } })
     end)
   end)
+
+  it('should mount and rerender components', function()
+    withbuf({}, function()
+      --- @type any
+      local leaked_state = { app = {}, c1 = {}, c2 = {} }
+
+      --- @param ctx u.renderer.FnComponentContext<any, { phase: string, count: integer }>
+      local function Counter(ctx)
+        if ctx.phase == 'mount' then ctx.state = { phase = ctx.phase, count = 1 } end
+        local state = assert(ctx.state)
+        state.phase = ctx.phase
+        leaked_state[ctx.props.id].ctx = ctx
+
+        return {
+          { 'Value: ', R.h.Number({}, tostring(state.count)) },
+        }
+      end
+
+      --- @param ctx u.renderer.FnComponentContext<any, {
+      ---   toggle1: boolean,
+      ---   show2: boolean
+      --- }>
+      function App(ctx)
+        if ctx.phase == 'mount' then ctx.state = { toggle1 = false, show2 = true } end
+        local state = assert(ctx.state)
+        leaked_state.app.ctx = ctx
+
+        return {
+          state.toggle1 and 'Toggle1' or R.h(Counter, { id = 'c1' }, {}),
+          '\n',
+
+          state.show2 and {
+            '\n',
+            R.h(Counter, { id = 'c2' }, {}),
+          },
+        }
+      end
+
+      local renderer = R.Renderer.new()
+      renderer:mount(R.h(App, {}, {}))
+
+      local Buffer = require 'u.buffer'
+      local buf = Buffer.current()
+      assert.are.same(buf:all():lines(), {
+        'Value: 1',
+        '',
+        'Value: 1',
+      })
+      assert.are.same(leaked_state.c1.ctx.state.phase, 'mount')
+      assert.are.same(leaked_state.c2.ctx.state.phase, 'mount')
+      leaked_state.app.ctx:update_immediate { toggle1 = true, show2 = true }
+      assert.are.same(buf:all():lines(), {
+        'Toggle1',
+        '',
+        'Value: 1',
+      })
+      assert.are.same(leaked_state.c1.ctx.state.phase, 'unmount')
+      assert.are.same(leaked_state.c2.ctx.state.phase, 'update')
+
+      leaked_state.app.ctx:update_immediate { toggle1 = true, show2 = false }
+      assert.are.same(buf:all():lines(), {
+        'Toggle1',
+        '',
+      })
+      assert.are.same(leaked_state.c1.ctx.state.phase, 'unmount')
+      assert.are.same(leaked_state.c2.ctx.state.phase, 'unmount')
+
+      leaked_state.app.ctx:update_immediate { toggle1 = false, show2 = true }
+      assert.are.same(buf:all():lines(), {
+        'Value: 1',
+        '',
+        'Value: 1',
+      })
+      assert.are.same(leaked_state.c1.ctx.state.phase, 'mount')
+      assert.are.same(leaked_state.c2.ctx.state.phase, 'mount')
+
+      leaked_state.c1.ctx:update_immediate { count = 2 }
+      assert.are.same(buf:all():lines(), {
+        'Value: 2',
+        '',
+        'Value: 1',
+      })
+      assert.are.same(leaked_state.c1.ctx.state.phase, 'update')
+      assert.are.same(leaked_state.c2.ctx.state.phase, 'update')
+
+      leaked_state.c2.ctx:update_immediate { count = 3 }
+      assert.are.same(buf:all():lines(), {
+        'Value: 2',
+        '',
+        'Value: 3',
+      })
+      assert.are.same(leaked_state.c1.ctx.state.phase, 'update')
+      assert.are.same(leaked_state.c2.ctx.state.phase, 'update')
+    end)
+  end)
 end)