package mod

import (
	"testing"
)

func TestParseArgument(t *testing.T) {
	cases := []struct {
		name     string
		in       string
		want     *Require
		hasError bool
	}{
		{
			name:     "Random",
			in:       "abcd/bla@:/xyz",
			hasError: true,
		},
		{
			name: "Dagger repo",
			in:   "github.com/dagger/dagger",
			want: &Require{
				repo:    "github.com/dagger/dagger",
				path:    "",
				version: "",
			},
		},
		{
			name: "Dagger repo with path",
			in:   "github.com/dagger/dagger/stdlib",
			want: &Require{
				repo:    "github.com/dagger/dagger",
				path:    "/stdlib",
				version: "",
			},
		},
		{
			name: "Dagger repo with longer path",
			in:   "github.com/dagger/dagger/stdlib/test/test",
			want: &Require{
				repo:    "github.com/dagger/dagger",
				path:    "/stdlib/test/test",
				version: "",
			},
		},
		{
			name: "Dagger repo with path and version",
			in:   "github.com/dagger/dagger/stdlib@v0.1",
			want: &Require{
				repo:    "github.com/dagger/dagger",
				path:    "/stdlib",
				version: "v0.1",
			},
		},
		{
			name: "Dagger repo with longer path and version tag",
			in:   "github.com/dagger/dagger/stdlib/test/test@v0.0.1",
			want: &Require{
				repo:    "github.com/dagger/dagger",
				path:    "/stdlib/test/test",
				version: "v0.0.1",
			},
		},
		{
			name: "Alpha Dagger repo with path",
			in:   "alpha.dagger.io/gcp/gke@v0.1.0-alpha.20",
			want: &Require{
				repo:    "alpha.dagger.io",
				path:    "/gcp/gke",
				version: "v0.1.0-alpha.20",

				cloneRepo: "github.com/dagger/dagger",
				clonePath: "/stdlib/gcp/gke",
			},
		},
		{
			name: "Alpha Dagger repo",
			in:   "alpha.dagger.io@v0.1.0-alpha.23",
			want: &Require{
				repo:    "alpha.dagger.io",
				path:    "",
				version: "v0.1.0-alpha.23",

				cloneRepo: "github.com/dagger/dagger",
				clonePath: "/stdlib",
			},
		},
		{
			name: "Dagger repo with longer path and commit version",
			in:   "github.com/dagger/dagger/stdlib/test/test@26a1d46d1b3c",
			want: &Require{
				repo:    "github.com/dagger/dagger",
				path:    "/stdlib/test/test",
				version: "26a1d46d1b3c",
			},
		},
		{
			name: "Unspecified provider without folder",
			in:   "dagger.io/dagger/universe.git@main",
			want: &Require{
				repo:    "dagger.io/dagger/universe.git",
				path:    "",
				version: "main",
			},
		},
		{
			name: "Unspecified provider without folder",
			in:   "dagger.io/dagger/universe.git/stdlib/alpha.dagger.io/dagger@v0.1.0",
			want: &Require{
				repo:    "dagger.io/dagger/universe.git",
				path:    "/stdlib/alpha.dagger.io/dagger",
				version: "v0.1.0",
			},
		},
		{
			name: "Unspecified provider without folder",
			in:   "dagger.io/dagger/universe.git/stdlib@v5",
			want: &Require{
				repo:    "dagger.io/dagger/universe.git",
				path:    "/stdlib",
				version: "v5",
			},
		},
		{
			name: "Unspecified provider without folder",
			in:   "dagger.io/dagger/universe.git",
			want: &Require{
				repo:    "dagger.io/dagger/universe.git",
				path:    "",
				version: "",
			},
		},
		{
			name: "Unspecified provider without folder",
			in:   "dagger.io/dagger/universe.git/stdlib/alpha.dagger.io/dagger",
			want: &Require{
				repo:    "dagger.io/dagger/universe.git",
				path:    "/stdlib/alpha.dagger.io/dagger",
				version: "",
			},
		},
		// TODO: Add more tests for ports!
	}

	for _, c := range cases {
		t.Run(c.name, func(t *testing.T) {
			got, err := newRequire(c.in, "")
			if err != nil && c.hasError {
				return
			}

			if err != nil {
				t.Fatal(err)
			}

			if got.repo != c.want.repo {
				t.Errorf("repos differ: want %s, got %s", c.want.repo, got.repo)
			}

			if got.path != c.want.path {
				t.Errorf("paths differ: want %s, got %s", c.want.path, got.path)
			}

			if got.version != c.want.version {
				t.Errorf("versions differ: want %s, got %s", c.want.version, got.version)
			}
		})
	}
}