<?php
/*.
	require_module 'standard';
	require_module 'pcre';
	require_module 'hash';
	require_module 'bitrix_main';
	require_module 'bitrix_clouds_classes_storage_service';
	require_module 'bitrix_clouds_classes_storage_bucket';
.*/
IncludeModuleLangFile(__FILE__);

class CCloudStorageUpload
{
	protected /*.string.*/ $_filePath = '';
	protected /*.string.*/ $_ID = '';
	protected /*.CCloudStorageBucket.*/ $obBucket;
	protected /*.int.*/ $_max_retries = 3;
	protected /*.array[string]string.*/ $_cache = null;

	/**
	 * @param string $filePath
	 * @return void
	*/
	public function __construct($filePath)
	{
		$this->_filePath = $filePath;
		$this->_ID = '1' . mb_substr(md5($filePath), 1);
	}

	/**
	 * @return array[string]string
	*/
	public function GetArray()
	{
		global $DB;

		if (!isset($this->_cache))
		{
			$rs = $DB->Query("
				SELECT *
				FROM b_clouds_file_upload
				WHERE ID = '" . $this->_ID . "'
			");
			$this->_cache = $rs->Fetch();
		}

		return $this->_cache;
	}

	/**
	 * @return bool
	*/
	public function isStarted()
	{
		return is_array($this->GetArray());
	}

	/**
	 * @return void
	*/
	public function Delete()
	{
		global $DB;
		//TODO: clean up temp files in Clodo
		$DB->Query("DELETE FROM b_clouds_file_upload WHERE ID = '" . $this->_ID . "'");
		unset($this->_cache);
	}

	/**
	 * @return void
	*/
	public function DeleteOld()
	{
		global $DB;
		$DB->Query('DELETE FROM b_clouds_file_upload WHERE TIMESTAMP_X < ' . $DB->CharToDateFunction(ConvertTimeStamp(time() - 24 * 60 * 60)));
	}

	/**
	 * @param int $bucket_id
	 * @param float $fileSize
	 * @param string $ContentType
	 * @return bool
	*/
	public function Start($bucket_id, $fileSize, $ContentType = 'binary/octet-stream', $tmpFileName = false)
	{
		global $DB;
		global $APPLICATION;

		if (is_object($bucket_id))
		{
			$obBucket = $bucket_id;
		}
		else
		{
			$obBucket = new CCloudStorageBucket(intval($bucket_id));
		}

		if (!$obBucket->Init())
		{
			return false;
		}

		if (!$this->isStarted())
		{
			$arUploadInfo = /*.(array[string]string).*/[];
			$bStarted = $obBucket->getService()->InitiateMultipartUpload(
				$obBucket->getBucketArray(),
				$arUploadInfo,
				$this->_filePath,
				$fileSize,
				$ContentType
			);
			if (!$bStarted && $obBucket->RenewToken())
			{
				$bStarted = $obBucket->getService()->InitiateMultipartUpload(
					$obBucket->getBucketArray(),
					$arUploadInfo,
					$this->_filePath,
					$fileSize,
					$ContentType
				);
			}

			if ($bStarted)
			{
				$bAdded = $DB->Add('b_clouds_file_upload', [
					'ID' => $this->_ID,
					'~TIMESTAMP_X' => $DB->CurrentTimeFunction(),
					'FILE_PATH' => $this->_filePath,
					'FILE_SIZE' => $fileSize,
					'TMP_FILE' => $tmpFileName,
					'BUCKET_ID' => intval($obBucket->ID),
					'PART_SIZE' => $obBucket->getService()->GetMinUploadPartSize(),
					'PART_NO' => 0,
					'PART_FAIL_COUNTER' => 0,
					'NEXT_STEP' => serialize($arUploadInfo),
				], ['NEXT_STEP']);
				unset($this->_cache);

				return $bAdded !== false;
			}
			else
			{
				$error = $obBucket->getService()->formatError();
				if ($error)
				{
					$APPLICATION->ThrowException($error);
				}
				else
				{
					$APPLICATION->ThrowException(GetMessage('CLO_STORAGE_UPLOAD_ERROR', ['#errno#' => 6]));
				}
			}
		}

		return false;
	}

	/**
	 * @param string $data
	 * @return bool
	*/
	public function Next($data, $obBucket = null)
	{
		global $APPLICATION;

		if ($this->isStarted())
		{
			$ar = $this->GetArray();

			if ($obBucket === null)
			{
				$obBucket = new CCloudStorageBucket(intval($ar['BUCKET_ID']));
			}

			if (!$obBucket->Init())
			{
				$APPLICATION->ThrowException(GetMessage('CLO_STORAGE_UPLOAD_ERROR', ['#errno#' => 1]));
				return false;
			}

			$arUploadInfo = unserialize($ar['NEXT_STEP'], ['allowed_classes' => false]);
			$bSuccess = $obBucket->getService()->UploadPart(
				$obBucket->getBucketArray(),
				$arUploadInfo,
				$data
			);

			if (!$bSuccess)
			{
				$error = $obBucket->getService()->formatError();
				if ($error)
				{
					$APPLICATION->ThrowException($error);
				}
			}

			if (!$this->UpdateProgress($arUploadInfo, $bSuccess))
			{
				$APPLICATION->ThrowException(GetMessage('CLO_STORAGE_UPLOAD_ERROR', ['#errno#' => 2]));
				return false;
			}

			return $bSuccess;
		}

		return false;
	}

	/**
	 * @param string $data
	 * @param int $part_no
	 * @return bool
	*/
	public function Part($data, $part_no, $obBucket = null)
	{
		global $APPLICATION;

		if ($this->isStarted())
		{
			$ar = $this->GetArray();

			if ($obBucket === null)
			{
				$obBucket = new CCloudStorageBucket(intval($ar['BUCKET_ID']));
			}

			if (!$obBucket->Init())
			{
				$APPLICATION->ThrowException(GetMessage('CLO_STORAGE_UPLOAD_ERROR', ['#errno#' => 3]));
				return false;
			}

			$arUploadInfo = unserialize($ar['NEXT_STEP'], ['allowed_classes' => false]);
			$bSuccess = $obBucket->getService()->UploadPartNo(
				$obBucket->getBucketArray(),
				$arUploadInfo,
				$data,
				$part_no
			);

			if (!$bSuccess)
			{
				$error = $obBucket->getService()->formatError();
				if ($error)
				{
					$APPLICATION->ThrowException($error);
				}
			}

			if (!$this->UpdateProgress($arUploadInfo, $bSuccess))
			{
				$APPLICATION->ThrowException(GetMessage('CLO_STORAGE_UPLOAD_ERROR', ['#errno#' => 5]));
				return false;
			}

			return $bSuccess;
		}

		return false;
	}

	/**
	 * @return bool
	*/
	public function Finish($obBucket = null)
	{
		global $APPLICATION;

		if ($this->isStarted())
		{
			$ar = $this->GetArray();

			if ($obBucket === null)
			{
				$obBucket = new CCloudStorageBucket(intval($ar['BUCKET_ID']));
			}
			if (!$obBucket->Init())
			{
				return false;
			}

			$arUploadInfo = unserialize($ar['NEXT_STEP'], ['allowed_classes' => false]);
			$bSuccess = $obBucket->getService()->CompleteMultipartUpload(
				$obBucket->getBucketArray(),
				$arUploadInfo
			);

			if ($bSuccess)
			{
				$this->Delete();

				if ($obBucket->getQueueFlag())
				{
					CCloudFailover::queueCopy($obBucket, $this->_filePath);
				}

				foreach (GetModuleEvents('clouds', 'OnAfterCompleteMultipartUpload', true) as $arEvent)
				{
					ExecuteModuleEventEx($arEvent, [$obBucket, ['size' => $ar['FILE_SIZE']], $this->_filePath]);
				}
			}
			else
			{
				$error = $obBucket->getService()->formatError();
				if ($error)
				{
					$APPLICATION->ThrowException($error);
				}
			}

			return $bSuccess;
		}

		return false;
	}

	/**
	 * @return int
	*/
	public function GetPartCount()
	{
		$ar = $this->GetArray();

		if (is_array($ar))
		{
			return intval($ar['PART_NO']);
		}
		else
		{
			return 0;
		}
	}

	/**
	 * @return float
	*/
	public function GetPos()
	{
		$ar = $this->GetArray();

		if (is_array($ar))
		{
			return intval($ar['PART_NO']) * doubleval($ar['PART_SIZE']);
		}
		else
		{
			return 0;
		}
	}

	/**
	 * @return int
	*/
	public function getPartSize()
	{
		$ar = $this->GetArray();

		if (is_array($ar))
		{
			return intval($ar['PART_SIZE']);
		}
		else
		{
			return 0;
		}
	}

	/**
	 * @return bool
	*/
	public function hasRetries()
	{
		$ar = $this->GetArray();
		return is_array($ar) && (intval($ar['PART_FAIL_COUNTER']) < $this->_max_retries);
	}

	/**
	 * @return string
	*/
	public function getTempFileName()
	{
		$ar = $this->GetArray();
		if (is_array($ar))
		{
			return $ar['TMP_FILE'];
		}
		else
		{
			return '';
		}
	}

	/**
	 * @param array $arUploadInfo
	 * @param bool $bSuccess
	 * @return bool
	*/
	protected function UpdateProgress($arUploadInfo, $bSuccess)
	{
		global $DB;

		if ($bSuccess)
		{
			$arFields = [
				'NEXT_STEP' => serialize($arUploadInfo),
				'~PART_NO' => 'PART_NO + 1',
				'PART_FAIL_COUNTER' => 0,
			];
			$arBinds = [
				'NEXT_STEP' => $arFields['NEXT_STEP'],
			];
		}
		else
		{
			$arFields = [
				'~PART_FAIL_COUNTER' => 'PART_FAIL_COUNTER + 1',
			];
			$arBinds = [
			];
		}

		$strUpdate = $DB->PrepareUpdate('b_clouds_file_upload', $arFields);
		if ($strUpdate != '')
		{
			$strSql = 'UPDATE b_clouds_file_upload SET ' . $strUpdate . " WHERE ID = '" . $this->_ID . "'";
			if (!$DB->QueryBind($strSql, $arBinds))
			{
				unset($this->_cache);
				return false;
			}
		}

		unset($this->_cache);
		return true;
	}

	public static function CleanUp($ID = '')
	{
		$connection = \Bitrix\Main\Application::getConnection();
		$helper = $connection->getSqlHelper();
		$rs = false;

		if ($ID)
		{
			$rs = $connection->query("
				SELECT ID, BUCKET_ID, NEXT_STEP
				FROM b_clouds_file_upload
				WHERE ID = '" . $helper->forSql($ID) . "'
			");
		}
		else
		{
			$days = COption::GetOptionInt('clouds', 'multipart_upload_keep_days');
			if ($days > 0)
			{
				$rs = $connection->query('
					SELECT ID, BUCKET_ID, NEXT_STEP
					FROM b_clouds_file_upload
					WHERE TIMESTAMP_X < ' . $helper->addDaysToDateTime(-$days)
				);
			}
		}

		if ($rs)
		{
			while ($arBucket = $rs->fetch())
			{
				$obBucket = new CCloudStorageBucket(intval($arBucket['BUCKET_ID']));
				if ($obBucket->Init())
				{
					$arUploadInfo = unserialize($arBucket['NEXT_STEP'], ['allowed_classes' => false]);
					$service = $obBucket->getService();
					$service->CancelMultipartUpload($obBucket->getBucketArray(), $arUploadInfo);
				}
				$connection->query("DELETE FROM b_clouds_file_upload WHERE ID = '" . $helper->forSql($arBucket['ID']) . "'");
			}
		}
	}
}