package repository

import (
	"app-cloudep-trade-service/internal/domain"
	"app-cloudep-trade-service/internal/domain/repository"
	"app-cloudep-trade-service/internal/model"
	"context"
	"fmt"
	"time"

	"github.com/shopspring/decimal"
	"github.com/zeromicro/go-zero/core/logx"
	"github.com/zeromicro/go-zero/core/stores/sqlx"
)

type WalletRepositoryParam struct {
	WalletModel model.WalletModel
	TxConn      sqlx.SqlConn
}

type WalletRepository struct {
	WalletRepositoryParam
}

// GetUserWalletOperator 取得本地操作使用者錢包的操作運算元
func (repo *WalletRepository) GetUserWalletOperator(uid, currency string, opts ...repository.Option) repository.UserWalletOperator {
	db := repo.TxConn

	// 看是否有最新的DB 連線要傳入，做 tx
	for _, fn := range opts {
		db = fn()
	}

	return NewUserWalletOperator(uid, currency, repo.WalletModel, db)
}

// Create 創建錢包，如果有相同的就跳過不建立
func (repo *WalletRepository) Create(ctx context.Context, uid, currency, brand string) ([]*model.Wallet, error) {
	wallets := make([]*model.Wallet, 0, len(domain.AllWalletType))
	// 建立個人所有種類的錢包
	now := time.Now().UTC().UnixNano()
	for _, item := range domain.AllWalletType {
		balance := decimal.Zero
		wallets = append(wallets, &model.Wallet{
			Brand:      brand,
			Currency:   currency,
			Uid:        uid,
			Balance:    balance,
			WalletType: item,
			CreatedAt:  now,
			UpdatedAt:  now,
		})
	}

	_, err := repo.WalletModel.InsertMany(ctx, wallets)
	if err != nil {
		// 錯誤代碼 06-021-20
		e := domain.CommentErrorL(
			domain.CreateWalletErrorCode,
			logx.WithContext(ctx),
			[]logx.LogField{
				{Key: "param", Value: fmt.Sprintf("uid: %s, currency:%s, brand:%s", uid, currency, brand)},
				{Key: "func", Value: "WalletModel.InsertMany"},
				{Key: "err", Value: err},
			},
			"failed to insert wallet into mysql:").Wrap(err)

		return nil, e
	}

	return wallets, nil
}

func (repo *WalletRepository) Balances(ctx context.Context, req repository.BalanceReq) ([]model.Wallet, error) {
	data, err := repo.WalletModel.Balances(ctx, model.BalanceReq{
		UID:      req.UID,
		Currency: req.Currency,
	})

	if err != nil {
		// 錯誤代碼 06-021-20
		e := domain.CommentErrorL(
			domain.CreateWalletErrorCode,
			logx.WithContext(ctx),
			[]logx.LogField{
				{Key: "param", Value: req},
				{Key: "func", Value: "WalletModel.Balances"},
				{Key: "err", Value: err},
			},
			"failed to find balance into mongo:").Wrap(err)

		return []model.Wallet{}, e
	}

	return data, nil
}

func (repo *WalletRepository) GetTxDatabaseConn() sqlx.SqlConn {
	return repo.TxConn
}

func NewWalletRepository(param WalletRepositoryParam) repository.WalletRepository {
	return &WalletRepository{
		WalletRepositoryParam: param,
	}
}