lonny-agent 0.2.4 → 0.2.7

package/src/tools/codebase/ast/tree-sitter-converters.ts

@@ -0,0 +1,480 @@
+import type * as WTS from 'web-tree-sitter'
+
+import type { Class, ClassMember, FuncDecl, Import, Statement, Variable } from './types.js'
+
+/** Export-related node types that may wrap function/class/variable declarations */
+const EXPORT_NODE_TYPES = new Set([
+  'export_statement',
+  'export_named_statement',
+  'export_default_statement',
+  'export_default',
+])
+
+export function walkStatements(node: WTS.Node, lang: string): Statement[] {
+  const statements: Statement[] = []
+  for (const child of node.children) {
+    const stmt = convertNode(child, lang)
+    if (stmt) {
+      statements.push(stmt)
+    }
+    // Extract inline functions from variable declarations (e.g. `const fn = () => {}`)
+    if (lang !== 'python') {
+      for (const decl of child.children) {
+        if (decl.type === 'variable_declarator') {
+          for (const valChild of decl.children) {
+            if (valChild.type === 'arrow_function' || valChild.type === 'function') {
+              statements.push(convertFunction(valChild))
+            }
+          }
+        }
+      }
+    }
+    // Recursively extract exports' inner declarations (function/class/variable)
+    // so `export function foo()` is captured both as Export and as FuncDecl.
+    if (EXPORT_NODE_TYPES.has(child.type)) {
+      const subStatements = walkStatements(child, lang)
+      for (const sub of subStatements) {
+        if (
+          !statements.some(
+            s => s.type === sub.type && 'name' in s && 'name' in sub && s.name === sub.name,
+          )
+        ) {
+          statements.push(sub)
+        }
+      }
+      // Handle `export { foo, bar }` / `export type { Foo }` named export lists
+      for (const clause of child.namedChildren) {
+        if (clause.type === 'export_clause' || clause.type === 'named_exports') {
+          for (const spec of clause.namedChildren) {
+            if (spec.type === 'export_specifier') {
+              const exportedName =
+                spec.childForFieldName('alias')?.text || // export { x as y } → y
+                spec.childForFieldName('name')?.text || // export { x } → x
+                spec.text
+              if (
+                !statements.some(s => s.type === 'Export' && 'name' in s && s.name === exportedName)
+              ) {
+                statements.push({ type: 'Export' as const, name: exportedName })
+              }
+            }
+          }
+        }
+      }
+    }
+  }
+  return statements
+}
+
+function convertNode(node: WTS.Node, lang: string): Statement | null {
+  const type = node.type
+
+  if (lang === 'typescript' || lang === 'javascript') {
+    if (type === 'function_declaration') return convertFunction(node)
+    if (type === 'arrow_function') return convertFunction(node)
+    if (type === 'class_declaration') return convertClass(node)
+    if (type === 'lexical_declaration') return convertLexicalDeclaration(node)
+    if (type === 'variable_declaration') return convertVariableDeclaration(node)
+    if (type === 'var_declaration') return convertVariableDeclaration(node)
+    if (type === 'import_statement') return convertImport(node)
+    if (type === 'import_declaration') return convertImport(node)
+    if (type === 'export_statement' || type === 'export_named_statement') return convertExport(node)
+    if (type === 'export_default_statement' || type === 'export_default')
+      return convertExportDefault(node)
+    if (type === 'generator_function_declaration') return convertFunction(node)
+    if (type === 'generator_function') return convertFunction(node)
+  }
+
+  if (lang === 'python') {
+    if (type === 'function_definition') return convertPythonFunction(node)
+    if (type === 'class_definition') return convertPythonClass(node)
+    if (type === 'assignment') return convertPythonAssignment(node)
+    if (type === 'import_statement') return convertPythonImport(node)
+    if (type === 'import_from_statement') return convertPythonImportFrom(node)
+    if (type === 'decorated_definition') return convertDecoratedDefinition(node)
+  }
+
+  return null
+}
+
+function convertFunction(node: WTS.Node): FuncDecl {
+  const nameNode = node.childForFieldName('name')
+  const paramsNode = node.childForFieldName('parameters')
+  const returnTypeNode = node.childForFieldName('return_type')
+  const asyncNode = node.children.find(c => c.type === 'async')
+  const isGenerator =
+    node.type === 'generator_function_declaration' ||
+    !!node.children.find(c => c.type === '*' || c.type === 'yield')
+
+  const parameters: { name: string }[] = []
+  if (paramsNode) {
+    for (const param of paramsNode.children) {
+      if (
+        param.type === 'identifier' ||
+        param.type === 'required_parameter' ||
+        param.type === 'optional_parameter'
+      ) {
+        const name =
+          param.childForFieldName('pattern')?.text ||
+          param.childForFieldName('name')?.text ||
+          param.text
+        parameters.push({ name })
+      } else if (param.type === 'assignment_pattern') {
+        const left = param.childForFieldName('left')
+        if (left) parameters.push({ name: left.text })
+      }
+    }
+  }
+
+  return {
+    type: 'Function',
+    name: nameNode?.text || null,
+    parameters,
+    returnType: returnTypeNode?.text || undefined,
+    isAsync: !!asyncNode,
+    isGenerator,
+    startLine: node.startPosition.row + 1,
+    endLine: node.endPosition.row + 1,
+  }
+}
+
+function convertMethod(node: WTS.Node): ClassMember {
+  const nameNode = node.childForFieldName('name')
+  const staticNode = node.namedChildren.find(c => c.type === 'static')
+  const modifierNode = node.namedChildren.find(c => c.type === 'accessibility_modifier')
+  const name = nameNode?.text || ''
+  const isConstructor =
+    name === 'constructor' || node.type === 'constructor_declaration' || node.type === 'constructor'
+
+  // Detect getter/setter by looking for `get`/`set` keyword children,
+  // not by fragile `includes()` on the node type string.
+  // `get`/`set` are unnamed keyword children, not named children.
+  const isGetter = node.children.some(c => c.type === 'get')
+  const isSetter = node.children.some(c => c.type === 'set')
+
+  return {
+    type: 'Method',
+    name,
+    kind: isConstructor ? 'constructor' : isGetter ? 'getter' : isSetter ? 'setter' : 'method',
+    visibility:
+      (modifierNode?.namedChildren[0]?.text as 'public' | 'private' | 'protected') || 'public',
+    isStatic: !!staticNode,
+    startLine: node.startPosition.row + 1,
+    endLine: node.endPosition.row + 1,
+  }
+}
+
+function convertClass(node: WTS.Node): Class {
+  const nameNode = node.childForFieldName('name')
+  const superNode = node.childForFieldName('superclass') || node.childForFieldName('super_class')
+  const bodyNode = node.childForFieldName('body')
+
+  const members: ClassMember[] = []
+  if (bodyNode) {
+    for (const child of bodyNode.namedChildren) {
+      if (child.type === 'method_definition' || child.type === 'method_signature') {
+        members.push(convertMethod(child))
+      } else if (
+        child.type === 'public_field_definition' ||
+        child.type === 'property_definition' ||
+        child.type === 'field_definition'
+      ) {
+        const mName = child.childForFieldName('name')
+        const staticNode = child.namedChildren.find(c => c.type === 'static')
+        members.push({
+          type: 'Property',
+          name: mName?.text || child.text,
+          kind: 'property',
+          visibility: 'public',
+          isStatic: !!staticNode,
+          startLine: child.startPosition.row + 1,
+          endLine: child.endPosition.row + 1,
+        })
+      }
+    }
+  }
+
+  return {
+    type: 'Class',
+    name: nameNode?.text || '',
+    superClass: superNode?.text || undefined,
+    members,
+    startLine: node.startPosition.row + 1,
+    endLine: node.endPosition.row + 1,
+  }
+}
+
+function convertLexicalDeclaration(node: WTS.Node): Variable | null {
+  const kind =
+    node.children.find(c => c.type === 'const' || c.type === 'let' || c.type === 'var')?.text ||
+    'const'
+  for (const declarator of node.children) {
+    if (declarator.type === 'variable_declarator') {
+      const nameNode =
+        declarator.childForFieldName('name') ||
+        declarator.children.find(c => c.type === 'identifier')
+      if (nameNode) {
+        return {
+          type: 'Variable',
+          name: nameNode.text,
+          kind: kind as 'const' | 'let' | 'var',
+          startLine: node.startPosition.row + 1,
+          endLine: node.endPosition.row + 1,
+        }
+      }
+    }
+  }
+  return null
+}
+
+function convertVariableDeclaration(node: WTS.Node): Variable | null {
+  for (const decl of node.children) {
+    if (decl.type === 'variable_declarator') {
+      const nameNode =
+        decl.childForFieldName('name') || decl.children.find(c => c.type === 'identifier')
+      if (nameNode) {
+        return {
+          type: 'Variable',
+          name: nameNode.text,
+          kind: 'var',
+          startLine: node.startPosition.row + 1,
+          endLine: node.endPosition.row + 1,
+        }
+      }
+    }
+  }
+  return null
+}
+
+function convertImport(node: WTS.Node): Import {
+  const sourceNode = node.children.find(c => c.type === 'string' || c.type === 'template_string')
+  const specifiers: { type: 'ImportSpecifier'; localName: string; importedName: string }[] = []
+
+  function extractSpecifiers(container: WTS.Node): void {
+    for (const c of container.children) {
+      if (c.type === 'import_specifier') {
+        const name = c.childForFieldName('name')
+        const alias = c.childForFieldName('alias')
+        const imported = name?.text || c.text
+        const local = alias?.text || imported
+        specifiers.push({ type: 'ImportSpecifier', localName: local, importedName: imported })
+      } else if (c.type === 'identifier') {
+        specifiers.push({ type: 'ImportSpecifier', localName: c.text, importedName: c.text })
+      } else if (c.type === 'namespace_import') {
+        const ns = c.namedChildren.find(n => n.type === 'identifier')
+        if (ns) specifiers.push({ type: 'ImportSpecifier', localName: ns.text, importedName: '*' })
+      } else if (c.type === 'named_imports' || c.type === 'import_clause') {
+        extractSpecifiers(c)
+      }
+    }
+  }
+
+  extractSpecifiers(node)
+
+  // Detect `import type { ... }` syntax
+  const isTypeOnly =
+    node.children.some(c => c.type === 'type') ||
+    node.namedChildren.some(
+      c => c.type === 'import_clause' && c.namedChildren.some(cc => cc.type === 'type'),
+    )
+
+  return {
+    type: 'Import',
+    source: sourceNode?.text || node.text,
+    specifiers,
+    isTypeOnly: isTypeOnly || undefined,
+  }
+}
+
+function convertExport(node: WTS.Node): { type: 'Export'; name: string } | null {
+  const sourceNode = node.namedChildren.find(
+    c => c.type === 'string' || c.type === 'template_string',
+  )
+  if (sourceNode) {
+    return { type: 'Export', name: sourceNode.text }
+  }
+  for (const child of node.namedChildren) {
+    if (child.type === 'function_declaration' || child.type === 'class_declaration') {
+      const nameNode = child.childForFieldName('name')
+      return { type: 'Export', name: nameNode?.text || '(unnamed)' }
+    }
+    if (child.type === 'lexical_declaration' || child.type === 'variable_declaration') {
+      const decl = child.namedChildren.find(c => c.type === 'variable_declarator')
+      const nameNode = decl?.childForFieldName('name')
+      return { type: 'Export', name: nameNode?.text || '(anonymous)' }
+    }
+    if (child.type === 'assignment') {
+      const left = child.childForFieldName('left')
+      return { type: 'Export', name: left?.text || '(assignment)' }
+    }
+    // Handle `export interface Foo { ... }` and `export type Foo = ...`
+    if (child.type === 'interface_declaration' || child.type === 'type_alias_declaration') {
+      const nameNode = child.childForFieldName('name')
+      return { type: 'Export', name: nameNode?.text || '(anonymous)' }
+    }
+  }
+  return { type: 'Export', name: '(anonymous)' }
+}
+
+function convertExportDefault(node: WTS.Node): { type: 'Export'; name: string } | null {
+  const valueNode = node.childForFieldName('value')
+  if (valueNode) {
+    const nameNode = valueNode.childForFieldName('name')
+    return { type: 'Export', name: nameNode?.text || 'default' }
+  }
+  return { type: 'Export', name: 'default' }
+}
+
+function convertPythonFunction(node: WTS.Node): FuncDecl {
+  const nameNode = node.childForFieldName('name')
+  const paramsNode = node.childForFieldName('parameters')
+  const returnTypeNode = node.childForFieldName('return_type')
+  const asyncNode = node.children.find(c => c.type === 'async')
+
+  const parameters: { name: string }[] = []
+  if (paramsNode) {
+    for (const param of paramsNode.children) {
+      if (param.type === 'identifier') {
+        parameters.push({ name: param.text })
+      } else if (param.type === 'typed_parameter' || param.type === 'default_parameter') {
+        const name = param.childForFieldName('name')?.text || param.text
+        parameters.push({ name })
+      } else if (param.type === 'list_splat_pattern') {
+        const name = param.childForFieldName('name')?.text || param.text
+        parameters.push({ name: `*${name}` })
+      } else if (param.type === 'keyword_splat_pattern') {
+        const name = param.childForFieldName('name')?.text || param.text
+        parameters.push({ name: `**${name}` })
+      }
+    }
+  }
+
+  return {
+    type: 'Function',
+    name: nameNode?.text || null,
+    parameters,
+    returnType: returnTypeNode?.text || undefined,
+    isAsync: !!asyncNode,
+    isGenerator: false,
+    startLine: node.startPosition.row + 1,
+    endLine: node.endPosition.row + 1,
+  }
+}
+
+function convertPythonClass(node: WTS.Node): Class {
+  const nameNode = node.childForFieldName('name')
+  const superNode = node.childForFieldName('superclasses')
+  const bodyNode = node.childForFieldName('body')
+
+  const members: ClassMember[] = []
+  if (bodyNode) {
+    for (const child of bodyNode.children) {
+      if (child.type === 'function_definition' || child.type === 'decorated_definition') {
+        const fnNode =
+          child.type === 'decorated_definition'
+            ? child.children.find(c => c.type === 'function_definition')
+            : child
+        const mName = fnNode?.childForFieldName('name')?.text || child.text
+        const isCtor = mName === '__init__' || mName === '__new__'
+        members.push({
+          type: 'Method',
+          name: mName,
+          kind: isCtor ? 'constructor' : 'method',
+          visibility: mName.startsWith('__') ? 'private' : 'public',
+          isStatic: false,
+          startLine: child.startPosition.row + 1,
+          endLine: child.endPosition.row + 1,
+        })
+      }
+    }
+  }
+
+  return {
+    type: 'Class',
+    name: nameNode?.text || '',
+    superClass: superNode?.text || undefined,
+    members,
+    startLine: node.startPosition.row + 1,
+    endLine: node.endPosition.row + 1,
+  }
+}
+
+function convertPythonAssignment(node: WTS.Node): Variable | null {
+  const leftNode = node.childForFieldName('left')
+  if (leftNode && leftNode.type === 'identifier') {
+    return {
+      type: 'Variable',
+      name: leftNode.text,
+      kind: 'let',
+      startLine: node.startPosition.row + 1,
+      endLine: node.endPosition.row + 1,
+    }
+  }
+  if (leftNode && leftNode.type === 'pattern_list') {
+    const first = leftNode.children.find(c => c.type === 'identifier')
+    if (first) {
+      return {
+        type: 'Variable',
+        name: first.text,
+        kind: 'let',
+        startLine: node.startPosition.row + 1,
+        endLine: node.endPosition.row + 1,
+      }
+    }
+  }
+  return null
+}
+
+function convertPythonImport(node: WTS.Node): Import {
+  const specifiers: Import['specifiers'] = []
+  const nameNode = node.childForFieldName('name')
+  if (nameNode) {
+    for (const child of nameNode.children) {
+      if (child.type === 'aliased_import') {
+        const original = child.childForFieldName('name')?.text || child.text
+        const alias = child.childForFieldName('alias')?.text || original
+        specifiers.push({ type: 'ImportSpecifier', localName: alias, importedName: original })
+      } else if (child.type === 'identifier') {
+        specifiers.push({
+          type: 'ImportSpecifier',
+          localName: child.text,
+          importedName: child.text,
+        })
+      }
+    }
+  }
+  return { type: 'Import', source: '', specifiers }
+}
+
+function convertPythonImportFrom(node: WTS.Node): Import {
+  const sourceNode = node.childForFieldName('module_name')
+  const nameNode = node.childForFieldName('name')
+  const names: { type: 'ImportSpecifier'; localName: string; importedName: string }[] = []
+
+  if (nameNode) {
+    for (const child of nameNode.children) {
+      if (child.type === 'aliased_import') {
+        const original = child.childForFieldName('name')?.text || child.text
+        const alias = child.childForFieldName('alias')?.text || original
+        names.push({ type: 'ImportSpecifier', localName: alias, importedName: original })
+      } else if (child.type === 'identifier') {
+        names.push({ type: 'ImportSpecifier', localName: child.text, importedName: child.text })
+      } else if (child.type === 'wildcard_import') {
+        names.push({ type: 'ImportSpecifier', localName: '*', importedName: '*' })
+      }
+    }
+  }
+
+  return {
+    type: 'Import',
+    source: sourceNode?.text || '',
+    specifiers: names,
+  }
+}
+
+function convertDecoratedDefinition(node: WTS.Node): Statement | null {
+  const inner = node.children.find(
+    c => c.type === 'function_definition' || c.type === 'class_definition',
+  )
+  if (inner) return convertNode(inner, 'python')
+  return null
+}

