<?php
namespace Bitrix\Sale;

/**
 * Class SalesZone
 * @package Bitrix\Sale *
 */
class SalesZone
{
	const CONN_ENTITY_NAME = 		'Bitrix\Sale\Location\SiteLocation';
	const LOCATION_ENTITY_NAME = 	'Bitrix\Sale\Location\Location';

	static $zoneCache = array(); // functionality is called hell number of times outside, so a little cache is provided

	/**
	 * @param string $lang - language Id
	 * @return array - list of all regions
	 */
	public static function getAllRegions($lang)
	{
		static $result = null;

		if($result === null)
		{
			$result = array();
			$dbRegionList = \CSaleLocation::GetRegionList(array(), array(), $lang);

			while ($arRegion = $dbRegionList->fetch())
				$result[$arRegion["ID"]] = $arRegion["NAME_LANG"];
		}

		return $result;
	}

	/**
	 * @param string $lang - language Id
	 * @return array - list of all cities
	 */
	public static function getAllCities($lang)
	{
		static $result = null;

		if($result === null)
		{
			$result = array();
			$dbCityList = \CSaleLocation::GetCityList(array(), array(), $lang);
			while($arCity = $dbCityList->fetch())
				$result[$arCity["ID"]] = $arCity["NAME_LANG"];
		}

		return $result;
	}

	/**
	 * Checks if country Id is in list of sales zone countries Ids.
	 * @param int $countryId
	 * @param string $siteId
	 * @return bool
	 */
	public static function checkCountryId($countryId, $siteId)
	{
		if(!strlen($siteId))
			return false;

		$cIds = static::getCountriesIds($siteId);
		return !$cIds || in_array($countryId, $cIds) || in_array("", $cIds);
	}

	/**
	 * Checks if regionId is in list of sales zone regions Ids
	 * @param int $regionId
	 * @param string $siteId
	 * @return bool
	 */
	public static function checkRegionId($regionId, $siteId)
	{
		if(!strlen($siteId))
			return false;

		$rIds = static::getRegionsIds($siteId);
		return !$rIds || in_array($regionId, $rIds) || in_array("", $rIds);
	}

	/**
	 * Checks if citiy Id is in list of sales zone cities Ids
	 * @param int $cityId
	 * @param string $siteId
	 * @return bool
	 */
	public static function checkCityId($cityId, $siteId)
	{
		if(!strlen($siteId))
			return false;

		$cIds = static::getCitiesIds($siteId);
		return !$cIds || in_array($cityId, $cIds) || in_array("", $cIds);
	}

	/**
	 * Checks if location id is in sales zone
	 * @param int $locationId
	 * @param string $siteId
	 * @return bool
	 */
	public static function checkLocationId($locationId, $siteId)
	{
		if(\CSaleLocation::isLocationProMigrated())
		{
			if(!intval($locationId) || !strlen($siteId))
				return false;

			return Location\SiteLocationTable::checkConnectionExists($siteId, $locationId);
		}
		else
		{

			$result = false;

			$arLocation = \CSaleLocation::GetByID($locationId);

			if(static::checkCountryId($arLocation["COUNTRY_ID"], $siteId)
				&& static::checkRegionId($arLocation["REGION_ID"], $siteId)
				&& static::checkCityId($arLocation["CITY_ID"], $siteId)
			)
			{
				$result = true;
			}

			return $result;

		}
	}

	private static function checkLocationIsInLinkedPart($locationId, $siteId)
	{
		$types = \CSaleLocation::getTypes();
		$class = self::CONN_ENTITY_NAME.'Table';

		if(!$class::checkLinkUsageAny($siteId))
			return true;

		if((string) $locationId == '')
			return false;

		$node = \Bitrix\Sale\Location\LocationTable::getList(array(
			'filter' => array('=ID' => $locationId),
			'select' => array('ID', 'LEFT_MARGIN', 'RIGHT_MARGIN')
		))->fetch();
		if(!is_array($node))
			return false;

		$stat = $class::getLinkStatusForMultipleNodes(array($node), $siteId);

		return $stat[$node['ID']] !== $class::LSTAT_IN_NOT_CONNECTED_BRANCH;
	}

