@oisheek_c/codeforge 1.0.1

package/packages/retrieval/parser/queries/csharp.js

@@ -0,0 +1,151 @@
+/**
+ * Tree-sitter queries for C#.
+ *
+ * Stable capture namespaces:
+ *
+ *  @symbol.namespace
+ *  @symbol.class
+ *  @symbol.struct
+ *  @symbol.interface
+ *  @symbol.enum
+ *  @symbol.record
+ *  @symbol.delegate
+ *  @symbol.constructor
+ *  @symbol.method
+ *  @symbol.property
+ *  @symbol.field
+ *  @symbol.event
+ *  @symbol.variable
+ *
+ *  @import
+ *  @export
+ *  @call
+ *  @comment
+ *  @todo
+ */
+
+export const SYMBOLS = `
+;; ============================================================
+;; Namespace
+;; ============================================================
+
+(namespace_declaration
+  name: (qualified_name) @symbol.namespace)
+
+(file_scoped_namespace_declaration
+  name: (qualified_name) @symbol.namespace)
+
+;; ============================================================
+;; Classes
+;; ============================================================
+
+(class_declaration
+  name: (identifier) @symbol.class)
+
+;; ============================================================
+;; Structs
+;; ============================================================
+
+(struct_declaration
+  name: (identifier) @symbol.struct)
+
+;; ============================================================
+;; Interfaces
+;; ============================================================
+
+(interface_declaration
+  name: (identifier) @symbol.interface)
+
+;; ============================================================
+;; Enums
+;; ============================================================
+
+(enum_declaration
+  name: (identifier) @symbol.enum)
+
+;; ============================================================
+;; Records
+;; ============================================================
+
+(record_declaration
+  name: (identifier) @symbol.record)
+
+;; ============================================================
+;; Delegates
+;; ============================================================
+
+(delegate_declaration
+  name: (identifier) @symbol.delegate)
+
+;; ============================================================
+;; Constructors
+;; ============================================================
+
+(constructor_declaration
+  name: (identifier) @symbol.constructor)
+
+;; ============================================================
+;; Methods
+;; ============================================================
+
+(method_declaration
+  name: (identifier) @symbol.method)
+
+;; ============================================================
+;; Properties
+;; ============================================================
+
+(property_declaration
+  name: (identifier) @symbol.property)
+
+;; ============================================================
+;; Fields
+;; ============================================================
+
+(field_declaration
+  (variable_declaration
+    (variable_declarator
+      name: (identifier) @symbol.field)))
+
+;; ============================================================
+;; Events
+;; ============================================================
+
+(event_declaration
+  name: (identifier) @symbol.event)
+
+;; ============================================================
+;; Local Variables
+;; ============================================================
+
+(local_declaration_statement
+  (variable_declaration
+    (variable_declarator
+      name: (identifier) @symbol.variable)))
+`;
+
+export const IMPORTS = `
+(using_directive) @import
+`;
+
+export const EXPORTS = `
+`;
+
+export const CALLS = `
+(invocation_expression
+  function: (identifier) @call)
+
+(invocation_expression
+  function:
+    (member_access_expression
+      name: (identifier) @call))
+`;
+
+export const COMMENTS = `
+(comment) @comment
+`;
+
+export const TODOS = `
+(comment) @todo
+(#match? @todo "(?i)(TODO|FIXME|BUG|HACK|XXX|NOTE)")
+`;

package/packages/retrieval/parser/queries/go.js

