import { expect } from "vitest";

const Parser = require("web-tree-sitter");

import { ScopeBuilder } from "../../../code-search/scope-graph/ScopeBuilder";
import { TestLanguageService } from "../../TestLanguageService";
import { testScopes } from "../../ScopeTestUtil";
import { TSLanguageService } from "../../../editor/language/service/TSLanguageService";
import { TreeSitterFile } from "../../../code-context/ast/TreeSitterFile";


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

describe('ScopeBuilderGo', () => {
	let parser: any;
	let language: any;
	let langConfig = LanguageProfileUtil.from("go")!!;
	let languageService: TSLanguageService;

	beforeEach(async () => {
		await Parser.init();
		parser = new Parser();
		languageService = new TestLanguageService(parser);

		language = await langConfig.grammar(languageService, "go")!!;
		parser.setLanguage(language);
		parser.setLogger(null);
	});

	it('build for node', async () => {
		const sourceCode = `
package main

const one uint64 = 1
const (
    two = 2
    three = 2
)

func four() {}

var five = 3

func six() {
    seven: for ;; {}
}

type eight struct {
    nine string
    ten uint64 
}

type eleven interface {}
`;

		parser.setTimeoutMicros(10 ** 6);
		let tree = parser.parse(sourceCode);
		const rootNode = tree.rootNode;
		const query = langConfig.scopeQuery.query(language);
		let scopeBuilder = new ScopeBuilder(query!!, rootNode, sourceCode, langConfig);
		let output = await scopeBuilder.build();

		const hoverRanges = output.hoverableRanges();
		expect(hoverRanges.length).toBe(11);

		const allText = hoverRanges.map((range) => range.getText()).join(", ");
		expect(allText).toBe("one, two, three, five, four, six, eight, eleven, nine, ten, seven");
	});

	it('test for main scopes', async () => {
		const sourceCode = `
func main() {
    var args = os.Args;
    var length = len(args);
    fmt.Printf("%d", l);
}
`;
		let tree = parser.parse(sourceCode);
		const tsf = new TreeSitterFile(sourceCode, tree, langConfig, parser, language, "");
		await testScopes("go", sourceCode, "", tsf);
	});
});