OCA\Files_Antivirus\RequestHelper

Complexity

Total Complexity

12

Size/Duplication

Total Lines	59
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	0

Test Coverage

Coverage

75%

Importance

Changes	2
Bugs	0	Features	0

3 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	3	1
A	isScriptName()	0	4	1
D	getUploadSize()	0	28	10

<?php
/**
 * ownCloud - files_antivirus
 *
 * This file is licensed under the Affero General Public License version 3 or
 * later. See the COPYING file.
 *
 * @author Viktar Dubiniuk <dubiniuk@owncloud.com>
 *
 * @copyright Viktar Dubiniuk 2017-2018
 * @license AGPL-3.0
 */

namespace OCA\Files_Antivirus;

use \OCP\IRequest;

class RequestHelper {

	/**
	 * @var  IRequest
	 */
	private $request;

	public function __construct(IRequest $request) {
		$this->request = $request;
	}

	/**
	 * Get current upload size
	 * returns null for chunks and when there is no upload
	 *
	 * @param string $path
	 *
	 * @return int|null
	 */
	public function getUploadSize($path) {
		$uploadSize = null;

		$requestMethod = $this->request->getMethod();
		$isRemoteScript = $this->isScriptName('remote.php');
		$isPublicScript = $this->isScriptName('public.php');
		// Are we uploading anything?
		if (\in_array($requestMethod, ['MOVE', 'PUT']) && $isRemoteScript) {
			// v1 && v2 Chunks are not scanned
			if ($requestMethod === 'PUT' &&  \strpos($path, 'uploads/') === 0) {
				return null;
			}

			if (\OC_FileChunking::isWebdavChunk() && \strpos($path, 'cache/') === 0) {
				return null;
			}

			if ($requestMethod === 'PUT') {
				$uploadSize = (int)$this->request->getHeader('CONTENT_LENGTH');
			} else {
				$uploadSize = (int)$this->request->getHeader('OC_TOTAL_LENGTH');
			}
		} else if ($requestMethod === 'PUT' && $isPublicScript) {
			$uploadSize = (int)$this->request->getHeader('CONTENT_LENGTH');
		}

		return $uploadSize;
	}

	/**
	 *
	 * @param string $string
	 *
	 * @return bool
	 */
	public function isScriptName($string) {
		$pattern = \sprintf('|/%s|', \preg_quote($string));
		return \preg_match($pattern, $this->request->getScriptName()) === 1;
	}
}