TypescriptAPIUtils: add findNodesByType function

Author: davidkudera

packages/typescript-api-utils/sourceFiles/findNodesByType.ts

@@ -0,0 +1,22 @@
+import * as ts from 'typescript';
+
+
+export function findNodesByType<T extends ts.Node>(kind: ts.SyntaxKind, sourceFile: ts.SourceFile): Array<T>
+{
+	const nodes: Array<T> = [];
+
+	function findLoop(parent: ts.Node): void
+	{
+		ts.forEachChild(parent, (child: ts.Node) => {
+			if (child.kind === kind) {
+				nodes.push(<T>child);
+			}
+
+			findLoop(child);
+		});
+	}
+
+	findLoop(sourceFile);
+
+	return nodes;
+}

packages/typescript-api-utils/sourceFiles/index.ts

@@ -1,3 +1,4 @@
 export * from './findNode';
 export * from './findLastNodeOfType';
+export * from './findNodesByType';
 export * from './lookupSourceFile';

packages/typescript-api-utils/tests/tests/sourceFiles/findNodesByType.ts

@@ -0,0 +1,27 @@
+import {findNodesByType} from '../../../';
+import {expect} from 'chai';
+import * as ts from 'typescript';
+
+
+describe('#sourceFiles/findNodesByType', () => {
+
+	describe('findNodesByType()', () => {
+
+		it('should return all nodes by type', () => {
+			const sourceFile = <ts.SourceFile>ts.createSourceFile(
+				'/main.ts',
+				'const a = 1; const b = 2;',
+				ts.ScriptTarget.Latest
+			);
+
+			const nodes = findNodesByType<ts.Identifier>(ts.SyntaxKind.Identifier, sourceFile);
+
+			console.log(nodes);
+			expect(nodes.length).to.be.equal(2);
+			expect(nodes[0].text).to.be.equal('a');
+			expect(nodes[1].text).to.be.equal('b');
+		});
+
+	});
+
+});