package compiler_test

import (
	"path/filepath"
	"slices"
	"strings"
	"testing"

	"efprojects.com/kitten-ipc/kitcom/internal/tsgo/bundled"
	"efprojects.com/kitten-ipc/kitcom/internal/tsgo/compiler"
	"efprojects.com/kitten-ipc/kitcom/internal/tsgo/core"
	"efprojects.com/kitten-ipc/kitcom/internal/tsgo/repo"
	"efprojects.com/kitten-ipc/kitcom/internal/tsgo/tsoptions"
	"efprojects.com/kitten-ipc/kitcom/internal/tsgo/tspath"
	"efprojects.com/kitten-ipc/kitcom/internal/tsgo/vfs/osvfs"
	"efprojects.com/kitten-ipc/kitcom/internal/tsgo/vfs/vfstest"
	"gotest.tools/v3/assert"
)

type testFile struct {
	fileName string
	contents string
}

type programTest struct {
	testName      string
	files         []testFile
	expectedFiles []string
	target        core.ScriptTarget
}

var esnextLibs = []string{
	"lib.es5.d.ts",
	"lib.es2015.d.ts",
	"lib.es2016.d.ts",
	"lib.es2017.d.ts",
	"lib.es2018.d.ts",
	"lib.es2019.d.ts",
	"lib.es2020.d.ts",
	"lib.es2021.d.ts",
	"lib.es2022.d.ts",
	"lib.es2023.d.ts",
	"lib.es2024.d.ts",
	"lib.esnext.d.ts",
	"lib.dom.d.ts",
	"lib.dom.iterable.d.ts",
	"lib.dom.asynciterable.d.ts",
	"lib.webworker.importscripts.d.ts",
	"lib.scripthost.d.ts",
	"lib.es2015.core.d.ts",
	"lib.es2015.collection.d.ts",
	"lib.es2015.generator.d.ts",
	"lib.es2015.iterable.d.ts",
	"lib.es2015.promise.d.ts",
	"lib.es2015.proxy.d.ts",
	"lib.es2015.reflect.d.ts",
	"lib.es2015.symbol.d.ts",
	"lib.es2015.symbol.wellknown.d.ts",
	"lib.es2016.array.include.d.ts",
	"lib.es2016.intl.d.ts",
	"lib.es2017.arraybuffer.d.ts",
	"lib.es2017.date.d.ts",
	"lib.es2017.object.d.ts",
	"lib.es2017.sharedmemory.d.ts",
	"lib.es2017.string.d.ts",
	"lib.es2017.intl.d.ts",
	"lib.es2017.typedarrays.d.ts",
	"lib.es2018.asyncgenerator.d.ts",
	"lib.es2018.asynciterable.d.ts",
	"lib.es2018.intl.d.ts",
	"lib.es2018.promise.d.ts",
	"lib.es2018.regexp.d.ts",
	"lib.es2019.array.d.ts",
	"lib.es2019.object.d.ts",
	"lib.es2019.string.d.ts",
	"lib.es2019.symbol.d.ts",
	"lib.es2019.intl.d.ts",
	"lib.es2020.bigint.d.ts",
	"lib.es2020.date.d.ts",
	"lib.es2020.promise.d.ts",
	"lib.es2020.sharedmemory.d.ts",
	"lib.es2020.string.d.ts",
	"lib.es2020.symbol.wellknown.d.ts",
	"lib.es2020.intl.d.ts",
	"lib.es2020.number.d.ts",
	"lib.es2021.promise.d.ts",
	"lib.es2021.string.d.ts",
	"lib.es2021.weakref.d.ts",
	"lib.es2021.intl.d.ts",
	"lib.es2022.array.d.ts",
	"lib.es2022.error.d.ts",
	"lib.es2022.intl.d.ts",
	"lib.es2022.object.d.ts",
	"lib.es2022.string.d.ts",
	"lib.es2022.regexp.d.ts",
	"lib.es2023.array.d.ts",
	"lib.es2023.collection.d.ts",
	"lib.es2023.intl.d.ts",
	"lib.es2024.arraybuffer.d.ts",
	"lib.es2024.collection.d.ts",
	"lib.es2024.object.d.ts",
	"lib.es2024.promise.d.ts",
	"lib.es2024.regexp.d.ts",
	"lib.es2024.sharedmemory.d.ts",
	"lib.es2024.string.d.ts",
	"lib.esnext.array.d.ts",
	"lib.esnext.collection.d.ts",
	"lib.esnext.intl.d.ts",
	"lib.esnext.disposable.d.ts",
	"lib.esnext.promise.d.ts",
	"lib.esnext.decorators.d.ts",
	"lib.esnext.iterator.d.ts",
	"lib.esnext.float16.d.ts",
	"lib.esnext.error.d.ts",
	"lib.esnext.sharedmemory.d.ts",
	"lib.decorators.d.ts",
	"lib.decorators.legacy.d.ts",
	"lib.esnext.full.d.ts",
}

