package service

import (
	"errors"
	"model-engine/db"
	"model-engine/pkg/logger"
)

func GetUnitsByIds(ids []string) (list []*db.DomainUnit, err error) {
	list = make([]*db.DomainUnit, 0)
	err = db.GetDB().Model(&db.DomainUnit{}).Where("id in ?", ids).Find(&list).Error
	return
}
func GetOrgIdsAndAreaIdsByDomainUnitIds(domainUnitIds []string) (targetOrgIds, targetAreaIds []interface{}, err error) {
	domains, err := GetUnitsByIds(domainUnitIds)
	if err != nil {
		return nil, nil, err
	}

	otherIds := make([]string, 0) //派出所以上的区域id
	for _, d := range domains {
		if d.DomainType == db.DomainTypeArea {
			targetAreaIds = append(targetAreaIds, d.ID)
		} else if d.DomainType == db.DomainTypeOrg {
			targetOrgIds = append(targetOrgIds, d.ID)
		} else {
			otherIds = append(otherIds, d.ID)
		}
	}

	if len(otherIds) > 0 {
		otherTargetOrgIds, err := GetTargetSubUnitsIds(otherIds, db.DomainTypeOrg)
		if err != nil {
			return nil, nil, err
		}
		for _, id := range otherTargetOrgIds {
			targetOrgIds = append(targetOrgIds, id)
		}
	}
	return
}

func GetTargetSubUnitsIds(ids []string, domainType db.DomainType) (targetIds []string, err error) {
	units, err := GetTargetSubUnits(ids, domainType)
	if err != nil {
		logger.Errorf("GetTargetSubUnits err:%v", err)
		return nil, err
	}
	if len(units) == 0 {
		logger.Warnf("GetTargetSubUnits empty result, ids: %v", ids)
		return nil, errors.New("GetTargetSubUnits empty result")
	}
	targetIds = make([]string, 0, len(units))
	for _, unit := range units {
		targetIds = append(targetIds, unit.ID)
	}
	return targetIds, nil

}

func GetTargetSubUnits(ids []string, domainType db.DomainType) (list []*db.DomainUnit, err error) {
	list = make([]*db.DomainUnit, 0)
	err = db.GetDB().Model(&db.DomainUnit{}).Where("parent_id in ?", ids).Find(&list).Error
	parentIds := make([]string, 0, len(list))
	for _, v := range list {
		if v.DomainType == domainType {
			parentIds = make([]string, 0, len(list))
			break
		} else if v.DomainType != db.DomainTypeArea {
			parentIds = append(parentIds, v.ID)
		}
	}

	if len(parentIds) > 0 {
		return GetTargetSubUnits(parentIds, domainType)
	}

	return list, nil

}

func GetUnitsMapByIds(ids []string) (m map[string]*db.DomainUnit, err error) {
	list := make([]*db.DomainUnit, 0)
	err = db.GetDB().Model(&db.DomainUnit{}).Where("id in ?", ids).Find(&list).Error
	if err != nil {
		return nil, err
	}
	m = make(map[string]*db.DomainUnit, len(list))
	for _, v := range list {
		m[v.ID] = v
	}
	return
}