sqlpp11/include/sqlpp11/update_list.h

#pragma once

/*
 * Copyright (c) 2013-2016, Roland Bock
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without modification,
 * are permitted provided that the following conditions are met:
 *
 *   Redistributions of source code must retain the above copyright notice, this
 *   list of conditions and the following disclaimer.
 *
 *   Redistributions in binary form must reproduce the above copyright notice, this
 *   list of conditions and the following disclaimer in the documentation and/or
 *   other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
 * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR
 * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON
 * ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#include <sqlpp11/detail/type_set.h>
#include <sqlpp11/interpret_tuple.h>
#include <sqlpp11/interpretable_list.h>
#include <sqlpp11/type_traits.h>

namespace sqlpp
{
  // UPDATE ASSIGNMENTS DATA
  template <typename Database, typename... Assignments>
  struct update_list_data_t
  {
    update_list_data_t(std::tuple<Assignments...> assignments) : _assignments(assignments)
    {
    }

    update_list_data_t(const update_list_data_t&) = default;
    update_list_data_t(update_list_data_t&&) = default;
    update_list_data_t& operator=(const update_list_data_t&) = default;
    update_list_data_t& operator=(update_list_data_t&&) = default;
    ~update_list_data_t() = default;

    std::tuple<Assignments...> _assignments;
    interpretable_list_t<Database> _dynamic_assignments;
  };

  SQLPP_PORTABLE_STATIC_ASSERT(
      assert_no_unknown_tables_in_update_assignments_t,
      "at least one update assignment requires a table which is otherwise not known in the statement");

  // UPDATE ASSIGNMENTS
  template <typename Database, typename... Assignments>
  struct update_list_t
  {
    using _traits = make_traits<no_value_t, tag::is_update_list>;
    using _nodes = detail::type_vector<Assignments...>;
    using _is_dynamic = is_database<Database>;

    // Data
    using _data_t = update_list_data_t<Database, Assignments...>;

    // Member implementation with data and methods
    template <typename Policies>
    struct _impl_t
    {
      // workaround for msvc bug https://connect.microsoft.com/VisualStudio/Feedback/Details/2173269
      _impl_t() = default;
      _impl_t(const _data_t& data) : _data(data)
      {
      }

      template <typename Assignment>
      void add(Assignment assignment)
      {
        static_assert(_is_dynamic::value, "add must not be called for static from()");
        static_assert(is_assignment_t<Assignment>::value, "invalid assignment argument in add()");
        using _assigned_columns = detail::make_type_set_t<lhs_t<Assignments>...>;
        static_assert(not _assigned_columns::template count<lhs_t<Assignment>>(),
                      "Must not assign value to column twice");
        static_assert(logic::not_t<must_not_update_t, lhs_t<Assignment>>::value, "add() argument must not be updated");
        static_assert(Policies::template _no_unknown_tables<Assignment>::value,
                      "assignment uses tables unknown to this statement in add()");
        using ok = logic::all_t<_is_dynamic::value, is_assignment_t<Assignment>::value>;

        _add_impl(assignment, ok());  // dispatch to prevent compile messages after the static_assert
      }

    private:
      template <typename Assignment>
      void _add_impl(Assignment assignment, const std::true_type& /*unused*/)
      {
        _data._dynamic_assignments.emplace_back(assignment);
      }

      template <typename Assignment>
      void _add_impl(Assignment assignment, const std::false_type&);

    public:
      _data_t _data;
    };

    // Base template to be inherited by the statement
    template <typename Policies>
    struct _base_t
    {
      using _data_t = update_list_data_t<Database, Assignments...>;

      // workaround for msvc bug https://connect.microsoft.com/VisualStudio/Feedback/Details/2173269
      template <typename... Args>
      _base_t(Args&&... args) : assignments{std::forward<Args>(args)...}
      {
      }

      _impl_t<Policies> assignments;
      _impl_t<Policies>& operator()()
      {
        return assignments;
      }
      const _impl_t<Policies>& operator()() const
      {
        return assignments;
      }

      template <typename T>
      static auto _get_member(T t) -> decltype(t.assignments)
      {
        return t.assignments;
      }

      using _consistency_check = typename std::conditional<Policies::template _no_unknown_tables<update_list_t>::value,
                                                           consistent_t,
                                                           assert_no_unknown_tables_in_update_assignments_t>::type;
    };
  };

  SQLPP_PORTABLE_STATIC_ASSERT(assert_update_assignments_t, "update assignments required, i.e. set(...)");

  SQLPP_PORTABLE_STATIC_ASSERT(assert_update_set_assignments_t, "at least one argument is not an assignment in set()");
  SQLPP_PORTABLE_STATIC_ASSERT(assert_update_set_no_duplicates_t, "at least one duplicate column detected in set()");
  SQLPP_PORTABLE_STATIC_ASSERT(assert_update_set_allowed_t,
                               "at least one assignment is prohibited by its column definition in set()");
  SQLPP_PORTABLE_STATIC_ASSERT(assert_update_set_single_table_t,
                               "set() contains assignments for columns from more than one table");
  SQLPP_PORTABLE_STATIC_ASSERT(assert_update_set_count_args_t, "at least one assignment expression required in set()");
  SQLPP_PORTABLE_STATIC_ASSERT(assert_update_dynamic_set_statement_dynamic_t,
                               "dynamic_set() must not be called in a static statement");
  namespace detail
  {
    template <typename Assignment>
    struct lhs_must_not_update
    {
      static constexpr auto value = detail::must_not_update_impl<typename lhs<Assignment>::type>::type::value;
    };
  }  // namespace detail

  template <typename... Assignments>
  using check_update_set_t = static_combined_check_t<
      static_check_t<logic::all_t<detail::is_assignment_impl<Assignments>::type::value...>::value,
                     assert_update_set_assignments_t>,
      static_check_t<not detail::has_duplicates<typename lhs<Assignments>::type...>::value,
                     assert_update_set_no_duplicates_t>,
      static_check_t<logic::none_t<detail::lhs_must_not_update<Assignments>::value...>::value,
                     assert_update_set_allowed_t>,
      static_check_t<sizeof...(Assignments) == 0 or detail::make_joined_set_t<required_tables_of<
                                                        typename lhs<Assignments>::type>...>::size::value == 1,
                     assert_update_set_single_table_t>>;

  template <typename... Assignments>
  struct check_update_static_set
  {
    using type = static_combined_check_t<check_update_set_t<Assignments...>,
                                         static_check_t<sizeof...(Assignments) != 0, assert_update_set_count_args_t>>;
  };

  template <typename... Assignments>
  using check_update_static_set_t = typename check_update_static_set<Assignments...>::type;

  template <typename Database, typename... Assignments>
  struct check_update_dynamic_set
  {
    using type = static_combined_check_t<
        static_check_t<not std::is_same<Database, void>::value, assert_update_dynamic_set_statement_dynamic_t>,
        check_update_set_t<Assignments...>>;
  };

  template <typename... Assignments>
  using check_update_dynamic_set_t = typename check_update_dynamic_set<Assignments...>::type;

  struct no_update_list_t
  {
    using _traits = make_traits<no_value_t, tag::is_where>;
    using _nodes = detail::type_vector<>;

    // Data
    using _data_t = no_data_t;

    // Member implementation with data and methods
    template <typename Policies>
    struct _impl_t
    {
      // workaround for msvc bug https://connect.microsoft.com/VisualStudio/Feedback/Details/2173269
      _impl_t() = default;
      _impl_t(const _data_t& data) : _data(data)
      {
      }

      _data_t _data;
    };

    // Base template to be inherited by the statement
    template <typename Policies>
    struct _base_t
    {
      using _data_t = no_data_t;

      // workaround for msvc bug https://connect.microsoft.com/VisualStudio/Feedback/Details/2173269
      template <typename... Args>
      _base_t(Args&&... args) : no_assignments{std::forward<Args>(args)...}
      {
      }

      _impl_t<Policies> no_assignments;
      _impl_t<Policies>& operator()()
      {
        return no_assignments;
      }
      const _impl_t<Policies>& operator()() const
      {
        return no_assignments;
      }

      template <typename T>
      static auto _get_member(T t) -> decltype(t.no_assignments)
      {
        return t.no_assignments;
      }

      using _database_t = typename Policies::_database_t;

      template <typename Check, typename T>
      using _new_statement_t = new_statement_t<Check, Policies, no_update_list_t, T>;

      using _consistency_check = assert_update_assignments_t;

      template <typename... Assignments>
      auto set(Assignments... assignments) const
          -> _new_statement_t<check_update_static_set_t<Assignments...>, update_list_t<void, Assignments...>>
      {
        using Check = check_update_static_set_t<Assignments...>;
        return _set_impl<void>(Check{}, std::make_tuple(assignments...));
      }

      template <typename... Assignments>
      auto set(std::tuple<Assignments...> assignments) const
          -> _new_statement_t<check_update_static_set_t<Assignments...>, update_list_t<void, Assignments...>>
      {
        using Check = check_update_static_set_t<Assignments...>;
        return _set_impl<void>(Check{}, assignments);
      }

      template <typename... Assignments>
      auto dynamic_set(Assignments... assignments) const
          -> _new_statement_t<check_update_dynamic_set_t<_database_t, Assignments...>,
                              update_list_t<_database_t, Assignments...>>
      {
        using Check = check_update_dynamic_set_t<_database_t, Assignments...>;
        return _set_impl<_database_t>(Check{}, std::make_tuple(assignments...));
      }

    private:
      template <typename Database, typename Check, typename... Assignments>
      auto _set_impl(Check, Assignments... assignments) const -> inconsistent<Check>;

      template <typename Database, typename... Assignments>
      auto _set_impl(consistent_t /*unused*/, std::tuple<Assignments...> assignments) const
          -> _new_statement_t<consistent_t, update_list_t<Database, Assignments...>>
      {
        return {static_cast<const derived_statement_t<Policies>&>(*this),
                update_list_data_t<Database, Assignments...>{assignments}};
      }
    };
  };

  // Interpreters
  template <typename Context, typename Database, typename... Assignments>
  Context& serialize(const update_list_data_t<Database, Assignments...>& t, Context& context)
  {
    context << " SET ";
    interpret_tuple(t._assignments, ",", context);
    if (sizeof...(Assignments) and not t._dynamic_assignments.empty())
    {
      context << ',';
    }
    interpret_list(t._dynamic_assignments, ',', context);
    return context;
  }
}  // namespace sqlpp