// file      : odb/wrapper-traits.hxx
// license   : GNU GPL v2; see accompanying LICENSE file

#ifndef ODB_WRAPPER_TRAITS_HXX
#define ODB_WRAPPER_TRAITS_HXX

#include <odb/pre.hxx>

#include <memory>      // std::unique_ptr, std::shared_ptr/weak_ptr
#include <type_traits> // std::remove_const

#include <odb/nullable.hxx>

namespace odb
{
  template <typename T>
  class wrapper_traits;

  // Sample specialization for raw pointers. It is not enabled by default
  // since it makes many assumptions that may not always hold true (such
  // as that instances are allocated with new and freed with delete).
  // This makes it too dangerous to be enabled unconditionally. If you
  // need this functionality, you can copy the below code into your
  // application. Also consider changing it to only specialize for
  // specific types instead of for any pointer (it will almost always
  // do the wrong thing for char*).
  //
#if 0
  template <typename T>
  class wrapper_traits<T*>
  {
  public:
    typedef T wrapped_type;
    typedef T* wrapper_type;

    // T can be const.
    //
    typedef typename std::remove_const<T>::type unrestricted_wrapped_type;

    static const bool null_handler = true;
    static const bool null_default = false;

    static bool
    get_null (const wrapper_type& p)
    {
      return p == 0;
    }

    static void
    set_null (wrapper_type& p)
    {
      delete p;
      p = 0;
    }

    static const wrapped_type&
    get_ref (const wrapper_type& p)
    {
      return *p;
    }

    static unrestricted_wrapped_type&
    set_ref (wrapper_type& p)
    {
      if (p == 0)
        p = new unrestricted_wrapped_type;

      return const_cast<unrestricted_wrapped_type&> (*p);
    }
  };
#endif

  // Specialization for C++11 std::unique_ptr.
  //
  template <typename T, typename D>
  class wrapper_traits<std::unique_ptr<T, D>>
  {
  public:
    // T can be const.
    //
    typedef T wrapped_type;
    typedef std::unique_ptr<T, D> wrapper_type;

    // T can be const.
    //
    typedef typename std::remove_const<T>::type unrestricted_wrapped_type;

    static const bool null_handler = true;
    static const bool null_default = false;

    static bool
    get_null (const wrapper_type& p)
    {
      return !p;
    }

    static void
    set_null (wrapper_type& p)
    {
      p.reset ();
    }

    static const wrapped_type&
    get_ref (const wrapper_type& p)
    {
      return *p;
    }

    static unrestricted_wrapped_type&
    set_ref (wrapper_type& p)
    {
      if (!p)
        p.reset (new unrestricted_wrapped_type ());

      return const_cast<unrestricted_wrapped_type&> (*p);
    }
  };

  // Specialization for C++11 std::shared_ptr.
  //
  template <typename T>
  class wrapper_traits<std::shared_ptr<T>>
  {
  public:
    typedef T wrapped_type;
    typedef std::shared_ptr<T> wrapper_type;

    // T can be const.
    //
    typedef typename std::remove_const<T>::type unrestricted_wrapped_type;

    static const bool null_handler = true;
    static const bool null_default = false;

    static bool
    get_null (const wrapper_type& p)
    {
      return !p;
    }

    static void
    set_null (wrapper_type& p)
    {
      p.reset ();
    }

    static const wrapped_type&
    get_ref (const wrapper_type& p)
    {
      return *p;
    }

    static unrestricted_wrapped_type&
    set_ref (wrapper_type& p)
    {
      if (!p)
        p.reset (new unrestricted_wrapped_type);

      return const_cast<unrestricted_wrapped_type&> (*p);
    }
  };

  // Specialization for odb::nullable.
  //
  template <typename T>
  class wrapper_traits< nullable<T> >
  {
  public:
    // T can be const.
    //
    typedef T wrapped_type;
    typedef nullable<T> wrapper_type;

    // T can be const.
    //
    typedef typename std::remove_const<T>::type unrestricted_wrapped_type;

    static const bool null_handler = true;
    static const bool null_default = true;

    static bool
    get_null (const wrapper_type& n)
    {
      return n.null ();
    }

    static void
    set_null (wrapper_type& n)
    {
      n.reset ();
    }

    static const wrapped_type&
    get_ref (const wrapper_type& n)
    {
      return *n;
    }

    static unrestricted_wrapped_type&
    set_ref (wrapper_type& n)
    {
      if (n.null ())
        n = unrestricted_wrapped_type ();

      return const_cast<unrestricted_wrapped_type&> (*n);
    }
  };
}

#include <odb/post.hxx>

#endif // ODB_WRAPPER_TRAITS_HXX
