From bf92abdb2c52b402b5b599e82cfde60b01f0301f Mon Sep 17 00:00:00 2001
From: ado <adnan.abd0124@gmail.com>
Date: Sun, 20 Dec 2020 03:16:17 +0100
Subject: [PATCH] add variant support

---
 include/ss/extract.hpp | 23 ++++++++++++++++++++++-
 1 file changed, 22 insertions(+), 1 deletion(-)

diff --git a/include/ss/extract.hpp b/include/ss/extract.hpp
index 4c84128..966f37b 100644
--- a/include/ss/extract.hpp
+++ b/include/ss/extract.hpp
@@ -7,6 +7,7 @@
 #include <optional>
 #include <stdexcept>
 #include <string>
+#include <variant>
 
 namespace ss {
 
@@ -315,7 +316,8 @@ struct unsupported_type {
 
 template <typename T>
 std::enable_if_t<!std::is_integral_v<T> && !std::is_floating_point_v<T> &&
-                     !is_instance_of<T, std::optional>::value,
+                     !is_instance_of<T, std::optional>::value &&
+                     !is_instance_of<T, std::variant>::value,
                  bool>
 extract(const char*, const char*, T&) {
         static_assert(error::unsupported_type<T>::value,
@@ -346,6 +348,25 @@ std::enable_if_t<is_instance_of<T, std::optional>::value, bool> extract(
         return true;
 }
 
+template <typename T, size_t I>
+bool extract_variant(const char* begin, const char* end, T& value) {
+        using IthType = std::variant_alternative_t<I, std::decay_t<T>>;
+        IthType ithValue;
+        if (extract<IthType>(begin, end, ithValue)) {
+                value = ithValue;
+                return true;
+        } else if constexpr (I + 1 < std::variant_size_v<T>) {
+                return extract_variant<T, I + 1>(begin, end, value);
+        }
+        return false;
+}
+
+template <typename T>
+std::enable_if_t<is_instance_of<T, std::variant>::value, bool> extract(
+    const char* begin, const char* end, T& value) {
+        return extract_variant<T, 0>(begin, end, value);
+}
+
 ////////////////
 // extract specialization
 ////////////////