castopod/app/Libraries/ActivityPub/Models/FavouriteModel.php

<?php

/**
 * @copyright  2021 Podlibre
 * @license    https://www.gnu.org/licenses/agpl-3.0.en.html AGPL3
 * @link       https://castopod.org/
 */

namespace ActivityPub\Models;

use ActivityPub\Activities\LikeActivity;
use ActivityPub\Activities\UndoActivity;
use CodeIgniter\Events\Events;

class FavouriteModel extends UuidModel
{
    protected $table = 'activitypub_favourites';
    protected $uuidFields = ['note_id'];

    protected $allowedFields = ['actor_id', 'note_id'];

    protected $returnType = \ActivityPub\Entities\Favourite::class;

    protected $useTimestamps = true;
    protected $updatedField = null;

    /**
     *
     * @param \ActivityPub\Entities\Actor $actor
     * @param \ActivityPub\Entities\Note $note
     * @param bool $registerActivity
     * @return void
     */
    public function addFavourite($actor, $note, $registerActivity = true)
    {
        $this->db->transStart();

        $this->insert([
            'actor_id' => $actor->id,
            'note_id' => $note->id,
        ]);

        model('NoteModel')
            ->where(
                'id',
                service('uuid')
                    ->fromString($note->id)
                    ->getBytes(),
            )
            ->increment('favourites_count');

        Events::trigger('on_note_favourite', $actor, $note);

        if ($registerActivity) {
            $likeActivity = new LikeActivity();
            $likeActivity->set('actor', $actor->uri)->set('object', $note->uri);

            $activityId = model('ActivityModel')->newActivity(
                'Like',
                $actor->id,
                null,
                $note->id,
                $likeActivity->toJSON(),
                $note->published_at,
                'queued',
            );

            $likeActivity->set(
                'id',
                url_to('activity', $actor->username, $activityId),
            );

            model('ActivityModel')->update($activityId, [
                'payload' => $likeActivity->toJSON(),
            ]);
        }

        $this->db->transComplete();
    }

    public function removeFavourite($actor, $note, $registerActivity = true)
    {
        $this->db->transStart();

        model('NoteModel')
            ->where(
                'id',
                service('uuid')
                    ->fromString($note->id)
                    ->getBytes(),
            )
            ->decrement('favourites_count');

        $this->table('activitypub_favourites')
            ->where([
                'actor_id' => $actor->id,
                'note_id' => service('uuid')
                    ->fromString($note->id)
                    ->getBytes(),
            ])
            ->delete();

        Events::trigger('on_note_undo_favourite', $actor, $note);

        if ($registerActivity) {
            $undoActivity = new UndoActivity();
            // get like activity
            $activity = model('ActivityModel')
                ->where([
                    'type' => 'Like',
                    'actor_id' => $actor->id,
                    'note_id' => service('uuid')
                        ->fromString($note->id)
                        ->getBytes(),
                ])
                ->first();

            $likeActivity = new LikeActivity();
            $likeActivity
                ->set(
                    'id',
                    base_url(
                        route_to('activity', $actor->username, $activity->id),
                    ),
                )
                ->set('actor', $actor->uri)
                ->set('object', $note->uri);

            $undoActivity
                ->set('actor', $actor->uri)
                ->set('object', $likeActivity);

            $activityId = model('ActivityModel')->newActivity(
                'Undo',
                $actor->id,
                null,
                $note->id,
                $undoActivity->toJSON(),
                $note->published_at,
                'queued',
            );

            $undoActivity->set(
                'id',
                url_to('activity', $actor->username, $activityId),
            );

            model('ActivityModel')->update($activityId, [
                'payload' => $undoActivity->toJSON(),
            ]);
        }

        $this->db->transComplete();
    }

    /**
     * Adds or removes favourite from database and increments count
     *
     * @param \ActivityPub\Entities\Actor $actor
     * @param \ActivityPub\Entities\Note $note
     * @return void
     */
    public function toggleFavourite($actor, $note)
    {
        if (
            $this->where([
                'actor_id' => $actor->id,
                'note_id' => service('uuid')
                    ->fromString($note->id)
                    ->getBytes(),
            ])->first()
        ) {
            $this->removeFavourite($actor, $note);
        } else {
            $this->addFavourite($actor, $note);
        }
    }
}