From 2fc3bdafe8842fe34d66ae15f1e0b383dbfc3f4b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Rimas=20Misevi=C4=8Dius?= <rmisev3@gmail.com>
Date: Sat, 26 Oct 2024 22:55:31 +0300
Subject: [PATCH] Add support for strings convertible to std::basic_string_view

---
 include/upa/str_arg.h    | 139 +++++++++++++++++++++++++++++----------
 include/upa/url_for_qt.h |  11 +++-
 2 files changed, 112 insertions(+), 38 deletions(-)

diff --git a/include/upa/str_arg.h b/include/upa/str_arg.h
index 0868461..5c19a3e 100644
--- a/include/upa/str_arg.h
+++ b/include/upa/str_arg.h
@@ -122,56 +122,123 @@ class str_arg {
 template<class T>
 using remove_cvptr_t = std::remove_cv_t<std::remove_pointer_t<T>>;
 
+template<class T>
+using remove_cvref_t = std::remove_cv_t<std::remove_reference_t<T>>;
+
 namespace detail {
-    // See: https://stackoverflow.com/a/9154394
-
-    // test class T has data() member
-    template<class T>
-    auto test_data(int) -> decltype(std::declval<T>().data());
-    template<class>
-    auto test_data(long) -> void;
-
-    // test class T has size() member
-    template<class T>
-    auto test_size(int) -> decltype(std::declval<T>().size());
-    template<class>
-    auto test_size(long) -> void;
-
-    // T::data() return type (void - if no such member)
-    template<class T>
-    using data_member_t = decltype(detail::test_data<T>(0));
-
-    // T::size() return type (void - if no such member)
-    template<class T>
-    using size_member_t = decltype(detail::test_size<T>(0));
+
+// See: https://stackoverflow.com/a/9154394
+
+// test class T has data() member
+template<class T>
+auto test_data(int) -> decltype(std::declval<T>().data());
+template<class>
+auto test_data(long) -> void;
+
+// test class T has size() member
+template<class T>
+auto test_size(int) -> decltype(std::declval<T>().size());
+template<class>
+auto test_size(long) -> void;
+
+// T::data() return type (void - if no such member)
+template<class T>
+using data_member_t = decltype(detail::test_data<T>(0));
+
+// T::size() return type (void - if no such member)
+template<class T>
+using size_member_t = decltype(detail::test_size<T>(0));
+
+// Check that StrT has data() and size() members of supported types
+
+template<class StrT>
+constexpr bool has_data_and_size_v =
+    std::is_pointer_v<detail::data_member_t<StrT>> &&
+    is_char_type_v<remove_cvptr_t<detail::data_member_t<StrT>>> &&
+    is_size_type_v<detail::size_member_t<StrT>>;
+
+// Check StrT is convertible to std::basic_string_view
+
+template<class StrT, typename CharT>
+constexpr bool convertible_to_string_view_v =
+    std::is_convertible_v<StrT, std::basic_string_view<CharT>> &&
+    !has_data_and_size_v<StrT> &&
+    !std::is_same_v<StrT, std::nullptr_t>;
+
+// Common class for converting input to str_arg
+
+template<typename CharT, class ArgT>
+struct str_arg_char_common {
+    using type = CharT;
+    static str_arg<CharT> to_str_arg(ArgT str) {
+        return { str.data(), str.size() };
+    }
+};
+
+// Default str_arg_char implementation
+
+template<class StrT, typename = void>
+struct str_arg_char_default {};
+
+// StrT has data() and size() members
+template<class StrT>
+struct str_arg_char_default<StrT, std::enable_if_t<
+    has_data_and_size_v<StrT>>>
+    : str_arg_char_common<
+    remove_cvptr_t<detail::data_member_t<StrT>>,
+    remove_cvref_t<StrT> const&> {};
+
+// StrT is convertible to std::basic_string_view
+template<class StrT>
+struct str_arg_char_default<StrT, std::enable_if_t<
+    convertible_to_string_view_v<StrT, char>>>
+    : str_arg_char_common<char, std::basic_string_view<char>> {};
+
+#ifdef __cpp_char8_t
+template<class StrT>
+struct str_arg_char_default<StrT, std::enable_if_t<
+    convertible_to_string_view_v<StrT, char8_t>>>
+    : str_arg_char_common<char8_t, std::basic_string_view<char8_t>> {};
+#endif
+
+template<class StrT>
+struct str_arg_char_default<StrT, std::enable_if_t<
+    convertible_to_string_view_v<StrT, char16_t>>>
+    : str_arg_char_common<char16_t, std::basic_string_view<char16_t>> {};
+
+template<class StrT>
+struct str_arg_char_default<StrT, std::enable_if_t<
+    convertible_to_string_view_v<StrT, char32_t>>>
+    : str_arg_char_common<char32_t, std::basic_string_view<char32_t>> {};
+
+template<class StrT>
+struct str_arg_char_default<StrT, std::enable_if_t<
+    convertible_to_string_view_v<StrT, wchar_t>>>
+    : str_arg_char_common<wchar_t, std::basic_string_view<wchar_t>> {};
+
 } // namespace detail
 
 
 // Requirements for string arguments
 
 template<class StrT, typename = void>
-struct str_arg_char {};
+struct str_arg_char : detail::str_arg_char_default<StrT> {};
 
 // Null terminated string
 template<class CharT>
-struct str_arg_char<CharT*> : std::remove_cv<CharT> {
-
-    template <typename T>
-    static str_arg<T> to_str_arg(const T* s) {
+struct str_arg_char<CharT*, std::enable_if_t<is_char_type_v<remove_cvref_t<CharT>>>> {
+    using type = remove_cvref_t<CharT>;
+    static str_arg<type> to_str_arg(const type* s) {
         return s;
     }
 };
 
-// String that has data() and size() members
-template<class StrT>
-struct str_arg_char<StrT> : std::enable_if<
-    std::is_pointer_v<detail::data_member_t<StrT>> &&
-    is_size_type_v<detail::size_member_t<StrT>>,
-    remove_cvptr_t<detail::data_member_t<StrT>>> {
-
-    template <class STR, typename T = typename STR::value_type>
-    static str_arg<T> to_str_arg(const STR& str) {
-        return { str.data(), str.size() };
+// str_arg input
+template<class CharT>
+struct str_arg_char<str_arg<CharT>> {
+    using type = CharT;
+    static str_arg<type> to_str_arg(str_arg<type> s) {
+        return s;
     }
 };
 
diff --git a/include/upa/url_for_qt.h b/include/upa/url_for_qt.h
index f461252..4f5afba 100644
--- a/include/upa/url_for_qt.h
+++ b/include/upa/url_for_qt.h
@@ -10,12 +10,17 @@
 #define UPA_URL_FOR_QT_H
 
 #include "url.h" // IWYU pragma: export
-#include <QString>
 #if __has_include(<QtVersionChecks>)
 # include <QtVersionChecks>
 #else
 # include <QtGlobal>
 #endif
+
+// As of version 6.7, Qt string classes are convertible to
+// std::basic_string_view, and such strings are supported in the
+// str_arg.h file.
+#if QT_VERSION < QT_VERSION_CHECK(6, 7, 0)
+#include <QString>
 #if QT_VERSION >= QT_VERSION_CHECK(5, 10, 0)
 # include <QStringView>
 #endif
@@ -32,7 +37,7 @@ struct str_arg_char_for_qt {
     using type = CharT;
 
     static str_arg<type> to_str_arg(const StrT& str) {
-        return { reinterpret_cast<const type*>(str.data()), str.length() };
+        return { reinterpret_cast<const type*>(str.data()), str.size() };
     }
 };
 
@@ -54,4 +59,6 @@ struct str_arg_char<QUtf8StringView> :
 
 } // namespace upa
 
+#endif // QT_VERSION < QT_VERSION_CHECK(6, 7, 0)
+
 #endif // UPA_URL_FOR_QT_H