From 7da2a287ba80d80b5b418ffe5e260f073a27d964 Mon Sep 17 00:00:00 2001
From: Zach Toogood <zrtoogood@gmail.com>
Date: Sat, 27 Aug 2022 20:59:47 +0300
Subject: [PATCH] Add formatter for std::exception

Co-authored-by: fekir <federico.kircheis@gmail.com>
Co-authored-by: Alexey Ochapov <alexez@alexez.com>
---
 include/fmt/std.h | 18 +++++++++++++++++-
 test/std-test.cc  | 20 +++++++++++++++++++-
 2 files changed, 36 insertions(+), 2 deletions(-)

diff --git a/include/fmt/std.h b/include/fmt/std.h
index 41d2b2838b6d7..d93764c33dc97 100644
--- 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,21 @@ 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<std::string> {
+  template <typename FormatContext>
+  auto format(const std::exception& ex, FormatContext& ctx) const ->
+      typename FormatContext::iterator {
+    return fmt::formatter<std::string>::format(ex.what(), ctx);
+  }
+};
+FMT_END_NAMESPACE
 
 #endif  // FMT_STD_H_
diff --git a/test/std-test.cc b/test/std-test.cc
index c22b3e38fb39b..4d0e5b5188786 100644
--- 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,21 @@ TEST(std_test, variant) {
   EXPECT_EQ(fmt::format("{}", v5), "variant(\"yes, this is variant\")");
 #endif
 }
+
+TEST(std_test, exception) {
+  try {
+    std::vector<int> vec;
+    std::ignore = vec.at(42);
+  } catch (const std::exception& ex) {
+    EXPECT_EQ(fmt::format("{}", ex), "invalid vector subscript");
+    EXPECT_EQ(fmt::format("{:?}", ex), "\"invalid vector subscript\"");
+  }
+
+  try {
+    std::vector<int> vec;
+    std::ignore = vec.at(42);
+  } catch (const std::out_of_range& ex) {
+    EXPECT_EQ(fmt::format("{}", ex), "invalid vector subscript");
+    EXPECT_EQ(fmt::format("{:?}", ex), "\"invalid vector subscript\"");
+  }
+}