// Copyright (c) Microsoft Corporation.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception

#include <algorithm>
#include <cassert>
#include <forward_list>
#include <ranges>
#include <span>
#include <type_traits>
#include <utility>
#include <vector>

#include <range_algorithm_support.hpp>
using namespace std;

#include <iostream>

// Test a silly precomposed range adaptor pipeline
template <int X>
constexpr auto is_less_than = [](const auto& x) { return x < X; };

using Pred = remove_const_t<decltype(is_less_than<3>)>;
static_assert(is_nothrow_copy_constructible_v<Pred> && is_nothrow_move_constructible_v<Pred>);

constexpr auto pipeline = views::take_while(is_less_than<3>) | views::take_while(is_less_than<3>)
                        | views::take_while(is_less_than<3>) | views::take_while(is_less_than<3>);

template <class Rng, class V = views::all_t<Rng>>
using pipeline_t = ranges::take_while_view<
    ranges::take_while_view<ranges::take_while_view<ranges::take_while_view<V, Pred>, Pred>, Pred>, Pred>;

template <class Rng>
concept CanViewTakeWhile = requires(Rng&& r) { views::take_while(forward<Rng>(r), is_less_than<3>); };

template <ranges::input_range Rng, ranges::random_access_range Expected>
constexpr bool test_one(Rng&& rng, Expected&& expected) {
    using ranges::take_while_view, ranges::bidirectional_range, ranges::common_range, ranges::contiguous_range,
        ranges::enable_borrowed_range, ranges::forward_range, ranges::iterator_t, ranges::prev,
        ranges::random_access_range;

    constexpr bool is_view = ranges::view<remove_cvref_t<Rng>>;

    using V = views::all_t<Rng>;
    using R = take_while_view<V, Pred>;
    static_assert(ranges::view<R>);
    static_assert(ranges::input_range<R>);
    static_assert(forward_range<R> == forward_range<Rng>);
    static_assert(bidirectional_range<R> == bidirectional_range<Rng>);
    static_assert(random_access_range<R> == random_access_range<Rng>);
    static_assert(contiguous_range<R> == contiguous_range<Rng>);

    // Validate range adaptor object and range adaptor closure
    constexpr auto closure = views::take_while(is_less_than<3>);

    // ... with lvalue argument
    static_assert(CanViewTakeWhile<Rng&> == (!is_view || copy_constructible<V>) );
    if constexpr (CanViewTakeWhile<Rng&>) { // Validate lvalue
        constexpr bool is_noexcept = !is_view || is_nothrow_copy_constructible_v<V>;

        static_assert(same_as<decltype(views::take_while(rng, is_less_than<3>)), R>);
        static_assert(noexcept(views::take_while(rng, is_less_than<3>)) == is_noexcept);

        static_assert(same_as<decltype(rng | closure), R>);
        static_assert(noexcept(rng | closure) == is_noexcept);

        static_assert(same_as<decltype(rng | pipeline), pipeline_t<Rng&>>);
        static_assert(noexcept(rng | pipeline) == is_noexcept);
    }

    // ... with const lvalue argument
    static_assert(CanViewTakeWhile<const remove_reference_t<Rng>&> == (!is_view || copy_constructible<V>) );
    if constexpr (is_view && copy_constructible<V>) {
        constexpr bool is_noexcept = is_nothrow_copy_constructible_v<V>;

        static_assert(same_as<decltype(views::take_while(as_const(rng), is_less_than<3>)), R>);
        static_assert(noexcept(views::take_while(as_const(rng), is_less_than<3>)) == is_noexcept);

        static_assert(same_as<decltype(as_const(rng) | closure), R>);
        static_assert(noexcept(as_const(rng) | closure) == is_noexcept);

        static_assert(same_as<decltype(as_const(rng) | pipeline), pipeline_t<const remove_reference_t<Rng>&>>);
        static_assert(noexcept(as_const(rng) | pipeline) == is_noexcept);
    } else if constexpr (!is_view) {
        using RC                   = take_while_view<ranges::ref_view<const remove_reference_t<Rng>>, Pred>;
        constexpr bool is_noexcept = true;

        static_assert(same_as<decltype(views::take_while(as_const(rng), is_less_than<3>)), RC>);
        static_assert(noexcept(views::take_while(as_const(rng), is_less_than<3>)) == is_noexcept);

        static_assert(same_as<decltype(as_const(rng) | closure), RC>);
        static_assert(noexcept(as_const(rng) | closure) == is_noexcept);

        static_assert(same_as<decltype(as_const(rng) | pipeline), pipeline_t<const remove_reference_t<Rng>&>>);
        static_assert(noexcept(as_const(rng) | pipeline) == is_noexcept);
    }

    // ... with rvalue argument
    static_assert(CanViewTakeWhile<remove_reference_t<Rng>> == (is_view || movable<remove_reference_t<Rng>>) );
    if constexpr (is_view) {
        constexpr bool is_noexcept = is_nothrow_move_constructible_v<V>;
        static_assert(same_as<decltype(views::take_while(move(rng), is_less_than<3>)), R>);
        static_assert(noexcept(views::take_while(move(rng), is_less_than<3>)) == is_noexcept);

        static_assert(same_as<decltype(move(rng) | closure), R>);
        static_assert(noexcept(move(rng) | closure) == is_noexcept);

        static_assert(same_as<decltype(move(rng) | pipeline), pipeline_t<remove_reference_t<Rng>>>);
        static_assert(noexcept(move(rng) | pipeline) == is_noexcept);
    } else if constexpr (movable<remove_reference_t<Rng>>) {
        using S                    = ranges::owning_view<remove_reference_t<Rng>>;
        using RS                   = take_while_view<S, Pred>;
        constexpr bool is_noexcept = is_nothrow_move_constructible_v<remove_reference_t<Rng>>;

        static_assert(same_as<decltype(views::take_while(move(rng), is_less_than<3>)), RS>);
        static_assert(noexcept(views::take_while(move(rng), is_less_than<3>)) == is_noexcept);

        static_assert(same_as<decltype(move(rng) | closure), RS>);
        static_assert(noexcept(move(rng) | closure) == is_noexcept);

        static_assert(same_as<decltype(move(rng) | pipeline), pipeline_t<remove_reference_t<Rng>>>);
        static_assert(noexcept(move(rng) | pipeline) == is_noexcept);
    }

    // ... with const rvalue argument
    static_assert(CanViewTakeWhile<const remove_reference_t<Rng>> == (is_view && copy_constructible<V>) );
    if constexpr (is_view && copy_constructible<V>) {
        constexpr bool is_noexcept = is_nothrow_copy_constructible_v<V>;

        static_assert(same_as<decltype(views::take_while(move(as_const(rng)), is_less_than<3>)), R>);
        static_assert(noexcept(views::take_while(move(as_const(rng)), is_less_than<3>)) == is_noexcept);

        static_assert(same_as<decltype(move(as_const(rng)) | closure), R>);
        static_assert(noexcept(move(as_const(rng)) | closure) == is_noexcept);

        static_assert(same_as<decltype(move(as_const(rng)) | pipeline), pipeline_t<const remove_reference_t<Rng>>>);
        static_assert(noexcept(move(as_const(rng)) | pipeline) == is_noexcept);
    }

    // Validate deduction guide
    same_as<R> auto r = take_while_view{forward<Rng>(rng), is_less_than<3>};
    assert(ranges::equal(r, expected));

    { // Validate take_while_view::pred
        [[maybe_unused]] same_as<Pred> auto pred_copy = as_const(r).pred();
        static_assert(noexcept(as_const(r).pred()));
    }

    const bool is_empty = ranges::empty(expected);

    // Validate view_interface::empty and operator bool
    static_assert(CanMemberEmpty<R> == forward_range<Rng>);
    static_assert(CanBool<R> == CanEmpty<R>);
    if constexpr (CanMemberEmpty<R>) {
        assert(r.empty() == is_empty);
        assert(static_cast<bool>(r) == !is_empty);
    } else {
        static_assert(CanEmpty<R> == CanSize<R>);
        if constexpr (CanEmpty<R>) {
            assert(ranges::empty(r) == is_empty);
            assert(static_cast<bool>(r) == !is_empty);
        }
    }

    static_assert(CanMemberEmpty<const R> == forward_range<const Rng>);
    static_assert(CanBool<const R> == CanEmpty<const R>);
    if constexpr (CanMemberEmpty<const R>) {
        assert(as_const(r).empty() == is_empty);
        assert(static_cast<bool>(as_const(r)) == !is_empty);
    } else {
        static_assert(CanEmpty<const R> == CanSize<const R>);
        if constexpr (CanEmpty<const R>) {
            assert(ranges::empty(as_const(r)) == is_empty);
            assert(static_cast<bool>(as_const(r)) == !is_empty);
        }
    }

    // Validate take_while_view::begin
    static_assert(CanMemberBegin<R>);
    static_assert(
        CanMemberBegin<const R> == ranges::range<const R> && indirect_unary_predicate<const Pred, iterator_t<const V>>);
    if (forward_range<V>) { // intentionally not if constexpr
        const same_as<iterator_t<R>> auto i = r.begin();
        if (!is_empty) {
            assert(*i == *begin(expected));
            assert(*r.begin() == *begin(expected));
            assert(*r.begin() == *begin(expected));
        }

        if constexpr (copyable<V>) {
            auto r2                              = r;
            const same_as<iterator_t<R>> auto i2 = r2.begin();
            if (!is_empty) {
                assert(*i2 == *i);
                assert(*r2.begin() == *i2);
                assert(*r2.begin() == *i2);
            }
        }

        const same_as<iterator_t<const R>> auto ic = as_const(r).begin();
        if (!is_empty) {
            assert(*ic == *begin(expected));
            assert(*as_const(r).begin() == *begin(expected));
            assert(*as_const(r).begin() == *begin(expected));
        }

        if constexpr (copyable<V>) {
            auto rc2                                    = as_const(r);
            const same_as<iterator_t<const R>> auto ic2 = rc2.begin();
            if (!is_empty) {
                assert(*ic2 == *ic);
                assert(*rc2.begin() == *ic2);
                assert(*rc2.begin() == *ic2);
            }
        }
    }

    // Validate take_while_view::end
    static_assert(CanMemberEnd<R>);
    static_assert(
        CanMemberEnd<const R> == ranges::range<const R> && indirect_unary_predicate<const Pred, iterator_t<const V>>);
    static_assert(CanEnd<const R&> == CanMemberEnd<const R>);
    static_assert(!common_range<R>);
    static_assert(!common_range<const R>);
    if (!is_empty) {
        same_as<ranges::sentinel_t<R>> auto s        = r.end();
        same_as<ranges::sentinel_t<const R>> auto sc = as_const(r).end();

        if (forward_range<V>) { // intentionally not if constexpr
            // Compare with const / non-const iterators
            assert(s != r.begin());
            assert(s != as_const(r).begin());
            assert(sc != r.begin());
            assert(sc != as_const(r).begin());

            // Compare with end of range
            if constexpr (common_range<V>) {
                assert(s == s.base());
                assert(s == sc.base());
                assert(sc == s.base());
                assert(sc == sc.base());
            } else if constexpr (forward_range<V> && is_lvalue_reference_v<Rng>) {
                auto full_range   = views::take_while(rng, [](const auto&) { return true; });
                const auto length = 8; // NB: depends on the test data
                assert(full_range.end() == next(full_range.begin(), length));
                assert(full_range.end() == next(as_const(full_range).begin(), length));
                assert(as_const(full_range).end() == next(full_range.begin(), length));
                assert(as_const(full_range).end() == next(as_const(full_range).begin(), length));
            }

            // Compare with iterator whose predicate evaluates to false
            if constexpr (forward_range<V>) {
                const auto length = 4; // NB: depends on the test data
                assert(s == next(r.begin(), length));
                assert(s == next(as_const(r).begin(), length));
                assert(sc == next(r.begin(), length));
                assert(sc == next(as_const(r).begin(), length));
            }
        }
    }

#if _HAS_CXX23
    using ranges::const_iterator_t, ranges::const_sentinel_t, ranges::cbegin, ranges::cend;

    // Validate view_interface::cbegin
    static_assert(CanMemberCBegin<R>);
    static_assert(CanMemberCBegin<const R> == ranges::range<const R>);
    static_assert(indirect_unary_predicate<const Pred, const_iterator_t<const V>>);
    if (forward_range<V>) { // intentionally not if constexpr
        const same_as<const_iterator_t<R>> auto i = r.cbegin();
        if (!is_empty) {
            assert(*i == *cbegin(expected));
            assert(*r.cbegin() == *cbegin(expected));
        }

        if constexpr (copyable<V>) {
            auto r2                                    = r;
            const same_as<const_iterator_t<R>> auto i2 = r2.cbegin();
            if (!is_empty) {
                assert(*i2 == *i);
                assert(*r2.cbegin() == *i2);
            }
        }

        const same_as<const_iterator_t<const R>> auto ic = as_const(r).cbegin();
        if (!is_empty) {
            assert(*ic == *cbegin(expected));
            assert(*as_const(r).cbegin() == *cbegin(expected));
        }

        if constexpr (copyable<V>) {
            auto rc2                                          = as_const(r);
            const same_as<const_iterator_t<const R>> auto ic2 = rc2.cbegin();
            if (!is_empty) {
                assert(*ic2 == *ic);
                assert(*rc2.cbegin() == *ic2);
            }
        }
    }

    // Validate view_interface::cend
    static_assert(CanMemberCEnd<R>);
    static_assert(CanMemberCEnd<const R> == ranges::range<const R>);
    static_assert(indirect_unary_predicate<const Pred, const_iterator_t<const V>>);
    static_assert(CanCEnd<const R&> == CanMemberCEnd<const R>);
    static_assert(!common_range<R>);
    static_assert(!common_range<const R>);
    if (!is_empty) {
        same_as<const_sentinel_t<R>> auto s        = r.cend();
        same_as<const_sentinel_t<const R>> auto sc = as_const(r).cend();

        if (forward_range<V>) { // intentionally not if constexpr
            // Compare with const / non-const iterators
            assert(s != r.cbegin());
            assert(s != as_const(r).cbegin());
            assert(sc != r.cbegin());
            assert(sc != as_const(r).cbegin());

            // Compare with end of range
            if constexpr (common_range<V>) {
                assert(s == s.base());
                assert(s == sc.base());
                assert(sc == s.base());
                assert(sc == sc.base());
            } else if constexpr (forward_range<V> && is_lvalue_reference_v<Rng>) {
                auto full_range   = views::take_while(rng, [](const auto&) { return true; });
                const auto length = 8; // NB: depends on the test data
                assert(full_range.cend() == next(full_range.cbegin(), length));
                assert(full_range.cend() == next(as_const(full_range).cbegin(), length));
                assert(as_const(full_range).cend() == next(full_range.cbegin(), length));
                assert(as_const(full_range).cend() == next(as_const(full_range).cbegin(), length));
            }

            // Compare with iterator whose predicate evaluates to false
            if constexpr (forward_range<V>) {
                const auto length = 4; // NB: depends on the test data
                assert(s == next(r.cbegin(), length));
                assert(s == next(as_const(r).cbegin(), length));
                assert(sc == next(r.cbegin(), length));
                assert(sc == next(as_const(r).cbegin(), length));
            }
        }
    }
#endif // _HAS_CXX23

    // Validate view_interface::data
    if constexpr (CanData<R&>) {
        const same_as<remove_reference_t<ranges::range_reference_t<V>>*> auto ptr1 = r.data();
        assert(ptr1 == to_address(r.begin()));
    }

    if constexpr (CanData<const R&>) {
        const same_as<remove_reference_t<ranges::range_reference_t<const V>>*> auto ptr2 = as_const(r).data();
        assert(ptr2 == to_address(as_const(r).begin()));
    }

    // Validate view_interface::size
    static_assert(!CanMemberSize<R>);
    static_assert(!CanMemberSize<const R>);

    // Validate view_interface::operator[]
    static_assert(CanIndex<R> == random_access_range<V>);
    if constexpr (CanIndex<R>) {
        if (!is_empty) {
            assert(r[0] == *r.begin());
        }
    }
    static_assert(CanIndex<const R> == random_access_range<const V>);
    if constexpr (CanIndex<const R>) {
        if (!is_empty) {
            assert(as_const(r)[0] == *as_const(r).begin());
        }
    }

    // Validate view_interface::front and back
    static_assert(CanMemberFront<R> == forward_range<V>);
    if constexpr (CanMemberFront<R>) {
        if (!is_empty) {
            assert(r.front() == *begin(expected));
        }
    }
    static_assert(CanMemberFront<const R> == forward_range<const V>);
    if constexpr (CanMemberFront<const R>) {
        if (!is_empty) {
            assert(as_const(r).front() == *begin(expected));
        }
    }

    static_assert(CanMemberBack<R> == (bidirectional_range<R> && common_range<R>) );
    if constexpr (CanMemberBack<R>) {
        if (!is_empty) {
            assert(r.back() == *prev(end(expected)));
        }
    }
    static_assert(CanMemberBack<const R> == (bidirectional_range<const R> && common_range<const R>) );
    if constexpr (CanMemberBack<const R>) {
        if (!is_empty) {
            assert(as_const(r).back() == *prev(end(expected)));
        }
    }

    // Validate take_while_view::base() const&
    static_assert(CanMemberBase<const R&> == copy_constructible<V>);
    if constexpr (copy_constructible<V> && forward_range<V>) {
        same_as<V> auto b1 = as_const(r).base();
        static_assert(noexcept(as_const(r).base()) == is_nothrow_copy_constructible_v<V>);
        if (!is_empty) {
            assert(*b1.begin() == 0); // NB: depends on the test data
            if constexpr (bidirectional_range<V> && common_range<V>) {
                assert(*prev(b1.end()) == 1); // NB: depends on the test data
            }
        }
    }

    // Validate take_while_view::base() && (NB: do this last since it leaves r moved-from)
    if (forward_range<V>) { // intentionally not if constexpr
        same_as<V> auto b2 = move(r).base();
        static_assert(noexcept(move(r).base()) == is_nothrow_move_constructible_v<V>);
        if (!is_empty) {
            assert(*b2.begin() == 0); // NB: depends on the test data
            if constexpr (bidirectional_range<V> && common_range<V>) {
                assert(*prev(b2.end()) == 1); // NB: depends on the test data
            }
        }
    }
    return true;
}

