fix(indents): indents for error block (css, lua) (#3207)

lua/nvim-treesitter/indent.lua

@@ -149,12 +149,19 @@ function M.get_indent(lnum)
     end
 
     -- do not indent for nodes that starts-and-ends on same line and starts on target line (lnum)
+    local should_process = not is_processed_by_row[srow]
+    local is_in_err = false
+    if should_process then
+      local parent = node:parent()
+      is_in_err = parent and parent:has_error()
+    end
     if
-      not is_processed_by_row[srow]
+      should_process
-      -- Dear stylua, please don't change the semantics of this statement!
+      and (
-      -- stylua: ignore start
+        q.indent[node:id()]
-      and (q.indent[node:id()] and srow ~= erow and ((srow ~= lnum - 1) or q.indent[node:id()].start_at_same_line))
+        and (srow ~= erow or is_in_err)
-      -- stylua: ignore end
+        and (srow ~= lnum - 1 or q.indent[node:id()].start_at_same_line)
+      )
     then
       indent = indent + indent_size
       is_processed = true

queries/lua/indents.scm

@@ -3,9 +3,11 @@
   (function_declaration)
   (field)
   (do_statement)
+  (method_index_expression)
   (while_statement)
   (repeat_statement)
   (if_statement)
+  "then"
   (for_statement)
   (return_statement)
   (table_constructor)

tests/indent/css_spec.lua

@@ -1,5 +1,4 @@
 local Runner = require("tests.indent.common").Runner
-local XFAIL = require("tests.indent.common").XFAIL
 
 local run = Runner:new(it, "tests/indent/css", {
   tabstop = 2,
@@ -17,12 +16,7 @@ describe("indent CSS:", function()
 
   describe("new line:", function()
     run:new_line("open_block.css", { on_line = 1, text = "}", indent = 0 })
-    run:new_line(
+    run:new_line("open_block.css", { on_line = 1, text = "color: green;", indent = 2 })
-      "open_block.css",
-      { on_line = 1, text = "color: green;", indent = 2 },
-      "might fail because tree is in a broken state",
-      XFAIL
-    )
     run:new_line("next_rule.css", { on_line = 3, text = ".next {", indent = 0 })
   end)
 end)

tests/indent/lua/cond.lua

@@ -10,3 +10,5 @@ else
   end
   x = 0
 end
+
+if x > 2 then

tests/indent/lua/method_index_expr.lua

@@ -0,0 +1,3 @@
+Node.new()
+  :call()
+  :build()

tests/indent/lua_spec.lua

@@ -36,8 +36,12 @@ describe("indent Lua:", function()
     run:new_line("cond.lua", { on_line = 8, text = "x = x + 1", indent = 4 })
     run:new_line("cond.lua", { on_line = 10, text = "x = x + 1", indent = 2 })
     run:new_line("cond.lua", { on_line = 12, text = "x = x + 1", indent = 0 })
+    run:new_line("cond.lua", { on_line = 14, text = "x = x + 1", indent = 2 })
+    run:new_line("cond.lua", { on_line = 14, text = "end", indent = 0 })
     run:new_line("no-indent-after-paren-pairs.lua", { on_line = 3, text = "x = x + 1", indent = 0 })
     run:new_line("no-indent-after-paren-pairs.lua", { on_line = 6, text = "x = x + 1", indent = 0 })
     run:new_line("nested-table.lua", { on_line = 5, text = "{}", indent = 4 })
+    run:new_line("method_index_expr.lua", { on_line = 1, text = ":test()", indent = 2 })
+    run:new_line("method_index_expr.lua", { on_line = 3, text = "local a = 1", indent = 0 })
   end)
 end)