1st draft

Author: numToStr

a.jsx

@@ -0,0 +1,57 @@
+function Yo() {
+  return (
+    <>
+      <div>hello</div>
+    </>
+  );
+}
+
+const Yoo = () => {
+  return (
+    <div>
+      <section>
+        <p>hello</p>
+        <p
+          attr={{
+            ...window,
+            hello: () => {
+              return (
+                <section>
+                  <p>IDK</p>
+                </section>
+              );
+            },
+          }}
+        >
+          hello
+        </p>
+      </section>
+    </div>
+  );
+};
+
+class Yooo {
+  render() {
+    return (
+      <>
+        <div>hello</div>
+      </>
+    );
+  }
+}
+
+const Yoooo = () => (
+  <>
+    <div>hello</div>
+  </>
+);
+
+const Yoooo = () => (
+  <section>
+    <div>hello</div>
+  </section>
+);
+
+function Yoooo() {
+  return <div>hello</div>;
+}

a.md

@@ -0,0 +1,83 @@
+# this is markdown
+
+```javascript
+function Yo() {
+  return (
+    <>
+      <div>hello</div>
+    </>
+  );
+}
+```
+
+```javascript
+const Yoo = () => {
+  return (
+    <div>
+      <section>
+        <p>hello</p>
+        <p
+          attr={{
+            ...window,
+            hello: () => {
+              return (
+                <section>
+                  <p>IDK</p>
+                </section>
+              );
+            },
+          }}
+        >
+          hello
+        </p>
+      </section>
+    </div>
+  );
+};
+```
+
+```tsx
+const Yoooo = () => (
+  <>
+    <div>hello</div>
+  </>
+);
+```
+
+```tsx
+const Yoooo = () => (
+  <section>
+    <div>hello</div>
+  </section>
+);
+```
+
+```tsx
+function Yo() {
+  return <div>hello</div>;
+}
+```
+
+```tsx
+function Yo() {
+  return <></>;
+}
+```
+
+```tsx
+class Yooo {
+  render() {
+    return (
+      <>
+        <div>hello</div>
+      </>
+    );
+  }
+}
+```
+
+<!-- This is block is just for testing -->
+
+```lua
+local a = 123
+```

lua/Comment/ft.lua

@@ -79,6 +79,7 @@ local L = {
     template = { M.double_hash },
     tmux = { M.hash },
     toml = { M.hash },
+    tsx = { M.cxx_l, M.cxx_b }, -- Alias for `typescriptreact` bcz `tsx` is the parser's name
     typescript = { M.cxx_l, M.cxx_b },
     typescriptreact = { M.cxx_l, M.cxx_b },
     vim = { '"%s' },

lua/Comment/jsx.lua

@@ -0,0 +1,57 @@
+local J = {
+    comment = '{/*%s*/}',
+}
+
+local query = [[
+    (parenthesized_expression
+        [(jsx_fragment) (jsx_element)] @jsx)
+
+    (return_statement
+        [(jsx_fragment) (jsx_element)] @jsx)
+]]
+
+local function is_jsx(lang)
+    -- Name of the treesitter parsers that supports jsx syntax
+    return lang == 'tsx' or lang == 'javascript'
+end
+
+local function capture(child, range)
+    local lang = child:lang()
+
+    if not is_jsx(lang) then
+        return
+    end
+
+    local Q = vim.treesitter.query.parse_query(lang, query)
+
+    for _, tree in ipairs(child:trees()) do
+        for _, node in Q:iter_captures(tree:root(), child:source()) do
+            local srow, _, erow = node:range()
+            -- Why subtracting 2?
+            -- 1. Lua indexes starts from 1
+            -- 2. Removing the `return` keyword from the range
+            if srow <= range.srow - 2 and erow >= range.erow then
+                return J.comment
+            end
+        end
+    end
+end
+
+function J.calculate(ctx)
+    local ok, P = pcall(vim.treesitter.get_parser)
+
+    if not ok then
+        return
+    end
+
+    for _, child in pairs(P:children()) do
+        local captured = capture(child, ctx.range)
+        if captured then
+            return captured
+        end
+    end
+
+    return capture(P, ctx.range)
+end
+
+return J