Add fennel treesitter queries and tests

lua/nvim-paredit/defaults.lua

@@ -119,11 +119,14 @@ M.defaults = {
     enabled = false,
     indentor = require("nvim-paredit.indentation.native").indentor,
   },
-  filetypes = { "clojure" },
+  filetypes = { "clojure", "fennel" },
   languages = {
     clojure = {
       whitespace_chars = { " ", "," },
     },
+    fennel = {
+      whitespace_chars = { " ", "," },
+    },
   },
   keys = {},
 }

queries/fennel/paredit/forms.scm

@@ -0,0 +1,3 @@
+(list) @form
+(sequence) @form
+(table) @form

tests/config.lua

@@ -11,6 +11,6 @@ vim.bo.swapfile = false
 
 require("nvim-treesitter.configs").setup({
   parser_install_dir = vim.fn.getcwd() .. "/.build/parsers",
-  ensure_installed = { "clojure" },
+  ensure_installed = { "clojure", "fennel" },
   sync_install = true,
 })

tests/nvim-paredit/fennel/element_raise_spec.lua

@@ -0,0 +1,71 @@
+local paredit = require("nvim-paredit.api")
+
+local prepare_buffer = require("tests.nvim-paredit.utils").prepare_buffer
+local expect = require("tests.nvim-paredit.utils").expect
+
+describe("element raising", function()
+  vim.api.nvim_set_option_value("filetype", "fennel", {
+    buf = 0,
+  })
+
+  it("should raise the element", function()
+    prepare_buffer({
+      content = "(a (b))",
+      cursor = { 1, 4 },
+    })
+    paredit.raise_element()
+    expect({
+      content = "(a b)",
+      cursor = { 1, 3 },
+    })
+  end)
+
+  it("should raise form elements when cursor is placed on edge", function()
+    prepare_buffer({
+      content = "(a (b))",
+      cursor = { 1, 3 },
+    })
+
+    paredit.raise_element()
+    expect({
+      content = "(b)",
+      cursor = { 1, 0 },
+    })
+
+    prepare_buffer({
+      content = "(a #(b))",
+      cursor = { 1, 3 },
+    })
+
+    paredit.raise_element()
+    expect({
+      content = "#(b)",
+      cursor = { 1, 0 },
+    })
+  end)
+
+  it("should raise a multi-line element", function()
+    prepare_buffer({
+      content = { "(a (b", " c))" },
+      cursor = { 1, 3 },
+    })
+
+    paredit.raise_element()
+    expect({
+      content = { "(b", " c)" },
+      cursor = { 1, 0 },
+    })
+  end)
+
+  it("should do nothing if it is a direct child of the document root", function()
+    prepare_buffer({
+      content = { "a", "b" },
+      cursor = { 1, 0 },
+    })
+    paredit.raise_form()
+    expect({
+      content = { "a", "b" },
+      cursor = { 1, 0 },
+    })
+  end)
+end)

tests/nvim-paredit/fennel/form_raise_spec.lua

@@ -0,0 +1,67 @@
+local paredit = require("nvim-paredit.api")
+
+local prepare_buffer = require("tests.nvim-paredit.utils").prepare_buffer
+local expect_all = require("tests.nvim-paredit.utils").expect_all
+local expect = require("tests.nvim-paredit.utils").expect
+
+describe("form raising", function()
+  vim.api.nvim_set_option_value("filetype", "fennel", {
+    buf = 0,
+  })
+
+  it("should raise the form", function()
+    expect_all(paredit.raise_form, {
+      {
+        "list",
+        before_content = "(a (b c))",
+        before_cursor = { 1, 6 },
+        after_content = "(b c)",
+        after_cursor = { 1, 0 },
+      },
+      {
+        "list with reader",
+        before_content = "(a #(b c))",
+        before_cursor = { 1, 5 },
+        after_content = "#(b c)",
+        after_cursor = { 1, 0 },
+      },
+    })
+  end)
+
+  it("should raise a multi-line form", function()
+    prepare_buffer({
+      content = { "(a (b ", "c))" },
+      cursor = { 1, 4 },
+    })
+
+    paredit.raise_form()
+    expect({
+      content = { "(b ", "c)" },
+      cursor = { 1, 0 },
+    })
+  end)
+
+  it("should do nothing if it is a direct child of the document root", function()
+    prepare_buffer({
+      content = { "(a)", "b" },
+      cursor = { 1, 1 },
+    })
+    paredit.raise_form()
+    expect({
+      content = { "(a)", "b" },
+      cursor = { 1, 1 },
+    })
+  end)
+
+  it("should do nothing if it is outside of a form", function()
+    prepare_buffer({
+      content = { "a", "b" },
+      cursor = { 1, 0 },
+    })
+    paredit.raise_form()
+    expect({
+      content = { "a", "b" },
+      cursor = { 1, 0 },
+    })
+  end)
+end)

