/*
 * Copyright (C) 2021 Emeric Poupon
 *
 * This file is part of LMS.
 *
 * LMS is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * LMS 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 General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with LMS.  If not, see <http://www.gnu.org/licenses/>.
 */

#include "InternalBackend.hpp"

#include "database/IDb.hpp"
#include "database/Session.hpp"
#include "database/objects/StarredArtist.hpp"
#include "database/objects/StarredRelease.hpp"
#include "database/objects/StarredTrack.hpp"

namespace lms::feedback
{
    namespace detail
    {
        template<typename StarredObjType>
        void onStarred(db::Session& session, typename StarredObjType::IdType id)
        {
            auto transaction{ session.createWriteTransaction() };

            if (auto starredObj{ StarredObjType::find(session, id) })
                starredObj.modify()->setSyncState(db::SyncState::Synchronized);
        }

        template<typename StarredObjType>
        void onUnstarred(db::Session& session, typename StarredObjType::IdType id)
        {
            auto transaction{ session.createWriteTransaction() };

            if (auto starredObj{ StarredObjType::find(session, id) })
                starredObj.remove();
        }
    } // namespace detail

    InternalBackend::InternalBackend(db::IDb& db)
        : _db{ db }
    {
    }

    void InternalBackend::onStarred(db::StarredArtistId artistId)
    {
        detail::onStarred<db::StarredArtist>(_db.getTLSSession(), artistId);
    }

    void InternalBackend::onUnstarred(db::StarredArtistId artistId)
    {
        detail::onUnstarred<db::StarredArtist>(_db.getTLSSession(), artistId);
    }

    void InternalBackend::onStarred(db::StarredReleaseId releaseId)
    {
        detail::onStarred<db::StarredRelease>(_db.getTLSSession(), releaseId);
    }

    void InternalBackend::onUnstarred(db::StarredReleaseId releaseId)
    {
        detail::onUnstarred<db::StarredRelease>(_db.getTLSSession(), releaseId);
    }

    void InternalBackend::onStarred(db::StarredTrackId trackId)
    {
        detail::onStarred<db::StarredTrack>(_db.getTLSSession(), trackId);
    }

    void InternalBackend::onUnstarred(db::StarredTrackId trackId)
    {
        detail::onUnstarred<db::StarredTrack>(_db.getTLSSession(), trackId);
    }
} // namespace lms::feedback
