nextcloud-recognize/lib/Db/FaceClusterMapper.php

<?php

/*
 * Copyright (c) 2022 The Recognize contributors.
 * This file is licensed under the Affero General Public License version 3 or later. See the COPYING file.
 */
declare(strict_types=1);
namespace OCA\Recognize\Db;

use OCP\AppFramework\Db\Entity;
use OCP\AppFramework\Db\QBMapper;
use OCP\DB\QueryBuilder\IQueryBuilder;
use OCP\IDBConnection;

/**
 * @template-extends QBMapper<FaceCluster>
 */
final class FaceClusterMapper extends QBMapper {
	public function __construct(IDBConnection $db) {
		parent::__construct($db, 'recognize_face_clusters', FaceCluster::class);
		$this->db = $db;
	}

	/**
	 * @throws \OCP\AppFramework\Db\DoesNotExistException
	 * @throws \OCP\AppFramework\Db\MultipleObjectsReturnedException
	 * @throws \OCP\DB\Exception
	 */
	public function find(int $id): FaceCluster {
		$qb = $this->db->getQueryBuilder();
		$qb->select(FaceCluster::$columns)
			->from('recognize_face_clusters')
			->where($qb->expr()->eq('id', $qb->createPositionalParameter($id)));
		return $this->findEntity($qb);
	}

	/**
	 * @throws \OCP\DB\Exception
	 * @return list<\OCA\Recognize\Db\FaceCluster>
	 */
	public function findByUserId(string $userId): array {
		$qb = $this->db->getQueryBuilder();
		$columns = array_map(fn ($c) => 'fc.'.$c, FaceCluster::$columns);
		$qb->selectDistinct($columns)
			->from('recognize_face_clusters', 'fc')
			->innerJoin('fc', 'recognize_face_detections', 'fd', 'fc.id = fd.cluster_id')
			->where($qb->expr()->eq('fc.user_id', $qb->createPositionalParameter($userId)));
		return $this->findEntities($qb);
	}


	/**
	 * @throws \OCP\AppFramework\Db\DoesNotExistException
	 * @throws \OCP\AppFramework\Db\MultipleObjectsReturnedException
	 * @throws \OCP\DB\Exception
	 */
	public function findByUserAndTitle(string $userId, string $title) : Entity {
		$qb = $this->db->getQueryBuilder();
		$qb->select(FaceCluster::$columns)
			->from('recognize_face_clusters')
			->where($qb->expr()->eq('user_id', $qb->createPositionalParameter($userId)));
		if (preg_match('/^[0-9]+$/', $title, $matches) != false) {
			$qb->andWhere($qb->expr()->orX(
				$qb->expr()->eq('title', $qb->createPositionalParameter($title)),
				$qb->expr()->andX(
					$qb->expr()->eq('title', $qb->createPositionalParameter('')),
					$qb->expr()->eq('id', $qb->createPositionalParameter((int) $title, IQueryBuilder::PARAM_INT))
				)
			));
		} else {
			$qb->andWhere($qb->expr()->eq('title', $qb->createPositionalParameter($title)));
		}
		return $this->findEntity($qb);
	}

	/**
	 * @param int $detectionId
	 * @throws \OCP\DB\Exception
	 * @return list<\OCA\Recognize\Db\FaceCluster>
	 */
	public function findByDetectionId(int $detectionId): array {
		$qb = $this->db->getQueryBuilder();
		$qb->select(array_map(function ($c) : string {
			return 'f.'.$c;
		}, FaceCluster::$columns))
			->from('recognize_face_clusters', 'f')
			->join('f', 'recognize_face_detections', 'd', 'f.id = d.cluster_id')
			->where($qb->expr()->eq('d.id', $qb->createPositionalParameter($detectionId)));
		return $this->findEntities($qb);
	}

	public function delete(Entity $entity): Entity {
		/** @var FaceCluster $entity */
		$qb = $this->db->getQueryBuilder();
		$qb->update('recognize_face_detections')
			->set('cluster_id', $qb->createPositionalParameter(-1, IQueryBuilder::PARAM_INT))
			->where($qb->expr()->eq('cluster_id', $qb->createPositionalParameter($entity->getId())));
		$qb->executeStatement();
		return parent::delete($entity);
	}

	/**
	 * @throws \OCP\DB\Exception
	 */
	public function deleteAll(): void {
		$qb = $this->db->getQueryBuilder();
		$qb->delete('recognize_face_clusters')
			->executeStatement();
	}
}