<?php

namespace Bitrix\Scale;

/**
 * Class ServersData
 * @package Bitrix\Scale *
 */
class ServersData
{
	protected static $bxInfo = array(); //wrapper_ansible_conf -a bx_info -H hostname

	/**
	 * @param $hostname
	 * @return array Server's params
	 * @throws \Bitrix\Main\ArgumentNullException
	 */
	public static function getServer($hostname)
	{
		if($hostname == '')
			throw new \Bitrix\Main\ArgumentNullException("hostname");

		$result = array();
		$servers = self::getList();

		if(isset($servers[$hostname]))
			$result = $servers[$hostname];

		return $result;
	}

	/**
	 * @return array List of servers & their params
	 */
	public static function getList()
	{
		$result = array();
		$shellAdapter = new ShellAdapter();
		$execRes = $shellAdapter->syncExec("sudo -u root /opt/webdir/bin/wrapper_ansible_conf -o json");
		$serversData = $shellAdapter->getLastOutput();

		if($execRes)
		{
			$arData = json_decode($serversData, true);

			//mgmt server must be first
			if(isset($arData["params"]) && is_array($arData["params"]))
			{
				foreach($arData["params"] as $hostname => $server)
				{
					try
					{
						$server["BX_ENV_VER"] = static::getBxEnvVer($hostname);
						$bxInfo = static::getBxInfo($hostname);
						$server["BX_INFO"] = $bxInfo;

						if(isset($bxInfo["bx_last_password_change"]))
							$server["LAST_PASSWORD_CHANGE"] = $bxInfo["bx_last_password_change"];

						if(!$server["BX_ENV_VER"] || !Helper::checkBxEnvVersion($server["BX_ENV_VER"]))
							$server["BX_ENV_NEED_UPDATE"] = true;
						else
							$server["BX_ENV_NEED_UPDATE"] = false;

					}
					catch(ServerBxInfoException $e)
					{
						$server["BX_INFO_ERROR"] = $e->getMessage();
					}

					$result[$hostname] = $server;
				}

				\sortByColumn($result, array( "host_id" => array(SORT_NUMERIC, SORT_ASC)));
			}
		}

		return $result;
	}

	/**
	 * @param $hostname
	 * @return array Server roles
	 * @throws \Bitrix\Main\ArgumentNullException
	 */
	public static function getServerRoles($hostname)
	{
		if($hostname == '')
			throw new \Bitrix\Main\ArgumentNullException("hostname");

		$result = array();
		$server = static::getServer($hostname);

		if(isset($server["roles"]))
			$result = $server["roles"];

		$result["SERVER"] = array();

		return $result;
	}

	public static function getDbList($hostname)
	{
		if($hostname == '')
			throw new \Bitrix\Main\ArgumentNullException("hostname");

		$dbList = array();

		$action =  new Action("get_db_list", array(
			"START_COMMAND_TEMPLATE" => "sudo -u root /opt/webdir/bin/wrapper_ansible_conf -a dbs_list -H ".$hostname." -o json",
			"LOG_LEVEL" => Logger::LOG_LEVEL_DISABLE
			),
			"",
			array()
		);

		$action->start();
		$actRes = $action->getResult();

		if(isset($actRes["get_db_list"]["OUTPUT"]["DATA"]["params"]["dbs_list"][$hostname]["dbs_list"]))
			$dbList = $actRes["get_db_list"]["OUTPUT"]["DATA"]["params"]["dbs_list"][$hostname]["dbs_list"];

		if(is_array($dbList))
			$result = $dbList;
		else
			$result = array();

		return $result;
	}

	/**
	 * @param string $hostname Server hostname.
	 * @return array server Info.
	 * @throws \Bitrix\Main\ArgumentNullException
	 * @throws ServerBxInfoException
	 */
	protected static function getBxInfo($hostname)
	{
		if($hostname == '')
			throw new \Bitrix\Main\ArgumentNullException("hostname");

		$result = array();

		if(isset(static::$bxInfo[$hostname]))
		{
			$result = static::$bxInfo[$hostname];
		}
		else
		{
			$action =  new Action("get_bx_info", array(
					"START_COMMAND_TEMPLATE" => "sudo -u root /opt/webdir/bin/wrapper_ansible_conf -a bx_info -H ".$hostname." -o json",
					"LOG_LEVEL" => Logger::LOG_LEVEL_DISABLE
				),
				"",
				array()
			);

			$action->start();
			$actRes = $action->getResult();

			if(isset($actRes["get_bx_info"]["OUTPUT"]["DATA"]["params"]["bx_variables"][$hostname]))
			{
				$result = static::$bxInfo[$hostname] = $actRes["get_bx_info"]["OUTPUT"]["DATA"]["params"]["bx_variables"][$hostname];
			}
			elseif(isset($actRes["get_bx_info"]["RESULT"])
				&& $actRes["get_bx_info"]["RESULT"] = "ERROR"
					&& $actRes["get_bx_info"]["ERROR"] <> ''
			)
			{
				throw new \Bitrix\Scale\ServerBxInfoException($actRes["get_bx_info"]["ERROR"], $hostname);
			}
		}

		return $result;
	}

	/**
	 * @param string $hostname Server hostname.
	 * @return bool|string - Version of bitrix environment.
	 * @throws \Bitrix\Main\ArgumentNullException
	 */
	public static function getBxEnvVer($hostname)
	{
		if($hostname == '')
			throw new \Bitrix\Main\ArgumentNullException("hostname");

		$bxInfo = static::getBxInfo($hostname);

		if(isset($bxInfo["bx_version"])
			&& $bxInfo["bx_version"] != "0"
		)
		{
			$result = $bxInfo["bx_version"];
		}
		else
		{
			$result = false;
		}

		return $result;
	}

	public static function getGraphCategories($hostname)
	{
		$result = array();
		$roles = static::getServerRoles($hostname);

		foreach($roles as $roleId => $role)
			$result = array_merge($result, \Bitrix\Scale\RolesData::getGraphsCategories($roleId));

		return $result;
	}

	public static function getDbMasterHostname()
	{
		$servers = static::getList();

		foreach($servers as $hostname => $server)
			if(isset($server["roles"]["mysql"]["type"]) && $server["roles"]["mysql"]["type"] == "master")
				return $hostname;

		return false;
	}
}