initial commit

1 files changed, 125 insertions, 0 deletions

diff --git a/panto-report-status/init.lua b/panto-report-status/init.lua
new file mode 100644
index 0000000..07c6110
--- /dev/null
+++ b/panto-report-status/init.lua
@@ -0,0 +1,125 @@
+local alphabet = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/"
+local function base64(text)
+   local out = {}
+   for i = 1, #text, 3 do
+      local a, b, c = text:byte(i, i + 2)
+      local n = a * 65536 + (b or 0) * 256 + (c or 0)
+      local function digit(shift)
+         local index = math.floor(n / 2 ^ shift) % 64 + 1
+         return alphabet:sub(index, index)
+      end
+      out[#out + 1] = digit(18) .. digit(12) .. (b and digit(6) or "=") .. (c and digit(0) or "=")
+   end
+   return table.concat(out)
+end
+
+-- Names are diagnostic labels, not arbitrary terminal text. Keep messages short,
+-- ASCII and control-free even when a tool or command supplies an unusual name.
+local function label(name)
+   return (name or "?"):gsub("[^ -~]", "?"):sub(1, 27)
+end
+
+local function command_status(name)
+   if name == "compact" then return "compacting" end
+   if name == "!" or name == "!!" then return "shell" end
+   return "cmd:" .. label(name)
+end
+
+return {
+   name = "report-status",
+   activate = function()
+      local uv = require("luv")
+      local function terminal_key()
+         if uv.guess_handle(1) ~= "tty" then return end
+         local stat = uv.fs_fstat(1)
+         if stat then return table.concat({ stat.dev, stat.ino, stat.rdev }, ":") end
+      end
+      local terminal = terminal_key()
+      if not terminal then return end
+      -- Preserve ownership across /new, but don't let a nested panto overwrite
+      -- or clear its parent's status. A different terminal may report normally.
+      local owner = terminal .. "@" .. string.format("%.0f", uv.os_getpid())
+      local inherited = os.getenv("PANTO_REPORT_STATUS_OWNER")
+      if inherited ~= owner and (inherited or ""):match("^(.*)@") == terminal then return end
+      uv.os_setenv("PANTO_REPORT_STATUS_OWNER", owner)
+
+      local panto = require("panto")
+      local last_state, last_status
+      local foreground_working, foreground_status = false, nil
+      local background_work = 0
+      local settled_state = "idle"
+      local ended = false
+
+      local function report(state, status)
+         if state == last_state and status == last_status then return end
+         -- Print mode can restore a redirected stdout after activation. Never
+         -- put control sequences in a pipe, file, or a different terminal.
+         if terminal_key() ~= terminal then return end
+         local sequence = "\27]7501;state=" .. state .. ":app=panto"
+         if status then sequence = sequence .. ":msg=" .. base64(status) end
+         sequence = sequence .. "\27\\"
+         local offset = 0
+         while offset < #sequence do
+            local written, err = uv.fs_write(1, sequence:sub(offset + 1), -1)
+            if not written or written == 0 then
+               error("panto-report-status: terminal write failed: " .. tostring(err or "short write"))
+            end
+            offset = offset + written
+         end
+         last_state, last_status = state, status
+      end
+
+      local function update()
+         if ended then return end
+         if foreground_working or background_work > 0 then
+            report("working", foreground_working and foreground_status or "background")
+         else
+            report(settled_state)
+         end
+      end
+
+      local function foreground(status)
+         foreground_working, foreground_status, settled_state = true, status, "done"
+         update()
+      end
+
+      local function finish(event)
+         if foreground_working then
+            local reason = event and event.reason
+            settled_state = reason == "failed" and "error"
+               or ((reason == "interrupted" or reason == "exit_requested") and "idle" or "done")
+         end
+         foreground_working, foreground_status = false, nil
+         update()
+      end
+
+      panto.ext.on("session_ready", function()
+         foreground_working, foreground_status, background_work, settled_state = false, nil, 0, "idle"
+         update()
+      end)
+      panto.ext.on("turn_start", function() foreground("thinking") end)
+      panto.ext.on("agent_submission", function() foreground("thinking") end)
+      panto.ext.on("render_thinking", function() foreground("thinking") end)
+      panto.ext.on("command_start", function(event) foreground(command_status(event.command_name)) end)
+      panto.ext.on("command_end", finish)
+      panto.ext.on("render_compaction", function()
+         if foreground_working then foreground("compacting") end
+      end)
+      panto.ext.on("render_tool_details", function(event) foreground("tool:" .. label(event.tool_name)) end)
+      panto.ext.on("render_assistant_text", function() foreground("responding") end)
+      panto.ext.on("turn_end", finish)
+      panto.ext.on("background_work_start", function()
+         if not foreground_working and background_work == 0 then settled_state = "done" end
+         background_work = background_work + 1
+         update()
+      end)
+      panto.ext.on("background_work_end", function()
+         background_work = math.max(0, background_work - 1)
+         update()
+      end)
+      panto.ext.on("session_end", function()
+         ended = true
+         report("clear")
+      end)
+   end,
+}