1 files changed, 109 insertions, 0 deletions
diff --git a/src/httpserver/qhttpserverviewtraits.h b/src/httpserver/qhttpserverviewtraits.h
new file mode 100644
index 0000000..0469ffb
--- /dev/null
+++ b/src/httpserver/qhttpserverviewtraits.h
@@ -0,0 +1,109 @@
+/****************************************************************************
+**
+** Copyright (C) 2020 Mikhail Svetkin <mikhail.svetkin@gmail.com>
+** Copyright (C) 2019 The Qt Company Ltd.
+** Contact: https://www.qt.io/licensing/
+**
+** This file is part of the QtHttpServer module of the Qt Toolkit.
+**
+** $QT_BEGIN_LICENSE:GPL$
+** Commercial License Usage
+** Licensees holding valid commercial Qt licenses may use this file in
+** accordance with the commercial license agreement provided with the
+** Software or, alternatively, in accordance with the terms contained in
+** a written agreement between you and The Qt Company. For licensing terms
+** and conditions see https://www.qt.io/terms-conditions. For further
+** information use the contact form at https://www.qt.io/contact-us.
+**
+** GNU General Public License Usage
+** Alternatively, this file may be used under the terms of the GNU
+** General Public License version 3 or (at your option) any later version
+** approved by the KDE Free Qt Foundation. The licenses are as published by
+** the Free Software Foundation and appearing in the file LICENSE.GPL3
+** included in the packaging of this file. Please review the following
+** information to ensure the GNU General Public License requirements will
+** be met: https://www.gnu.org/licenses/gpl-3.0.html.
+**
+** $QT_END_LICENSE$
+**
+****************************************************************************/
+
+#ifndef QHTTPSERVERVIEWTRAITS_H
+#define QHTTPSERVERVIEWTRAITS_H
+
+#include <QtHttpServer/qhttpserverviewtraits_impl.h>
+
+QT_BEGIN_NAMESPACE
+
+class QHttpServerRequest;
+class QHttpServerResponse;
+
+namespace QtPrivate {
+
+template <typename ViewHandler, bool DisableStaticAssert>
+struct AfterRequestViewTraitsHelper : ViewTraits<ViewHandler, DisableStaticAssert> {
+    using VTraits = ViewTraits<ViewHandler, DisableStaticAssert>;
+    using FunctionTraits = typename VTraits::FTraits;
+
+    static_assert(DisableStaticAssert ||
+                  FunctionTraits::ArgumentCount == 2 ||
+                  FunctionTraits::ArgumentCount == 1,
+                  "ViewHandler arguments error: "
+                  "afterRequest can only accept QHttpServerResponse and QHttpServerRequest");
+
+    static_assert(DisableStaticAssert ||
+                  std::is_same<typename FunctionTraits::ReturnType,
+                               QHttpServerResponse>::value,
+                  "ViewHandler return type error: "
+                  "Return type can only be QHttpServerResponse");
+
+    template<int I>
+    struct ArgumentChecker {
+        using IsRequest = typename VTraits::template Special<I, const QHttpServerRequest &>;
+        static_assert(IsRequest::AssertCondition,
+                      "ViewHandler arguments error: "
+                      "QHttpServerRequest can only be passed as a const reference");
+
+        using IsResponse = typename VTraits::template Special<I, QHttpServerResponse &&>;
+        static_assert(IsResponse::AssertCondition,
+                      "ViewHandler arguments error: "
+                      "QHttpServerResponse can only be passed as a universal reference");
+
+        using IsSpecial = CheckAny<IsRequest, IsResponse>;
+
+        static constexpr bool Valid = IsSpecial::Valid;
+        static constexpr bool StaticAssert = IsSpecial::StaticAssert;
+    };
+
+    struct Arguments {
+        template<int ... I>
+        struct ArgumentsReturn {
+            template<int Idx>
+            using Arg = ArgumentChecker<Idx>;
+            static constexpr bool Valid = QtPrivate::StaticMath::And::eval(Arg<I>::Valid...);
+            static constexpr bool StaticAssert = QtPrivate::StaticMath::And::eval(
+                    Arg<I>::StaticAssert...);
+            using Last = Arg<FunctionTraits::ArgumentIndexMax>;
+            static constexpr std::size_t Count = FunctionTraits::ArgumentCount;
+        };
+
+        template<int ... I>
+        static constexpr ArgumentsReturn<I...> eval(QtPrivate::IndexesList<I...>) noexcept
+        {
+            return ArgumentsReturn<I...>{};
+        }
+    };
+};
+
+} // namespace QtPrivate
+
+template <typename ViewHandler, bool DisableStaticAssert = false>
+struct QHttpServerAfterRequestViewTraits
+{
+    using Helpers = typename QtPrivate::AfterRequestViewTraitsHelper<ViewHandler, DisableStaticAssert>;
+    using Arguments = decltype(Helpers::Arguments::eval(typename Helpers::ArgumentIndexes{}));
+};
+
+QT_END_NAMESPACE
+
+#endif  // QHTTPSERVERVIEWTRAITS_H