var programTestCases = []programTest{
	{
		testName: "BasicFileOrdering",
		files: []testFile{
			{fileName: "c:/dev/src/index.ts", contents: "/// <reference path='c:/dev/src2/a/5.ts' />\n/// <reference path='c:/dev/src2/a/10.ts' />"},
			{fileName: "c:/dev/src2/a/5.ts", contents: "/// <reference path='4.ts' />"},
			{fileName: "c:/dev/src2/a/4.ts", contents: "/// <reference path='b/3.ts' />"},
			{fileName: "c:/dev/src2/a/b/3.ts", contents: "/// <reference path='2.ts' />"},
			{fileName: "c:/dev/src2/a/b/2.ts", contents: "/// <reference path='c/1.ts' />"},
			{fileName: "c:/dev/src2/a/b/c/1.ts", contents: "console.log('hello');"},
			{fileName: "c:/dev/src2/a/10.ts", contents: "/// <reference path='b/c/d/9.ts' />"},
			{fileName: "c:/dev/src2/a/b/c/d/9.ts", contents: "/// <reference path='e/8.ts' />"},
			{fileName: "c:/dev/src2/a/b/c/d/e/8.ts", contents: "/// <reference path='7.ts' />"},
			{fileName: "c:/dev/src2/a/b/c/d/e/7.ts", contents: "/// <reference path='f/6.ts' />"},
			{fileName: "c:/dev/src2/a/b/c/d/e/f/6.ts", contents: "console.log('world!');"},
		},
		expectedFiles: slices.Concat(esnextLibs,
			[]string{
				"c:/dev/src2/a/b/c/1.ts",
				"c:/dev/src2/a/b/2.ts",
				"c:/dev/src2/a/b/3.ts",
				"c:/dev/src2/a/4.ts",
				"c:/dev/src2/a/5.ts",
				"c:/dev/src2/a/b/c/d/e/f/6.ts",
				"c:/dev/src2/a/b/c/d/e/7.ts",
				"c:/dev/src2/a/b/c/d/e/8.ts",
				"c:/dev/src2/a/b/c/d/9.ts",
				"c:/dev/src2/a/10.ts",
				"c:/dev/src/index.ts",
			}),
		target: core.ScriptTargetESNext,
	},
	{
		testName: "FileOrderingImports",
		files: []testFile{
			{fileName: "c:/dev/src/index.ts", contents: "import * as five from '../src2/a/5.ts';\nimport * as ten from '../src2/a/10.ts';"},
			{fileName: "c:/dev/src2/a/5.ts", contents: "import * as four from './4.ts';"},
			{fileName: "c:/dev/src2/a/4.ts", contents: "import * as three from './b/3.ts';"},
			{fileName: "c:/dev/src2/a/b/3.ts", contents: "import * as two from './2.ts';"},
			{fileName: "c:/dev/src2/a/b/2.ts", contents: "import * as one from './c/1.ts';"},
			{fileName: "c:/dev/src2/a/b/c/1.ts", contents: "console.log('hello');"},
			{fileName: "c:/dev/src2/a/10.ts", contents: "import * as nine from './b/c/d/9.ts';"},
			{fileName: "c:/dev/src2/a/b/c/d/9.ts", contents: "import * as eight from './e/8.ts';"},
			{fileName: "c:/dev/src2/a/b/c/d/e/8.ts", contents: "import * as seven from './7.ts';"},
			{fileName: "c:/dev/src2/a/b/c/d/e/7.ts", contents: "import * as six from './f/6.ts';"},
			{fileName: "c:/dev/src2/a/b/c/d/e/f/6.ts", contents: "console.log('world!');"},
		},
		expectedFiles: slices.Concat(esnextLibs,
			[]string{
				"c:/dev/src2/a/b/c/1.ts",
				"c:/dev/src2/a/b/2.ts",
				"c:/dev/src2/a/b/3.ts",
				"c:/dev/src2/a/4.ts",
				"c:/dev/src2/a/5.ts",
				"c:/dev/src2/a/b/c/d/e/f/6.ts",
				"c:/dev/src2/a/b/c/d/e/7.ts",
				"c:/dev/src2/a/b/c/d/e/8.ts",
				"c:/dev/src2/a/b/c/d/9.ts",
				"c:/dev/src2/a/10.ts",
				"c:/dev/src/index.ts",
			}),
		target: core.ScriptTargetESNext,
	},
	{
		testName: "FileOrderingCycles",
		files: []testFile{
			{fileName: "c:/dev/src/index.ts", contents: "import * as five from '../src2/a/5.ts';\nimport * as ten from '../src2/a/10.ts';"},
			{fileName: "c:/dev/src2/a/5.ts", contents: "import * as four from './4.ts';"},
			{fileName: "c:/dev/src2/a/4.ts", contents: "import * as three from './b/3.ts';"},
			{fileName: "c:/dev/src2/a/b/3.ts", contents: "import * as two from './2.ts';\nimport * as cycle from 'c:/dev/src/index.ts'; "},
			{fileName: "c:/dev/src2/a/b/2.ts", contents: "import * as one from './c/1.ts';"},
			{fileName: "c:/dev/src2/a/b/c/1.ts", contents: "console.log('hello');"},
			{fileName: "c:/dev/src2/a/10.ts", contents: "import * as nine from './b/c/d/9.ts';"},
			{fileName: "c:/dev/src2/a/b/c/d/9.ts", contents: "import * as eight from './e/8.ts';\nimport * as cycle from 'c:/dev/src/index.ts';"},
			{fileName: "c:/dev/src2/a/b/c/d/e/8.ts", contents: "import * as seven from './7.ts';"},
			{fileName: "c:/dev/src2/a/b/c/d/e/7.ts", contents: "import * as six from './f/6.ts';"},
			{fileName: "c:/dev/src2/a/b/c/d/e/f/6.ts", contents: "console.log('world!');"},
		},
		expectedFiles: slices.Concat(esnextLibs,
			[]string{
				"c:/dev/src2/a/b/c/1.ts",
				"c:/dev/src2/a/b/2.ts",
				"c:/dev/src2/a/b/3.ts",
				"c:/dev/src2/a/4.ts",
				"c:/dev/src2/a/5.ts",
				"c:/dev/src2/a/b/c/d/e/f/6.ts",
				"c:/dev/src2/a/b/c/d/e/7.ts",
				"c:/dev/src2/a/b/c/d/e/8.ts",
				"c:/dev/src2/a/b/c/d/9.ts",
				"c:/dev/src2/a/10.ts",
				"c:/dev/src/index.ts",
			}),
		target: core.ScriptTargetESNext,
	},
}

