package stripe

import (
	"github.com/gin-gonic/gin"
	"github.com/google/uuid"
	"github.com/stripe/stripe-go/v72"
	"github.com/stripe/stripe-go/v72/checkout/session"
	"github.com/stripe/stripe-go/v72/paymentintent"
	"log"
	"payment-poc/database"
	"payment-poc/state"
)

type Service struct {
	Provider   *database.PaymentEntryProvider
	ApiKey     string
	BackendUrl string
}

func (s *Service) CreatePaymentUrl(amount int64) (url string, err error) {
	entry, err := s.Provider.CreateEntry(database.PaymentEntry{
		Gateway:     state.GatewayStripe,
		State:       state.StateInitialized,
		TotalAmount: amount,
	})
	if err != nil {
		return "", err
	}
	entry, url, err = s.InitializePayment(entry)
	if err != nil {
		return "", err
	}
	entry, err = s.Provider.UpdateEntry(entry)
	if err != nil {
		return "", err
	}
	return url, nil
}

func (s *Service) InitializePayment(entry database.PaymentEntry) (database.PaymentEntry, string, error) {

	currency := string(stripe.CurrencyEUR)
	productName := "Example product"
	productDescription := "Simple example product"

	params := &stripe.CheckoutSessionParams{
		LineItems: []*stripe.CheckoutSessionLineItemParams{
			{
				PriceData: &stripe.CheckoutSessionLineItemPriceDataParams{
					Currency: &currency,
					ProductData: &stripe.CheckoutSessionLineItemPriceDataProductDataParams{
						Name:        &productName,
						Description: &productDescription,
					},
					UnitAmount: &entry.TotalAmount,
				},
				Quantity: stripe.Int64(1),
			},
		},
		Mode: stripe.String(string(stripe.CheckoutSessionModePayment)),
		PaymentIntentData: &stripe.CheckoutSessionPaymentIntentDataParams{
			CaptureMethod: stripe.String("manual"),
		},
		SuccessURL: stripe.String(s.BackendUrl + "/stripe/success?token=" + entry.Id.String()),
		CancelURL:  stripe.String(s.BackendUrl + "/stripe/cancel?token=" + entry.Id.String()),
	}
	result, err := session.New(params)
	if err != nil {
		return database.PaymentEntry{}, "", err
	}
	entry.PaymentIntentId = &result.PaymentIntent.ID

	return entry, result.URL, nil
}

func (s *Service) CompleteTransaction(entry database.PaymentEntry, amount int64) (database.PaymentEntry, error) {
	params := &stripe.PaymentIntentCaptureParams{
		AmountToCapture: stripe.Int64(amount),
	}
	pi, err := paymentintent.Capture(*entry.PaymentIntentId, params)
	if err != nil {
		return database.PaymentEntry{}, err
	}
	log.Printf("received state on completion: %v", pi.Status)
	if pi.Status == stripe.PaymentIntentStatusSucceeded || pi.Status == stripe.PaymentIntentStatusProcessing {
		entry.Amount = &pi.AmountReceived
		entry.State = state.StateCompleted
	}
	return entry, nil
}

func (s *Service) CancelTransaction(entry database.PaymentEntry) (database.PaymentEntry, error) {
	params := &stripe.PaymentIntentCancelParams{}
	pi, err := paymentintent.Cancel(*entry.PaymentIntentId, params)
	if err != nil {
		return database.PaymentEntry{}, err
	}
	log.Printf("received state on completion: %v", pi.Status)
	if pi.Status == stripe.PaymentIntentStatusCanceled {
		entry.State = state.StateCanceled
	}
	return entry, nil
}

func (s *Service) HandleResponse(c *gin.Context, paymentState state.PaymentState) (string, error) {
	id := uuid.MustParse(c.Query("token"))
	entry, err := s.Provider.FetchById(id)
	if err != nil {
		return "", err
	}
	entry.State = paymentState
	s.Provider.UpdateEntry(entry)
	return "/entries/" + entry.Id.String(), nil
}