package tests_test

import (
	"os"
	"os/exec"
	"path/filepath"
	"testing"

	"git.akyoto.dev/cli/q/src/build"
	"git.akyoto.dev/go/assert"
)

var programs = []struct {
	Name             string
	ExpectedOutput   string
	ExpectedExitCode int
}{
	{"empty.q", "", 0},
	{"math.q", "", 10},
	{"precedence.q", "", 10},
	{"square-sum.q", "", 25},
	{"chained-calls.q", "", 9},
	{"nested-calls.q", "", 4},
	{"overwrite.q", "", 3},
	{"reuse.q", "", 3},
	{"return.q", "", 6},
	{"reassign.q", "", 2},
	{"branch.q", "", 0},
	{"branch-and.q", "", 0},
	{"branch-or.q", "", 0},
	{"branch-combined.q", "", 0},
}

func TestPrograms(t *testing.T) {
	for _, test := range programs {
		t.Run(test.Name, func(t *testing.T) {
			run(t, filepath.Join("programs", test.Name), test.ExpectedOutput, test.ExpectedExitCode)
		})
	}
}

func BenchmarkPrograms(b *testing.B) {
	for _, test := range programs {
		b.Run(test.Name, func(b *testing.B) {
			compiler := build.New(filepath.Join("programs", test.Name))

			for i := 0; i < b.N; i++ {
				_, err := compiler.Run()
				assert.Nil(b, err)
			}
		})
	}
}

// run builds and runs the file to check if the output matches the expected output.
func run(t *testing.T, name string, expectedOutput string, expectedExitCode int) {
	b := build.New(name)
	assert.True(t, len(b.Executable()) > 0)

	t.Run("Compile", func(t *testing.T) {
		result, err := b.Run()
		assert.Nil(t, err)

		err = result.Write(b.Executable())
		assert.Nil(t, err)

		stat, err := os.Stat(b.Executable())
		assert.Nil(t, err)
		assert.True(t, stat.Size() > 0)
	})

	t.Run("Output", func(t *testing.T) {
		cmd := exec.Command(b.Executable())
		output, err := cmd.Output()
		exitCode := 0

		if err != nil {
			exitError, ok := err.(*exec.ExitError)

			if !ok {
				t.Fatal(exitError)
			}

			exitCode = exitError.ExitCode()
		}

		assert.Equal(t, exitCode, expectedExitCode)
		assert.DeepEqual(t, string(output), expectedOutput)
	})
}