static constexpr int some_ints[]               = {0, 1, 2, 3, 4, 3, 2, 1};
static constexpr int expected_output[]         = {0, 1, 2};
static constexpr int expected_output_reverse[] = {2, 1};

struct instantiator {
    template <ranges::input_range R>
    static constexpr void call() {
        R r{some_ints};
        test_one(r, expected_output);
    }
};

template <class Category, test::Common IsCommon>
using test_range = test::range<Category, const int, test::Sized::no,
    test::CanDifference{derived_from<Category, random_access_iterator_tag>}, IsCommon,
    test::CanCompare{derived_from<Category, forward_iterator_tag> || IsCommon == test::Common::yes},
    test::ProxyRef{!derived_from<Category, contiguous_iterator_tag>}>;

constexpr void instantiation_test() {
#ifdef TEST_EVERYTHING
    test_in<instantiator, const int>();
#else // ^^^ test all input range permutations / test only "interesting" permutations vvv
    // The view is sensitive to category and commonality, but oblivious to size, differencing, and proxyness.
    using test::Common;

    instantiator::call<test_range<input_iterator_tag, Common::no>>();
    instantiator::call<test_range<input_iterator_tag, Common::yes>>();
    instantiator::call<test_range<forward_iterator_tag, Common::no>>();
    instantiator::call<test_range<forward_iterator_tag, Common::yes>>();
    instantiator::call<test_range<bidirectional_iterator_tag, Common::no>>();
    instantiator::call<test_range<bidirectional_iterator_tag, Common::yes>>();
    instantiator::call<test_range<random_access_iterator_tag, Common::no>>();
    instantiator::call<test_range<random_access_iterator_tag, Common::yes>>();
    instantiator::call<test_range<contiguous_iterator_tag, Common::no>>();
    instantiator::call<test_range<contiguous_iterator_tag, Common::yes>>();
#endif // TEST_EVERYTHING
}

