passman/lib/Db/FileMapper.php

<?php
/**
 * Nextcloud - passman
 *
 * This file is licensed under the Affero General Public License version 3 or
 * later. See the COPYING file.
 *
 * @author Sander Brand <brantje@gmail.com>
 * @copyright Sander Brand 2016
 */
namespace OCA\Passman\Db;

use OCA\Passman\Utility\Utils;
use OCP\IDBConnection;
use OCP\AppFramework\Db\Mapper;

class FileMapper extends Mapper {
	private $utils;

	public function __construct(IDBConnection $db, Utils $utils) {
		parent::__construct($db, 'passman_files');
		$this->utils = $utils;
	}


	/**
	 * @param $file_id
	 * @param null $user_id
	 * @return File
	 * @throws \OCP\AppFramework\Db\DoesNotExistException if not found
	 * @throws \OCP\AppFramework\Db\MultipleObjectsReturnedException if more than one result
	 */
	public function getFile($file_id, $user_id = null) {
		$sql = 'SELECT * FROM `*PREFIX*passman_files` ' .
			'WHERE `id` = ?';
		$params = [$file_id];
		if ($user_id !== null) {
			$sql .= ' and `user_id` = ? ';
			array_push($params, $user_id);
		}
		return $this->findEntity($sql, $params);
	}
	/**
	 * @param $file_id
	 * @param null $user_id
	 * @return File
	 * @throws \OCP\AppFramework\Db\DoesNotExistException if not found
	 * @throws \OCP\AppFramework\Db\MultipleObjectsReturnedException if more than one result
	 */
	public function getFileByGuid($file_guid, $user_id = null) {
		$sql = 'SELECT * FROM `*PREFIX*passman_files` ' .
			'WHERE `guid` = ?';
		$params = [$file_guid];
		if ($user_id !== null) {
			$sql .= ' and `user_id` = ? ';
			array_push($params, $user_id);
		}
		return $this->findEntity($sql, $params);
	}

	/**
	 * @param $file_raw
	 * @param $userId
	 * @return File
	 */
	public function create($file_raw, $userId) {
		$file = new File();
		$file->setGuid($this->utils->GUID());
		$file->setUserId($userId);
		$file->setFilename($file_raw['filename']);
		$file->setSize($file_raw['size']);
		$file->setCreated($this->utils->getTime());
		$file->setFileData($file_raw['file_data']);
		$file->setMimetype($file_raw['mimetype']);


		return $this->insert($file);
	}

	public function deleteFile($file_id, $userId) {
		$file = new File();
		$file->setId($file_id);
		$file->setUserId($userId);
		$this->delete($file);
	}

	public function updateFile(File $file) {
		return $this->update($file);
	}
}