package app

import (
	"context"
	"strings"
	"testing"

	"github.com/google/uuid"
	"google.golang.org/grpc"
	"google.golang.org/grpc/codes"
	"google.golang.org/grpc/metadata"
	"google.golang.org/grpc/status"
	"stream.api/internal/database/model"
	appv1 "stream.api/internal/gen/proto/app/v1"
	"stream.api/internal/middleware"
	"stream.api/internal/modules/common"
)

func TestCreatePayment(t *testing.T) {
	t.Run("plan không tồn tại", func(t *testing.T) {
		db := newTestDB(t)
		services := newTestAppServices(t, db)
		user := seedTestUser(t, db, model.User{ID: uuid.NewString(), Email: "user@example.com", Role: ptrString("USER")})
		conn, cleanup := newTestGRPCServer(t, services)
		defer cleanup()
		client := newPaymentsClient(conn)
		_, err := client.CreatePayment(testActorOutgoingContext(user.ID, "USER"), &appv1.CreatePaymentRequest{PlanId: uuid.NewString(), TermMonths: 1, PaymentMethod: common.PaymentMethodWallet})
		assertGRPCCode(t, err, codes.NotFound)
	})

	t.Run("plan inactive", func(t *testing.T) {
		db := newTestDB(t)
		services := newTestAppServices(t, db)
		user := seedTestUser(t, db, model.User{ID: uuid.NewString(), Email: "user@example.com", Role: ptrString("USER")})
		plan := seedTestPlan(t, db, model.Plan{ID: uuid.NewString(), Name: "Starter", Price: 10, Cycle: "monthly", StorageLimit: 10, UploadLimit: 1, QualityLimit: "720p", IsActive: ptrBool(false)})
		conn, cleanup := newTestGRPCServer(t, services)
		defer cleanup()
		client := newPaymentsClient(conn)
		_, err := client.CreatePayment(testActorOutgoingContext(user.ID, "USER"), &appv1.CreatePaymentRequest{PlanId: plan.ID, TermMonths: 1, PaymentMethod: common.PaymentMethodWallet})
		assertGRPCCode(t, err, codes.InvalidArgument)
	})

	t.Run("term không hợp lệ", func(t *testing.T) {
		db := newTestDB(t)
		services := newTestAppServices(t, db)
		user := seedTestUser(t, db, model.User{ID: uuid.NewString(), Email: "user@example.com", Role: ptrString("USER")})
		plan := seedTestPlan(t, db, model.Plan{ID: uuid.NewString(), Name: "Starter", Price: 10, Cycle: "monthly", StorageLimit: 10, UploadLimit: 1, QualityLimit: "720p", IsActive: ptrBool(true)})
		conn, cleanup := newTestGRPCServer(t, services)
		defer cleanup()
		client := newPaymentsClient(conn)
		_, err := client.CreatePayment(testActorOutgoingContext(user.ID, "USER"), &appv1.CreatePaymentRequest{PlanId: plan.ID, TermMonths: 2, PaymentMethod: common.PaymentMethodWallet})
		assertGRPCCode(t, err, codes.InvalidArgument)
	})

	t.Run("payment method không hợp lệ", func(t *testing.T) {
		db := newTestDB(t)
		services := newTestAppServices(t, db)
		user := seedTestUser(t, db, model.User{ID: uuid.NewString(), Email: "user@example.com", Role: ptrString("USER")})
		plan := seedTestPlan(t, db, model.Plan{ID: uuid.NewString(), Name: "Starter", Price: 10, Cycle: "monthly", StorageLimit: 10, UploadLimit: 1, QualityLimit: "720p", IsActive: ptrBool(true)})
		conn, cleanup := newTestGRPCServer(t, services)
		defer cleanup()
		client := newPaymentsClient(conn)
		_, err := client.CreatePayment(testActorOutgoingContext(user.ID, "USER"), &appv1.CreatePaymentRequest{PlanId: plan.ID, TermMonths: 1, PaymentMethod: "bank_transfer"})
		assertGRPCCode(t, err, codes.InvalidArgument)
	})

	t.Run("wallet thiếu tiền giữ trailer", func(t *testing.T) {
		db := newTestDB(t)
		services := newTestAppServices(t, db)
		user := seedTestUser(t, db, model.User{ID: uuid.NewString(), Email: "user@example.com", Role: ptrString("USER")})
		plan := seedTestPlan(t, db, model.Plan{ID: uuid.NewString(), Name: "Pro", Price: 50, Cycle: "monthly", StorageLimit: 100, UploadLimit: 10, QualityLimit: "1080p", IsActive: ptrBool(true)})
		seedWalletTransaction(t, db, model.WalletTransaction{ID: uuid.NewString(), UserID: user.ID, Type: common.WalletTransactionTypeTopup, Amount: 10, Currency: ptrString("USD")})
		conn, cleanup := newTestGRPCServer(t, services)
		defer cleanup()
		client := newPaymentsClient(conn)
		var trailer metadata.MD
		_, err := client.CreatePayment(testActorOutgoingContext(user.ID, "USER"), &appv1.CreatePaymentRequest{PlanId: plan.ID, TermMonths: 1, PaymentMethod: common.PaymentMethodWallet}, grpc.Trailer(&trailer))
		assertGRPCCode(t, err, codes.InvalidArgument)
		body := firstTestMetadataValue(trailer, "x-error-body")
		if body == "" {
			t.Fatal("expected x-error-body trailer")
		}
		if !strings.Contains(body, "Insufficient wallet balance") {
			t.Fatalf("x-error-body = %q, want insufficient wallet balance", body)
		}
	})

	t.Run("happy path user", func(t *testing.T) {
		db := newTestDB(t)
		services := newTestAppServices(t, db)
		user := seedTestUser(t, db, model.User{ID: uuid.NewString(), Email: "user@example.com", Role: ptrString("USER")})
		plan := seedTestPlan(t, db, model.Plan{ID: uuid.NewString(), Name: "Pro", Price: 20, Cycle: "monthly", StorageLimit: 100, UploadLimit: 10, QualityLimit: "1080p", IsActive: ptrBool(true)})
		seedWalletTransaction(t, db, model.WalletTransaction{ID: uuid.NewString(), UserID: user.ID, Type: common.WalletTransactionTypeTopup, Amount: 5, Currency: ptrString("USD")})
		conn, cleanup := newTestGRPCServer(t, services)
		defer cleanup()
		client := newPaymentsClient(conn)
		resp, err := client.CreatePayment(testActorOutgoingContext(user.ID, "USER"), &appv1.CreatePaymentRequest{PlanId: plan.ID, TermMonths: 1, PaymentMethod: common.PaymentMethodTopup, TopupAmount: ptrFloat64(15)})
		if err != nil {
			t.Fatalf("CreatePayment() error = %v", err)
		}
		if resp.Payment == nil || resp.Subscription == nil {
			t.Fatalf("CreatePayment() response incomplete: %#v", resp)
		}
		if resp.InvoiceId != common.BuildInvoiceID(resp.Payment.Id) {
			t.Fatalf("invoice id = %q, want %q", resp.InvoiceId, common.BuildInvoiceID(resp.Payment.Id))
		}
		if resp.Subscription.PaymentMethod != common.PaymentMethodTopup {
			t.Fatalf("subscription payment method = %q, want %q", resp.Subscription.PaymentMethod, common.PaymentMethodTopup)
		}
		if resp.Subscription.WalletAmount != 20 {
			t.Fatalf("subscription wallet amount = %v, want 20", resp.Subscription.WalletAmount)
		}
		if resp.Subscription.TopupAmount != 15 {
			t.Fatalf("subscription topup amount = %v, want 15", resp.Subscription.TopupAmount)
		}
		if resp.WalletBalance != 0 {
			t.Fatalf("wallet balance = %v, want 0", resp.WalletBalance)
		}
		payment := mustLoadPayment(t, db, resp.Payment.Id)
		if payment.Amount != 20 {
			t.Fatalf("payment amount = %v, want 20", payment.Amount)
		}
	})
}

func testActorOutgoingContext(userID, role string) context.Context {
	return metadata.NewOutgoingContext(context.Background(), metadata.Pairs(
		middleware.ActorMarkerMetadataKey, testTrustedMarker,
		middleware.ActorIDMetadataKey, userID,
		middleware.ActorRoleMetadataKey, role,
		middleware.ActorEmailMetadataKey, strings.ToLower(role)+"@example.com",
	))
}

func assertGRPCCode(t *testing.T, err error, want codes.Code) {
	t.Helper()
	if err == nil {
		t.Fatalf("grpc error = nil, want %v", want)
	}
	if got := status.Code(err); got != want {
		t.Fatalf("grpc code = %v, want %v, err=%v", got, want, err)
	}
}