@@ -0,0 +1,124 @@
+/**
+ * Tree-sitter queries for Go.
+ *
+ * Stable capture namespaces:
+ *
+ *  @symbol.package
+ *  @symbol.struct
+ *  @symbol.interface
+ *  @symbol.function
+ *  @symbol.method
+ *  @symbol.type
+ *  @symbol.variable
+ *  @symbol.constant
+ *
+ *  @import
+ *  @export
+ *  @call
+ *  @comment
+ *  @todo
+ */
+
+export const SYMBOLS = `
+;; ============================================================
+;; Package
+;; ============================================================
+
+(package_clause
+  (package_identifier) @symbol.package)
+
+;; ============================================================
+;; Structs
+;; ============================================================
+
+(type_declaration
+  (type_spec
+    name: (type_identifier) @symbol.struct
+    type: (struct_type)))
+
+;; ============================================================
+;; Interfaces
+;; ============================================================
+
+(type_declaration
+  (type_spec
+    name: (type_identifier) @symbol.interface
+    type: (interface_type)))
+
+;; ============================================================
+;; Type aliases
+;; ============================================================
+
+(type_declaration
+  (type_spec
+    name: (type_identifier) @symbol.type))
+
+;; ============================================================
+;; Functions
+;; ============================================================
+
+(function_declaration
+  name: (identifier) @symbol.function)
+
+;; ============================================================
+;; Methods
+;; ============================================================
+
+(method_declaration
+  name: (field_identifier) @symbol.method)
+
+;; ============================================================
+;; Variables
+;; ============================================================
+
+(var_declaration
+  (var_spec
+    name: (identifier) @symbol.variable))
+
+(short_var_declaration
+  left: (expression_list
+    (identifier) @symbol.variable))
+
+;; ============================================================
+;; Constants
+;; ============================================================
+
+(const_declaration
+  (const_spec
+    name: (identifier) @symbol.constant))
+`;
+
+export const IMPORTS = `
+(import_declaration) @import
+`;
+
+export const EXPORTS = `
+`;
+
+export const CALLS = `
+;; ============================================================
+;; foo()
+;; ============================================================
+
+(call_expression
+  function: (identifier) @call)
+
+;; ============================================================
+;; pkg.Func()
+;; obj.Method()
+;; ============================================================
+
+(call_expression
+  function:
+    (selector_expression
+      field: (field_identifier) @call))
+`;
+
+export const COMMENTS = `
+(comment) @comment
+`;
+
+export const TODOS = `
+(comment) @todo
+(#match? @todo "(?i)(TODO|FIXME|BUG|HACK|XXX|NOTE)")
+`;

package/packages/retrieval/parser/queries/index.js

@@ -0,0 +1,35 @@
+import * as javascript from "./javascript.js";
+import * as typescript from "./typescript.js";
+import * as python from "./python.js";
+import * as java from "./java.js";
+import * as go from "./go.js";
+import * as rust from "./rust.js";
+import * as c from "./c.js";
+import * as cpp from "./cpp.js";
+import * as csharp from "./csharp.js";
+import * as php from "./php.js";
+import * as ruby from "./ruby.js";
+import * as swift from "./swift.js";
+
+const QUERIES = {
+    javascript,
+    jsx: javascript,
+
+    typescript,
+    tsx: typescript,
+
+    python,
+    java,
+    go,
+    rust,
+    c,
+    cpp,
+    csharp,
+    php,
+    ruby,
+    swift,
+};
+
+export function getQuerySet(language) {
+    return QUERIES[language] ?? null;
+}

package/packages/retrieval/parser/queries/java.js

@@ -0,0 +1,128 @@
+/**
+ * Tree-sitter queries for Java.
+ *
+ * Stable capture namespaces:
+ *
+ *  @symbol.class
+ *  @symbol.interface
+ *  @symbol.enum
+ *  @symbol.record
+ *  @symbol.constructor
+ *  @symbol.method
+ *  @symbol.field
+ *  @symbol.variable
+ *
+ *  @import
+ *  @export
+ *  @call
+ *  @comment
+ *  @todo
+ */
+
+export const SYMBOLS = `
+;; ============================================================
+;; Classes
+;; ============================================================
+
+(class_declaration
+  name: (identifier) @symbol.class)
+
+;; ============================================================
+;; Interfaces
+;; ============================================================
+
+(interface_declaration
+  name: (identifier) @symbol.interface)
+
+;; ============================================================
+;; Enums
+;; ============================================================
+
+(enum_declaration
+  name: (identifier) @symbol.enum)
+
+;; ============================================================
+;; Records (Java 16+)
+;; ============================================================
+
+(record_declaration
+  name: (identifier) @symbol.record)
+
+;; ============================================================
+;; Constructors
+;; ============================================================
+
+(constructor_declaration
+  name: (identifier) @symbol.constructor)
+
+;; ============================================================
+;; Methods
+;; ============================================================
+
+(method_declaration
+  name: (identifier) @symbol.method)
+
+;; ============================================================
+;; Fields
+;; ============================================================
+
+(field_declaration
+  declarator:
+    (variable_declarator
+      name: (identifier) @symbol.field))
+
+;; ============================================================
+;; Local Variables
+;; ============================================================
+
+(local_variable_declaration
+  declarator:
+    (variable_declarator
+      name: (identifier) @symbol.variable))
+`;
+
+export const IMPORTS = `
+(import_declaration) @import
+`;
+
+export const EXPORTS = `
+`;
+
+export const CALLS = `
+;; foo()
+
+(method_invocation
+  name: (identifier) @call)
+
+;; obj.foo()
+
+(method_invocation
+  object: (_)
+  name: (identifier) @call)
+
+;; this.foo()
+
+(method_invocation
+  object: (this)
+  name: (identifier) @call)
+
+;; super.foo()
+
+(method_invocation
+  object: (super)
+  name: (identifier) @call)
+`;
+
+export const COMMENTS = `
+(line_comment) @comment
+
+(block_comment) @comment
+`;
+
+export const TODOS = `
+(line_comment) @todo
+(#match? @todo "(?i)(TODO|FIXME|BUG|HACK|XXX|NOTE)")
+
+(block_comment) @todo
+(#match? @todo "(?i)(TODO|FIXME|BUG|HACK|XXX|NOTE)")
+`;

