package test

import (
	"context"
	"os"
	"strings"
	"testing"
	"time"

	"github.com/authorizerdev/authorizer/server/constants"
	"github.com/authorizerdev/authorizer/server/db"
	"github.com/authorizerdev/authorizer/server/db/models"
	"github.com/authorizerdev/authorizer/server/env"
	"github.com/authorizerdev/authorizer/server/memorystore"
	"github.com/authorizerdev/authorizer/server/utils"
)

func TestResolvers(t *testing.T) {
	databases := map[string]string{
		constants.DbTypeSqlite:      "../../test.db",
		constants.DbTypeArangodb:    "http://localhost:8529",
		constants.DbTypeMongodb:     "mongodb://localhost:27017",
		constants.DbTypeScyllaDB:    "127.0.0.1:9042",
		constants.DbTypeDynamoDB:    "http://0.0.0.0:8000",
		constants.DbTypeCouchbaseDB: "couchbase://127.0.0.1",
	}

	testDBs := strings.Split(os.Getenv("TEST_DBS"), ",")
	t.Log("Running tests for following dbs: ", testDBs)
	for dbType := range databases {
		if !utils.StringSliceContains(testDBs, dbType) {
			delete(databases, dbType)
		}
	}

	if utils.StringSliceContains(testDBs, constants.DbTypeSqlite) && len(testDBs) == 1 {
		// do nothing
	} else {
		t.Log("waiting for docker containers to start...")
		// wait for docker containers to spun up
		time.Sleep(30 * time.Second)
	}

	testDb := "authorizer_test"
	s := testSetup()
	defer s.Server.Close()

	for dbType, dbURL := range databases {
		ctx := context.Background()
		memorystore.Provider.UpdateEnvVariable(constants.EnvKeyDatabaseURL, dbURL)
		memorystore.Provider.UpdateEnvVariable(constants.EnvKeyDatabaseType, dbType)
		memorystore.Provider.UpdateEnvVariable(constants.EnvKeyDatabaseName, testDb)
		os.Setenv(constants.EnvKeyDatabaseURL, dbURL)
		os.Setenv(constants.EnvKeyDatabaseType, dbType)
		os.Setenv(constants.EnvKeyDatabaseName, testDb)

		if dbType == constants.DbTypeDynamoDB {
			memorystore.Provider.UpdateEnvVariable(constants.EnvAwsRegion, "ap-south-1")
			os.Setenv(constants.EnvAwsRegion, "ap-south-1")
			os.Unsetenv(constants.EnvAwsAccessKeyID)
			os.Unsetenv(constants.EnvAwsSecretAccessKey)
			// Remove aws credentials from env, so that local dynamodb can be used
			memorystore.Provider.UpdateEnvVariable(constants.EnvAwsAccessKeyID, "")
			memorystore.Provider.UpdateEnvVariable(constants.EnvAwsSecretAccessKey, "")
		}
		if dbType == constants.DbTypeCouchbaseDB {
			memorystore.Provider.UpdateEnvVariable(constants.EnvKeyDatabaseUsername, "Administrator")
			os.Setenv(constants.EnvKeyDatabaseUsername, "Administrator")
			memorystore.Provider.UpdateEnvVariable(constants.EnvKeyDatabasePassword, "password")
			os.Setenv(constants.EnvKeyDatabasePassword, "password")
		}

		memorystore.InitRequiredEnv()

		err := db.InitDB()
		if err != nil {
			t.Logf("Error initializing database: %s", err.Error())
		}

		// clean the persisted config for test to use fresh config
		envData, err := db.Provider.GetEnv(ctx)
		if err == nil && envData == nil {
			envData = &models.Env{
				EnvData: "",
			}
			_, err = db.Provider.UpdateEnv(ctx, envData)
			if err != nil {
				t.Logf("Error updating env: %s", err.Error())
			}
		} else if err != nil {
			t.Logf("Error getting env: %s", err.Error())
		}
		err = env.PersistEnv()
		if err != nil {
			t.Logf("Error persisting env: %s", err.Error())
		}

		memorystore.Provider.UpdateEnvVariable(constants.EnvKeyEnv, "test")
		memorystore.Provider.UpdateEnvVariable(constants.EnvKeyIsProd, false)
		t.Run("should pass tests for "+dbType, func(t *testing.T) {
			// admin resolvers tests
			adminSignupTests(t, s)
			addWebhookTest(t, s) // add webhooks for all the system events
			testEndpointTest(t, s)
			verificationRequestsTest(t, s)
			updateWebhookTest(t, s)
			webhookTest(t, s)
			webhooksTest(t, s)
			//usersTest(t, s)
			userTest(t, s)
			deleteUserTest(t, s)
			//updateUserTest(t, s)
			adminLoginTests(t, s)
			adminLogoutTests(t, s)
			adminSessionTests(t, s)
			updateEnvTests(t, s)
			envTests(t, s)
			revokeAccessTest(t, s)
			enableAccessTest(t, s)
			generateJWTkeyTest(t, s)
			addEmailTemplateTest(t, s)
			updateEmailTemplateTest(t, s)
			emailTemplatesTest(t, s)
			deleteEmailTemplateTest(t, s)

			// user resolvers tests
			loginTests(t, s)
			signupTests(t, s)
			mobileSingupTest(t, s)
			mobileLoginTests(t, s)
			totpLoginTest(t, s)
			forgotPasswordTest(t, s)
			resendVerifyEmailTests(t, s)
			resetPasswordTest(t, s)
			verifyEmailTest(t, s)
			sessionTests(t, s)
			profileTests(t, s)
			updateProfileTests(t, s)
			magicLinkLoginTests(t, s)
			logoutTests(t, s)
			metaTests(t, s)
			inviteUserTest(t, s)
			validateJwtTokenTest(t, s)
			verifyOTPTest(t, s)
			resendOTPTest(t, s)
			validateSessionTests(t, s)
			deactivateAccountTests(t, s)

			updateAllUsersTest(t, s)
			webhookLogsTest(t, s)   // get logs after above resolver tests are done
			deleteWebhookTest(t, s) // delete webhooks (admin resolver)
		})
	}
}