const Parser = require("web-tree-sitter");
import { SyntaxNode } from "web-tree-sitter";
import 'reflect-metadata';

import { LanguageProfileUtil } from "../../../code-context/_base/LanguageProfile";
import { TestLanguageService } from "../../TestLanguageService";

describe('TypeScriptLangConfig', () => {
	let parser: any;
	let grammar: any;
	let langConfig = LanguageProfileUtil.from("typescript")!!;

	beforeEach(async () => {
		await Parser.init();
		parser = new Parser();
		const languageService = new TestLanguageService(parser);
		grammar = await langConfig.grammar(languageService, "typescript")!!;
		parser.setLanguage(grammar);
	});

	it('should identify methods of TypeScript', async () => {
		const sampleCode = `function hello() {
  console.log("hello, world");
}

class Greeter {
  greeting: string;
 
  constructor(message: string) {
    this.greeting = message;
  }
 
  greet() {
    return "Hello, " + this.greeting;
  }
}
`;

		const query = langConfig.methodQuery.query(grammar);
		const root = parser.parse(sampleCode).rootNode;
		const matches = query!!.matches(root);

		expect(matches).toBeDefined();
		expect(matches.length).toBe(3);

		const bodyNodes: SyntaxNode[] = matches.map((match: any) => match.captures[0].node);
		expect(bodyNodes[0].text).toBe("function hello() {\n  console.log(\"hello, world\");\n}");
		expect(bodyNodes[1].text).toBe("constructor(message: string) {\n    this.greeting = message;\n  }");
		expect(bodyNodes[2].text).toBe("greet() {\n    return \"Hello, \" + this.greeting;\n  }");

		const methodNameNodes: SyntaxNode[] = matches.map((match: any) => match.captures[1].node);
		expect(methodNameNodes[0].text).toBe("hello");
		expect(methodNameNodes[1].text).toBe("constructor");
		expect(methodNameNodes[2].text).toBe("greet");
	});
});