hedera-web/rest/dms/add.php

<?php

use Vn\Lib;

/**
 * Adds a document to the Document Management System.
 */
class Add extends Vn\Web\JsonRequest
{
	function run ($db)
	{
		// XXX: Uncomment only to test the script
		//$_REQUEST['description'] = 'description';
		
		$description = empty ($_REQUEST['description']) ?
			NULL : $_REQUEST['description']; 
		
		$baseDir = _DATA_DIR .'/'. $this->app->getName ();
		$docsDir = "$baseDir/dms";
		$tempDir = "$baseDir/.dms";

		$digXDir = 3;
		$zerosDir = '';
		
		for ($i = 0; $i < $digXDir; $i++)
			$zerosDir .= '0';
		
		// Checks document restrictions
		
		if (empty ($_FILES['doc']['name']))
			throw new Lib\UserException ('File not choosed');

		$maxSize = $db->getValue ('SELECT max_size FROM dms_config');

		if ($_FILES['doc']['size'] > $maxSize * 1048576)
			throw new Lib\UserException (sprintf ('File size exceeds size: %d MB', $maxSize)));

		try {
			// Registers the document in the database

			$db->query ('START TRANSACTION');

			$db->query ('INSERT INTO dms_document SET description = #', [$description]);	
			$docId = (string) $db->getValue ('SELECT LAST_INSERT_ID()');
		
			$len = strlen ($docId);
			$neededLevels = ceil ($len / $digXDir) - 1;

			$dirLevels = $db->getValue (
				'SELECT dir_levels FROM dms_config LOCK IN SHARE MODE');

			if ($dirLevels > $neededLevels)
				$neededLevels = $dirLevels;

			// Reorganizes the file repository if necessary

			if ($dirLevels < $neededLevels)
				$dirLevels = $db->getValue (
					'SELECT dir_levels FROM dms_config FOR UPDATE');

			if ($dirLevels < $neededLevels)
			{
				if (is_dir ($docsDir))
				{
					$dif = ($neededLevels - $dirLevels) - 1;
					$newDir = $docsDir;
				
					for ($i = 0; $i < $dif; $i++)
						$newDir .= "/$zerosDir";
		
					$success = rename ($docsDir, $tempDir)
						&& mkdir ($newDir, 0770, TRUE)
						&& rename ($tempDir, "$newDir/$zerosDir");

					if (!$success)
						throw new Exception ('Error while reorganizing directory tree');
				}

				$curLevels = $db->query ('UPDATE dms_config SET dir_levels = #',
					[$neededLevels]);
			}

			// Saves the document to the repository

			$padLen = ($neededLevels + 1) * $digXDir;
			$paddedId = str_pad ($docId, $padLen, '0', STR_PAD_LEFT);
		
			$saveDir = $docsDir;
		
			for ($i = 0; $i < $neededLevels; $i++)
				$saveDir .= '/'. substr ($paddedId, $i * $digXDir, $digXDir);

			if (!file_exists ($saveDir))
				mkdir ($saveDir, 0770, TRUE);
			
			$savePath = "$saveDir/". substr ($paddedId, -$digXDir);
	
			move_uploaded_file ($_FILES['doc']['tmp_name'], $savePath);

			$db->query ('COMMIT');

			return $docId;
		}
		catch (Exception $e)
		{
			$db->query ('ROLLBACK');
			throw $e;
		}
	}
}