//
//  maybe_null.hpp
//  pointer
//
//  Created by Sam Jaffe on 9/24/15.
//
//

#pragma once

#include <stdexcept>
#include <memory>

#include "pointer_fwd.hpp"

#include "not_null.hpp"

class unchecked_pointer_exception : public std::logic_error {
  using std::logic_error::logic_error;
};

template <typename P> class maybe_null<not_null<P>>; // not permitted

template <typename T>
class maybe_null {
public:
  using element_type = typename std::pointer_traits<T>::element_type;
  using pointer = element_type *;
  using reference = element_type &;
  
  maybe_null() : _ptr(nullptr) {}
  template <typename... Args> maybe_null(Args && ...args) : _ptr(std::forward<Args>(args)...) { }
  maybe_null(maybe_null const&) = default;
  template <typename Y> maybe_null(maybe_null<Y> const& other) : _ptr(other.ptr_) { }
  template <typename Y> maybe_null(not_null<Y> const& other) : _ptr(other.get()) { }
  
  maybe_null& operator=(T ptr) { return operator=(maybe_null<T>(ptr)); }
  maybe_null& operator=(maybe_null const&) = default;
  template <typename Y> maybe_null& operator=(maybe_null<Y> const&other) {
    if (_ptr != other.get()) {
      _ptr = other.get();
#if defined( DEBUG )
      tested_ = other.tested_;
#endif
    }
    return *this;
  }
  
  template <typename Y> maybe_null& operator=(not_null<Y> const&other) {
    _ptr = other.get();
#if defined( DEBUG )
    tested_ = true;
#endif
    return *this;
  }
  
  
  operator bool() const {
#if defined( DEBUG )
    tested_ = true;
#endif
    return bool(_ptr);
  }
  
  pointer get() const {
#if defined( DEBUG )
    if ( !tested_ ) {
      throw unchecked_pointer_exception{"did not verify that pointer was non-null"};
    }
#endif
    return std::addressof(operator*());
  }
  //  operator T() const { return get(); }
  pointer operator->() const { return get(); }
  reference operator*() const { return *_ptr; }
  
  bool operator==(maybe_null const&rhs) const { return _ptr == rhs._ptr; }
  bool operator!=(maybe_null const&rhs) const { return _ptr != rhs._ptr; }
  bool operator<=(maybe_null const&rhs) const { return _ptr <= rhs._ptr; }
  bool operator>=(maybe_null const&rhs) const { return _ptr >= rhs._ptr; }
  bool operator< (maybe_null const&rhs) const { return _ptr <  rhs._ptr; }
  bool operator> (maybe_null const&rhs) const { return _ptr >  rhs._ptr; }
private:
  T _ptr;
#if defined( DEBUG )
  mutable bool tested_ = false;
#endif
};