package pkg

import (
	"context"
	"log"
	"strings"

	"github.com/google/uuid"

	ory "github.com/ory/kratos-client-go"
)

func RandomCredentials() (email, password string) {
	email = "dev+" + uuid.New().String() + "@ory.sh"
	password = strings.ReplaceAll(uuid.New().String(), "-", "")
	return
}

// CreateIdentityWithSession creates an identity and an Ory Session Token for it.
func CreateIdentityWithSession(c *ory.APIClient, email, password string) (*ory.Session, string) {
	ctx := context.Background()

	if email == "" {
		email, _ = RandomCredentials()
	}

	if password == "" {
		_, password = RandomCredentials()
	}

	// Initialize a registration flow
	flow, _, err := c.FrontendAPI.CreateNativeRegistrationFlow(ctx).Execute()

	result, res, err := c.FrontendAPI.UpdateRegistrationFlow(ctx).Flow(flow.Id).UpdateRegistrationFlowBody(
		ory.UpdateRegistrationFlowWithPasswordMethodAsUpdateRegistrationFlowBody(
			&ory.UpdateRegistrationFlowWithPasswordMethod{
				Method:   "password",
				Password: password,
				Traits: map[string]interface{}{
					"email":       email,
					"first_name":  "Jon",
					"last_name":   "Doe",
					"description": "nothing",
				},
			},
		),
	).Execute()
	SDKExitOnError(err, res)

	if result.Session == nil {
		log.Fatalf("The server is expected to create sessions for new registrations.")
	}

	return result.Session, *result.SessionToken
}

func CreateIdentity(c *ory.APIClient) *ory.Identity {
	ctx := context.Background()

	email, _ := RandomCredentials()

	identity, _, err := c.IdentityAPI.CreateIdentity(ctx).CreateIdentityBody(*ory.NewCreateIdentityBody("default", map[string]interface{}{"email": email,}})).Execute()
	ExitOnError(err)
	
	return identity
}