Change to ConstStringRef

Made it use std::visit() instead of std::holds_alternative()
ArthurSonzogni · Aug 10, 2023 · d3887e4 · d3887e4
1 parent 6a6b861
commit d3887e4
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/include/ftxui/util/ref.hpp b/include/ftxui/util/ref.hpp
@@ -105,12 +105,13 @@ class ConstStringRef {
 
   const std::string_view getStringView() const
   {
-    if ( std::holds_alternative<std::string_view>(variant_) )
-      return std::get<std::string_view>(variant_);
-    else if ( std::holds_alternative<std::string *>(variant_) )
-      return *std::get<std::string *>(variant_);
-    else if ( std::holds_alternative<std::string>(variant_) )
-      return std::get<std::string>(variant_);
+    struct Fn {
+      std::string_view operator()(std::string_view x) const { return x; }
+      std::string_view operator()(std::string &x) const { return x; }
+      std::string_view operator()(std::string *x) const { return *x; }
+    };
+
+    return std::visit(Fn(), variant_);
   }
 };