	public static function setSelectedIds($siteId, $ids)
	{
		static::$zoneCache[$siteId] = $ids;
	}

	public static function getSelectedIds($siteId)
	{
		$typesAll = \CSaleLocation::getTypes();

		$types = array();
		if(isset($typesAll['COUNTRY']))
			$types[] = "'".intval($typesAll['COUNTRY'])."'";
		if(isset($typesAll['REGION']))
			$types[] = "'".intval($typesAll['REGION'])."'";
		if(isset($typesAll['CITY']))
			$types[] = "'".intval($typesAll['CITY'])."'";

		$typesAll = array_flip($typesAll);

		if((string) $siteId != '' && \Bitrix\Sale\Location\SiteLocationTable::checkLinkUsageAny($siteId) && !empty($types))
		{
			$result = array();

			$sql = \Bitrix\Sale\Location\SiteLocationTable::getConnectedLocationsSql(
				$siteId,
				array('select' => array('ID', 'LEFT_MARGIN', 'RIGHT_MARGIN')),
				array('GET_LINKED_THROUGH_GROUPS' => true)
			);

			if((string) $sql != '')
			{
				$res = $GLOBALS['DB']->query("

					select SL.ID, SL.TYPE_ID from b_sale_location SL 
						inner join (
						".$sql."
						) as TT on 
							(
								(
									(
										SL.LEFT_MARGIN >= TT.LEFT_MARGIN 
										and 
										SL.RIGHT_MARGIN <= TT.RIGHT_MARGIN
									)
									or
									(
										SL.LEFT_MARGIN <= TT.LEFT_MARGIN 
										and 
										SL.RIGHT_MARGIN >= TT.RIGHT_MARGIN
									)
								)
								and
								SL.TYPE_ID in (".implode(', ', $types).")
							)

					group by SL.ID
				");

				while($item = $res->fetch())
				{
					$typeId = $item['TYPE_ID'];
					unset($item['TYPE_ID']);

					$result[$typesAll[$typeId]][$item['ID']] = $item['ID'];
				}

				// special case: when all types are actually selected, an empty string ('') SHOULD be present among $index[$siteId][$type]

				$res = Location\LocationTable::getList(array(
					'filter' => array(
						'TYPE_ID' => $types
					),
					'runtime' => array(
						'CNT' => array(
							'data_type' => 'integer',
							'expression' => array('COUNT(*)')
						)
					),
					'select' => array(
						'CNT',
						'TYPE_ID'
					)
				));
				while($item = $res->fetch())
				{
					if(intval($item['TYPE_ID']))
					{
						$typeCode = $typesAll[$item['TYPE_ID']];
						if(isset($result[$typeCode]) && $item['CNT'] == count($result[$typeCode]))
						{
							$result[$typeCode][] = '';
						}
					}
				}
			}

			return $result;
		}
		else
		{
			return array(
				'COUNTRY' => array(''),
				'REGION' => array(''),
				'CITY' => array(''),
			); // means "all"
		}
	}

	// returns a list of IDs of locations that are linked with $siteId and have type of $type
	private static function getSelectedTypeIds($type, $siteId)
	{
		$index =& static::$zoneCache;

		if(!isset($index[$siteId]))
		{
			$result = array();

			$index[$siteId] = static::getSelectedIds($siteId);
		}

		return $index[$siteId][$type];
	}

	/**
	 * @param string $siteId
	 * @return array - sales zones cities Ids
	 */
	public static function getCitiesIds($siteId)
	{
		if(\CSaleLocation::isLocationProMigrated())
			return self::getSelectedTypeIds('CITY', $siteId);

		return explode(":" , \COption::GetOptionString('sale', 'sales_zone_cities', '', $siteId));
	}

	/**
	 * @param string $siteId
	 * @return array - sales zones regions Ids
	 */
	public static function getRegionsIds($siteId)
	{
		if(\CSaleLocation::isLocationProMigrated())
			return self::getSelectedTypeIds('REGION', $siteId);

		return explode(":" , \COption::GetOptionString('sale', 'sales_zone_regions', '', $siteId));
	}

	/**
	 * @param string $siteId
	 * @return array - sales zones countries Ids
	 */
	public static function getCountriesIds($siteId)
	{
		if(\CSaleLocation::isLocationProMigrated())
			return self::getSelectedTypeIds('COUNTRY', $siteId);

		return explode(":" , \COption::GetOptionString('sale', 'sales_zone_countries', '', $siteId));
	}

	/**
	 * A very important function. Here we decide what locations we need to take, 
	 * making a descision based on $_REQUEST from sales zone selector.
	 * 
	 * Then we normalize the selection and store to database.
	 * 
	 * Also this function is used in data migrator.
	 */
	public static function saveSelectedTypes($typeList, $siteId)
	{
		$types = \CSaleLocation::getTypes();

		$locations = array(
			Location\Connector::DB_LOCATION_FLAG => array(),
			Location\Connector::DB_GROUP_FLAG => array()
		);
		if(is_array($typeList['COUNTRY']) && !empty($typeList['COUNTRY']))
			$typeList['COUNTRY'] = array_flip($typeList['COUNTRY']);

		if(is_array($typeList['REGION']) && !empty($typeList['REGION']))
			$typeList['REGION'] = array_flip($typeList['REGION']);

		if(is_array($typeList['CITY']) && !empty($typeList['CITY']))
			$typeList['CITY'] = array_flip($typeList['CITY']);

		$allCountries = isset($typeList['COUNTRY']['']);
		$allRegions = isset($typeList['REGION']['']);
		$allCities = isset($typeList['CITY']['']);

		// no countries
		$noCountry = isset($typeList['COUNTRY']['NULL']);
		$noRegion = isset($typeList['REGION']['NULL']);

		// make up list of ids
		$res = Location\LocationTable::getList(array('select' => array(
			'ID', 
			'COUNTRY_ID', 
			'REGION_ID', 
			'CITY_ID',
			'TYPE_ID',
			//'LNAME' => 'NAME.NAME'
		), 'filter' => array(
			//'=NAME.LANGUAGE_ID' => LANGUAGE_ID
		)));
		while($item = $res->fetch())
		{
			$id = $item['ID'];
			$countryId = intval($item['COUNTRY_ID']);
			$regionId = intval($item['REGION_ID']);
			$cityId = intval($item['CITY_ID']);
			$typeId = intval($item['TYPE_ID']);

			$take = false;
			$countryTaken = false;
			$regionTaken = false;

			if($typeId == $types['COUNTRY']) // it is a country
			{
				if($allCountries // we take all countries
					|| // or..
					isset($typeList['COUNTRY'][$countryId]) // we manually selected this country
				)
				{
					$take = true;
					$countryTaken = true;
				}
			}

			if($typeId == $types['REGION']) // it is a region
			{
				if((
						$allRegions // we take all regions (of selected countries)
						&& // and
						$countryTaken // country is selected already
					) 
					|| // or ..
					isset($typeList['REGION'][$regionId]) // we manually selected this region
					|| // or ..
					($noCountry && !$countryId) // we also accept regions without countries, and this region actually dont have a country
				)
				{
					$take = true;
					$regionTaken = true;
				}
			}

			if($typeId == $types['CITY']) // it is a city
			{
				if((
						$allCities // we take all cities (of selected regions of selected countries)
						&& // and
						$regionTaken // region is selected already
					) 
					|| // or..
					isset($typeList['REGION'][$regionId]) // we manually selected this city
					|| // or ..
					($noRegion && !$regionId) // we also accept cities without regions, and this city actually dont have a region
					// it seems that cities without a country is not supported with the current logic
				)
				{
					$take = true;
				}
			}

			if(isset($typeList['CITY'][$cityId]) && $typeId == $types['CITY']) // this is a city and it is in list - take it
				$take = true;

			if($take)
				$locations[Location\Connector::DB_LOCATION_FLAG][$id] = true;
		}

		// normalize
		$class = self::CONN_ENTITY_NAME.'Table';
		$locations[Location\Connector::DB_LOCATION_FLAG] = array_keys($locations[Location\Connector::DB_LOCATION_FLAG]);

		$locations[Location\Connector::DB_LOCATION_FLAG] = $class::normalizeLocationList($locations[Location\Connector::DB_LOCATION_FLAG]);

		// store to database
		$class::resetMultipleForOwner(strlen($siteId) ? $siteId : $class::ALL_SITES, $locations);
	}

	/**
	 * Returns filter for using in queries such as in bitrix/modules/sale/install/components/bitrix/sale.ajax.locations/search.php
	 * @param string $object (city|region|country)
	 * @param string $siteId
	 * @return array
	 */
	public static function makeSearchFilter($object, $siteId)
	{
		$result = array();

		$countries = static::getCountriesIds($siteId);
		$regions = static::getRegionsIds($siteId);
		$cities = static::getCitiesIds($siteId);

		if(!in_array("", $cities) && $object == "city")
			$result = array("CITY_ID" => $cities);
		elseif(!in_array("", $regions) && ($object == "city"  || $object == "region"))
			$result = array("REGION_ID" => $regions);
		elseif(!in_array("", $countries))
			$result = array("COUNTRY_ID" => $countries);

		return $result;
	}

	/**
	 * @param array $countriesIds
	 * @param string $lang
	 * @return array - regions from sales zone
	 */
	public static function getRegions($countriesIds = array(), $lang = LANGUAGE_ID)
	{

		$regions = array();
		$regionsList = static::getAllRegions($lang);
		$getCountryNull = in_array("NULL", $countriesIds) ? true : false;
		$filter = in_array("", $countriesIds) ? array() : array(($getCountryNull ? "+" : "")."COUNTRY_ID" => $countriesIds);

		$dbLocationsList = \CSaleLocation::GetList(
			array("SORT"=>"ASC", "REGION_NAME_LANG"=>"ASC"),
			$filter,
			array("REGION_ID", "COUNTRY_ID")
		);

		while($arRegion = $dbLocationsList->GetNext())
		{
			if(strlen($arRegion["REGION_ID"]) > 0 && $arRegion["REGION_ID"] != "0")
				$regions[$arRegion["REGION_ID"]] = $regionsList[$arRegion["REGION_ID"]];
		}

		return $regions;
	}

	/**
	 * @param array $countriesIds
	 * @param array $regionsIds
	 * @param string $lang
	 * @return array cities list from sales zone
	 */
	public static function getCities($countriesIds = array(), $regionsIds = array(), $lang )
	{
		$cities = array();
		$citiesList = static::getAllCities($lang);
		$getRegionNull = in_array("NULL", $regionsIds) ? true : false;
		$getRegionAll = in_array("", $regionsIds) ? true : false;
		$getCountryNull = in_array("NULL", $countriesIds) ? true : false;
		$getCountryAll = in_array("", $countriesIds) ? true : false;

		$filter = in_array("", $regionsIds) ? array() : array(($getRegionNull ? "+" : "")."REGION_ID" => $regionsIds);

		foreach($countriesIds as $countryId)
		{
			if(($getRegionNull || $getRegionAll) && !$getCountryAll)
				$filter[($getCountryNull ? "+" : "")."COUNTRY_ID"] = $countryId;

			$dbLocationsList = \CSaleLocation::GetList(
				array("SORT"=>"ASC", "CITY_NAME_LANG"=>"ASC"),
				$filter,
				array("CITY_ID")
			);

			while($arCity = $dbLocationsList->GetNext())
				if(strlen($arCity["CITY_ID"]) > 0)
					$cities[$arCity["CITY_ID"]] =  $citiesList[$arCity["CITY_ID"]];
		}

		return  $cities;
	}
}