424 lines
13 KiB
Go
424 lines
13 KiB
Go
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"fmt"
|
|
"log"
|
|
"net/http"
|
|
"strconv"
|
|
"time"
|
|
|
|
"git.tornberg.me/go-cart-actor/pkg/actor"
|
|
messages "git.tornberg.me/go-cart-actor/pkg/messages"
|
|
"github.com/gogo/protobuf/proto"
|
|
)
|
|
|
|
type PoolServer struct {
|
|
pod_name string
|
|
pool actor.GrainPool[*CartGrain]
|
|
klarnaClient *KlarnaClient
|
|
}
|
|
|
|
func NewPoolServer(pool actor.GrainPool[*CartGrain], pod_name string, klarnaClient *KlarnaClient) *PoolServer {
|
|
return &PoolServer{
|
|
pod_name: pod_name,
|
|
pool: pool,
|
|
klarnaClient: klarnaClient,
|
|
}
|
|
}
|
|
|
|
func (s *PoolServer) ApplyLocal(id CartId, mutation proto.Message) (*CartGrain, error) {
|
|
return s.pool.Apply(uint64(id), mutation)
|
|
}
|
|
|
|
func (s *PoolServer) HandleGet(w http.ResponseWriter, r *http.Request, id CartId) error {
|
|
grain, err := s.pool.Get(uint64(id))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return s.WriteResult(w, grain)
|
|
}
|
|
|
|
func (s *PoolServer) HandleAddSku(w http.ResponseWriter, r *http.Request, id CartId) error {
|
|
sku := r.PathValue("sku")
|
|
msg, err := GetItemAddMessage(sku, 1, getCountryFromHost(r.Host), nil)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
data, err := s.ApplyLocal(id, msg)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return s.WriteResult(w, data)
|
|
}
|
|
|
|
func (s *PoolServer) WriteResult(w http.ResponseWriter, result *CartGrain) error {
|
|
w.Header().Set("Content-Type", "application/json")
|
|
w.Header().Set("Cache-Control", "no-cache")
|
|
w.Header().Set("Access-Control-Allow-Origin", "*")
|
|
w.Header().Set("X-Pod-Name", s.pod_name)
|
|
if result == nil {
|
|
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
|
|
return nil
|
|
}
|
|
w.WriteHeader(http.StatusOK)
|
|
enc := json.NewEncoder(w)
|
|
err := enc.Encode(result)
|
|
return err
|
|
}
|
|
|
|
func (s *PoolServer) HandleDeleteItem(w http.ResponseWriter, r *http.Request, id CartId) error {
|
|
|
|
itemIdString := r.PathValue("itemId")
|
|
itemId, err := strconv.Atoi(itemIdString)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
data, err := s.ApplyLocal(id, &messages.RemoveItem{Id: int64(itemId)})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return s.WriteResult(w, data)
|
|
}
|
|
|
|
type SetDeliveryRequest struct {
|
|
Provider string `json:"provider"`
|
|
Items []int64 `json:"items"`
|
|
PickupPoint *messages.PickupPoint `json:"pickupPoint,omitempty"`
|
|
}
|
|
|
|
func (s *PoolServer) HandleSetDelivery(w http.ResponseWriter, r *http.Request, id CartId) error {
|
|
|
|
delivery := SetDeliveryRequest{}
|
|
err := json.NewDecoder(r.Body).Decode(&delivery)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
data, err := s.ApplyLocal(id, &messages.SetDelivery{
|
|
Provider: delivery.Provider,
|
|
Items: delivery.Items,
|
|
PickupPoint: delivery.PickupPoint,
|
|
})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return s.WriteResult(w, data)
|
|
}
|
|
|
|
func (s *PoolServer) HandleSetPickupPoint(w http.ResponseWriter, r *http.Request, id CartId) error {
|
|
|
|
deliveryIdString := r.PathValue("deliveryId")
|
|
deliveryId, err := strconv.Atoi(deliveryIdString)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
pickupPoint := messages.PickupPoint{}
|
|
err = json.NewDecoder(r.Body).Decode(&pickupPoint)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
reply, err := s.ApplyLocal(id, &messages.SetPickupPoint{
|
|
DeliveryId: int64(deliveryId),
|
|
Id: pickupPoint.Id,
|
|
Name: pickupPoint.Name,
|
|
Address: pickupPoint.Address,
|
|
City: pickupPoint.City,
|
|
Zip: pickupPoint.Zip,
|
|
Country: pickupPoint.Country,
|
|
})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return s.WriteResult(w, reply)
|
|
}
|
|
|
|
func (s *PoolServer) HandleRemoveDelivery(w http.ResponseWriter, r *http.Request, id CartId) error {
|
|
|
|
deliveryIdString := r.PathValue("deliveryId")
|
|
deliveryId, err := strconv.Atoi(deliveryIdString)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
reply, err := s.ApplyLocal(id, &messages.RemoveDelivery{Id: int64(deliveryId)})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return s.WriteResult(w, reply)
|
|
}
|
|
|
|
func (s *PoolServer) HandleQuantityChange(w http.ResponseWriter, r *http.Request, id CartId) error {
|
|
changeQuantity := messages.ChangeQuantity{}
|
|
err := json.NewDecoder(r.Body).Decode(&changeQuantity)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
reply, err := s.ApplyLocal(id, &changeQuantity)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return s.WriteResult(w, reply)
|
|
}
|
|
|
|
func (s *PoolServer) HandleSetCartItems(w http.ResponseWriter, r *http.Request, id CartId) error {
|
|
setCartItems := messages.SetCartRequest{}
|
|
err := json.NewDecoder(r.Body).Decode(&setCartItems)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
reply, err := s.ApplyLocal(id, &setCartItems)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return s.WriteResult(w, reply)
|
|
}
|
|
|
|
func (s *PoolServer) HandleAddRequest(w http.ResponseWriter, r *http.Request, id CartId) error {
|
|
addRequest := messages.AddRequest{}
|
|
err := json.NewDecoder(r.Body).Decode(&addRequest)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
msg, err := GetItemAddMessage(addRequest.Sku, int(addRequest.Quantity), addRequest.Country, addRequest.StoreId)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
reply, err := s.ApplyLocal(id, msg)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return s.WriteResult(w, reply)
|
|
}
|
|
|
|
// func (s *PoolServer) HandleConfirmation(w http.ResponseWriter, r *http.Request, id CartId) error {
|
|
// orderId := r.PathValue("orderId")
|
|
// if orderId == "" {
|
|
// return fmt.Errorf("orderId is empty")
|
|
// }
|
|
// order, err := KlarnaInstance.GetOrder(orderId)
|
|
|
|
// if err != nil {
|
|
// return err
|
|
// }
|
|
|
|
// w.Header().Set("Content-Type", "application/json")
|
|
// w.Header().Set("X-Pod-Name", s.pod_name)
|
|
// w.Header().Set("Cache-Control", "no-cache")
|
|
// w.Header().Set("Access-Control-Allow-Origin", "*")
|
|
// w.WriteHeader(http.StatusOK)
|
|
// return json.NewEncoder(w).Encode(order)
|
|
// }
|
|
|
|
func getCurrency(country string) string {
|
|
if country == "no" {
|
|
return "NOK"
|
|
}
|
|
return "SEK"
|
|
}
|
|
|
|
func getLocale(country string) string {
|
|
if country == "no" {
|
|
return "nb-no"
|
|
}
|
|
return "sv-se"
|
|
}
|
|
|
|
func (s *PoolServer) CreateOrUpdateCheckout(host string, id CartId) (*CheckoutOrder, error) {
|
|
country := getCountryFromHost(host)
|
|
meta := &CheckoutMeta{
|
|
Terms: fmt.Sprintf("https://%s/terms", host),
|
|
Checkout: fmt.Sprintf("https://%s/checkout?order_id={checkout.order.id}", host),
|
|
Confirmation: fmt.Sprintf("https://%s/confirmation/{checkout.order.id}", host),
|
|
Validation: fmt.Sprintf("https://%s/validate", host),
|
|
Push: fmt.Sprintf("https://%s/push?order_id={checkout.order.id}", host),
|
|
Country: country,
|
|
Currency: getCurrency(country),
|
|
Locale: getLocale(country),
|
|
}
|
|
|
|
// Get current grain state (may be local or remote)
|
|
grain, err := s.pool.Get(uint64(id))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// Build pure checkout payload
|
|
payload, _, err := BuildCheckoutOrderPayload(grain, meta)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if grain.OrderReference != "" {
|
|
return s.klarnaClient.UpdateOrder(grain.OrderReference, bytes.NewReader(payload))
|
|
} else {
|
|
return s.klarnaClient.CreateOrder(bytes.NewReader(payload))
|
|
}
|
|
}
|
|
|
|
func (s *PoolServer) ApplyCheckoutStarted(klarnaOrder *CheckoutOrder, id CartId) (*CartGrain, error) {
|
|
// Persist initialization state via mutation (best-effort)
|
|
return s.pool.Apply(uint64(id), &messages.InitializeCheckout{
|
|
OrderId: klarnaOrder.ID,
|
|
Status: klarnaOrder.Status,
|
|
PaymentInProgress: true,
|
|
})
|
|
}
|
|
|
|
// func (s *PoolServer) HandleCheckout(w http.ResponseWriter, r *http.Request, id CartId) error {
|
|
// klarnaOrder, err := s.CreateOrUpdateCheckout(r.Host, id)
|
|
// if err != nil {
|
|
// return err
|
|
// }
|
|
|
|
// s.ApplyCheckoutStarted(klarnaOrder, id)
|
|
|
|
// w.Header().Set("Content-Type", "application/json")
|
|
// return json.NewEncoder(w).Encode(klarnaOrder)
|
|
// }
|
|
//
|
|
|
|
func CookieCartIdHandler(fn func(cartId CartId, w http.ResponseWriter, r *http.Request) error) func(w http.ResponseWriter, r *http.Request) {
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
|
var id CartId
|
|
cookie, err := r.Cookie("cartid")
|
|
if err != nil || cookie.Value == "" {
|
|
id = MustNewCartId()
|
|
http.SetCookie(w, &http.Cookie{
|
|
Name: "cartid",
|
|
Value: id.String(),
|
|
Secure: r.TLS != nil,
|
|
HttpOnly: true,
|
|
Path: "/",
|
|
Expires: time.Now().AddDate(0, 0, 14),
|
|
SameSite: http.SameSiteLaxMode,
|
|
})
|
|
w.Header().Set("Set-Cart-Id", id.String())
|
|
} else {
|
|
parsed, ok := ParseCartId(cookie.Value)
|
|
if !ok {
|
|
id = MustNewCartId()
|
|
http.SetCookie(w, &http.Cookie{
|
|
Name: "cartid",
|
|
Value: id.String(),
|
|
Secure: r.TLS != nil,
|
|
HttpOnly: true,
|
|
Path: "/",
|
|
Expires: time.Now().AddDate(0, 0, 14),
|
|
SameSite: http.SameSiteLaxMode,
|
|
})
|
|
w.Header().Set("Set-Cart-Id", id.String())
|
|
} else {
|
|
id = parsed
|
|
}
|
|
}
|
|
|
|
err = fn(id, w, r)
|
|
if err != nil {
|
|
log.Printf("Server error, not remote error: %v\n", err)
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
w.Write([]byte(err.Error()))
|
|
}
|
|
|
|
}
|
|
}
|
|
|
|
// Removed leftover legacy block after CookieCartIdHandler (obsolete code referencing cid/legacy)
|
|
|
|
func (s *PoolServer) RemoveCartCookie(w http.ResponseWriter, r *http.Request, cartId CartId) error {
|
|
// Clear cart cookie (breaking change: do not issue a new legacy id here)
|
|
http.SetCookie(w, &http.Cookie{
|
|
Name: "cartid",
|
|
Value: "",
|
|
Path: "/",
|
|
Secure: r.TLS != nil,
|
|
HttpOnly: true,
|
|
Expires: time.Unix(0, 0),
|
|
SameSite: http.SameSiteLaxMode,
|
|
})
|
|
w.WriteHeader(http.StatusOK)
|
|
return nil
|
|
}
|
|
|
|
func CartIdHandler(fn func(cartId CartId, w http.ResponseWriter, r *http.Request) error) func(w http.ResponseWriter, r *http.Request) {
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
|
var id CartId
|
|
raw := r.PathValue("id")
|
|
// If no id supplied, generate a new one
|
|
if raw == "" {
|
|
id := MustNewCartId()
|
|
w.Header().Set("Set-Cart-Id", id.String())
|
|
} else {
|
|
// Parse base62 cart id
|
|
if parsedId, ok := ParseCartId(raw); !ok {
|
|
w.WriteHeader(http.StatusBadRequest)
|
|
w.Write([]byte("cart id is invalid"))
|
|
return
|
|
} else {
|
|
id = parsedId
|
|
}
|
|
}
|
|
|
|
err := fn(id, w, r)
|
|
if err != nil {
|
|
log.Printf("Server error, not remote error: %v\n", err)
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
w.Write([]byte(err.Error()))
|
|
}
|
|
}
|
|
}
|
|
|
|
func (s *PoolServer) ProxyHandler(fn func(w http.ResponseWriter, r *http.Request, cartId CartId) error) func(cartId CartId, w http.ResponseWriter, r *http.Request) error {
|
|
return func(cartId CartId, w http.ResponseWriter, r *http.Request) error {
|
|
if ownerHost, ok := s.pool.OwnerHost(uint64(cartId)); ok {
|
|
handled, err := ownerHost.Proxy(uint64(cartId), w, r)
|
|
if err == nil && handled {
|
|
return nil
|
|
}
|
|
}
|
|
|
|
return fn(w, r, cartId)
|
|
|
|
}
|
|
}
|
|
|
|
func (s *PoolServer) Serve() *http.ServeMux {
|
|
|
|
mux := http.NewServeMux()
|
|
mux.HandleFunc("OPTIONS /", func(w http.ResponseWriter, r *http.Request) {
|
|
w.Header().Set("Access-Control-Allow-Origin", "*")
|
|
w.Header().Set("Access-Control-Allow-Methods", "GET, PUT, POST, DELETE")
|
|
w.Header().Set("Access-Control-Allow-Headers", "Content-Type")
|
|
w.WriteHeader(http.StatusOK)
|
|
})
|
|
|
|
mux.HandleFunc("GET /", CookieCartIdHandler(s.ProxyHandler(s.HandleGet)))
|
|
mux.HandleFunc("GET /add/{sku}", CookieCartIdHandler(s.ProxyHandler(s.HandleAddSku)))
|
|
mux.HandleFunc("POST /", CookieCartIdHandler(s.ProxyHandler(s.HandleAddRequest)))
|
|
mux.HandleFunc("POST /set", CookieCartIdHandler(s.ProxyHandler(s.HandleSetCartItems)))
|
|
mux.HandleFunc("DELETE /{itemId}", CookieCartIdHandler(s.ProxyHandler(s.HandleDeleteItem)))
|
|
mux.HandleFunc("PUT /", CookieCartIdHandler(s.ProxyHandler(s.HandleQuantityChange)))
|
|
mux.HandleFunc("DELETE /", CookieCartIdHandler(s.ProxyHandler(s.RemoveCartCookie)))
|
|
mux.HandleFunc("POST /delivery", CookieCartIdHandler(s.ProxyHandler(s.HandleSetDelivery)))
|
|
mux.HandleFunc("DELETE /delivery/{deliveryId}", CookieCartIdHandler(s.ProxyHandler(s.HandleRemoveDelivery)))
|
|
mux.HandleFunc("PUT /delivery/{deliveryId}/pickupPoint", CookieCartIdHandler(s.ProxyHandler(s.HandleSetPickupPoint)))
|
|
//mux.HandleFunc("GET /checkout", CookieCartIdHandler(s.ProxyHandler(s.HandleCheckout)))
|
|
//mux.HandleFunc("GET /confirmation/{orderId}", CookieCartIdHandler(s.ProxyHandler(s.HandleConfirmation)))
|
|
|
|
mux.HandleFunc("GET /byid/{id}", CartIdHandler(s.ProxyHandler(s.HandleGet)))
|
|
mux.HandleFunc("GET /byid/{id}/add/{sku}", CartIdHandler(s.ProxyHandler(s.HandleAddSku)))
|
|
mux.HandleFunc("POST /byid/{id}", CartIdHandler(s.ProxyHandler(s.HandleAddRequest)))
|
|
mux.HandleFunc("DELETE /byid/{id}/{itemId}", CartIdHandler(s.ProxyHandler(s.HandleDeleteItem)))
|
|
mux.HandleFunc("PUT /byid/{id}", CartIdHandler(s.ProxyHandler(s.HandleQuantityChange)))
|
|
mux.HandleFunc("POST /byid/{id}/delivery", CartIdHandler(s.ProxyHandler(s.HandleSetDelivery)))
|
|
mux.HandleFunc("DELETE /byid/{id}/delivery/{deliveryId}", CartIdHandler(s.ProxyHandler(s.HandleRemoveDelivery)))
|
|
mux.HandleFunc("PUT /byid/{id}/delivery/{deliveryId}/pickupPoint", CartIdHandler(s.ProxyHandler(s.HandleSetPickupPoint)))
|
|
//mux.HandleFunc("GET /byid/{id}/checkout", CartIdHandler(s.ProxyHandler(s.HandleCheckout)))
|
|
//mux.HandleFunc("GET /byid/{id}/confirmation", CartIdHandler(s.ProxyHandler(s.HandleConfirmation)))
|
|
|
|
return mux
|
|
}
|