package policy

import "github.com/pkg/errors"

// location should be country code
func (h *PolicyChecker) CheckWithPolicyLocation(countryCode string, clientIp string) (bool, error) {
	geoInfo, err := h.geoClient.QueryLite(clientIp)
	if err != nil {
		return false, errors.Errorf("geo ip[%s] query err: %+v", clientIp, err)
	}
	if geoInfo == nil {
		return false, errors.Errorf("geo ip[%s] is nil", clientIp)
	}

	return geoInfo.GetCountryCode() == countryCode, nil
}