template <class Category, test::Common IsCommon, bool is_random = derived_from<Category, random_access_iterator_tag>>
using move_only_view = test::range<Category, const int, test::Sized{is_random}, test::CanDifference{is_random},
    IsCommon, test::CanCompare{derived_from<Category, forward_iterator_tag>},
    test::ProxyRef{!derived_from<Category, contiguous_iterator_tag>}, test::CanView::yes, test::Copyability::move_only>;

int main() {
    // Validate views
    { // ... copyable
        constexpr span<const int> s{some_ints};
        static_assert(test_one(s, expected_output));
        test_one(s, expected_output);
    }
    { // ... move-only
        test_one(move_only_view<input_iterator_tag, test::Common::no>{some_ints}, expected_output);
        test_one(move_only_view<forward_iterator_tag, test::Common::no>{some_ints}, expected_output);
        test_one(move_only_view<forward_iterator_tag, test::Common::yes>{some_ints}, expected_output);
        test_one(move_only_view<bidirectional_iterator_tag, test::Common::no>{some_ints}, expected_output);
        test_one(move_only_view<bidirectional_iterator_tag, test::Common::yes>{some_ints}, expected_output);
        test_one(move_only_view<random_access_iterator_tag, test::Common::no>{some_ints}, expected_output);
        test_one(move_only_view<random_access_iterator_tag, test::Common::yes>{some_ints}, expected_output);
    }

    // Validate non-views
    {
        static_assert(test_one(some_ints, expected_output));
        test_one(some_ints, expected_output);
    }
    {
        vector vec(ranges::begin(some_ints), ranges::end(some_ints));
        test_one(vec, expected_output);
    }
    {
        forward_list lst(ranges::begin(some_ints), ranges::end(some_ints));
        test_one(lst, expected_output);
    }

    // take_while/reverse interaction test
    {
        auto twr_pipe = views::take_while(is_less_than<3>) | views::reverse;
        auto rtw_pipe = views::reverse | views::take_while(is_less_than<3>);

        auto r0  = some_ints | twr_pipe;
        using R0 = decltype(r0);
        static_assert(ranges::bidirectional_range<R0> && ranges::view<R0>);
        assert(ranges::equal(r0, views::reverse(expected_output)));

        auto r1  = some_ints | rtw_pipe;
        using R1 = decltype(r1);
        static_assert(ranges::bidirectional_range<R1> && ranges::view<R1>);
        assert(ranges::equal(r1, views::reverse(expected_output_reverse)));
    }

    { // empty range
        static_assert(test_one(span<const int, 0>{}, span<const int, 0>{}));
        test_one(span<const int, 0>{}, span<const int, 0>{});
    }

    static_assert((instantiation_test(), true));
    instantiation_test();

    { // Validate guarantee that predicates are moved into the range adaptor closure, and into the view
        // object from an rvalue closure
        struct Fn {
            Fn()     = default;
            Fn(Fn&&) = default;
            Fn(const Fn&) {
                assert(false);
            }
            Fn& operator=(Fn&&) = default;

            Fn& operator=(const Fn&) {
                assert(false);
                return *this;
            }

            bool operator()(int) const {
                return true;
            }
        };

        (void) views::take_while(Fn{})(span<int>{});
    }
}
