package model

import (
	"context"
	"fmt"
	"git.noahlan.cn/northlan/ntools-go/gorm-zero/gormc"
	"github.com/pkg/errors"
	"gorm.io/gorm"
)

var _ UserIntegralModel = (*customUserIntegralModel)(nil)

type (
	// UserIntegralModel is an interface to be customized, add more methods here,
	// and implement the added methods in customUserIntegralModel.
	UserIntegralModel interface {
		userIntegralModel
		Transact(ctx context.Context, tx *gorm.DB, fn func(tx *gorm.DB) error) error
		InsertTx(ctx context.Context, tx *gorm.DB, data *UserIntegral) error
		UpdateIntegralTx(ctx context.Context, tx *gorm.DB, userId, addon int64) error
		FindIntegral(ctx context.Context, tx *gorm.DB, userId int64) (int64, error)
	}

	customUserIntegralModel struct {
		*defaultUserIntegralModel
	}
)

// NewUserIntegralModel returns a model for the database table.
func NewUserIntegralModel(conn *gorm.DB) UserIntegralModel {
	return &customUserIntegralModel{
		defaultUserIntegralModel: newUserIntegralModel(conn),
	}
}

func (m *customUserIntegralModel) Transact(ctx context.Context, tx *gorm.DB, fn func(tx *gorm.DB) error) error {
	return withTx(ctx, m.conn, tx).Transaction(fn)
}

func (m *customUserIntegralModel) InsertTx(ctx context.Context, tx *gorm.DB, data *UserIntegral) error {
	err := withTx(ctx, m.conn, tx).Create(&data).Error
	return err
}

func (m *customUserIntegralModel) UpdateIntegralTx(ctx context.Context, tx *gorm.DB, userId, change int64) error {
	if change < 0 {
		return errors.New("无法将积分更新至负数")
	}
	db := withTx(ctx, m.conn, tx)

	result := db.Table(m.table).
		Where("user_id = ?", userId).
		Update("integral", change)
	if result.Error != nil {
		return result.Error
	}
	if result.RowsAffected == 0 {
		return ErrRowsAffectedZero
	}
	return nil
}

func (m *customUserIntegralModel) FindIntegral(ctx context.Context, tx *gorm.DB, userId int64) (int64, error) {
	var resp int64
	err := withTx(ctx, m.conn, tx).Table(m.table).
		Select(fmt.Sprintf("%s.integral", m.table)).
		Where("user_id = ?", userId).Take(&resp).Error
	switch err {
	case nil:
		return resp, nil
	case gormc.ErrNotFound:
		return 0, ErrNotFound
	default:
		return 0, err
	}
}