feat: show message count in session selection (#37)

Author: azorng

README.md

@@ -81,7 +81,7 @@ require('goose').setup({
     window = {
       submit = '<cr>',                     -- Submit prompt
       close = '<esc>',                     -- Close UI windows
-      stop = '<C-c>',                      -- Stop a running job
+      stop = '<C-c>',                      -- Stop goose while it is running
       next_message = ']]',                 -- Navigate to next message in the conversation
       prev_message = '[[',                 -- Navigate to previous message in the conversation
       mention_file = '@',                  -- Pick a file and add to context. See File Mentions section

lua/goose/session.lua

@@ -14,14 +14,15 @@ function M.get_all_sessions()
     return {
       workspace = session.metadata.working_dir,
       description = session.metadata.description,
+      message_count = session.metadata.message_count,
+      tokens = session.metadata.total_tokens,
       modified = session.modified,
       name = session.id,
       path = session.path
     }
   end, sessions)
 end
 
--- Helper function to get all sessions as JSON
 function M.get_all_workspace_sessions()
   local sessions = M.get_all_sessions()
   if not sessions then return nil end

lua/goose/ui/output_renderer.lua

@@ -261,7 +261,13 @@ end
 
 function M.render_session_bar()
   local function update_winbar(desc)
-    -- content
+    --[[ ** TODO: use this right side win bar text with something useful
+      local winwidth = vim.api.nvim_win_get_width(state.windows.output_win)
+      local txt = "txt content here"
+      local padding = string.rep(" ", winwidth - #desc - #txt - 1)
+      local right_side_txt = padding .. txt
+    ]]
+
     vim.wo[state.windows.output_win].winbar = " " .. desc
 
     -- Add our winbar highlights while preserving existing highlights

lua/goose/ui/session_formatter.lua

@@ -62,12 +62,13 @@ function M._format_message(message)
 
   for _, part in ipairs(message.content) do
     if part.type == 'text' and part.text and part.text ~= "" then
+      local text = vim.trim(part.text)
       has_content = true
 
       if message.role == 'user' then
-        M._format_user_message(lines, part.text)
+        M._format_user_message(lines, text)
       elseif message.role == 'assistant' then
-        for _, line in ipairs(vim.split(part.text, "\n")) do
+        for _, line in ipairs(vim.split(text, "\n")) do
           table.insert(lines, line)
         end
       end

lua/goose/ui/ui.lua

@@ -170,12 +170,22 @@ function M.select_session(sessions, cb)
   vim.ui.select(sessions, {
     prompt = "",
     format_item = function(session)
-      if not session.modified then
-        return session.description
+      local parts = {}
+
+      if session.description then
+        table.insert(parts, session.description)
+      end
+
+      if session.message_count then
+        table.insert(parts, session.message_count .. " messages")
       end
 
       local modified = util.time_ago(session.modified)
-      return session.description .. " ~ " .. modified
+      if modified then
+        table.insert(parts, modified)
+      end
+
+      return table.concat(parts, " ~ ")
     end
   }, function(session_choice)
     cb(session_choice)