package repository

import (
	"errors"
	"fmt"

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

type PartnerInfoRepository struct {
	transactionContext *transaction.TransactionContext
}

var (
	_ domain.PartnerInfoRepository = (*PartnerInfoRepository)(nil)
)

func (repository *PartnerInfoRepository) Save(dm *domain.PartnerInfo) error {
	var (
		err error
		tx  = repository.transactionContext.PgTx
	)

	var categories []domain.PartnerCategory
	for _, v := range dm.PartnerCategoryInfos {
		categories = append(categories, domain.PartnerCategory{
			Id:   v.Id,
			Name: v.Name,
			Code: v.Code,
		})
	}
	m := &models.PartnerInfo{
		Id:                   dm.Partner.Id,
		PartnerName:          dm.Partner.PartnerName,
		Account:              dm.Partner.Account,
		Password:             dm.Password,
		Status:               dm.Status,
		PartnerCategory:      dm.PartnerCategory,
		Salesman:             dm.Salesman,
		RegionInfo:           dm.RegionInfo,
		CooperateTime:        dm.CooperateTime,
		CompanyId:            dm.CompanyId,
		PartnerCategoryInfos: categories,
		Remark:               dm.Remark,
	}
	if m.Id == 0 {
		err = tx.Insert(m)
		dm.Partner.Id = m.Id
		if err != nil {
			return err
		}
	} else {
		_, err = tx.Model(m).WherePK().
			Column("partner_name", "account", "password", "status", "partner_category", "salesman",
				"region_info", "cooperate_time", "update_at", "partner_category_infos", "remark").
			Update()
		if err != nil {
			return err
		}
	}
	return nil
}

func (repository *PartnerInfoRepository) FindOne(queryOptions domain.PartnerFindOneQuery) (*domain.PartnerInfo, error) {
	tx := repository.transactionContext.PgTx
	PartnerInfoModel := new(models.PartnerInfo)
	var hasCondition bool
	query := tx.Model(PartnerInfoModel)
	if len(queryOptions.AccountEqual) > 0 {
		hasCondition = true
		query = query.Where("account=?", queryOptions.AccountEqual)
	}
	if queryOptions.UserId > 0 {
		hasCondition = true
		query = query.Where("id=?", queryOptions.UserId)
	}
	if queryOptions.CompanyId > 0 {
		hasCondition = true
		query = query.Where("company_id=?", queryOptions.CompanyId)
	}
	if !hasCondition {
		return nil, errors.New("FindOne 必须要有查询条件")
	}
	if err := query.First(); err != nil {
		return nil, err
	}
	result, err := repository.transformPgModelToDomainModel(PartnerInfoModel)
	return &result, err
}

func (repository *PartnerInfoRepository) Find(queryOption domain.PartnerFindQuery) ([]domain.PartnerInfo, error) {
	db := repository.transactionContext.PgTx
	var partnerModels []models.PartnerInfo
	query := db.Model(&partnerModels)
	if len(queryOption.PartnerName) > 0 {
		query = query.Where("partner_name like ?", "%"+queryOption.PartnerName+"%")
	}
	if queryOption.RegionInfo != nil {
		query = query.Where("region_info::jsonb->>'regionName' like ?", "%"+queryOption.RegionInfo.RegionName+"%")
	}
	//合伙人类型
	if len(queryOption.PartnerCategory) > 0 {
		query = query.WhereGroup(func(q *orm.Query) (*orm.Query, error) {
			for _, value := range queryOption.PartnerCategory {
				q = q.WhereOr(`partner_category_infos@> '[{"id":?}]'`, value)
			}
			return q, nil
		})
	}
	if queryOption.CompanyId > 0 {
		query = query.Where("company_id=?", queryOption.CompanyId)
	}
	if len(queryOption.Ids) > 0 {
		query = query.WhereIn("id in(?)", queryOption.Ids)
	}
	if queryOption.Offset > -1 {
		query = query.Offset(queryOption.Offset)
	}
	if queryOption.Limit > 0 {
		query = query.Limit(queryOption.Limit)
	} else {
		query = query.Limit(1000)
	}
	var (
		err           error
		partnerReturn = make([]domain.PartnerInfo, 0)
	)
	query = query.Order("create_at DESC")
	err = query.Select()
	if err != nil {
		return partnerReturn, err
	}
	for i := range partnerModels {
		domainPartner, err := repository.transformPgModelToDomainModel(&partnerModels[i])
		if err != nil {
			return partnerReturn, err
		}
		partnerReturn = append(partnerReturn, domainPartner)
	}
	return partnerReturn, nil
}

/**
 * @Author SteveChan
 * @Description // 移除合伙人
 * @Date 15:33 2020/12/29
 * @Param
 * @return
 **/
func (repository *PartnerInfoRepository) Remove(id int64) error {
	var (
		err error
		tx  = repository.transactionContext.PgTx
	)
	m := &models.PartnerInfo{
		Id: id,
	}
	_, err = tx.Model(m).
		Where("id=?", id).
		Delete()
	return err
}

func (repository *PartnerInfoRepository) CountAll(queryOption domain.PartnerFindQuery) (int, error) {
	db := repository.transactionContext.PgTx
	partnerModels := models.PartnerInfo{}
	query := db.Model(&partnerModels)
	if len(queryOption.PartnerName) > 0 {
		query = query.Where("partner_name like ?", "%"+queryOption.PartnerName+"%")
	}
	if queryOption.RegionInfo != nil {
		query = query.Where("region_info::jsonb->>'regionName' like ?", "%"+queryOption.RegionInfo.RegionName+"%")
	}
	//合伙人类型
	if len(queryOption.PartnerCategory) > 0 {
		query = query.WhereGroup(func(q *orm.Query) (*orm.Query, error) {
			for _, value := range queryOption.PartnerCategory {
				q = q.WhereOr(`partner_category_infos@> '[{"id":?}]'`, value)
			}
			return q, nil
		})
	}
	if queryOption.CompanyId > 0 {
		query = query.Where("company_id=?", queryOption.CompanyId)
	}
	if len(queryOption.Ids) > 0 {
		query = query.WhereIn("id in(?)", queryOption.Ids)
	}
	cnt, err := query.Count()
	return cnt, err
}

func (repository *PartnerInfoRepository) transformPgModelToDomainModel(partnerInfoModel *models.PartnerInfo) (domain.PartnerInfo, error) {
	m := domain.PartnerInfo{
		Partner: domain.Partner{
			Id:          partnerInfoModel.Id,
			PartnerName: partnerInfoModel.PartnerName,
			Account:     partnerInfoModel.Account,
		},
		PartnerCategory: partnerInfoModel.PartnerCategory,
		Password:        partnerInfoModel.Password,
		Status:          partnerInfoModel.Status,
		CreateAt:        partnerInfoModel.CreateAt,
		Salesman:        partnerInfoModel.Salesman,
		RegionInfo:      partnerInfoModel.RegionInfo,
		CooperateTime:   partnerInfoModel.CooperateTime,
		CompanyId:       partnerInfoModel.CompanyId,
		Remark:          partnerInfoModel.Remark,
	}
	var p []domain.PartnerCategory
	for _, v := range partnerInfoModel.PartnerCategoryInfos {
		category := domain.PartnerCategory{
			Id:   v.Id,
			Name: v.Name,
			Code: v.Code,
		}
		p = append(p, category)
	}
	m.PartnerCategoryInfos = p
	return m, nil
}

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