Finalize 2

Author: Colin McNeil

functions/tree_sitter/main.js

@@ -1,19 +1,49 @@
 const TreeSitter = require('tree-sitter');
 const Python = require('tree-sitter-python');
+const TypeScript = require('tree-sitter-typescript');
+const JavaScript = require('tree-sitter-javascript');
 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);
+const ext = args.path.split('.').pop();
+
+let language
+
+if (ext === 'py') {
+    language = Python
+}
+else if (ext === 'ts') {
+    language = TypeScript.typescript
+}
+else if (ext === 'tsx') {
+    language = TypeScript.tsx
+}
+else if (ext === 'js' || ext === 'jsx') {
+    language = JavaScript
+}
+
+parser.setLanguage(language);
+
 // 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
+const line_to_grab = args.line + 1
+
+if (line_to_grab === undefined) {
+    console.log('No line number provided')
+    process.exit(1)
+}
+
+if (line_to_grab > codeContent.split('\n').length) {
+    console.log('Line number provided is greater than the number of lines in the file')
+    process.exit(1)
+}
 
 // Look for node where node.startPosition.row and node.endPosition.row are equal to line_to_grab
 const search_node = (node) => {
@@ -31,16 +61,23 @@ const search_node = (node) => {
 
 const line_node = search_node(parsed.rootNode)
 
+if (!line_node) {
+    console.log('No node found for line', line_to_grab)
+    console.log('Code:', codeContent.split('\n').slice(line_to_grab - 5, line_to_grab + 5).join('\n'))
+    process.exit(1)
+}
+
 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')
+if (parent.text.length > 256) {
+    console.log('Parent node text is too long, truncating')
+    const index = parent.text.indexOf(line_node.text)
+    const start = index - 128
+    const end = index + 128
+    parent.text = parent.text.slice(start, end)
 }
 
+
 console.log({
     offending_line: line_node.text,
     line_node: line_node,

prompts/eslint/020_user_prompt.md

@@ -1,3 +1,3 @@
-Lint my project's JS/TS files.
+Lint my project's JS/TS files using `summary` output.
 
 Report what you did, broken down by each tool.

prompts/eslint_fix/020_user_prompt.md

@@ -1 +1,3 @@
 Fix any lint violations you can using the tools provided.
+
+Please only fix `no-unused-vars`

prompts/eslint_fix/fixing.md

@@ -1,49 +1,11 @@
 1. **Get the line**: Use read_eslint with the `path` arg to get all of the violations for a file.
-2. **Get the code**: Use tree_sitter with the path and line to read the surrounding code if necessary. Repeat for all violations in the file.
-3. **Make the correction** Respond in the following format:
+2. **Get the code**: Use tree_sitter with the path and line to read the surrounding code if necessary. 
+3. **Make the correction**: Suggest an edit in the following format:
 
 ```json
 {
-    "start": [1, 4],
-    "end": [2, 4],
-    "edit": "Lorem ipsum"
+    "start": [1, 4], // row,col for start character
+    "end": [2, 4], // row,col for end character
+    "edit": "Lorem ipsum" // The edit to make
 }
-```
-
-Once you have fixed one file, move on to the next.
-
-You are able to fix the following violations:
-
-constructor-super:
-no-compare-neg-0:
-no-const-assign
-no-control-regex
-no-debugger
-no-empty-character-classes
-no-empty-pattern
-no-new-native-constructors
-no-obj-calls
-no-self-assign
-getter-return
-no-async-promise-executer
-no-class-assign
-no-constant-binary-expression
-no-dupe-args
-no-dupe-else-if
-no-dupe-keys
-no-duplicate-case
-no-ex-assign
-no-func-assign
-no-import-assign
-no-irregular-whitespace
-no-loss-of-precision
-no-misleading-character-class
-no-this-before-super
-no-undef
-no-unexpected-multiline
-no-unreachable
-no-unsafe-finally
-no-unsafe-negation
-no-unused-private-class-members
-no-unused-vars
-no-useless-backreference
+```