fix: c files with wonky formatting now still work

.. where wonky means putting your curly on a newline after your
function declaration. Aka you operate within a religion that my
religion does not tolerate. Why not just switch to emacs? It probably
has better tree walking natively.

Closes #34

Author: aaronik

lua/treewalker/nodes.lua

@@ -4,13 +4,14 @@ local util = require "treewalker.util"
 -- These are regexes but just happen to be real simple so far
 local TARGET_BLACKLIST_TYPE_MATCHERS = {
   "comment",
-  "attribute_item",   -- decorators (rust)
-  "decorat",          -- decorators (py)
-  "else",             -- else/elseif statements (lua)
-  "elif",             -- else/elseif statements (py)
-  "end_tag",          -- html closing tags
-  "block",            -- C# puts their blocks under their fn names like a psycho
-  "declaration_list", -- C# class blocks
+  "attribute_item",     -- decorators (rust)
+  "decorat",            -- decorators (py)
+  "else",               -- else/elseif statements (lua)
+  "elif",               -- else/elseif statements (py)
+  "end_tag",            -- html closing tags
+  "block",              -- C# puts their blocks under their fn names like a psycho
+  "declaration_list",   -- C# class blocks
+  "compound_statement", -- C blocks when defined under their fn names like a psycho
 }
 
 local HIGHLIGHT_BLACKLIST_TYPE_MATCHERS = {

tests/fixtures/c.c

@@ -60,3 +60,21 @@ int main() {
     return 0;
 }
 
+// Some psychos actually structure their code like this
+void doOtherThings(Account* account)
+{
+    printf("Account Number: %d\nBalance: $%.2f\n", account->accountNumber, account->balance);
+}
+
+void doOtherThingsAgain(Account* account)
+{
+    printf("Account Number: %d\nBalance: $%.2f\n", account->accountNumber, account->balance);
+}
+
+#define PI 3.14159  // Define a constant PI (preprocessor directive)
+
+void doOtherThingsAgainAgain(Account* account)
+{
+    printf("Account Number: %d\nBalance: $%.2f\n", account->accountNumber, account->balance);
+}
+

tests/treewalker/c_spec.lua

@@ -4,6 +4,24 @@ local tw = require 'treewalker'
 local h = require 'tests.treewalker.helpers'
 local lines = require 'treewalker.lines'
 
+describe("Movement in a c file:", function()
+  before_each(function()
+    load_fixture("/c.c")
+  end)
+
+  h.ensure_has_parser()
+
+  it("Moves around", function()
+    vim.fn.cursor(46, 1)
+    tw.move_down()
+    h.assert_cursor_at(50, 1, "int main")
+    tw.move_down()
+    h.assert_cursor_at(64, 1)
+    tw.move_down()
+    h.assert_cursor_at(69, 1)
+  end)
+end)
+
 describe("Swapping in a c file:", function()
   before_each(function()
     load_fixture("/c.c")
@@ -24,4 +42,22 @@ describe("Swapping in a c file:", function()
     tw.swap_left()
     assert.same('        printf("\\ntwo\\n", "one\\n");', lines.get_line(17))
   end)
+
+  it("swaps down on next line bracket structured functions", function()
+    local first_block = lines.get_lines(63, 67)
+    local second_block = lines.get_lines(69, 72)
+    vim.fn.cursor(64, 1)
+    tw.swap_down()
+    assert.same(first_block, lines.get_lines(68, 72))
+    assert.same(second_block, lines.get_lines(63, 66))
+  end)
+
+  it("swaps up on next line bracket structured functions", function()
+    local first_block = lines.get_lines(63, 67)
+    local second_block = lines.get_lines(69, 72)
+    vim.fn.cursor(69, 1)
+    tw.swap_up()
+    assert.same(first_block, lines.get_lines(68, 72))
+    assert.same(second_block, lines.get_lines(63, 66))
+  end)
 end)