aoc_lib: make pretty_print::repr aware of aoc::as_number

This will apply the `char_as_number` flag for the entire argument of
`as_number`.

Author: yut23

aoc_lib/src/lib.hpp

@@ -402,7 +402,7 @@ ExpectInputHelper<std::string> expect_input(const char *expected) {
 // I/O manipulator that extracts an 8-bit number into a char instead of
 // a single ASCII character, and inserts a char as an 8-bit number.
 template <typename T>
-class as_number {
+struct as_number {
     T &dest;
     friend std::istream &operator>>(std::istream &is, as_number h) {
         static_assert(!std::is_const_v<T>,
@@ -435,7 +435,6 @@ class as_number {
         return os;
     }
 
-  public:
     explicit as_number(T &dest) : dest(dest) {}
 };
 

aoc_lib/src/test_pretty_print.cpp

@@ -76,7 +76,10 @@ std::size_t test_repr() {
     test('\x10', R"(16)", {.char_as_number = true});
     test(std::initializer_list<std::int8_t>{-2, 0, 15, -42, 127, -128},
          "{-2, 0, 15, -42, 127, -128}", {.char_as_number = true});
-    test(std::vector<std::uint8_t>{254, 0, 15, 214, 127, 128}, "{254, 0, 15, 214, 127, 128}", {.char_as_number = true});
+    std::vector<std::uint8_t> uint8_vec{254, 0, 15, 214, 127, 128};
+    test(uint8_vec, "{254, 0, 15, 214, 127, 128}", {.char_as_number = true});
+    test(aoc::as_number(uint8_vec), "{254, 0, 15, 214, 127, 128}");
+
     return suite.done(), suite.num_failed();
 }
 

aoc_lib/src/unit_test/pretty_print.hpp

@@ -1,6 +1,7 @@
 #ifndef PRETTY_PRINT_HPP
 #define PRETTY_PRINT_HPP
 
+#include "lib.hpp" // for as_number
 #include "util/concepts.hpp"
 #include <array>       // for array
 #include <cctype>      // for isprint
@@ -154,6 +155,15 @@ std::ostream &print_repr(std::ostream &os, const std::optional<T> &opt,
     return os;
 }
 
+// aoc::as_number
+template <pretty_print::has_print_repr T>
+std::ostream &print_repr(std::ostream &os, const aoc::as_number<T> &h,
+                         pretty_print::repr_state state) {
+    state.char_as_number = true;
+    os << pretty_print::repr(h.dest, state);
+    return os;
+}
+
 /*
  * Base/leaf types
  */
@@ -328,6 +338,8 @@ void _lint_helper_template(std::ostream &os, const T &t = T()) {
     _lint_helper_template<std::tuple<std::string, long, double>>(os);
     _lint_helper_template<std::optional<int>>(os);
     _lint_helper_template(os, std::strong_ordering::equal);
+    std::vector<char> vec{'a', 'b', 'c'};
+    _lint_helper_template(os, aoc::as_number(vec));
 }
 } // namespace