#pragma once

#include <memory>
#include <vector>
#include "tach_sensor.hpp"
#include "trust_group.hpp"
#include "types.hpp"

namespace phosphor
{
namespace fan
{
namespace trust
{

/**
 * @class Manager
 *
 * The sensor trust manager class.  It can be asked if a tach sensor's
 * reading can be trusted or not, based on the trust groups the sensor
 * is in.
 *
 * When it finds a group's trust status changing, it will either stop or
 * start the tach error timers for the group's sensors accordingly.
 *
 * See the trust::Group documentation for more details on sensor trust.
 */
class Manager
{
    public:

        Manager() = delete;
        Manager(const Manager&) = delete;
        Manager& operator=(const Manager&) = delete;
        Manager(Manager&&) = default;
        Manager& operator=(Manager&&) = default;
        ~Manager() = default;

        /**
         * Constructor
         *
         * @param[in] functions - trust group creation function vector
         */
        explicit Manager(const std::vector<monitor::CreateGroupFunction>& functions)
        {
            for (auto& create : functions)
            {
                groups.emplace_back(create());
            }
        }

        /**
         * Says if trust groups have been created and
         * need to be checked.
         *
         * @return bool - If there are any trust groups
         */
        inline bool active() const
        {
            return !groups.empty();
        }

        /**
         * Checks if a sensor value can be trusted
         *
         * Checks if the sensor is trusted in each group
         * it belongs to.  Only considered trusted if it is
         * trusted in all groups it belongs to.
         *
         * While checking group trust, the code will also check
         * if the trust status has just changed.  If the status
         * just changed to false, it will stop the tach error
         * timers for that group so these untrusted sensors won't
         * cause errors.  If changed to true, it will start those timers
         * back up again.
         *
         * Note this means groups should be designed such that
         * in the same call to this function a sensor shouldn't
         * make one group change to trusted and another to untrusted.
         *
         * @param[in] sensor - the sensor to check
         *
         * @return bool - if sensor is trusted in all groups or not
         */
        bool checkTrust(
                const monitor::TachSensor& sensor)
        {
            auto trusted = true;

            for (auto& group : groups)
            {
                if (group->inGroup(sensor))
                {
                    bool trust, changed;
                    std::tie(trust, changed) = group->checkTrust(sensor);

                    if (!trust)
                    {
                        trusted = false;

                        if (changed)
                        {
                            group->stopTimers();
                        }
                    }
                    else
                    {
                        if (changed)
                        {
                            group->startTimers();
                        }
                    }
                }
            }

            return trusted;
        }

        /**
         * Registers a sensor with any trust groups that are interested
         *
         * @param[in] sensor - the sensor to register
         */
        void registerSensor(
                std::shared_ptr<monitor::TachSensor>& sensor)
        {
            std::for_each(
                    groups.begin(),
                    groups.end(),
                    [&sensor](auto& group)
                    {
                        group->registerSensor(sensor);
                    });
        }

    private:

        /**
         * The list of sensor trust groups
         */
        std::vector<std::unique_ptr<Group>> groups;
};

}
}
}