jj mini/snacks rice

nvim/lua/plugins/lib/snacks_jj.lua

@@ -0,0 +1,133 @@
+local M = {}
+
+function M.status()
+  local function get_files()
+    local status_raw = vim.fn.system('jj diff --no-pager --quiet --summary')
+    local files = {}
+
+    for status in status_raw:gmatch('[^\r\n]+') do
+      local state, text = string.match(status, '^(%a)%s(.+)$')
+
+      if state and text then
+        local file = text
+
+        local hl = ''
+        if state == 'A' then
+          hl = 'SnacksPickerGitStatusAdded'
+        elseif state == 'M' then
+          hl = 'SnacksPickerGitStatusModified'
+        elseif state == 'D' then
+          hl = 'SnacksPickerGitStatusDeleted'
+        elseif state == 'R' then
+          hl = 'SnacksPickerGitStatusRenamed'
+          file = string.match(text, '{.-=>%s*(.-)}')
+        end
+
+        local diff = vim.fn.system('jj diff ' .. file .. ' --no-pager --stat --git')
+        table.insert(files, {
+          text = text,
+          file = file,
+          filename_hl = hl,
+          state = state,
+          diff = diff,
+        })
+      end
+    end
+
+    return files
+  end
+
+  local files = get_files()
+
+  Snacks.picker.pick {
+    source = 'jj_status',
+    items = files,
+    format = 'file',
+    title = 'jj status',
+    preview = function(ctx)
+      if ctx.item.file then
+        Snacks.picker.preview.diff(ctx)
+      else
+        ctx.preview:reset()
+        ctx.preview:set_title('No preview')
+      end
+    end,
+  }
+end
+
+function M.revs()
+  local function jj_new()
+    return function(picker, item)
+      picker:close()
+      if item then
+        if not item.rev then
+          Snacks.notify.warn('No branch or commit found', { title = 'Snacks Picker' })
+          return
+        end
+        local cmd = { 'jj', 'new', '-r', item.rev }
+        Snacks.picker.util.cmd(cmd, function()
+          Snacks.notify('Checking out revision: ' .. item.rev, { title = 'Snacks Picker' })
+          vim.cmd.checktime()
+        end, { cwd = item.cwd })
+      end
+    end
+  end
+
+  local function jj_rev_cmd(rev)
+    if rev ~= nil then
+      return vim.fn.system { 'jj', 'show', '--git', '-r', rev }
+    else
+      return 'No preview available.'
+    end
+  end
+
+  local function jj_log(revset)
+    if revset == nil then
+      revset = '-r @'
+    else
+      revset = '-r ' .. revset
+    end
+    local status_raw = vim.fn.system(
+      'jj log '
+      .. revset
+      ..
+      ' --template \'if(root, format_root_commit(self), label(if(current_working_copy, "working_copy"), concat( format_short_commit_header(self) ++ " ", separate(" ", if(empty, label("empty", "(empty)")), if(description, description.first_line(), label(if(empty, "empty"), description_placeholder),),) ++ "\n",),))\''
+    )
+    local lines = {}
+
+    for line in status_raw:gmatch('[^\r\n]+') do
+      local sign, rev, description = string.match(line, '(.)%s*(%a+)(.*)')
+      table.insert(lines, {
+        text = line,
+        file = line,
+        sign = sign,
+        rev = rev,
+        hl = 'SnacksPickerGitMsg',
+        description = description,
+        diff = jj_rev_cmd(rev),
+      })
+    end
+
+    return lines
+  end
+
+  local lines = jj_log('::@')
+
+  Snacks.picker.pick {
+    source = 'jj_revs',
+    items = lines,
+    format = 'text',
+    title = 'jj log',
+    confirm = jj_new(),
+    preview = function(ctx)
+      if ctx.item.file then
+        Snacks.picker.preview.diff(ctx)
+      else
+        ctx.preview:reset()
+        ctx.preview:set_title('No rev found')
+      end
+    end,
+  }
+end
+
+return M