GNUsocial/gnu-social
4
0
Fork 1
Code Issues Releases Wiki Activity

[ENTITY][Note] Relive isVisibleTo method

Browse Source
This commit is contained in:
Diogo Peralta Cordeiro 2021-12-26 19:50:21 +00:00
parent 78cc9c4659
commit 0c421116a6
Signed by: diogo
GPG Key ID: 18D2D35001FBFAB0
2 changed files with 36 additions and 46 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
components/Feed/Util/FeedController.php
View File

@ -34,10 +34,10 @@ namespace Component\Feed\Util;
use App\Core\Controller; use App\Core\Controller;
use App\Core\Event; use App\Core\Event;
use App\Core\Log;
use App\Core\VisibilityScope;
use App\Entity\Actor; use App\Entity\Actor;
use App\Entity\Note;
use App\Util\Common; use App\Util\Common;
use Functional as F;
abstract class FeedController extends Controller abstract class FeedController extends Controller
{ {
@ -62,37 +62,6 @@ abstract class FeedController extends Controller
private static function enforceScope(array &$notes, ?Actor $actor): void private static function enforceScope(array &$notes, ?Actor $actor): void
{ {
$filtered_notes = []; $notes = F\select($notes, fn (Note $n) => $n->isVisibleTo($actor));
foreach ($notes as $note) {
switch ($note->getScope()) {
case VisibilityScope::LOCAL: // The controller handles it if private
case VisibilityScope::EVERYWHERE:
$filtered_notes[] = $note;
break;
case VisibilityScope::ADDRESSEE:
// If the actor is logged in and
if (!\is_null($actor)
&& (
// Is either the author Or
$note->getActorId() == $actor->getId()
// one of the targets
|| \in_array($actor->getId(), $note->getNotificationTargetIds())
)) {
$filtered_notes[] = $note;
}
break;
case VisibilityScope::GROUP:
// Only for the group to see
break;
case VisibilityScope::COLLECTION:
case VisibilityScope::MESSAGE:
// Only for the collection to see (they will only find it in their notifications)
break;
default:
Log::warning("Unknown scope found: {$note->getScope()}.");
}
}
// Replace notes with filtered ones I/O
$notes = $filtered_notes;
} }
} }

45
src/Entity/Note.php
View File

@ -27,6 +27,7 @@ use App\Core\Cache;
use App\Core\DB\DB; use App\Core\DB\DB;
use App\Core\Entity; use App\Core\Entity;
use App\Core\Event; use App\Core\Event;
use App\Core\Log;
use App\Core\Router\Router; use App\Core\Router\Router;
use App\Core\VisibilityScope; use App\Core\VisibilityScope;
use App\Util\Formatting; use App\Util\Formatting;
@ -358,24 +359,44 @@ class Note extends Entity
/** /**
* Whether this note is visible to the given actor * Whether this note is visible to the given actor
*/ */
public function isVisibleTo(null|Actor|LocalUser $a): bool public function isVisibleTo(null|Actor|LocalUser $actor): bool
{ {
// TODO cache this // TODO: cache this
$scope = VisibilityScope::create($this->scope); switch ($this->getScope()) {
return $scope->public case VisibilityScope::LOCAL: // The controller handles it if private
|| (!\is_null($a) && ( case VisibilityScope::EVERYWHERE:
($scope->subscriber && 0 != DB::count('subscription', ['subscriber' => $a->getId(), 'subscribed' => $this->actor_id])) return true;
|| ($scope->addressee && 0 != DB::count('notification', ['activity_id' => $this->id, 'actor_id' => $a->getId()])) case VisibilityScope::ADDRESSEE:
|| ($scope->group && [] != DB::dql( // If the actor is logged in and
<<<'EOF' if (!\is_null($actor)
&& (
// Is either the author Or
$this->getActorId() == $actor->getId()
// one of the targets
|| \in_array($actor->getId(), $this->getNotificationTargetIds())
)) {
return true;
}
return false;
case VisibilityScope::GROUP:
// Only for the group to see
return DB::dql(
<<<'EOF'
select m from group_member m select m from group_member m
join group_inbox i with m.group_id = i.group_id join group_inbox i with m.group_id = i.group_id
join note n with i.activity_id = n.id join note n with i.activity_id = n.id
where n.id = :note_id and m.actor_id = :actor_id where n.id = :note_id and m.actor_id = :actor_id
EOF, EOF,
['note_id' => $this->id, 'actor_id' => $a->getId()], ['note_id' => $this->id, 'actor_id' => $actor->getId()],
)) ) !== [];
)); case VisibilityScope::COLLECTION:
case VisibilityScope::MESSAGE:
// Only for the collection to see
return in_array($actor->getId(), $this->getNotificationTargetIds());
default:
Log::error("Unknown scope found: {$this->getScope()}.");
}
return false;
} }
/** /**