package/src/tools/codebase/ast/types.ts

@@ -0,0 +1,102 @@
+export interface Module {
+  type: 'Module'
+  filePath: string
+  language: 'typescript' | 'javascript' | 'python'
+  imports: Import[]
+  body: Statement[]
+  exports: string[]
+}
+
+export interface Import {
+  type: 'Import'
+  source: string
+  specifiers: ImportSpecifier[]
+  isTypeOnly?: boolean
+}
+
+export interface ImportSpecifier {
+  type: 'ImportSpecifier'
+  localName: string
+  importedName: string
+}
+
+export interface FuncDecl {
+  type: 'Function'
+  name: string | null
+  parameters: Parameter[]
+  returnType?: string
+  isAsync: boolean
+  isGenerator: boolean
+  startLine: number
+  endLine: number
+}
+
+export interface Parameter {
+  name: string
+}
+
+export interface Class {
+  type: 'Class'
+  name: string
+  superClass?: string
+  members: ClassMember[]
+  startLine: number
+  endLine: number
+}
+
+export interface Variable {
+  type: 'Variable'
+  name: string
+  kind: 'const' | 'let' | 'var'
+  startLine: number
+  endLine: number
+}
+
+export interface ClassMember {
+  type: 'Property' | 'Method'
+  name: string
+  kind: 'constructor' | 'method' | 'getter' | 'setter' | 'property'
+  visibility: 'public' | 'private' | 'protected'
+  isStatic: boolean
+  startLine: number
+  endLine: number
+}
+
+export interface Reference {
+  type: 'Reference'
+  name: string
+  line: number
+  column: number
+  context: string
+}
+
+export interface StructureOverview {
+  filePath: string
+  language: string
+  imports: { source: string; names: string[] }[]
+  exports: string[]
+  functions: { name: string | null; line: number }[]
+  classes: { name: string; line: number; methods: string[] }[]
+  variables: { name: string; line: number }[]
+}
+
+export interface AstEdit {
+  type: 'insert' | 'replace' | 'delete'
+  startOffset: number
+  endOffset?: number
+  text?: string
+}
+
+export interface TextEditResult {
+  success: boolean
+  source: string
+  editsApplied: number
+  error?: string
+}
+
+export type Statement = FuncDecl | Class | Variable | Import | Export
+
+export interface Export {
+  type: 'Export'
+  name: string
+}