package/packages/retrieval/parser/queries/javascript.js

@@ -0,0 +1,181 @@
+/**
+ * Tree-sitter queries for JavaScript.
+ *
+ * Every capture uses a stable namespace:
+ *
+ *  @symbol.class
+ *  @symbol.function
+ *  @symbol.method
+ *  @symbol.variable
+ *
+ *  @import
+ *  @export
+ *  @call
+ *  @comment
+ *  @todo
+ */
+
+export const SYMBOLS = `
+;; ------------------------------------------------------------
+;; Classes
+;; ------------------------------------------------------------
+
+(class_declaration
+  name: (identifier) @symbol.class)
+
+;; ------------------------------------------------------------
+;; Functions
+;; ------------------------------------------------------------
+
+(function_declaration
+  name: (identifier) @symbol.function)
+
+(generator_function_declaration
+  name: (identifier) @symbol.function)
+
+;; ------------------------------------------------------------
+;; Methods
+;; ------------------------------------------------------------
+
+(method_definition
+  name: (property_identifier) @symbol.method)
+
+;; ------------------------------------------------------------
+;; Arrow functions
+;; ------------------------------------------------------------
+
+(variable_declarator
+  name: (identifier) @symbol.function
+  value: (arrow_function))
+
+(variable_declarator
+  name: (identifier) @symbol.function
+  value: (function_expression))
+
+;; ------------------------------------------------------------
+;; Variables
+;; ------------------------------------------------------------
+
+(variable_declarator
+  name: (identifier) @symbol.variable)
+`;
+
+export const IMPORTS = `
+;; ES modules
+
+(import_statement) @import
+
+;; require(...)
+
+(call_expression
+  function: (identifier) @_require
+  (#eq? @_require "require"))
+@import
+`;
+
+export const EXPORTS = `
+;; ------------------------------------------------------------
+;; ES module declaration exports
+;; ------------------------------------------------------------
+
+(export_statement
+  declaration:
+    (function_declaration
+      name: (identifier) @export.named))
+
+(export_statement
+  declaration:
+    (generator_function_declaration
+      name: (identifier) @export.named))
+
+(export_statement
+  declaration:
+    (class_declaration
+      name: (identifier) @export.named))
+
+(export_statement
+  declaration:
+    (lexical_declaration
+      (variable_declarator
+        name: (identifier) @export.named)))
+
+;; ------------------------------------------------------------
+;; ES module named exports
+;;
+;; export { foo };
+;; export { foo as bar };
+;; export { foo } from "./module.js";
+;; ------------------------------------------------------------
+
+(export_statement
+  (export_clause
+    (export_specifier
+      alias: (identifier) @export.named)))
+
+(export_statement
+  (export_clause
+    (export_specifier
+      name: (identifier) @export.named)))
+
+;; ------------------------------------------------------------
+;; Default exports with named declarations
+;;
+;; export default function foo() {}
+;; export default class Foo {}
+;; ------------------------------------------------------------
+
+(export_statement
+  declaration:
+    (function_declaration
+      name: (identifier) @export.default))
+
+(export_statement
+  declaration:
+    (class_declaration
+      name: (identifier) @export.default))
+
+;; ------------------------------------------------------------
+;; CommonJS: module.exports = ...
+;;
+;; Store "exports" as the export marker instead of capturing
+;; the complete assignment expression.
+;; ------------------------------------------------------------
+
+(assignment_expression
+  left:
+    (member_expression
+      object: (identifier) @_module
+      property: (property_identifier) @export.commonjs)
+  (#eq? @_module "module")
+  (#eq? @export.commonjs "exports"))
+
+;; ------------------------------------------------------------
+;; CommonJS: exports.foo = ...
+;; ------------------------------------------------------------
+
+(assignment_expression
+  left:
+    (member_expression
+      object: (identifier) @_exports
+      property: (property_identifier) @export.commonjs)
+  (#eq? @_exports "exports"))
+`;
+
+export const CALLS = `
+(call_expression
+  function: (identifier) @call)
+
+(call_expression
+  function:
+    (member_expression
+      property: (property_identifier) @call))
+`;
+
+export const COMMENTS = `
+(comment) @comment
+`;
+
+export const TODOS = `
+(comment) @todo
+(#match? @todo "(TODO|FIXME|BUG|HACK|XXX|NOTE)")
+`;