func TestProgram(t *testing.T) {
	t.Parallel()

	if !bundled.Embedded {
		// Without embedding, we'd need to read all of the lib files out from disk into the MapFS.
		// Just skip this for now.
		t.Skip("bundled files are not embedded")
	}

	for _, testCase := range programTestCases {
		t.Run(testCase.testName, func(t *testing.T) {
			t.Parallel()
			libPrefix := bundled.LibPath() + "/"
			fs := vfstest.FromMap[any](nil, false /*useCaseSensitiveFileNames*/)
			fs = bundled.WrapFS(fs)

			for _, testFile := range testCase.files {
				_ = fs.WriteFile(testFile.fileName, testFile.contents, false)
			}

			opts := core.CompilerOptions{Target: testCase.target}

			program := compiler.NewProgram(compiler.ProgramOptions{
				Config: &tsoptions.ParsedCommandLine{
					ParsedConfig: &core.ParsedOptions{
						FileNames:       []string{"c:/dev/src/index.ts"},
						CompilerOptions: &opts,
					},
				},
				Host: compiler.NewCompilerHost("c:/dev/src", fs, bundled.LibPath(), nil, nil),
			})

			actualFiles := []string{}
			for _, file := range program.GetSourceFiles() {
				actualFiles = append(actualFiles, strings.TrimPrefix(file.FileName(), libPrefix))
			}

			assert.DeepEqual(t, testCase.expectedFiles, actualFiles)
		})
	}
}

func BenchmarkNewProgram(b *testing.B) {
	if !bundled.Embedded {
		// Without embedding, we'd need to read all of the lib files out from disk into the MapFS.
		// Just skip this for now.
		b.Skip("bundled files are not embedded")
	}

	for _, testCase := range programTestCases {
		b.Run(testCase.testName, func(b *testing.B) {
			fs := vfstest.FromMap[any](nil, false /*useCaseSensitiveFileNames*/)
			fs = bundled.WrapFS(fs)

			for _, testFile := range testCase.files {
				_ = fs.WriteFile(testFile.fileName, testFile.contents, false)
			}

			opts := core.CompilerOptions{Target: testCase.target}
			programOpts := compiler.ProgramOptions{
				Config: &tsoptions.ParsedCommandLine{
					ParsedConfig: &core.ParsedOptions{
						FileNames:       []string{"c:/dev/src/index.ts"},
						CompilerOptions: &opts,
					},
				},
				Host: compiler.NewCompilerHost("c:/dev/src", fs, bundled.LibPath(), nil, nil),
			}

			for b.Loop() {
				compiler.NewProgram(programOpts)
			}
		})
	}

	b.Run("compiler", func(b *testing.B) {
		repo.SkipIfNoTypeScriptSubmodule(b)

		rootPath := tspath.NormalizeSlashes(filepath.Join(repo.TypeScriptSubmodulePath, "src", "compiler"))

		fs := osvfs.FS()
		fs = bundled.WrapFS(fs)

		host := compiler.NewCompilerHost(rootPath, fs, bundled.LibPath(), nil, nil)

		parsed, errors := tsoptions.GetParsedCommandLineOfConfigFile(tspath.CombinePaths(rootPath, "tsconfig.json"), nil, host, nil)
		assert.Equal(b, len(errors), 0, "Expected no errors in parsed command line")

		opts := compiler.ProgramOptions{
			Config: parsed,
			Host:   host,
		}

		for b.Loop() {
			compiler.NewProgram(opts)
		}
	})
}