undo-ransomware / ransomware_detection

lib/Db/RecoveredFileOperationMapper.php

<?php

/**
 * @copyright Copyright (c) 2020 Matthias Held <matthias.held@uni-konstanz.de>
 * @author Matthias Held <matthias.held@uni-konstanz.de>
 * @license GNU AGPL version 3 or any later version
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as
 * published by the Free Software Foundation, either version 3 of the
 * License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <https://www.gnu.org/licenses/>.
 */

namespace OCA\RansomwareDetection\Db;

use OCP\IDBConnection;

The type OCP\IDBConnection was not found. Maybe you did not declare it correctly or list all dependencies?

use OCP\AppFramework\Db\Mapper;

The type OCP\AppFramework\Db\Mapper was not found. Maybe you did not declare it correctly or list all dependencies?

class RecoveredFileOperationMapper extends Mapper
{
    /**
     * @param IDBConnection $db
     */
    public function __construct(
        IDBConnection $db
    ) {
        parent::__construct($db, 'rd_recovery');
    }

    /**
     * Find one by id.
     *
     * @throws \OCP\AppFramework\Db\DoesNotExistException            if not found
     * @throws \OCP\AppFramework\Db\MultipleObjectsReturnedException if more than one result
     *
     * @param int $id
     *
     * @return Entity

The type OCA\RansomwareDetection\Db\Entity was not found. Maybe you did not declare it correctly or list all dependencies?

     */
    public function find($id, $userId)
    {
        $sql = 'SELECT * FROM `*PREFIX*rd_recovery` '.
            'WHERE `id` = ? AND `user_id` = ?';

        return $this->findEntity($sql, [$id, $userId]);
    }

    /**
     * Find one by file name.
     *
     * @throws \OCP\AppFramework\Db\DoesNotExistException            if not found
     * @throws \OCP\AppFramework\Db\MultipleObjectsReturnedException if more than one result
     *
     * @param string $name
     *
     * @return Entity
     */
    public function findOneByFileName($name, $userId)
    {
        $sql = 'SELECT * FROM `*PREFIX*rd_recovery` '.
            'WHERE `original_name` = ? AND `user_id` = ?';

        return $this->findEntity($sql, [$name, $userId]);
    }

    /**
     * Find the one with the highest id.
     *
     * @throws \OCP\AppFramework\Db\DoesNotExistException            if not found
     * @throws \OCP\AppFramework\Db\MultipleObjectsReturnedException if more than one result
     *
     * @return Entity
     */
    public function findOneWithHighestId($userId)
    {
        $sql = 'SELECT * FROM `*PREFIX*rd_recovery` WHERE `user_id` = ?'.
            'ORDER BY id DESC LIMIT 1';

        return $this->findEntity($sql, [$userId]);
    }

    /**
     * Find all.
     *
     * @param int $limit
     * @param int $offset
     *
     * @return array
     */
    public function findAll(array $params = [], $limit = null, $offset = null)
    {
        $sql = 'SELECT * FROM `*PREFIX*rd_recovery` WHERE `user_id` = ?';

        return $this->findEntities($sql, $params, $limit, $offset);
    }

    /**
     * Find a sequence by its id.
     *
     * @param array $params
     * @param int   $limit
     * @param int   $offset
     *
     * @return array
     */
    public function findSequenceById(array $params = [], $limit = null, $offset = null)
    {
        $sql = 'SELECT * FROM `*PREFIX*rd_recovery` WHERE `sequence` = ? AND `user_id` = ?';

        return $this->findEntities($sql, $params, $limit, $offset);
    }

    /**
     * Delete entity by id.
     *
     * @param int $id
     */
    public function deleteById($id, $userId)
    {
        $sql = 'DELETE FROM `*PREFIX*rd_recovery` WHERE `id` = ? AND `user_id` = ?';
        $stmt = $this->execute($sql, [$id, $userId]);
        $stmt->closeCursor();
    }

    /**
     * Deletes a sequence of file operations.
     *
     * @param int $sequence
     */
    public function deleteSequenceById($sequence, $userId)
    {
        $sql = 'DELETE FROM `*PREFIX*rd_recovery` WHERE `sequence` = ? AND `user_id` = ?';
        $stmt = $this->execute($sql, [$sequence, $userId]);
        $stmt->closeCursor();
    }

    /**
     * Delete all entries before $timestamp.
     *
     * @param int $timestamp
     */
    public function deleteFileOperationsBefore($timestamp)
    {
        $sql = 'DELETE FROM `*PREFIX*rd_recovery` WHERE `timestamp` < ?';
        $stmt = $this->execute($sql, [$timestamp]);
        $stmt->closeCursor();
    }
}