package acctest

import (
	"context"
	"fmt"
	"github.com/filipowm/go-unifi/unifi"
	"github.com/filipowm/terraform-provider-unifi/internal/provider"
	pt "github.com/filipowm/terraform-provider-unifi/internal/provider/testing"
	"github.com/hashicorp/go-version"
	"github.com/hashicorp/terraform-plugin-framework/providerserver"
	"github.com/hashicorp/terraform-plugin-go/tfprotov5"
	"github.com/hashicorp/terraform-plugin-go/tfprotov6"
	"github.com/hashicorp/terraform-plugin-mux/tf5to6server"
	"github.com/hashicorp/terraform-plugin-mux/tf6muxserver"
	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
	"github.com/hashicorp/terraform-plugin-testing/helper/resource"
	"os"
	"testing"
)

type providersMap map[string]func() (tfprotov6.ProviderServer, error)

var (
	providers  = createProviders()
	testClient unifi.Client
)

type Steps []resource.TestStep

type AcceptanceTestCase struct {
	CheckDestroy      resource.TestCheckFunc
	VersionConstraint string
	MinVersion        *version.Version
	PreCheck          func()
	Steps             Steps
}

func AcceptanceTest(t *testing.T, testCase AcceptanceTestCase) {
	t.Helper()
	if len(testCase.Steps) == 0 {
		t.Fatal("missing test steps")
	}
	resource.ParallelTest(t, resource.TestCase{
		PreCheck: func() {
			pt.PreCheck(t)
			if testCase.VersionConstraint != "" {
				PreCheckVersionConstraint(t, testCase.VersionConstraint)
			}
			if testCase.MinVersion != nil {
				PreCheckMinVersion(t, testCase.MinVersion)
			}
			if testCase.PreCheck != nil {
				testCase.PreCheck()
			}
		},
		ProtoV6ProviderFactories: providers,
		CheckDestroy:             testCase.CheckDestroy,
		Steps:                    testCase.Steps,
	})
}

func TestMain(m *testing.M) {
	providers = createProviders()
	os.Exit(pt.Run(m, func(env *pt.TestEnvironment) {
		testClient = env.Client
	}))
}

func createProviders() providersMap {
	ctx := context.Background()
	// Init mux servers
	return map[string]func() (tfprotov6.ProviderServer, error){
		"unifi": func() (tfprotov6.ProviderServer, error) {
			return tf6muxserver.NewMuxServer(ctx,
				providerserver.NewProtocol6(provider.NewV2("acctestv2")()),
				func() tfprotov6.ProviderServer {
					sdkV2Provider, err := tf5to6server.UpgradeServer(
						ctx,
						func() tfprotov5.ProviderServer {
							return schema.NewGRPCProviderServer(
								provider.New("acctestv1")(),
							)
						},
					)
					if err != nil {
						panic(fmt.Errorf("failed to create test providers: %w", err))
					}

					return sdkV2Provider
				},
			)
		},
	}
}