feat(javascript): support if statement

Goose97 · Nov 21, 2024 · ce490ac · ce490ac
1 parent 6f44821
commit ce490ac
Show file tree

Hide file tree

Showing 3 changed files with 87 additions and 0 deletions.
diff --git a/queries/javascript/neolog-log-container.scm b/queries/javascript/neolog-log-container.scm
@@ -16,3 +16,8 @@
     body: (statement_block) @logable_range (#adjust-range! @logable_range 1 -1)
   )
 )
+
+(if_statement
+  condition: (parenthesized_expression) @log_container
+  consequence: (statement_block) @logable_range (#adjust-range! @logable_range 1 -1)
+)
diff --git a/tests/neolog/actions/lang/javascript_spec.lua b/tests/neolog/actions/lang/javascript_spec.lua
@@ -434,6 +434,47 @@ describe("javascript", function()
     end)
   end)
 
+  it("supports if clause", function()
+    local actions = require("neolog.actions")
+
+    local input = [[
+      if (fo|o > 1 && bar < baz) {
+        return null
+      }
+    ]]
+
+    helper.assert_scenario({
+      input = input,
+      filetype = "javascript",
+      action = function()
+        actions.add_log({ log_template = [[console.log("%identifier", %identifier)]], position = "below" })
+      end,
+      expected = [[
+        if (foo > 1 && bar < baz) {
+          console.log("foo", foo)
+          return null
+        }
+      ]],
+    })
+
+    helper.assert_scenario({
+      input = input,
+      filetype = "javascript",
+      action = function()
+        vim.cmd("normal! vi(")
+        actions.add_log({ log_template = [[console.log("%identifier", %identifier)]], position = "below" })
+      end,
+      expected = [[
+        if (foo > 1 && bar < baz) {
+          console.log("foo", foo)
+          console.log("bar", bar)
+          console.log("baz", baz)
+          return null
+        }
+      ]],
+    })
+  end)
+
   describe("supports import statements", function()
     it("supports plain imports", function()
       local actions = require("neolog.actions")

diff --git a/tests/neolog/actions/lang/typescript_spec.lua b/tests/neolog/actions/lang/typescript_spec.lua
@@ -486,6 +486,47 @@ describe("typescript", function()
     end)
   end)
 
+  it("supports if clause", function()
+    local actions = require("neolog.actions")
+
+    local input = [[
+      if (fo|o > 1 && bar < baz) {
+        return null
+      }
+    ]]
+
+    helper.assert_scenario({
+      input = input,
+      filetype = "typescript",
+      action = function()
+        actions.add_log({ log_template = [[console.log("%identifier", %identifier)]], position = "below" })
+      end,
+      expected = [[
+        if (foo > 1 && bar < baz) {
+          console.log("foo", foo)
+          return null
+        }
+      ]],
+    })
+
+    helper.assert_scenario({
+      input = input,
+      filetype = "typescript",
+      action = function()
+        vim.cmd("normal! vi(")
+        actions.add_log({ log_template = [[console.log("%identifier", %identifier)]], position = "below" })
+      end,
+      expected = [[
+        if (foo > 1 && bar < baz) {
+          console.log("foo", foo)
+          console.log("bar", bar)
+          console.log("baz", baz)
+          return null
+        }
+      ]],
+    })
+  end)
+
   describe("supports import statements", function()
     it("supports plain imports", function()
       local actions = require("neolog.actions")