capture and split notification if it has multiple hosts

cmd/cart/pool-server.go

@@ -4,7 +4,6 @@ import (
 	"bytes"
 	"context"
 	"encoding/json"
-	"fmt"
 	"log"
 	"net/http"
 	"os"
@@ -15,10 +14,13 @@ import (
 	"git.k6n.net/go-cart-actor/pkg/actor"
 	"git.k6n.net/go-cart-actor/pkg/cart"
 	messages "git.k6n.net/go-cart-actor/pkg/messages"
+	"git.k6n.net/go-cart-actor/pkg/proxy"
 	"git.k6n.net/go-cart-actor/pkg/voucher"
-	"github.com/adyen/adyen-go-api-library/v14/src/adyen"
-	"github.com/adyen/adyen-go-api-library/v14/src/hmacvalidator"
-	"github.com/adyen/adyen-go-api-library/v14/src/webhook"
+	"github.com/adyen/adyen-go-api-library/v21/src/adyen"
+	"github.com/adyen/adyen-go-api-library/v21/src/checkout"
+	"github.com/adyen/adyen-go-api-library/v21/src/hmacvalidator"
+	"github.com/adyen/adyen-go-api-library/v21/src/webhook"
+	"github.com/google/uuid"
 	"github.com/matst80/go-redis-inventory/pkg/inventory"
 	"github.com/prometheus/client_golang/prometheus"
 	"github.com/prometheus/client_golang/prometheus/promauto"
@@ -375,15 +377,7 @@ func getClientIp(r *http.Request) string {
 }
 
 func (s *PoolServer) CreateOrUpdateCheckout(r *http.Request, id cart.CartId) (*CheckoutOrder, error) {
-	host := getOriginalHost(r)
-	country := getCountryFromHost(host)
-	meta := &CheckoutMeta{
-		ClientIp: getClientIp(r),
-		SiteUrl:  fmt.Sprintf("https://%s", host),
-		Country:  country,
-		Currency: getCurrency(country),
-		Locale:   getLocale(country),
-	}
+	meta := GetCheckoutMetaFromRequest(r)
 
 	// Get current grain state (may be local or remote)
 	grain, err := s.Get(r.Context(), uint64(id))
@@ -534,7 +528,7 @@ func (s *PoolServer) ProxyHandler(fn func(w http.ResponseWriter, r *http.Request
 			span.SetAttributes(hostAttr)
 			logger.InfoContext(ctx, "cart proxyed", "result", ownerHost.Name())
 			proxyCalls.Add(ctx, 1, metric.WithAttributes(hostAttr))
-			handled, err := ownerHost.Proxy(uint64(cartId), w, r)
+			handled, err := ownerHost.Proxy(uint64(cartId), w, r, nil)
 
 			grainLookups.Inc()
 			if err == nil && handled {
@@ -735,15 +729,8 @@ func (s *PoolServer) AdyenSessionHandler(w http.ResponseWriter, r *http.Request,
 	if err != nil {
 		return err
 	}
-	host := getOriginalHost(r)
-	country := getCountryFromHost(host)
-	meta := &CheckoutMeta{
-		ClientIp: getClientIp(r),
-		SiteUrl:  fmt.Sprintf("https://%s", host),
-		Country:  country,
-		Currency: getCurrency(country),
-		Locale:   getLocale(country),
-	}
+
+	meta := GetCheckoutMetaFromRequest(r)
 	sessionData, err := BuildAdyenCheckoutSession(grain, meta)
 	if err != nil {
 		return err
@@ -751,12 +738,103 @@ func (s *PoolServer) AdyenSessionHandler(w http.ResponseWriter, r *http.Request,
 	service := s.adyenClient.Checkout()
 	req := service.PaymentsApi.SessionsInput().CreateCheckoutSessionRequest(*sessionData)
 	res, _, err := service.PaymentsApi.Sessions(r.Context(), req)
+	// apply checkout started
 	if err != nil {
 		return err
 	}
 	return s.WriteResult(w, res)
 }
 
+func (s *PoolServer) AdyenHookHandler(w http.ResponseWriter, r *http.Request) {
+	var notificationRequest webhook.Webhook
+	service := s.adyenClient.Checkout()
+	if err := json.NewDecoder(r.Body).Decode(&notificationRequest); err != nil {
+		http.Error(w, err.Error(), http.StatusBadRequest)
+		return
+	}
+	cartHostMap := make(map[actor.Host][]webhook.NotificationItem)
+	for _, notificationItem := range *notificationRequest.NotificationItems {
+		item := notificationItem.NotificationRequestItem
+		log.Printf("Recieved notification event code: %s, %v", item.EventCode, item)
+		cartId, ok := cart.ParseCartId(item.OriginalReference)
+		if !ok {
+			log.Printf("invalid cart id %s", item.OriginalReference)
+			http.Error(w, "Invalid cart id", http.StatusBadRequest)
+			return
+		}
+		if host, ok := s.OwnerHost(uint64(cartId)); ok {
+			cartHostMap[host] = append(cartHostMap[host], notificationItem)
+			continue
+		}
+
+		isValid := hmacvalidator.ValidateHmac(item, hmacKey)
+		if !isValid {
+			log.Printf("notification hmac not valid %s, %v", item.EventCode, item)
+			http.Error(w, "Invalid HMAC", http.StatusUnauthorized)
+			return
+		} else {
+			switch item.EventCode {
+			case "AUTHORISATION":
+
+				grain, err := s.Get(r.Context(), uint64(cartId))
+				if err != nil {
+					log.Printf("Error getting cart: %v", err)
+					http.Error(w, "Cart not found", http.StatusBadRequest)
+					return
+				}
+				meta := GetCheckoutMetaFromRequest(r)
+				pspReference := item.PspReference
+				uid := uuid.New().String()
+				ref := uuid.New().String()
+				req := service.ModificationsApi.CaptureAuthorisedPaymentInput(pspReference).IdempotencyKey(uid).PaymentCaptureRequest(checkout.PaymentCaptureRequest{
+					Amount: checkout.Amount{
+						Currency: meta.Currency,
+						Value:    grain.TotalPrice.IncVat,
+					},
+					MerchantAccount: "ElgigantenECOM",
+					Reference:       &ref,
+				})
+				res, _, err := service.ModificationsApi.CaptureAuthorisedPayment(r.Context(), req)
+				if err != nil {
+					log.Printf("Error capturing payment: %v", err)
+				} else {
+					log.Printf("Payment captured successfully: %v", res)
+					s.Apply(r.Context(), uint64(cartId), &messages.OrderCreated{
+						OrderId: res.PaymentPspReference,
+						Status:  item.EventCode,
+					})
+				}
+			}
+		}
+	}
+	var failed bool = false
+	var lastMock *proxy.MockResponseWriter
+	for host, items := range cartHostMap {
+		notificationRequest.NotificationItems = &items
+		bodyBytes, err := json.Marshal(notificationRequest)
+		if err != nil {
+			log.Printf("error marshaling notification: %v", err)
+			continue
+		}
+		customBody := bytes.NewReader(bodyBytes)
+		mockW := proxy.NewMockResponseWriter()
+		handled, err := host.Proxy(0, mockW, r, customBody)
+		if err != nil {
+			log.Printf("proxy failed for %s: %v", host.Name(), err)
+			failed = true
+			lastMock = mockW
+		} else if handled {
+			log.Printf("notification proxied to %s", host.Name())
+		}
+	}
+	if failed {
+		w.WriteHeader(lastMock.StatusCode)
+		w.Write(lastMock.Body.Bytes())
+	} else {
+		w.WriteHeader(http.StatusAccepted)
+	}
+}
+
 func (s *PoolServer) Serve(mux *http.ServeMux) {
 
 	// mux.HandleFunc("OPTIONS /cart", func(w http.ResponseWriter, r *http.Request) {
@@ -780,34 +858,7 @@ func (s *PoolServer) Serve(mux *http.ServeMux) {
 		}))
 	}
 
-	handleFunc("/adyen_hook", func(w http.ResponseWriter, r *http.Request) {
-		var notificationRequest webhook.Webhook
-		if err := json.NewDecoder(r.Body).Decode(&notificationRequest); err != nil {
-			http.Error(w, err.Error(), http.StatusBadRequest)
-			return
-		}
-		for _, notification := range notificationRequest.GetNotificationItems() {
-			isValid := hmacvalidator.ValidateHmac(*notification, hmacKey)
-			if !isValid {
-				http.Error(w, "Invalid HMAC", http.StatusUnauthorized)
-				return
-			} else {
-
-				cartId, ok := cart.ParseCartId(notification.OriginalReference)
-				log.Printf("Recieved notification event code: %s, %v", notification.EventCode, notification)
-				if ok {
-					host, ok := s.OwnerHost(uint64(cartId))
-					if ok {
-						log.Printf("Not owner of %d, owner: %s", cartId, host.Name())
-						//host.Apply(r.Context(), cartId)
-					} else {
-						log.Printf("I'm the owner of %d", cartId)
-					}
-				}
-			}
-		}
-		w.WriteHeader(http.StatusAccepted)
-	})
+	handleFunc("/adyen_hook", s.AdyenHookHandler)
 
 	handleFunc("GET /cart", CookieCartIdHandler(s.ProxyHandler(s.GetCartHandler)))
 	handleFunc("GET /cart/add/{sku}", CookieCartIdHandler(s.ProxyHandler(s.AddSkuToCartHandler)))