Tree sitter tool

Author: Colin McNeil

functions/tree_sitter/Dockerfile

@@ -0,0 +1,18 @@
+FROM node:alpine3.20 AS build
+
+WORKDIR /tree-sitter
+
+# Install python/pip
+ENV PYTHONUNBUFFERED=1
+RUN apk add --update --no-cache python3 make gcc g++ && ln -sf python3 /usr/bin/python
+
+RUN npm install tree-sitter tree-sitter-python tree-sitter-typescript tree-sitter-javascript
+
+
+# Copy tree-sitter dir to new image
+FROM node:alpine3.20
+
+COPY main.js /main.js
+COPY --from=build /tree-sitter/node_modules /node_modules
+
+ENTRYPOINT [ "node", "main.js" ] 

functions/tree_sitter/main.js

@@ -0,0 +1,49 @@
+const TreeSitter = require('tree-sitter');
+const Python = require('tree-sitter-python');
+const fs = require('fs');
+
+const args = JSON.parse(process.argv[2])
+
+// Load the Python parser
+const parser = new TreeSitter();
+// Set the language to the parser
+parser.setLanguage(Python);
+// Read the code file content
+const codeContent = fs.readFileSync(args.path, 'utf8');
+// Parse the code using the chosen parser
+const parsed = parser.parse(codeContent);
+
+const line_to_grab = args.line
+
+// Look for node where node.startPosition.row and node.endPosition.row are equal to line_to_grab
+const search_node = (node) => {
+    if (node.startPosition.row === line_to_grab && node.endPosition.row === line_to_grab) {
+        return node
+    }
+    for (const child of node.children) {
+        const result = search_node(child)
+        if (result) {
+            return result
+        }
+    }
+    return null
+}
+
+const line_node = search_node(parsed.rootNode)
+
+const parent = line_node.parent
+
+if (parent) {
+    const start_line = parent.startPosition.row
+    const end_line = parent.endPosition.row
+    // Return codeContent from start_line to end_line
+    const lines = codeContent.split('\n').slice(start_line, end_line + 1)
+    parent.content = lines.join('\n')
+}
+
+console.log({
+    offending_line: line_node.text,
+    line_node: line_node,
+    parent: parent,
+    parent_text: parent.text
+})

functions/tree_sitter/readme.md

@@ -0,0 +1,4 @@
+```sh
+docker build . -t vonwig/tree_sitter
+docker run --mount type=bind,source=$PWD,target=/project --workdir /project vonwig/tree_sitter '{"path": "test_file.py", "line": 32}'
+```

functions/tree_sitter/test_file.py

@@ -0,0 +1,40 @@
+from pprint import pprint
+from textwrap import wrap
+from flask import Flask, jsonify, request
+
+app = Flask(__name__)
+methods = ["GET", "POST", "PATCH", "DELETE"]
+
+
+@app.route("/", methods=methods, defaults={"path": ""})
+@app.route("/<path:path>", methods=methods)
+def hello_world(path):
+    divider = "================================================================"
+    j = request.get_json()
+
+    print(divider)
+    print(f"*** Received data at: {path}")
+
+    print("\n** data:")
+    print("\n".join(wrap(request.data.decode())))
+
+    print("\n** form:")
+    pprint(request.form)
+
+    print("\n** json:")
+    pprint(j)
+
+    print(f"{divider}\n\n")
+
+    return jsonify(
+        {
+            "endpoint": path,
+            "data": request.data.decode("utf-8"),
+            "form": request.form,
+            "json": request.get_json(),
+        }
+    )
+
+
+if __name__ == "__main__":
+    app.run()