// +build integration

package integration

import (
	"context"
	"io"
	"io/ioutil"
	"os"
	"strings"
	"testing"

	"github.com/docker/docker/api/types"
	"github.com/docker/docker/client"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"

	"github.com/aquasecurity/trivy/pkg/commands"
)

func TestRun_WithDockerEngine(t *testing.T) {
	testCases := []struct {
		name                string
		withImageSubcommand bool
		imageTag            string
		invalidImage        bool
		ignoreUnfixed       bool
		severity            []string
		ignoreIDs           []string
		testfile            string
		expectedOutputFile  string
		expectedError       string
	}{
		// All of these cases should pass for either
		// $ trivy <args>
		// $ trivy image <args>
		{
			name:               "happy path, valid image path, alpine:3.10",
			imageTag:           "alpine:3.10",
			expectedOutputFile: "testdata/alpine-310.json.golden",
			testfile:           "testdata/fixtures/alpine-310.tar.gz",
		},
		{
			name:                "happy path, valid image path, with image subcommand, alpine:3.10",
			withImageSubcommand: true,
			imageTag:            "alpine:3.10",
			expectedOutputFile:  "testdata/alpine-310.json.golden",
			testfile:            "testdata/fixtures/alpine-310.tar.gz",
		},
		{
			name:               "happy path, valid image path, alpine:3.10, ignore unfixed",
			ignoreUnfixed:      true,
			imageTag:           "alpine:3.10",
			expectedOutputFile: "testdata/alpine-310-ignore-unfixed.json.golden",
			testfile:           "testdata/fixtures/alpine-310.tar.gz",
		},
		{
			name:               "happy path, valid image path, alpine:3.10, ignore unfixed, with medium and high severity",
			ignoreUnfixed:      true,
			severity:           []string{"MEDIUM", "HIGH"},
			imageTag:           "alpine:3.10",
			expectedOutputFile: "testdata/alpine-310-medium-high.json.golden",
			testfile:           "testdata/fixtures/alpine-310.tar.gz",
		},
		{
			name:               "happy path, valid image path, alpine:3.10, with .trivyignore",
			imageTag:           "alpine:3.10",
			ignoreIDs:          []string{"CVE-2019-1549", "CVE-2019-1563"},
			expectedOutputFile: "testdata/alpine-310-ignore-cveids.json.golden",
			testfile:           "testdata/fixtures/alpine-310.tar.gz",
		},
		{
			name:               "happy path, valid image path, alpine:3.9",
			imageTag:           "alpine:3.9",
			expectedOutputFile: "testdata/alpine-39.json.golden",
			testfile:           "testdata/fixtures/alpine-39.tar.gz",
		},
		{
			name:               "happy path, valid image path, amazonlinux:1",
			imageTag:           "amazonlinux:1",
			expectedOutputFile: "testdata/amazon-1.json.golden",
			testfile:           "testdata/fixtures/amazon-1.tar.gz",
		},
		{
			name:               "happy path, valid image path, amazonlinux:2",
			imageTag:           "amazonlinux:2",
			expectedOutputFile: "testdata/amazon-2.json.golden",
			testfile:           "testdata/fixtures/amazon-2.tar.gz",
		},
		{
			name:               "happy path, valid image path, centos:6",
			imageTag:           "centos:6",
			expectedOutputFile: "testdata/centos-6.json.golden",
			testfile:           "testdata/fixtures/centos-6.tar.gz",
		},
		{
			name:               "happy path, valid image path, centos:7",
			imageTag:           "centos:7",
			expectedOutputFile: "testdata/centos-7.json.golden",
			testfile:           "testdata/fixtures/centos-7.tar.gz",
		},
		{
			name:               "happy path, valid image path, centos:7, with --ignore-unfixed option",
			imageTag:           "centos:7",
			ignoreUnfixed:      true,
			expectedOutputFile: "testdata/centos-7-ignore-unfixed.json.golden",
			testfile:           "testdata/fixtures/centos-7.tar.gz",
		},
		{
			name:               "happy path, valid image path, centos:7, with --ignore-unfixed option, with low and high severity",
			imageTag:           "centos:7",
			ignoreUnfixed:      true,
			severity:           []string{"LOW", "HIGH"},
			expectedOutputFile: "testdata/centos-7-low-high.json.golden",
			testfile:           "testdata/fixtures/centos-7.tar.gz",
		},
		{
			name:               "happy path, valid image path, debian:buster",
			imageTag:           "debian:buster",
			expectedOutputFile: "testdata/debian-buster.json.golden",
			testfile:           "testdata/fixtures/debian-buster.tar.gz",
		},
		{
			name:               "happy path, valid image path, debian:buster, with --ignore-unfixed option",
			ignoreUnfixed:      true,
			imageTag:           "debian:buster",
			expectedOutputFile: "testdata/debian-buster-ignore-unfixed.json.golden",
			testfile:           "testdata/fixtures/debian-buster.tar.gz",
		},
		{
			name:               "happy path, valid image path, debian:stretch",
			imageTag:           "debian:stretch",
			expectedOutputFile: "testdata/debian-stretch.json.golden",
			testfile:           "testdata/fixtures/debian-stretch.tar.gz",
		},
		{
			name:               "happy path, valid image path, distroless:base",
			imageTag:           "gcr.io/distroless/base:latest",
			expectedOutputFile: "testdata/distroless-base.json.golden",
			testfile:           "testdata/fixtures/distroless-base.tar.gz",
		},
		{
			name:               "happy path, valid image path, distroless:base",
			imageTag:           "gcr.io/distroless/base:latest",
			expectedOutputFile: "testdata/distroless-base.json.golden",
			testfile:           "testdata/fixtures/distroless-base.tar.gz",
		},
		{
			name:               "happy path, valid image path, distroless:base, with --ignore-unfixed option",
			imageTag:           "gcr.io/distroless/base:latest",
			ignoreUnfixed:      true,
			expectedOutputFile: "testdata/distroless-base-ignore-unfixed.json.golden",
			testfile:           "testdata/fixtures/distroless-base.tar.gz",
		},
		{
			name:               "happy path, valid image path, distroless:python2.7",
			imageTag:           "gcr.io/distroless/python2.7:latest",
			expectedOutputFile: "testdata/distroless-python27.json.golden",
			testfile:           "testdata/fixtures/distroless-python27.tar.gz",
		},
		{
			name:               "happy path, valid image path, oraclelinux:6-slim",
			imageTag:           "oraclelinux:6-slim",
			expectedOutputFile: "testdata/oraclelinux-6-slim.json.golden",
			testfile:           "testdata/fixtures/oraclelinux-6-slim.tar.gz",
		},
		{
			name:               "happy path, valid image path, oraclelinux:7-slim",
			imageTag:           "oraclelinux:7-slim",
			expectedOutputFile: "testdata/oraclelinux-7-slim.json.golden",
			testfile:           "testdata/fixtures/oraclelinux-7-slim.tar.gz",
		},
		{
			name:               "happy path, valid image path, oraclelinux:8-slim",
			imageTag:           "oraclelinux:8-slim",
			expectedOutputFile: "testdata/oraclelinux-8-slim.json.golden",
			testfile:           "testdata/fixtures/oraclelinux-8-slim.tar.gz",
		},
		{
			name:               "happy path, valid image path, ubuntu:16.04",
			imageTag:           "ubuntu:16.04",
			expectedOutputFile: "testdata/ubuntu-1604.json.golden",
			testfile:           "testdata/fixtures/ubuntu-1604.tar.gz",
		},
		{
			name:               "happy path, valid image path, ubuntu:18.04",
			imageTag:           "ubuntu:18.04",
			expectedOutputFile: "testdata/ubuntu-1804.json.golden",
			testfile:           "testdata/fixtures/ubuntu-1804.tar.gz",
		},
		{
			name:               "happy path, valid image path, ubuntu:18.04, with --ignore-unfixed option",
			imageTag:           "ubuntu:18.04",
			ignoreUnfixed:      true,
			expectedOutputFile: "testdata/ubuntu-1804-ignore-unfixed.json.golden",
			testfile:           "testdata/fixtures/ubuntu-1804.tar.gz",
		},
		{
			name:               "happy path, valid image path, registry.redhat.io/ubi7",
			imageTag:           "registry.redhat.io/ubi7",
			expectedOutputFile: "testdata/ubi-7.json.golden",
			testfile:           "testdata/fixtures/ubi-7.tar.gz",
		},
		{
			name:               "happy path, valid image path, opensuse leap 15.1",
			imageTag:           "opensuse/leap:latest",
			expectedOutputFile: "testdata/opensuse-leap-151.json.golden",
			testfile:           "testdata/fixtures/opensuse-leap-151.tar.gz",
		},
		{
			name:               "happy path, valid image path, opensuse leap 42.3",
			imageTag:           "opensuse/leap:42.3",
			expectedOutputFile: "testdata/opensuse-leap-423.json.golden",
			testfile:           "testdata/fixtures/opensuse-leap-423.tar.gz",
		},
		{
			name:               "happy path, valid image path, photon 1.0",
			imageTag:           "photon:1.0-20190823",
			expectedOutputFile: "testdata/photon-10.json.golden",
			testfile:           "testdata/fixtures/photon-10.tar.gz",
		},
		{
			name:               "happy path, valid image path, photon 2.0",
			imageTag:           "photon:2.0-20190726",
			expectedOutputFile: "testdata/photon-20.json.golden",
			testfile:           "testdata/fixtures/photon-20.tar.gz",
		},
		{
			name:               "happy path, valid image path, photon 3.0",
			imageTag:           "photon:3.0-20190823",
			expectedOutputFile: "testdata/photon-30.json.golden",
			testfile:           "testdata/fixtures/photon-30.tar.gz",
		},
		{
			name:               "buxybox with Cargo.lock integration",
			imageTag:           "busy-cargo:latest",
			expectedOutputFile: "testdata/busybox-with-lockfile.json.golden",
			testfile:           "testdata/fixtures/busybox-with-lockfile.tar.gz",
		},
		{
			name:          "sad path, invalid image",
			invalidImage:  true,
			testfile:      "badimage:latest",
			expectedError: "unable to inspect the image (index.docker.io/library/badimage:latest)",
		},
	}

	// Copy DB file
	cacheDir, err := gunzipDB()
	require.NoError(t, err)
	defer os.RemoveAll(cacheDir)

	ctx := context.Background()
	defer ctx.Done()

	cli, err := client.NewClientWithOpts(client.FromEnv, client.WithAPIVersionNegotiation())
	require.NoError(t, err)

	for _, tc := range testCases {
		t.Run(tc.name, func(t *testing.T) {
			if !tc.invalidImage {
				testfile, err := os.Open(tc.testfile)
				require.NoError(t, err, tc.name)

				// ensure image doesnt already exists
				_, _ = cli.ImageRemove(ctx, tc.testfile, types.ImageRemoveOptions{
					Force:         true,
					PruneChildren: true,
				})

				// load image into docker engine
				res, err := cli.ImageLoad(ctx, testfile, true)
				require.NoError(t, err, tc.name)
				io.Copy(ioutil.Discard, res.Body)

				// tag our image to something unique
				err = cli.ImageTag(ctx, tc.imageTag, tc.testfile)
				require.NoError(t, err, tc.name)
			}

			of, err := ioutil.TempFile("", "integration-docker-engine-output-file-*")
			require.NoError(t, err, tc.name)
			defer os.Remove(of.Name())

			// run trivy
			app := commands.NewApp("dev")
			trivyArgs := []string{"trivy"}
			trivyArgs = append(trivyArgs, "--cache-dir", cacheDir)
			if tc.withImageSubcommand {
				trivyArgs = append(trivyArgs, "image")
			}

			trivyArgs = append(trivyArgs, []string{"--skip-update", "--format=json", "--output", of.Name()}...)

			if tc.ignoreUnfixed {
				trivyArgs = append(trivyArgs, "--ignore-unfixed")
			}
			if len(tc.severity) != 0 {
				trivyArgs = append(trivyArgs,
					[]string{"--severity", strings.Join(tc.severity, ",")}...,
				)
			}
			if len(tc.ignoreIDs) != 0 {
				trivyIgnore := ".trivyignore"
				err := ioutil.WriteFile(trivyIgnore, []byte(strings.Join(tc.ignoreIDs, "\n")), 0444)
				assert.NoError(t, err, "failed to write .trivyignore")
				defer os.Remove(trivyIgnore)
			}
			trivyArgs = append(trivyArgs, tc.testfile)

			err = app.Run(trivyArgs)
			switch {
			case tc.expectedError != "":
				require.NotNil(t, err)
				assert.Contains(t, err.Error(), tc.expectedError, tc.name)
				return
			default:
				assert.NoError(t, err, tc.name)
			}

			// check for vulnerability output info
			got, err := ioutil.ReadAll(of)
			assert.NoError(t, err, tc.name)
			want, err := ioutil.ReadFile(tc.expectedOutputFile)
			assert.NoError(t, err, tc.name)
			assert.JSONEq(t, string(want), string(got), tc.name)

			// cleanup
			_, err = cli.ImageRemove(ctx, tc.testfile, types.ImageRemoveOptions{
				Force:         true,
				PruneChildren: true,
			})
			_, err = cli.ImageRemove(ctx, tc.imageTag, types.ImageRemoveOptions{
				Force:         true,
				PruneChildren: true,
			})
			assert.NoError(t, err, tc.name)
		})
	}
}
