either.h

Go to the documentation of this file.

/**********************************************************************
 * LeechCraft - modular cross-platform feature rich internet client.
 * Copyright (C) 2006-2014  Georg Rudoy
 *
 * Distributed under the Boost Software License, Version 1.0.
 * (See accompanying file LICENSE or copy at https://www.boost.org/LICENSE_1_0.txt)
 **********************************************************************/

#pragma once

#include <source_location>
#include <QtDebug>
#include <util/sll/either.h>
#include <util/sll/void.h>
#include "task.h"

namespace LC::Util
{
    struct IgnoreLeft {};

    namespace detail
    {
        template<typename Promise>
        [[noreturn]]
        void TerminateLeftyCoroutine (std::coroutine_handle<Promise> handle, const auto& error)
        {
            auto& promise = handle.promise ();
            if constexpr (Promise::IsVoid)
                promise.return_void ();
            else
                promise.return_value (Left { error });

            throw EitherFailureAbort {};
        }

        template<typename L, typename R, typename ErrorHandler>
        struct EitherAwaiter
        {
            Either<L, R> Either_;
            ErrorHandler Handler_;

            using HandlerReturn_t = decltype (Handler_ (Either_.GetLeft ()));

            bool await_ready () noexcept
            {
                if (Either_.IsRight ())
                    return true;

                if constexpr (std::is_same_v<IgnoreLeft, HandlerReturn_t>)
                {
                    static_assert (std::is_default_constructible_v<R>);
                    Handler_ (Either_.GetLeft ());
                    Either_ = R {};
                    return true;
                }

                return false;
            }

            auto await_suspend (auto handle)
            {
                if constexpr (std::is_same_v<void, HandlerReturn_t>)
                {
                    Handler_ (Either_.GetLeft ());
                    TerminateLeftyCoroutine (handle, Either_.GetLeft ());
                }
                else if constexpr (std::is_same_v<IgnoreLeft, HandlerReturn_t>)
                    qFatal () << "shall never suspend when ignoring left";
                else
                    TerminateLeftyCoroutine (handle, Handler_ (Either_.GetLeft ()));
            }

            R await_resume () const noexcept
            {
                return Either_.GetRight ();
            }
        };
    }

    template<typename L, typename R, typename F>
        requires std::invocable<F, const L&>
    detail::EitherAwaiter<L, R, F> WithHandler (const Either<L, R>& either, F&& errorHandler)
    {
        return { either, std::forward<F> (errorHandler) };
    }

    template<typename L, typename R>
    auto WithHandler (const Either<L, R>& either, IgnoreLeft)
    {
        return WithHandler (either, [] (const auto&) { return IgnoreLeft {}; });
    }

    template<typename T>
    Either<Void, T> NonEmpty (const T& t, auto&& msg, std::source_location loc = std::source_location::current ())
    {
        if (t)
            return t;

        QMessageLogger { loc.file_name (), static_cast<int> (loc.line ()), loc.function_name () }.warning () << msg;
        return { AsLeft, Void {} };
    }

    template<typename T, typename... Msgs>
    Either<Void, T> NonEmpty (const T& t, const std::tuple<Msgs...>& msgsTuple,
            std::source_location loc = std::source_location::current ())
    {
        if (t)
            return t;

        std::apply ([&]<typename... AMsgs> (AMsgs&&... amsgs)
        {
            const QMessageLogger log { loc.file_name (), static_cast<int> (loc.line ()), loc.function_name () };
            (log.warning () << ... << std::forward<AMsgs> (amsgs));
        }, msgsTuple);
        return { AsLeft, Void {} };
    }
}

namespace LC
{
    template<typename L, typename R>
    Util::detail::EitherAwaiter<L, R, std::identity> operator co_await (const Util::Either<L, R>& either)
    {
        return { either, {} };
    }
}