mirror of
https://github.com/nextcloud/recognize.git
synced 2026-01-13 20:25:35 +00:00
110 lines
3.5 KiB
PHP
110 lines
3.5 KiB
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;
|
|
|
|
/**
|
|
* @psalm-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 {
|
|
$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();
|
|
}
|
|
}
|