package apple

import (
	"context"
	"crypto/sha256"
	"encoding/hex"
	"fmt"
	"time"

	iapmodel "github.com/perfect-panel/server/internal/model/iap/apple"
	"github.com/perfect-panel/server/internal/model/user"
	"github.com/perfect-panel/server/internal/svc"
	"github.com/perfect-panel/server/internal/types"
	"github.com/perfect-panel/server/pkg/constant"
	iapapple "github.com/perfect-panel/server/pkg/iap/apple"
	"github.com/perfect-panel/server/pkg/logger"
	"github.com/perfect-panel/server/pkg/xerr"
	"github.com/pkg/errors"
	"gorm.io/gorm"
	"github.com/google/uuid"
)

type AttachTransactionLogic struct {
	logger.Logger
	ctx    context.Context
	svcCtx *svc.ServiceContext
}

func NewAttachTransactionLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AttachTransactionLogic {
	return &AttachTransactionLogic{
		Logger: logger.WithContext(ctx),
		ctx:    ctx,
		svcCtx: svcCtx,
	}
}

func (l *AttachTransactionLogic) Attach(req *types.AttachAppleTransactionRequest) (*types.AttachAppleTransactionResponse, error) {
	u, ok := l.ctx.Value(constant.CtxKeyUser).(*user.User)
	if !ok || u == nil {
		return nil, errors.Wrapf(xerr.NewErrCode(xerr.InvalidAccess), "invalid access")
	}
	txPayload, err := iapapple.ParseTransactionJWS(req.SignedTransactionJWS)
	if err != nil {
		return nil, errors.Wrapf(xerr.NewErrCode(xerr.ERROR), "invalid jws")
	}
	pm, _ := iapapple.ParseProductMap(l.svcCtx.Config.Site.CustomData)
	m, ok := pm.Items[txPayload.ProductId]
	var duration int64
	var tier string
	var subscribeId int64
	if ok {
		duration = m.DurationDays
		tier = m.Tier
		subscribeId = m.SubscribeId
	} else {
		if req.DurationDays <= 0 || req.SubscribeId <= 0 {
			return nil, errors.Wrapf(xerr.NewErrCode(xerr.ERROR), "unknown product")
		}
		duration = req.DurationDays
		tier = req.Tier
		subscribeId = req.SubscribeId
	}
	exp := iapapple.CalcExpire(txPayload.PurchaseDate, duration)
	sum := sha256.Sum256([]byte(req.SignedTransactionJWS))
	jwsHash := hex.EncodeToString(sum[:])
	iapTx := &iapmodel.Transaction{
		UserId:                u.Id,
		OriginalTransactionId: txPayload.OriginalTransactionId,
		TransactionId:         txPayload.TransactionId,
		ProductId:             txPayload.ProductId,
		PurchaseAt:            txPayload.PurchaseDate,
		RevocationAt:          txPayload.RevocationDate,
		JWSHash:               jwsHash,
	}
	err = l.svcCtx.DB.Transaction(func(tx *gorm.DB) error {
		if e := tx.Model(&iapmodel.Transaction{}).Create(iapTx).Error; e != nil {
			return e
		}
		// insert user_subscribe
		userSub := user.Subscribe{
			UserId:      u.Id,
			SubscribeId: subscribeId,
			StartTime:   time.Now(),
			ExpireTime:  exp,
			Traffic:     0,
			Download:    0,
			Upload:      0,
			Token:       fmt.Sprintf("iap:%s", txPayload.OriginalTransactionId),
			UUID:        uuid.New().String(),
			Status:      1,
		}
		return l.svcCtx.UserModel.InsertSubscribe(l.ctx, &userSub, tx)
	})
	if err != nil {
		return nil, errors.Wrapf(xerr.NewErrCode(xerr.DatabaseInsertError), "insert error: %v", err.Error())
	}
	return &types.AttachAppleTransactionResponse{
		ExpiresAt: exp.Unix(),
		Tier:      tier,
	}, nil
}