feat(neojj): add neojj (jujutsu) integration (#187)

spec/neojj_integration_spec.lua

@@ -0,0 +1,173 @@
+require('spec.helpers')
+
+vim.g.diffs = { integrations = { neojj = true } }
+
+local diffs = require('diffs')
+local parser = require('diffs.parser')
+
+local function create_buffer(lines)
+  local bufnr = vim.api.nvim_create_buf(false, true)
+  vim.api.nvim_buf_set_lines(bufnr, 0, -1, false, lines or {})
+  return bufnr
+end
+
+local function delete_buffer(bufnr)
+  if vim.api.nvim_buf_is_valid(bufnr) then
+    vim.api.nvim_buf_delete(bufnr, { force = true })
+  end
+end
+
+describe('neojj_integration', function()
+  describe('neojj_disable_hunk_highlight', function()
+    it('sets neojj_disable_hunk_highlight on NeojjStatus buffer after attach', function()
+      local bufnr = create_buffer({
+        'modified   test.lua',
+        '@@ -1,1 +1,2 @@',
+        ' local x = 1',
+        '+local y = 2',
+      })
+      vim.api.nvim_set_option_value('filetype', 'NeojjStatus', { buf = bufnr })
+      diffs.attach(bufnr)
+
+      assert.is_true(vim.b[bufnr].neojj_disable_hunk_highlight)
+
+      delete_buffer(bufnr)
+    end)
+
+    it('does not set neojj_disable_hunk_highlight on non-Neojj buffer', function()
+      local bufnr = create_buffer({})
+      vim.api.nvim_set_option_value('filetype', 'git', { buf = bufnr })
+      diffs.attach(bufnr)
+
+      assert.is_not_true(vim.b[bufnr].neojj_disable_hunk_highlight)
+
+      delete_buffer(bufnr)
+    end)
+  end)
+
+  describe('NeojjStatus buffer attach', function()
+    it('populates hunk_cache for NeojjStatus buffer with diff content', function()
+      local bufnr = create_buffer({
+        'modified   hello.lua',
+        '@@ -1,2 +1,3 @@',
+        ' local M = {}',
+        '+local x = 1',
+        ' return M',
+      })
+      vim.api.nvim_set_option_value('filetype', 'NeojjStatus', { buf = bufnr })
+      diffs.attach(bufnr)
+      local entry = diffs._test.hunk_cache[bufnr]
+      assert.is_not_nil(entry)
+      assert.is_table(entry.hunks)
+      assert.are.equal(1, #entry.hunks)
+      assert.are.equal('hello.lua', entry.hunks[1].filename)
+      delete_buffer(bufnr)
+    end)
+
+    it('populates hunk_cache for NeojjDiffView buffer', function()
+      local bufnr = create_buffer({
+        'new file   newmod.lua',
+        '@@ -0,0 +1,2 @@',
+        '+local M = {}',
+        '+return M',
+      })
+      vim.api.nvim_set_option_value('filetype', 'NeojjDiffView', { buf = bufnr })
+      diffs.attach(bufnr)
+      local entry = diffs._test.hunk_cache[bufnr]
+      assert.is_not_nil(entry)
+      assert.is_table(entry.hunks)
+      assert.are.equal(1, #entry.hunks)
+      delete_buffer(bufnr)
+    end)
+  end)
+
+  describe('parser neojj patterns', function()
+    it('detects added prefix via parser', function()
+      local bufnr = create_buffer({
+        'added   utils.py',
+        '@@ -0,0 +1,2 @@',
+        '+def hello():',
+        '+    pass',
+      })
+      local hunks = parser.parse_buffer(bufnr)
+      assert.are.equal(1, #hunks)
+      assert.are.equal('utils.py', hunks[1].filename)
+      delete_buffer(bufnr)
+    end)
+
+    it('detects updated prefix via parser', function()
+      local bufnr = create_buffer({
+        'updated   config.toml',
+        '@@ -1,2 +1,3 @@',
+        ' [section]',
+        '+key = "val"',
+        ' other = 1',
+      })
+      local hunks = parser.parse_buffer(bufnr)
+      assert.are.equal(1, #hunks)
+      assert.are.equal('config.toml', hunks[1].filename)
+      delete_buffer(bufnr)
+    end)
+
+    it('detects changed prefix via parser', function()
+      local bufnr = create_buffer({
+        'changed   main.rs',
+        '@@ -1,1 +1,2 @@',
+        ' fn main() {}',
+        '+fn helper() {}',
+      })
+      local hunks = parser.parse_buffer(bufnr)
+      assert.are.equal(1, #hunks)
+      assert.are.equal('main.rs', hunks[1].filename)
+      delete_buffer(bufnr)
+    end)
+
+    it('detects unmerged prefix via parser', function()
+      local bufnr = create_buffer({
+        'unmerged   conflict.lua',
+        '@@ -1,1 +1,2 @@',
+        ' local x = 1',
+        '+local y = 2',
+      })
+      local hunks = parser.parse_buffer(bufnr)
+      assert.are.equal(1, #hunks)
+      assert.are.equal('conflict.lua', hunks[1].filename)
+      delete_buffer(bufnr)
+    end)
+
+    it('parses multi-file neojj buffer with modified and added', function()
+      local bufnr = create_buffer({
+        'modified   test.lua',
+        '@@ -1,2 +1,3 @@',
+        ' local M = {}',
+        '+local x = 1',
+        ' return M',
+        'added   utils.py',
+        '@@ -0,0 +1,2 @@',
+        '+def hello():',
+        '+    pass',
+      })
+      local hunks = parser.parse_buffer(bufnr)
+      assert.are.equal(2, #hunks)
+      assert.are.equal('test.lua', hunks[1].filename)
+      assert.are.equal('utils.py', hunks[2].filename)
+      delete_buffer(bufnr)
+    end)
+  end)
+
+  describe('compute_filetypes', function()
+    it('includes Neojj filetypes when neojj integration is enabled', function()
+      local fts = diffs.compute_filetypes({ integrations = { neojj = true } })
+      assert.is_true(vim.tbl_contains(fts, 'NeojjStatus'))
+      assert.is_true(vim.tbl_contains(fts, 'NeojjCommitView'))
+      assert.is_true(vim.tbl_contains(fts, 'NeojjDiffView'))
+    end)
+
+    it('excludes Neojj filetypes when neojj integration is disabled', function()
+      local fts = diffs.compute_filetypes({ integrations = { neojj = false } })
+      assert.is_false(vim.tbl_contains(fts, 'NeojjStatus'))
+      assert.is_false(vim.tbl_contains(fts, 'NeojjCommitView'))
+      assert.is_false(vim.tbl_contains(fts, 'NeojjDiffView'))
+    end)
+  end)
+end)