gnu-social/plugins/TreeNotes/TreeNotes.php
Hugo Sales a4cb90ba12
[EVENTS] Change FormatNoteList do separate in and out arguments
This is necessary due to some weird problem which I wasn't able to figure out (but which doesn't matter)
that somehow causes the event to be called twice during testing, and thus the function was exploding
2021-08-14 16:12:58 +01:00

51 lines
1.7 KiB
PHP

<?php
// {{{ License
// This file is part of GNU social - https://www.gnu.org/software/social
//
// GNU social 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.
//
// GNU social 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 GNU social. If not, see <http://www.gnu.org/licenses/>.
// }}}
namespace Plugin\TreeNotes;
use App\Core\Modules\Plugin;
use App\Entity\Note;
class TreeNotes extends Plugin
{
/**
* Format the given $notes_in_trees_out in a list of reply trees
*/
public function onFormatNoteList(array $notes_in, ?array &$notes_out)
{
$roots = array_filter($notes_in, function (Note $note) { return $note->getReplyTo() == null; });
$notes_out = $this->build_tree($roots, $notes_in);
}
private function build_tree(array $parents, array $notes)
{
$subtree = [];
foreach ($parents as $p) {
$subtree[] = $this->build_subtree($p, $notes);
}
return $subtree;
}
private function build_subtree(Note $parent, array $notes)
{
$children = array_filter($notes, function (Note $n) use ($parent) { return $parent->getId() == $n->getReplyTo(); });
return ['note' => $parent, 'replies' => $this->build_tree($children, $notes)];
}
}