package/src/tools/{find.ts → codebase/find.ts}

@@ -1,9 +1,11 @@
-import { fmtErr } from './errors.js'
-import type { Tool, ToolDefinition } from './types.js'
+import { fmtErr } from '../errors.js'
+import type { Tool, ToolDefinition } from '../types.js'
 
 export const createFindTool = (cwd: string): Tool => {
   const definition: ToolDefinition = {
     name: 'find',
+    category: 'Codebase',
+    group: 'Search/By Pattern',
     description: 'Find files by name pattern. Uses glob internally. Returns matching file paths.',
     parameters: {
       pattern: {

package/src/tools/{glob.ts → codebase/glob.ts}

@@ -1,10 +1,12 @@
 import { glob } from 'node:fs/promises'
-import { fmtErr } from './errors.js'
-import type { Tool, ToolResult } from './types.js'
+import { fmtErr } from '../errors.js'
+import type { Tool, ToolResult } from '../types.js'
 
 export const globTool: Tool = {
   definition: {
     name: 'glob',
+    category: 'Codebase',
+    group: 'Search/By Pattern',
     description: 'Find files by glob pattern (e.g. "src/**/*.ts"). Returns matching file paths.',
     parameters: {
       pattern: { type: 'string', description: 'Glob pattern to search for', required: true },