Add formatter for std::exception (#3062)

Co-authored-by: fekir <[email protected]> Co-authored-by: Alexey Ochapov <[email protected]> Co-authored-by: Vladislav Shchapov <[email protected]> Co-authored-by: fekir <[email protected]> Co-authored-by: Alexey Ochapov <[email protected]> Co-authored-by: Vladislav Shchapov <[email protected]>
fmtlib · Sep 2, 2022 · 4191477 · 4191477
1 parent 75383a8
commit 4191477
Show file tree

Hide file tree

Showing 2 changed files with 36 additions and 2 deletions.
diff --git a/include/fmt/std.h b/include/fmt/std.h
@@ -8,6 +8,7 @@
 #ifndef FMT_STD_H_
 #define FMT_STD_H_
 
+#include <exception>
 #include <thread>
 #include <type_traits>
 #include <utility>
@@ -166,6 +167,20 @@ struct formatter<
   }
 };
 FMT_END_NAMESPACE
-#endif
+#endif  // __cpp_lib_variant
+
+FMT_BEGIN_NAMESPACE
+template <typename T, typename Char>
+struct formatter<
+    T, Char,
+    typename std::enable_if<std::is_base_of<std::exception, T>::value>::type>
+    : formatter<string_view> {
+  template <typename FormatContext>
+  auto format(const std::exception& ex, FormatContext& ctx) const ->
+      typename FormatContext::iterator {
+    return fmt::formatter<string_view>::format(ex.what(), ctx);
+  }
+};
+FMT_END_NAMESPACE
 
 #endif  // FMT_STD_H_
diff --git a/test/std-test.cc b/test/std-test.cc
@@ -6,11 +6,11 @@
 // For the license information refer to format.h.
 
 #include "fmt/std.h"
-#include "fmt/ranges.h"
 
 #include <string>
 #include <vector>
 
+#include "fmt/ranges.h"
 #include "gtest/gtest.h"
 
 TEST(std_test, path) {
@@ -77,3 +77,22 @@ TEST(std_test, variant) {
   EXPECT_EQ(fmt::format("{}", v5), "variant(\"yes, this is variant\")");
 #endif
 }
+
+TEST(std_test, exception) {
+  std::string str("Test Exception");
+  std::string escstr = fmt::format("\"{}\"", str);
+
+  try {
+    throw std::runtime_error(str);
+  } catch (const std::exception& ex) {
+    EXPECT_EQ(fmt::format("{}", ex), str);
+    EXPECT_EQ(fmt::format("{:?}", ex), escstr);
+  }
+
+  try {
+    throw std::runtime_error(str);
+  } catch (const std::runtime_error& ex) {
+    EXPECT_EQ(fmt::format("{}", ex), str);
+    EXPECT_EQ(fmt::format("{:?}", ex), escstr);
+  }
+}