nextcloud / richdocuments

use OCP\IDBConnection;

use OCP\Security\ISecureRandom;

class AssetMapper extends Mapper {

	/** @var int Limetime of a token is 10 minutes */

	const tokenLifeTime = 600;

	/** @var ISecureRandom */

	private $random;

	/** @var ITimeFactory */

	private $time;

	public function __construct(IDBConnection $db, ISecureRandom $random, ITimeFactory $timeFactory) {

		parent::__construct($db, 'richdocuments_assets', Asset::class);

		$this->random = $random;

		$this->time = $timeFactory;

	}

	/**

	 * @param $uid

	 * @param $fileid

	 * @return Asset

	 */

	public function newAsset($uid, $fileid) {

		$asset = new Asset();

		$asset->setUid($uid);

		$asset->setFileid($fileid);

		$asset->setTimestamp($this->time->getTime());

		$asset->setToken($this->random->generate(64, ISecureRandom::CHAR_UPPER . ISecureRandom::CHAR_LOWER . ISecureRandom::CHAR_DIGITS));

		$asset = $this->insert($asset);

		return $asset;

	}

	/**

	 * @param $token

	 * @return Asset

	 * @throws DoesNotExistException

	 */

	public function getAssetByToken($token) {

		$qb = $this->db->getQueryBuilder();

		$qb->select('*')

			->from('richdocuments_assets')

			->where($qb->expr()->eq('token', $qb->createNamedParameter($token)));

		$cursor = $qb->execute();

		$data = $cursor->fetch();

		$cursor->closeCursor();

		//There can only be one row since it is a unqiue field

		if ($data === false) {

			throw new DoesNotExistException('No asset for token found');

		}

		$asset = Asset::fromRow($data);

		// Check the token lifetime

		if ($asset->getTimestamp() + self::tokenLifeTime < $this->time->getTime()) {

			$this->delete($asset);

			throw new DoesNotExistException('No asset for token found');

		}

		return $asset;

	}

}

use OCP\Security\ISecureRandom;

use PhpParser\Node\Scalar\MagicConst\Dir;

class DirectMapper extends Mapper {

	/** @var int Limetime of a token is 10 minutes */

	const tokenLifeTime = 600;

	/** @var ISecureRandom */

	protected $random;

	/**@var ITimeFactory */

	protected $timeFactory;

	public function __construct(IDBConnection $db,

								ISecureRandom $random,

								ITimeFactory $timeFactory) {

		parent::__construct($db, 'richdocuments_direct', Direct::class);

		$this->random = $random;

		$this->timeFactory = $timeFactory;

	}

	/**

	 * @param string $uid

	 * @param int $fileid

	 * @param int $destination

	 * @return Direct

	 */

	public function newDirect($uid, $fileid, $destination = null) {

		$direct = new Direct();

		$direct->setUid($uid);

		$direct->setFileid($fileid);

		$direct->setToken($this->random->generate(64, ISecureRandom::CHAR_DIGITS . ISecureRandom::CHAR_LOWER . ISecureRandom::CHAR_UPPER));

		$direct->setTimestamp($this->timeFactory->getTime());

		$direct->setTemplateDestination($destination);

		$direct = $this->insert($direct);

		return $direct;

	}

	/**

	 * @param string $token

	 * @return Direct

	 */

	public function getByToken($token) {

		$qb = $this->db->getQueryBuilder();

		$qb->select('*')

			->from('richdocuments_direct')

			->where($qb->expr()->eq('token', $qb->createNamedParameter($token)));

		$cursor = $qb->execute();

		$row = $cursor->fetch();

		$cursor->closeCursor();

		//There can only be one as the token is unique

		if ($row === false) {

			throw new DoesNotExistException('Could not find token.');

		}

		$direct = Direct::fromRow($row);

		if (($direct->getTimestamp() + self::tokenLifeTime) < $this->timeFactory->getTime()) {

			$this->delete($direct);

			throw new DoesNotExistException('Could not find token.');

		}

		return $direct;

	}

}