/* 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/. */

/* Iterator over ranges of integers */

#ifndef mozilla_IntegerRange_h
#define mozilla_IntegerRange_h

#include "mozilla/Assertions.h"
#include "mozilla/ReverseIterator.h"

#include <iterator>
#include <type_traits>

namespace mozilla {

namespace detail {

template <typename IntTypeT>
class IntegerIterator {
 public:
  // It is disputable whether these type definitions are correct, since
  // operator* doesn't return a reference at all. Also, the iterator_category
  // can be at most std::input_iterator_tag (rather than
  // std::bidrectional_iterator_tag, as it might seem), because it is a stashing
  // iterator. See also, e.g.,
  // https://stackoverflow.com/questions/50909701/what-should-be-iterator-category-for-a-stashing-iterator
  using value_type = const IntTypeT;
  using pointer = const value_type*;
  using reference = const value_type&;
  using difference_type = std::make_signed_t<IntTypeT>;
  using iterator_category = std::input_iterator_tag;

  template <typename IntType>
  explicit IntegerIterator(IntType aCurrent) : mCurrent(aCurrent) {}

  template <typename IntType>
  explicit IntegerIterator(const IntegerIterator<IntType>& aOther)
      : mCurrent(aOther.mCurrent) {}

  // This intentionally returns a value rather than a reference, to make
  // mozilla::ReverseIterator work with it. Still, std::reverse_iterator cannot
  // be used with IntegerIterator because it still is a "stashing iterator". See
  // Bug 1175485.
  IntTypeT operator*() const { return mCurrent; }

  /* Increment and decrement operators */

  IntegerIterator& operator++() {
    ++mCurrent;
    return *this;
  }
  IntegerIterator& operator--() {
    --mCurrent;
    return *this;
  }
  IntegerIterator operator++(int) {
    auto ret = *this;
    ++mCurrent;
    return ret;
  }
  IntegerIterator operator--(int) {
    auto ret = *this;
    --mCurrent;
    return ret;
  }

  /* Comparison operators */

  friend bool operator==(const IntegerIterator<IntTypeT>& aIter1,
                         const IntegerIterator<IntTypeT>& aIter2) {
    return aIter1.mCurrent == aIter2.mCurrent;
  }
  friend bool operator!=(const IntegerIterator<IntTypeT>& aIter1,
                         const IntegerIterator<IntTypeT>& aIter2) {
    return aIter1.mCurrent != aIter2.mCurrent;
  }
  friend bool operator<(const IntegerIterator<IntTypeT>& aIter1,
                        const IntegerIterator<IntTypeT>& aIter2) {
    return aIter1.mCurrent < aIter2.mCurrent;
  }
  friend bool operator<=(const IntegerIterator<IntTypeT>& aIter1,
                         const IntegerIterator<IntTypeT>& aIter2) {
    return aIter1.mCurrent <= aIter2.mCurrent;
  }
  friend bool operator>(const IntegerIterator<IntTypeT>& aIter1,
                        const IntegerIterator<IntTypeT>& aIter2) {
    return aIter1.mCurrent > aIter2.mCurrent;
  }
  friend bool operator>=(const IntegerIterator<IntTypeT>& aIter1,
                         const IntegerIterator<IntTypeT>& aIter2) {
    return aIter1.mCurrent >= aIter2.mCurrent;
  }

 private:
  IntTypeT mCurrent;
};

template <typename IntTypeT>
class IntegerRange {
 public:
  typedef IntegerIterator<IntTypeT> iterator;
  typedef IntegerIterator<IntTypeT> const_iterator;
  typedef ReverseIterator<IntegerIterator<IntTypeT>> reverse_iterator;
  typedef ReverseIterator<IntegerIterator<IntTypeT>> const_reverse_iterator;

  template <typename IntType>
  explicit IntegerRange(IntType aEnd) : mBegin(0), mEnd(aEnd) {}

  template <typename IntType1, typename IntType2>
  IntegerRange(IntType1 aBegin, IntType2 aEnd) : mBegin(aBegin), mEnd(aEnd) {}

  iterator begin() const { return iterator(mBegin); }
  const_iterator cbegin() const { return begin(); }
  iterator end() const { return iterator(mEnd); }
  const_iterator cend() const { return end(); }
  reverse_iterator rbegin() const { return reverse_iterator(iterator(mEnd)); }
  const_reverse_iterator crbegin() const { return rbegin(); }
  reverse_iterator rend() const { return reverse_iterator(iterator(mBegin)); }
  const_reverse_iterator crend() const { return rend(); }

 private:
  IntTypeT mBegin;
  IntTypeT mEnd;
};

}  // namespace detail

template <typename IntType>
detail::IntegerRange<IntType> IntegerRange(IntType aEnd) {
  static_assert(std::is_integral_v<IntType>, "value must be integral");
  if constexpr (std::is_signed_v<IntType>) {
    MOZ_ASSERT(aEnd >= 0, "Should never have negative value here");
  }
  return detail::IntegerRange<IntType>(aEnd);
}

template <typename IntType1, typename IntType2>
detail::IntegerRange<IntType2> IntegerRange(IntType1 aBegin, IntType2 aEnd) {
  static_assert(std::is_integral_v<IntType1> && std::is_integral_v<IntType2>,
                "values must both be integral");
  static_assert(std::is_signed_v<IntType1> == std::is_signed_v<IntType2>,
                "signed/unsigned mismatch");
  MOZ_ASSERT(aEnd >= aBegin, "End value should be larger than begin value");
  return detail::IntegerRange<IntType2>(aBegin, aEnd);
}

}  // namespace mozilla

#endif  // mozilla_IntegerRange_h