/*
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/.
 *
 * Copyright 2011-2015 Danny Robson <danny@nerdcruft.net>
 */

#ifndef CRUFT_UTIL_SIGNAL_HPP
#define CRUFT_UTIL_SIGNAL_HPP

#include "types/traits.hpp"
#include "debug.hpp"
#include "nocopy.hpp"

#include <algorithm>
#include <functional>
#include <list>

namespace util {
    namespace combine {
        template <typename F>
        struct logical_and {
            using R = typename func_traits<F>::return_type;

            template <typename T, typename ...Args>
            R operator() (T first, T last, Args&&... args)
            {
                while (first != last)
                    if (!(*first++)(args...))
                        return false;

                return true;
            }
        };

        template <typename F>
        struct logical_or {
            using R = typename func_traits<F>::return_type;

            template <typename T, typename ...Args>
            R operator() (T first, T last, Args&&... args)
            {
                while (first != last)
                    if ((*first++)(args...))
                        return true;

                return false;
            }
        };

        template <typename F>
        struct noop {
            using R = void;

            template <typename T, typename ...Args>
            R operator() (T first, T last, Args&&... args)
            {
                while (first != last) {
                    (*first++)(args...);
                }
            }
        };
    }

    template <typename F, template <typename> class C = combine::noop>
    class signal {
        public:
            using R = typename C<F>::R;
            using callback = std::function<F>;

            struct cookie;

        public:
            signal () = default;
            ~signal ()
            {
                CHECK (empty ());
            }

            /// Add a callback to list.
            cookie connect [[nodiscard]] (callback &&_cb)
            {
                return cookie (
                    m_children.insert (
                        m_children.end (),
                        std::move (_cb)
                    ),
                    *this
                );
            }

            cookie connect [[nodiscard]] (const callback &_cb)
            {
                {
                    return cookie (
                        m_children.insert (
                            m_children.end (),
                            std::move (_cb)
                        ),
                        *this
                    );
                }
            }

            void disconnect (cookie &c)
            {
                m_children.erase (c.m_position);
                c.m_position = m_children.end ();
            }

            /// Disconnect all callbacks
            void clear (void)
            {
                m_children.clear ();
            }

            /// Returns the number of callbacks connected.
            size_t size (void) const
            {
                return m_children.size ();
            }

            bool empty (void) const
            {
                return m_children.empty ();
            }

            /// Execute all callbacks
            template <typename ...Args>
            R
            operator() (Args&&... tail)
            {
                if (m_children.empty ())
                    return R();

                C<F> combiner;
                return combiner (
                    m_children.begin (),
                    m_children.end (),
                    std::forward<Args> (tail)...
                );
            }

        private:
            typedef std::list<callback> group;
            group m_children;
    };


    ///////////////////////////////////////////////////////////////////////////
    template <typename F, template <typename> class C>
    struct signal<F,C>::cookie : public nocopy {
        cookie (typename group::iterator _position,
                signal<F,C> &_parent):
            m_position (_position),
            m_parent (_parent)
        { ; }

        cookie (cookie &&rhs):
            m_position (rhs.m_position),
            m_parent (rhs.m_parent)
        {
            rhs.m_position = rhs.m_parent.m_children.end ();
        }

        ~cookie ()
        {
            if (m_parent.m_children.end () != m_position)
                m_parent.disconnect (*this);
        }

        void reset (callback &&cb)
        {
            *m_position = std::move (cb);
        }

        typename group::iterator m_position;
        signal<F,C> &m_parent;
    };



    ///////////////////////////////////////////////////////////////////////////
    // wrap a value in a signal and trigger on assignment
    //template <typename T, template <typename> class C>
    template <typename T>
    class value_signal : public signal<void(T)> {
    public:
        explicit value_signal (T t): m_value (t) { ; }
        value_signal () = default;

        operator const T&() const { return m_value; }

        value_signal<T>&
        operator= (const T &t)
        {
            m_value = t;
            (*this) (m_value);
            return *this;
        }

    private:
        T m_value;
    };
}

#endif