package wxapp

import (
	"errors"

	"gitee.com/taotechip/modules/database"
	"gorm.io/gorm"
)

type wxappRepo struct {
	database.GeneralRepo
}

func WxappRepo(transactionalConn *gorm.DB) (repo *wxappRepo) {
	repo = database.NewRepo(transactionalConn, &wxappRepo{})
	return
}

func (repo *wxappRepo) FindWechatAccountByOpenId(openId string) (account *WechatAccount, err error) {
	account = &WechatAccount{}
	err = repo.GetConn().Where("open_id = ?", openId).First(account).Error
	return
}

func (repo *wxappRepo) FindWechatAccountByUnionId(unionId string) (account *WechatAccount, err error) {
	account = &WechatAccount{}
	err = repo.GetConn().Where("union_id = ?", unionId).First(account).Error
	if errors.Is(err, gorm.ErrRecordNotFound) {
		return nil, nil
	}
	return
}

func (repo *wxappRepo) FindWechatAccountByAccountId(accountId string) (account *WechatAccount, err error) {
	account = &WechatAccount{}
	err = repo.GetConn().Where("account_id = ?", accountId).First(account).Error
	if errors.Is(err, gorm.ErrRecordNotFound) {
		return nil, nil
	}
	return
}

func (repo *wxappRepo) CreateWechatAccount(account *WechatAccount) error {
	return repo.GetConn().Create(account).Error
}

func (repo *wxappRepo) UpdateWechatAccount(account *WechatAccount, columns ...string) error {
	return repo.GetConn().Model(account).Select(columns).Updates(account).Error
}

func (repo *wxappRepo) DeleteWechatAccount(id string) error {
	return repo.GetConn().Delete(&WechatAccount{}, id).Error
}