package dao

import (
	"fmt"

	"gitlab.fjmaimaimai.com/mmm-go/partnermg/pkg/infrastructure/pg/models"
	"gitlab.fjmaimaimai.com/mmm-go/partnermg/pkg/infrastructure/pg/transaction"
)

type PartnerInfoDao struct {
	transactionContext *transaction.TransactionContext
}

func NewPartnerInfoDao(transactionContext *transaction.TransactionContext) (*PartnerInfoDao, error) {
	if transactionContext == nil {
		return nil, fmt.Errorf("transactionContext参数不能为nil")
	} else {
		return &PartnerInfoDao{
			transactionContext: transactionContext,
		}, nil
	}
}

func (dao PartnerInfoDao) PartnerAccountExist(account string, companyId int64) (bool, error) {
	tx := dao.transactionContext.PgDd
	m := &models.PartnerInfo{}
	ok, err := tx.Model(m).
		Where("account=?", account).
		Where("company_id=?", companyId).
		Exists()

	return ok, err
}

func (dao PartnerInfoDao) UpdatePartnerStatus(ids []int64, companyId int64, status int) error {
	tx := dao.transactionContext.PgDd
	m := &models.PartnerInfo{}
	_, err := tx.Model(m).
		WhereIn("id in (?)", ids).
		Where("company_id=?", companyId).
		Set("status=?", status).
		Update()
	return err
}