package test

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

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"

	"github.com/aquasecurity/trivy/pkg/iac/scanners/helm/parser"
)

func Test_helm_parser_with_options_with_values_file(t *testing.T) {

	tests := []struct {
		testName   string
		chartName  string
		valuesFile string
	}{
		{
			testName:   "Parsing directory 'testchart'",
			chartName:  "testchart",
			valuesFile: "values/values.yaml",
		},
	}

	for _, test := range tests {
		t.Run(test.testName, func(t *testing.T) {
			chartName := test.chartName

			t.Logf("Running test: %s", test.testName)

			var opts []parser.Option

			if test.valuesFile != "" {
				opts = append(opts, parser.OptionWithValuesFile(test.valuesFile))
			}

			helmParser, err := parser.New(chartName, opts...)
			require.NoError(t, err)
			require.NoError(t, helmParser.ParseFS(t.Context(), os.DirFS(filepath.Join("testdata", chartName)), "."))
			manifests, err := helmParser.RenderedChartFiles()
			require.NoError(t, err)

			assert.Len(t, manifests, 3)

			for _, manifest := range manifests {
				expectedPath := filepath.Join("testdata", "expected", "options", chartName, manifest.TemplateFilePath)

				expectedContent, err := os.ReadFile(expectedPath)
				require.NoError(t, err)

				cleanExpected := strings.ReplaceAll(string(expectedContent), "\r\n", "\n")
				cleanActual := strings.ReplaceAll(manifest.ManifestContent, "\r\n", "\n")

				assert.Equal(t, cleanExpected, cleanActual)
			}
		})
	}
}

func Test_helm_parser_with_options_with_set_value(t *testing.T) {

	tests := []struct {
		testName   string
		chartName  string
		valuesFile string
		values     string
	}{
		{
			testName:  "Parsing directory 'testchart'",
			chartName: "testchart",
			values:    "securityContext.runAsUser=0",
		},
	}

	for _, test := range tests {
		t.Run(test.testName, func(t *testing.T) {
			chartName := test.chartName

			t.Logf("Running test: %s", test.testName)

			var opts []parser.Option

			if test.valuesFile != "" {
				opts = append(opts, parser.OptionWithValuesFile(test.valuesFile))
			}

			if test.values != "" {
				opts = append(opts, parser.OptionWithValues(test.values))
			}

			helmParser, err := parser.New(chartName, opts...)
			require.NoError(t, err)
			err = helmParser.ParseFS(t.Context(), os.DirFS(filepath.Join("testdata", chartName)), ".")
			require.NoError(t, err)
			manifests, err := helmParser.RenderedChartFiles()
			require.NoError(t, err)

			assert.Len(t, manifests, 3)

			for _, manifest := range manifests {
				expectedPath := filepath.Join("testdata", "expected", "options", chartName, manifest.TemplateFilePath)

				expectedContent, err := os.ReadFile(expectedPath)
				require.NoError(t, err)

				cleanExpected := strings.ReplaceAll(string(expectedContent), "\r\n", "\n")
				cleanActual := strings.ReplaceAll(manifest.ManifestContent, "\r\n", "\n")

				assert.Equal(t, cleanExpected, cleanActual)
			}
		})
	}
}

func Test_helm_parser_with_options_with_api_versions(t *testing.T) {

	tests := []struct {
		testName    string
		chartName   string
		apiVersions []string
	}{
		{
			testName:    "Parsing directory 'with-api-version'",
			chartName:   "with-api-version",
			apiVersions: []string{"policy/v1/PodDisruptionBudget"},
		},
	}

	for _, test := range tests {
		t.Run(test.testName, func(t *testing.T) {
			chartName := test.chartName

			t.Logf("Running test: %s", test.testName)

			var opts []parser.Option

			if len(test.apiVersions) > 0 {
				opts = append(opts, parser.OptionWithAPIVersions(test.apiVersions...))
			}

			helmParser, err := parser.New(chartName, opts...)
			require.NoError(t, err)
			err = helmParser.ParseFS(t.Context(), os.DirFS(filepath.Join("testdata", chartName)), ".")
			require.NoError(t, err)
			manifests, err := helmParser.RenderedChartFiles()
			require.NoError(t, err)

			assert.Len(t, manifests, 1)

			for _, manifest := range manifests {
				expectedPath := filepath.Join("testdata", "expected", "options", chartName, manifest.TemplateFilePath)

				expectedContent, err := os.ReadFile(expectedPath)
				require.NoError(t, err)

				cleanExpected := strings.TrimSpace(strings.ReplaceAll(string(expectedContent), "\r\n", "\n"))
				cleanActual := strings.TrimSpace(strings.ReplaceAll(manifest.ManifestContent, "\r\n", "\n"))

				assert.Equal(t, cleanExpected, cleanActual)
			}
		})
	}
}

func Test_helm_parser_with_options_with_kube_versions(t *testing.T) {

	tests := []struct {
		testName      string
		chartName     string
		kubeVersion   string
		expectedError string
	}{
		{
			testName:    "Parsing directory 'with-kube-version'",
			chartName:   "with-kube-version",
			kubeVersion: "1.60",
		},
		{
			testName:      "Parsing directory 'with-kube-version' with invalid kube version",
			chartName:     "with-kube-version",
			kubeVersion:   "a.b.c",
			expectedError: "could not parse \"a.b.c\" as version",
		},
	}

	for _, test := range tests {
		t.Run(test.testName, func(t *testing.T) {
			chartName := test.chartName

			t.Logf("Running test: %s", test.testName)

			var opts []parser.Option

			opts = append(opts, parser.OptionWithKubeVersion(test.kubeVersion))

			helmParser, err := parser.New(chartName, opts...)
			if test.expectedError != "" {
				require.EqualError(t, err, test.expectedError)
				return
			}
			require.NoError(t, err)
			require.NoError(t, helmParser.ParseFS(t.Context(), os.DirFS(filepath.Join("testdata", chartName)), "."))
			manifests, err := helmParser.RenderedChartFiles()
			require.NoError(t, err)

			assert.Len(t, manifests, 1)

			for _, manifest := range manifests {
				expectedPath := filepath.Join("testdata", "expected", "options", chartName, manifest.TemplateFilePath)

				expectedContent, err := os.ReadFile(expectedPath)
				require.NoError(t, err)

				cleanExpected := strings.TrimSpace(strings.ReplaceAll(string(expectedContent), "\r\n", "\n"))
				cleanActual := strings.TrimSpace(strings.ReplaceAll(manifest.ManifestContent, "\r\n", "\n"))

				assert.Equal(t, cleanExpected, cleanActual)
			}
		})
	}
}
