Skip to content

TypeScript AST query library with CLI

License

Notifications You must be signed in to change notification settings

aztack/tsquery

 
 

Repository files navigation

TSQuery

TSQuery is a port of the ESQuery API for TypeScript! TSQuery allows you to query a TypeScript AST for patterns of syntax using a CSS style selector system.

Installation

npm install @aztack/tsquery --save-dev
# or install tsquery cli to global
npm install -g @aztack/tsquery

Examples

Say we want to select all instances of an identifier with name "Animal", e.g. the identifier in the class declaration, and the identifier in the extends declaration.

We would do something like the following:

import { tsquery } from '@aztack/tsquery';

const typescript = `

class Animal {
    constructor(public name: string) { }
    move(distanceInMeters: number = 0) {
        console.log(\`\${this.name} moved \${distanceInMeters}m.\`);
    }
}

class Snake extends Animal {
    constructor(name: string) { super(name); }
    move(distanceInMeters = 5) {
        console.log("Slithering...");
        super.move(distanceInMeters);
    }
}

`

const ast = tsquery.ast(typescript);
const nodes = tsquery(ast, 'Identifier[name="Animal"]');
console.log(nodes.length); // 2

CLI

# Usage:
$ tsquery <glob-pattern> <ts-query-string> [inline-script | js-file-path]
// ./src/a.ts
class ClassA {}

// ./src/b.ts
class ClassB {}

// ./process.tsq.js
module.exprots = function ($, $filepath, $filename, $files, $tsq) {
    // matched: files matched with glob-pattern
    // index: current file index
    // $files = {matched, index}
    
    // ast: ast of current ts code
    // tsquery: the tsquery instance
    // code: code string read file current file
    // ts: the typescript library
    // $tsq = {ast, tsquery, code, ts}
    return $.name.escapedText;
}
$ tsquery "./src/*.ts" "ClassDeclaration" "return $.name.escapedText"
# or
$ tsquery "./src/*.ts" "ClassDeclaration" "./process.tsq.js"

will output

{
  query: "ClassDeclaration",
  result: [{
    file: './a.ts',
    basename: 'a',
    count: 1,
    result: [ 'ClassA' ]
  },{
    file: './b.ts',
    basename: 'b',
    count: 1,
    result: [ 'ClassB' ]
  }],
  count: 2
  script: "inline",
}

Selectors

The following selectors are supported:

Common AST node types

  • Identifier - any identifier (name of a function, class, variable, etc)
  • IfStatement, ForStatement, WhileStatement, DoStatement - control flow
  • FunctionDeclaration, ClassDeclaration, ArrowFunction - declarations
  • VariableStatement - var, const, let.
  • ImportDeclaration - any import statement
  • StringLiteral - any string
  • TrueKeyword, FalseKeyword, NullKeyword, AnyKeyword - various keywords
  • CallExpression - function call
  • NumericLiteral - any numeric constant
  • NoSubstitutionTemplateLiteral, TemplateExpression - template strings and expressions

About

TypeScript AST query library with CLI

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • JavaScript 99.1%
  • TypeScript 0.9%