tests/nvim-paredit/fennel/motion_spec.lua

@@ -0,0 +1,49 @@
+local paredit = require("nvim-paredit.api")
+
+local prepare_buffer = require("tests.nvim-paredit.utils").prepare_buffer
+local expect = require("tests.nvim-paredit.utils").expect
+
+describe("motions", function()
+  vim.api.nvim_set_option_value("filetype", "fennel", {
+    buf = 0,
+  })
+
+  it("should jump to next element in form (tail)", function()
+    prepare_buffer({
+      content = "(aa (bb) #(cc))",
+      cursor = { 1, 2 },
+    })
+
+    paredit.move_to_next_element_tail()
+    expect({
+      cursor = { 1, 7 },
+    })
+
+    paredit.move_to_next_element_tail()
+    expect({
+      cursor = { 1, 13 },
+    })
+
+    paredit.move_to_next_element_tail()
+    expect({
+      cursor = { 1, 13 },
+    })
+  end)
+
+  it("should jump to next element in form (head)", function()
+    prepare_buffer({
+      content = "(aa (bb) #(cc))",
+      cursor = { 1, 2 },
+    })
+
+    paredit.move_to_next_element_head()
+    expect({
+      cursor = { 1, 4 },
+    })
+
+    paredit.move_to_next_element_head()
+    expect({
+      cursor = { 1, 9 },
+    })
+  end)
+end)

tests/nvim-paredit/fennel/slurp_spec.lua

@@ -0,0 +1,107 @@
+local paredit = require("nvim-paredit.api")
+
+local prepare_buffer = require("tests.nvim-paredit.utils").prepare_buffer
+local expect_all = require("tests.nvim-paredit.utils").expect_all
+local expect = require("tests.nvim-paredit.utils").expect
+
+describe("slurping forward", function()
+  vim.api.nvim_set_option_value("filetype", "fennel", {
+    buf = 0,
+  })
+  local parser = vim.treesitter.get_parser(0)
+  if not parser then
+    return error("Failed to get parser for fennel")
+  end
+
+  it("should slurp forward different form types", function()
+    expect_all(paredit.slurp_forwards, {
+      {
+        "list",
+        before_content = "() a",
+        before_cursor = { 1, 1 },
+        after_content = "( a)",
+        after_cursor = { 1, 1 },
+      },
+      {
+        "vector",
+        before_content = "[] a",
+        before_cursor = { 1, 1 },
+        after_content = "[ a]",
+        after_cursor = { 1, 1 },
+      },
+      {
+        "quoted list",
+        before_content = "`() a",
+        before_cursor = { 1, 2 },
+        after_content = "`( a)",
+        after_cursor = { 1, 2 },
+      },
+    })
+  end)
+
+  it("should skip comments", function()
+    prepare_buffer({
+      content = { "()", ";; comment", "a" },
+      cursor = { 1, 1 },
+    })
+    paredit.slurp_forwards()
+    expect({
+      content = { "(", ";; comment", "a)" },
+      cursor = { 1, 0 },
+    })
+  end)
+
+  it("should recursively slurp the next sibling", function()
+    prepare_buffer({
+      content = "(()) 1 2",
+      cursor = { 1, 2 },
+    })
+
+    paredit.slurp_forwards()
+    expect({
+      content = "(() 1) 2",
+      cursor = { 1, 2 },
+    })
+
+    parser:parse()
+
+    paredit.slurp_forwards()
+    expect({
+      content = "(( 1)) 2",
+      cursor = { 1, 2 },
+    })
+
+    parser:parse()
+
+    prepare_buffer({
+      content = "(( 1)) 2",
+      cursor = { 1, 2 },
+    })
+
+    paredit.slurp_forwards()
+    expect({
+      content = "(( 1) 2)",
+      cursor = { 1, 2 },
+    })
+
+    parser:parse()
+
+    paredit.slurp_forwards()
+    expect({
+      content = "(( 1 2))",
+      cursor = { 1, 2 },
+    })
+  end)
+
+  it("should follow the bracket with cursor", function()
+    prepare_buffer({
+      content = "() 1 2",
+      cursor = { 1, 1 },
+    })
+    paredit.slurp_forwards({ cursor_behaviour = "follow" })
+    expect({
+      content = "( 1) 2",
+      cursor = { 1, 3 },
+    })
+  end)
+end)