Skip to content

Commit

Permalink
fix aftermath of #3716
Browse files Browse the repository at this point in the history
Signed-off-by: dartcafe <[email protected]>
  • Loading branch information
dartcafe committed Sep 24, 2024
1 parent 0198ce0 commit f1b2499
Show file tree
Hide file tree
Showing 4 changed files with 51 additions and 11 deletions.
13 changes: 9 additions & 4 deletions lib/Db/CommentMapper.php
Original file line number Diff line number Diff line change
Expand Up @@ -53,12 +53,17 @@ public function findByPoll(int $pollId, bool $getDeleted = false): array {
/**
* @return void
*/
public function renameUserId(string $userId, string $replacementName): void {
public function renameUserId(string $userId, string $replacementId, int|null $pollId = null): void {
$query = $this->db->getQueryBuilder();
$query->update($this->getTableName(), self::TABLE)
->set('user_id', $query->createNamedParameter($replacementName))
->where($query->expr()->eq('user_id', $query->createNamedParameter($userId)))
->executeStatement();
->set('user_id', $query->createNamedParameter($replacementId))
->where($query->expr()->eq('user_id', $query->createNamedParameter($userId)));

if ($pollId !== null) {
$query->andWhere($query->expr()->eq('poll_id', $query->createNamedParameter($pollId, IQueryBuilder::PARAM_INT)));
}

$query->executeStatement();
}

public function purgeDeletedComments(int $offset): void {
Expand Down
11 changes: 8 additions & 3 deletions lib/Db/OptionMapper.php
Original file line number Diff line number Diff line change
Expand Up @@ -128,12 +128,17 @@ public function findOptionsWithDuration(): array {
return $this->findEntities($qb);
}

public function renameUserId(string $userId, string $replacementName): void {
public function renameUserId(string $userId, string $replacementName, int|null $pollId = null): void {
$query = $this->db->getQueryBuilder();
$query->update($this->getTableName())
->set('owner', $query->createNamedParameter($replacementName))
->where($query->expr()->eq('owner', $query->createNamedParameter($userId)))
->executeStatement();
->where($query->expr()->eq('owner', $query->createNamedParameter($userId)));

if ($pollId !== null) {
$query->andWhere($query->expr()->eq('poll_id', $query->createNamedParameter($pollId, IQueryBuilder::PARAM_INT)));
}

$query->executeStatement();
}

public function purgeDeletedOptions(int $offset): void {
Expand Down
13 changes: 9 additions & 4 deletions lib/Db/VoteMapper.php
Original file line number Diff line number Diff line change
Expand Up @@ -115,12 +115,17 @@ public function deleteByPollAndUserId(int $pollId, string $userId): void {
$qb->executeStatement();
}

public function renameUserId(string $userId, string $replacementName): void {
public function renameUserId(string $userId, string $replacementId, int|null $pollId = null): void {
$query = $this->db->getQueryBuilder();
$query->update($this->getTableName())
->set('user_id', $query->createNamedParameter($replacementName))
->where($query->expr()->eq('user_id', $query->createNamedParameter($userId)))
->executeStatement();
->set('user_id', $query->createNamedParameter($replacementId))
->where($query->expr()->eq('user_id', $query->createNamedParameter($userId)));

if ($pollId !== null) {
$query->andWhere($query->expr()->eq('poll_id', $query->createNamedParameter($pollId, IQueryBuilder::PARAM_INT)));
}

$query->executeStatement();
}

public function fixVoteOptionText(int $pollId, int $optionId, string $searchOptionText, string $replaceOptionText): void {
Expand Down
25 changes: 25 additions & 0 deletions lib/Service/ShareService.php
Original file line number Diff line number Diff line change
Expand Up @@ -8,11 +8,14 @@

namespace OCA\Polls\Service;

use OCA\Polls\Db\CommentMapper;
use OCA\Polls\Db\OptionMapper;
use OCA\Polls\Db\Poll;
use OCA\Polls\Db\PollMapper;
use OCA\Polls\Db\Share;
use OCA\Polls\Db\ShareMapper;
use OCA\Polls\Db\UserMapper;
use OCA\Polls\Db\VoteMapper;
use OCA\Polls\Event\ShareChangedDisplayNameEvent;
use OCA\Polls\Event\ShareChangedEmailEvent;
use OCA\Polls\Event\ShareChangedLabelEvent;
Expand Down Expand Up @@ -64,6 +67,9 @@ public function __construct(
private PollMapper $pollMapper,
private UserMapper $userMapper,
private UserSession $userSession,
private VoteMapper $voteMapper,
private OptionMapper $optionMapper,
private CommentMapper $commentMapper,
) {
$this->shares = [];
}
Expand Down Expand Up @@ -304,6 +310,7 @@ private function convertPersonalPublicShareToExternalShare(
return;
}

$initialUserId = $this->share->getUserId();
$this->share->setUserId($userId ?? $this->generatePublicUserId());
$this->share->setDisplayName($displayName ?? $this->share->getDisplayName());
$this->share->setTimeZoneName($timeZone ?? $this->share->getTimeZoneName());
Expand All @@ -322,6 +329,24 @@ private function convertPersonalPublicShareToExternalShare(
// remove personal information from user id
$this->share->setUserId($this->generatePublicUserId());
$this->share = $this->shareMapper->update($this->share);
$this->convertDependingObjects($initialUserId, $this->share->getUserId(), $this->share->getPollId());
}

/**
* Rename userId as a follow up on renaming share's userId
* This methods covers the situation, where a userId of a share
* is changed, but there are already existing votes or options
* belonging to the renamed user
*
* This situation could occur, if a user already registered before the update to 7.2.4 and
* already voted, commented or suggested an option and reenters the pol lwith an email or contact share
*
* added in Polls 7.2.4 (can be removed later)
*/
private function convertDependingObjects(string $userId, string $replacementId, int $pollId) {
$this->voteMapper->renameUserId($userId, $replacementId, $pollId);
$this->optionMapper->renameUserId($userId, $replacementId, $pollId);
$this->commentMapper->renameUserId($userId, $replacementId, $pollId);
}

/**
Expand Down

0 comments on commit f1b2499

Please sign in to comment.