/*
 * Copyright (C) 2013-2016 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 "services/database/Cluster.hpp"

#include "services/database/Artist.hpp"
#include "services/database/Release.hpp"
#include "services/database/ScanSettings.hpp"
#include "services/database/Session.hpp"
#include "services/database/Track.hpp"
#include "IdTypeTraits.hpp"
#include "SqlQuery.hpp"
#include "Utils.hpp"

namespace Database
{
    namespace
    {
        template <typename ResultType>
        Wt::Dbo::Query<ResultType> createQuery(Session& session, std::string_view itemToSelect, const Cluster::FindParameters& params)
        {
            session.checkReadTransaction();

            auto query{ session.getDboSession().query<ResultType>("SELECT DISTINCT " + std::string{ itemToSelect } + " FROM cluster c") };

            if (params.track.isValid() || params.release.isValid())
            {
                query.join("track_cluster t_c ON t_c.cluster_id = c.id");
                query.join("track t ON t.id = t_c.track_id");
            }

            if (params.track.isValid())
                query.where("t.id = ?").bind(params.track);
            if (params.release.isValid())
                query.where("t.release_id = ?").bind(params.release);

            if (params.clusterType.isValid())
                query.where("c.cluster_type_id = ?").bind(params.clusterType);

            return query;
        }

        template <typename ResultType>
        Wt::Dbo::Query<ResultType> createQuery(Session& session, const Cluster::FindParameters& params)
        {
            std::string_view itemToSelect;
            
            if constexpr (std::is_same_v<ResultType, ClusterId>)
                itemToSelect = "c.id";
            else if constexpr (std::is_same_v<ResultType, Wt::Dbo::ptr<Cluster>>)
                itemToSelect = "c";
            else
                static_assert("Unhandled type");

            return createQuery<ResultType>(session, itemToSelect, params);
        }
    }

    Cluster::Cluster(ObjectPtr<ClusterType> type, std::string_view name)
        : _name{ std::string {name, 0, _maxNameLength} },
        _clusterType{ getDboPtr(type) }
    {
    }

    Cluster::pointer Cluster::create(Session& session, ObjectPtr<ClusterType> type, std::string_view name)
    {
        return session.getDboSession().add(std::unique_ptr<Cluster> {new Cluster{ type, name }});
    }

    std::size_t Cluster::getCount(Session& session)
    {
        session.checkReadTransaction();

        return session.getDboSession().query<int>("SELECT COUNT(*) FROM cluster");
    }

   RangeResults<ClusterId> Cluster::findIds(Session& session, const FindParameters& params)
    {
        session.checkReadTransaction();
        auto query{ createQuery<ClusterId>(session, params) };

        return Utils::execQuery<ClusterId>(query, params.range);
    }

    RangeResults<Cluster::pointer> Cluster::find(Session& session, const FindParameters& params)
    {
        session.checkReadTransaction();
        auto query{ createQuery<Wt::Dbo::ptr<Cluster>>(session, params) };

        return Utils::execQuery<Cluster::pointer>(query, params.range);
    }

    RangeResults<ClusterId> Cluster::findOrphanIds(Session& session, std::optional<Range> range)
    {
        session.checkReadTransaction();
        auto query{ session.getDboSession().query<ClusterId>("SELECT DISTINCT c.id FROM cluster c WHERE NOT EXISTS(SELECT 1 FROM track_cluster t_c WHERE t_c.cluster_id = c.id)") };

        return Utils::execQuery<ClusterId>(query, range);
    }

    Cluster::pointer Cluster::find(Session& session, ClusterId id)
    {
        session.checkReadTransaction();

        return session.getDboSession().find<Cluster>().where("id = ?").bind(id).resultValue();
    }

    std::size_t Cluster::computeTrackCount(Session& session, ClusterId id)
    {
        session.checkReadTransaction();

        return session.getDboSession().query<int>("SELECT COUNT(t.id) FROM track t INNER JOIN track_cluster t_c ON t_c.track_id = t.id")
            .where("t_c.cluster_id = ?").bind(id).resultValue();
    }

    std::size_t Cluster::computeReleaseCount(Session& session, ClusterId id)
    {
        session.checkReadTransaction();

        return session.getDboSession().query<int>("SELECT COUNT(DISTINCT r.id) FROM release r INNER JOIN track t on t.release_id = r.id INNER JOIN track_cluster t_c ON t_c.track_id = t.id")
            .where("t_c.cluster_id = ?").bind(id).resultValue();
    }

    void Cluster::addTrack(ObjectPtr<Track> track)
    {
        _tracks.insert(getDboPtr(track));
    }

    RangeResults<TrackId> Cluster::getTracks(std::optional<Range> range) const
    {
        assert(session());

        auto query{ session()->query<TrackId>("SELECT t.id FROM track t INNER JOIN cluster c ON c.id = t_c.cluster_id INNER JOIN track_cluster t_c ON t_c.track_id = t.id")
                .where("c.id = ?").bind(getId()) };

        return Utils::execQuery<TrackId>(query, range);
    }

    ClusterType::ClusterType(std::string_view name)
        : _name{ name }
    {
    }

    ClusterType::pointer ClusterType::create(Session& session, const std::string& name)
    {
        return session.getDboSession().add(std::unique_ptr<ClusterType> {new ClusterType{ name }});
    }

    std::size_t ClusterType::getCount(Session& session)
    {
        session.checkReadTransaction();

        return session.getDboSession().query<int>("SELECT COUNT(*) FROM cluster_type");
    }


    RangeResults<ClusterTypeId> ClusterType::findOrphans(Session& session, std::optional<Range> range)
    {
        session.checkReadTransaction();

        auto query{ session.getDboSession().query<ClusterTypeId>(
                "SELECT c_t.id from cluster_type c_t"
                " LEFT OUTER JOIN cluster c ON c_t.id = c.cluster_type_id")
            .where("c.id IS NULL") };

        return Utils::execQuery<ClusterTypeId>(query, range);
    }

    RangeResults<ClusterTypeId> ClusterType::findUsed(Session& session, std::optional<Range> range)
    {
        session.checkReadTransaction();

        auto query{ session.getDboSession().query<ClusterTypeId>(
                "SELECT DISTINCT c_t.id from cluster_type c_t")
            .join("cluster c ON c_t.id = c.cluster_type_id") };

        return Utils::execQuery<ClusterTypeId>(query, range);
    }

    ClusterType::pointer ClusterType::find(Session& session, std::string_view name)
    {
        session.checkReadTransaction();

        return session.getDboSession().find<ClusterType>().where("name = ?").bind(std::string{ name }).resultValue();
    }

    ClusterType::pointer ClusterType::find(Session& session, ClusterTypeId id)
    {
        session.checkReadTransaction();

        return session.getDboSession().find<ClusterType>().where("id = ?").bind(id).resultValue();
    }

    RangeResults<ClusterTypeId> ClusterType::find(Session& session, std::optional<Range> range)
    {
        session.checkReadTransaction();

        auto query{ session.getDboSession().query<ClusterTypeId>("SELECT id from cluster_type") };

        return Utils::execQuery<ClusterTypeId>(query, range);
    }

    Cluster::pointer ClusterType::getCluster(const std::string& name) const
    {
        assert(self());
        assert(session());

        return session()->find<Cluster>()
            .where("name = ?").bind(name)
            .where("cluster_type_id = ?").bind(getId()).resultValue();
    }

    std::vector<Cluster::pointer> ClusterType::getClusters() const
    {
        assert(self());
        assert(session());

        auto res = session()->find<Cluster>()
            .where("cluster_type_id = ?").bind(getId())
            .orderBy("name")
            .resultList();

        return std::vector<Cluster::pointer>(res.begin(), res.end());
    }
} // namespace Database