1 //
2 // Copyright 2018 The Abseil Authors.
3 //
4 // Licensed under the Apache License, Version 2.0 (the "License");
5 // you may not use this file except in compliance with the License.
6 // You may obtain a copy of the License at
7 //
8 // https://www.apache.org/licenses/LICENSE-2.0
9 //
10 // Unless required by applicable law or agreed to in writing, software
11 // distributed under the License is distributed on an "AS IS" BASIS,
12 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 // See the License for the specific language governing permissions and
14 // limitations under the License.
15 //
16 // -----------------------------------------------------------------------------
17 // File: str_format.h
18 // -----------------------------------------------------------------------------
19 //
20 // The `str_format` library is a typesafe replacement for the family of
21 // `printf()` string formatting routines within the `<cstdio>` standard library
22 // header. Like the `printf` family, `str_format` uses a "format string" to
23 // perform argument substitutions based on types. See the `FormatSpec` section
24 // below for format string documentation.
25 //
26 // Example:
27 //
28 // std::string s = absl::StrFormat(
29 // "%s %s You have $%d!", "Hello", name, dollars);
30 //
31 // The library consists of the following basic utilities:
32 //
33 // * `absl::StrFormat()`, a type-safe replacement for `std::sprintf()`, to
34 // write a format string to a `string` value.
35 // * `absl::StrAppendFormat()` to append a format string to a `string`
36 // * `absl::StreamFormat()` to more efficiently write a format string to a
37 // stream, such as`std::cout`.
38 // * `absl::PrintF()`, `absl::FPrintF()` and `absl::SNPrintF()` as
39 // replacements for `std::printf()`, `std::fprintf()` and `std::snprintf()`.
40 //
41 // Note: a version of `std::sprintf()` is not supported as it is
42 // generally unsafe due to buffer overflows.
43 //
44 // Additionally, you can provide a format string (and its associated arguments)
45 // using one of the following abstractions:
46 //
47 // * A `FormatSpec` class template fully encapsulates a format string and its
48 // type arguments and is usually provided to `str_format` functions as a
49 // variadic argument of type `FormatSpec<Arg...>`. The `FormatSpec<Args...>`
50 // template is evaluated at compile-time, providing type safety.
51 // * A `ParsedFormat` instance, which encapsulates a specific, pre-compiled
52 // format string for a specific set of type(s), and which can be passed
53 // between API boundaries. (The `FormatSpec` type should not be used
54 // directly except as an argument type for wrapper functions.)
55 //
56 // The `str_format` library provides the ability to output its format strings to
57 // arbitrary sink types:
58 //
59 // * A generic `Format()` function to write outputs to arbitrary sink types,
60 // which must implement a `FormatRawSink` interface.
61 //
62 // * A `FormatUntyped()` function that is similar to `Format()` except it is
63 // loosely typed. `FormatUntyped()` is not a template and does not perform
64 // any compile-time checking of the format string; instead, it returns a
65 // boolean from a runtime check.
66 //
67 // In addition, the `str_format` library provides extension points for
68 // augmenting formatting to new types. See "StrFormat Extensions" below.
69
70 #ifndef ABSL_STRINGS_STR_FORMAT_H_
71 #define ABSL_STRINGS_STR_FORMAT_H_
72
73 #include <cstdio>
74 #include <string>
75
76 #include "absl/strings/internal/str_format/arg.h" // IWYU pragma: export
77 #include "absl/strings/internal/str_format/bind.h" // IWYU pragma: export
78 #include "absl/strings/internal/str_format/checker.h" // IWYU pragma: export
79 #include "absl/strings/internal/str_format/extension.h" // IWYU pragma: export
80 #include "absl/strings/internal/str_format/parser.h" // IWYU pragma: export
81
82 namespace absl {
83 ABSL_NAMESPACE_BEGIN
84
85 // UntypedFormatSpec
86 //
87 // A type-erased class that can be used directly within untyped API entry
88 // points. An `UntypedFormatSpec` is specifically used as an argument to
89 // `FormatUntyped()`.
90 //
91 // Example:
92 //
93 // absl::UntypedFormatSpec format("%d");
94 // std::string out;
95 // CHECK(absl::FormatUntyped(&out, format, {absl::FormatArg(1)}));
96 class UntypedFormatSpec {
97 public:
98 UntypedFormatSpec() = delete;
99 UntypedFormatSpec(const UntypedFormatSpec&) = delete;
100 UntypedFormatSpec& operator=(const UntypedFormatSpec&) = delete;
101
UntypedFormatSpec(string_view s)102 explicit UntypedFormatSpec(string_view s) : spec_(s) {}
103
104 protected:
UntypedFormatSpec(const str_format_internal::ParsedFormatBase * pc)105 explicit UntypedFormatSpec(const str_format_internal::ParsedFormatBase* pc)
106 : spec_(pc) {}
107
108 private:
109 friend str_format_internal::UntypedFormatSpecImpl;
110 str_format_internal::UntypedFormatSpecImpl spec_;
111 };
112
113 // FormatStreamed()
114 //
115 // Takes a streamable argument and returns an object that can print it
116 // with '%s'. Allows printing of types that have an `operator<<` but no
117 // intrinsic type support within `StrFormat()` itself.
118 //
119 // Example:
120 //
121 // absl::StrFormat("%s", absl::FormatStreamed(obj));
122 template <typename T>
FormatStreamed(const T & v)123 str_format_internal::StreamedWrapper<T> FormatStreamed(const T& v) {
124 return str_format_internal::StreamedWrapper<T>(v);
125 }
126
127 // FormatCountCapture
128 //
129 // This class provides a way to safely wrap `StrFormat()` captures of `%n`
130 // conversions, which denote the number of characters written by a formatting
131 // operation to this point, into an integer value.
132 //
133 // This wrapper is designed to allow safe usage of `%n` within `StrFormat(); in
134 // the `printf()` family of functions, `%n` is not safe to use, as the `int *`
135 // buffer can be used to capture arbitrary data.
136 //
137 // Example:
138 //
139 // int n = 0;
140 // std::string s = absl::StrFormat("%s%d%n", "hello", 123,
141 // absl::FormatCountCapture(&n));
142 // EXPECT_EQ(8, n);
143 class FormatCountCapture {
144 public:
FormatCountCapture(int * p)145 explicit FormatCountCapture(int* p) : p_(p) {}
146
147 private:
148 // FormatCountCaptureHelper is used to define FormatConvertImpl() for this
149 // class.
150 friend struct str_format_internal::FormatCountCaptureHelper;
151 // Unused() is here because of the false positive from -Wunused-private-field
152 // p_ is used in the templated function of the friend FormatCountCaptureHelper
153 // class.
Unused()154 int* Unused() { return p_; }
155 int* p_;
156 };
157
158 // FormatSpec
159 //
160 // The `FormatSpec` type defines the makeup of a format string within the
161 // `str_format` library. It is a variadic class template that is evaluated at
162 // compile-time, according to the format string and arguments that are passed to
163 // it.
164 //
165 // You should not need to manipulate this type directly. You should only name it
166 // if you are writing wrapper functions which accept format arguments that will
167 // be provided unmodified to functions in this library. Such a wrapper function
168 // might be a class method that provides format arguments and/or internally uses
169 // the result of formatting.
170 //
171 // For a `FormatSpec` to be valid at compile-time, it must be provided as
172 // either:
173 //
174 // * A `constexpr` literal or `absl::string_view`, which is how it most often
175 // used.
176 // * A `ParsedFormat` instantiation, which ensures the format string is
177 // valid before use. (See below.)
178 //
179 // Example:
180 //
181 // // Provided as a string literal.
182 // absl::StrFormat("Welcome to %s, Number %d!", "The Village", 6);
183 //
184 // // Provided as a constexpr absl::string_view.
185 // constexpr absl::string_view formatString = "Welcome to %s, Number %d!";
186 // absl::StrFormat(formatString, "The Village", 6);
187 //
188 // // Provided as a pre-compiled ParsedFormat object.
189 // // Note that this example is useful only for illustration purposes.
190 // absl::ParsedFormat<'s', 'd'> formatString("Welcome to %s, Number %d!");
191 // absl::StrFormat(formatString, "TheVillage", 6);
192 //
193 // A format string generally follows the POSIX syntax as used within the POSIX
194 // `printf` specification.
195 //
196 // (See http://pubs.opengroup.org/onlinepubs/9699919799/functions/fprintf.html.)
197 //
198 // In specific, the `FormatSpec` supports the following type specifiers:
199 // * `c` for characters
200 // * `s` for strings
201 // * `d` or `i` for integers
202 // * `o` for unsigned integer conversions into octal
203 // * `x` or `X` for unsigned integer conversions into hex
204 // * `u` for unsigned integers
205 // * `f` or `F` for floating point values into decimal notation
206 // * `e` or `E` for floating point values into exponential notation
207 // * `a` or `A` for floating point values into hex exponential notation
208 // * `g` or `G` for floating point values into decimal or exponential
209 // notation based on their precision
210 // * `p` for pointer address values
211 // * `n` for the special case of writing out the number of characters
212 // written to this point. The resulting value must be captured within an
213 // `absl::FormatCountCapture` type.
214 //
215 // Implementation-defined behavior:
216 // * A null pointer provided to "%s" or "%p" is output as "(nil)".
217 // * A non-null pointer provided to "%p" is output in hex as if by %#x or
218 // %#lx.
219 //
220 // NOTE: `o`, `x\X` and `u` will convert signed values to their unsigned
221 // counterpart before formatting.
222 //
223 // Examples:
224 // "%c", 'a' -> "a"
225 // "%c", 32 -> " "
226 // "%s", "C" -> "C"
227 // "%s", std::string("C++") -> "C++"
228 // "%d", -10 -> "-10"
229 // "%o", 10 -> "12"
230 // "%x", 16 -> "10"
231 // "%f", 123456789 -> "123456789.000000"
232 // "%e", .01 -> "1.00000e-2"
233 // "%a", -3.0 -> "-0x1.8p+1"
234 // "%g", .01 -> "1e-2"
235 // "%p", (void*)&value -> "0x7ffdeb6ad2a4"
236 //
237 // int n = 0;
238 // std::string s = absl::StrFormat(
239 // "%s%d%n", "hello", 123, absl::FormatCountCapture(&n));
240 // EXPECT_EQ(8, n);
241 //
242 // The `FormatSpec` intrinsically supports all of these fundamental C++ types:
243 //
244 // * Characters: `char`, `signed char`, `unsigned char`
245 // * Integers: `int`, `short`, `unsigned short`, `unsigned`, `long`,
246 // `unsigned long`, `long long`, `unsigned long long`
247 // * Floating-point: `float`, `double`, `long double`
248 //
249 // However, in the `str_format` library, a format conversion specifies a broader
250 // C++ conceptual category instead of an exact type. For example, `%s` binds to
251 // any string-like argument, so `std::string`, `absl::string_view`, and
252 // `const char*` are all accepted. Likewise, `%d` accepts any integer-like
253 // argument, etc.
254
255 template <typename... Args>
256 using FormatSpec = str_format_internal::FormatSpecTemplate<
257 str_format_internal::ArgumentToConv<Args>()...>;
258
259 // ParsedFormat
260 //
261 // A `ParsedFormat` is a class template representing a preparsed `FormatSpec`,
262 // with template arguments specifying the conversion characters used within the
263 // format string. Such characters must be valid format type specifiers, and
264 // these type specifiers are checked at compile-time.
265 //
266 // Instances of `ParsedFormat` can be created, copied, and reused to speed up
267 // formatting loops. A `ParsedFormat` may either be constructed statically, or
268 // dynamically through its `New()` factory function, which only constructs a
269 // runtime object if the format is valid at that time.
270 //
271 // Example:
272 //
273 // // Verified at compile time.
274 // absl::ParsedFormat<'s', 'd'> formatString("Welcome to %s, Number %d!");
275 // absl::StrFormat(formatString, "TheVillage", 6);
276 //
277 // // Verified at runtime.
278 // auto format_runtime = absl::ParsedFormat<'d'>::New(format_string);
279 // if (format_runtime) {
280 // value = absl::StrFormat(*format_runtime, i);
281 // } else {
282 // ... error case ...
283 // }
284
285 #if defined(__cpp_nontype_template_parameter_auto)
286 // If C++17 is available, an 'extended' format is also allowed that can specify
287 // multiple conversion characters per format argument, using a combination of
288 // `absl::FormatConversionCharSet` enum values (logically a set union)
289 // via the `|` operator. (Single character-based arguments are still accepted,
290 // but cannot be combined). Some common conversions also have predefined enum
291 // values, such as `absl::FormatConversionCharSet::kIntegral`.
292 //
293 // Example:
294 // // Extended format supports multiple conversion characters per argument,
295 // // specified via a combination of `FormatConversionCharSet` enums.
296 // using MyFormat = absl::ParsedFormat<absl::FormatConversionCharSet::d |
297 // absl::FormatConversionCharSet::x>;
298 // MyFormat GetFormat(bool use_hex) {
299 // if (use_hex) return MyFormat("foo %x bar");
300 // return MyFormat("foo %d bar");
301 // }
302 // // `format` can be used with any value that supports 'd' and 'x',
303 // // like `int`.
304 // auto format = GetFormat(use_hex);
305 // value = StringF(format, i);
306 template <auto... Conv>
307 using ParsedFormat = absl::str_format_internal::ExtendedParsedFormat<
308 absl::str_format_internal::ToFormatConversionCharSet(Conv)...>;
309 #else
310 template <char... Conv>
311 using ParsedFormat = str_format_internal::ExtendedParsedFormat<
312 absl::str_format_internal::ToFormatConversionCharSet(Conv)...>;
313 #endif // defined(__cpp_nontype_template_parameter_auto)
314
315 // StrFormat()
316 //
317 // Returns a `string` given a `printf()`-style format string and zero or more
318 // additional arguments. Use it as you would `sprintf()`. `StrFormat()` is the
319 // primary formatting function within the `str_format` library, and should be
320 // used in most cases where you need type-safe conversion of types into
321 // formatted strings.
322 //
323 // The format string generally consists of ordinary character data along with
324 // one or more format conversion specifiers (denoted by the `%` character).
325 // Ordinary character data is returned unchanged into the result string, while
326 // each conversion specification performs a type substitution from
327 // `StrFormat()`'s other arguments. See the comments for `FormatSpec` for full
328 // information on the makeup of this format string.
329 //
330 // Example:
331 //
332 // std::string s = absl::StrFormat(
333 // "Welcome to %s, Number %d!", "The Village", 6);
334 // EXPECT_EQ("Welcome to The Village, Number 6!", s);
335 //
336 // Returns an empty string in case of error.
337 template <typename... Args>
StrFormat(const FormatSpec<Args...> & format,const Args &...args)338 ABSL_MUST_USE_RESULT std::string StrFormat(const FormatSpec<Args...>& format,
339 const Args&... args) {
340 return str_format_internal::FormatPack(
341 str_format_internal::UntypedFormatSpecImpl::Extract(format),
342 {str_format_internal::FormatArgImpl(args)...});
343 }
344
345 // StrAppendFormat()
346 //
347 // Appends to a `dst` string given a format string, and zero or more additional
348 // arguments, returning `*dst` as a convenience for chaining purposes. Appends
349 // nothing in case of error (but possibly alters its capacity).
350 //
351 // Example:
352 //
353 // std::string orig("For example PI is approximately ");
354 // std::cout << StrAppendFormat(&orig, "%12.6f", 3.14);
355 template <typename... Args>
StrAppendFormat(std::string * dst,const FormatSpec<Args...> & format,const Args &...args)356 std::string& StrAppendFormat(std::string* dst,
357 const FormatSpec<Args...>& format,
358 const Args&... args) {
359 return str_format_internal::AppendPack(
360 dst, str_format_internal::UntypedFormatSpecImpl::Extract(format),
361 {str_format_internal::FormatArgImpl(args)...});
362 }
363
364 // StreamFormat()
365 //
366 // Writes to an output stream given a format string and zero or more arguments,
367 // generally in a manner that is more efficient than streaming the result of
368 // `absl:: StrFormat()`. The returned object must be streamed before the full
369 // expression ends.
370 //
371 // Example:
372 //
373 // std::cout << StreamFormat("%12.6f", 3.14);
374 template <typename... Args>
StreamFormat(const FormatSpec<Args...> & format,const Args &...args)375 ABSL_MUST_USE_RESULT str_format_internal::Streamable StreamFormat(
376 const FormatSpec<Args...>& format, const Args&... args) {
377 return str_format_internal::Streamable(
378 str_format_internal::UntypedFormatSpecImpl::Extract(format),
379 {str_format_internal::FormatArgImpl(args)...});
380 }
381
382 // PrintF()
383 //
384 // Writes to stdout given a format string and zero or more arguments. This
385 // function is functionally equivalent to `std::printf()` (and type-safe);
386 // prefer `absl::PrintF()` over `std::printf()`.
387 //
388 // Example:
389 //
390 // std::string_view s = "Ulaanbaatar";
391 // absl::PrintF("The capital of Mongolia is %s", s);
392 //
393 // Outputs: "The capital of Mongolia is Ulaanbaatar"
394 //
395 template <typename... Args>
PrintF(const FormatSpec<Args...> & format,const Args &...args)396 int PrintF(const FormatSpec<Args...>& format, const Args&... args) {
397 return str_format_internal::FprintF(
398 stdout, str_format_internal::UntypedFormatSpecImpl::Extract(format),
399 {str_format_internal::FormatArgImpl(args)...});
400 }
401
402 // FPrintF()
403 //
404 // Writes to a file given a format string and zero or more arguments. This
405 // function is functionally equivalent to `std::fprintf()` (and type-safe);
406 // prefer `absl::FPrintF()` over `std::fprintf()`.
407 //
408 // Example:
409 //
410 // std::string_view s = "Ulaanbaatar";
411 // absl::FPrintF(stdout, "The capital of Mongolia is %s", s);
412 //
413 // Outputs: "The capital of Mongolia is Ulaanbaatar"
414 //
415 template <typename... Args>
FPrintF(std::FILE * output,const FormatSpec<Args...> & format,const Args &...args)416 int FPrintF(std::FILE* output, const FormatSpec<Args...>& format,
417 const Args&... args) {
418 return str_format_internal::FprintF(
419 output, str_format_internal::UntypedFormatSpecImpl::Extract(format),
420 {str_format_internal::FormatArgImpl(args)...});
421 }
422
423 // SNPrintF()
424 //
425 // Writes to a sized buffer given a format string and zero or more arguments.
426 // This function is functionally equivalent to `std::snprintf()` (and
427 // type-safe); prefer `absl::SNPrintF()` over `std::snprintf()`.
428 //
429 // In particular, a successful call to `absl::SNPrintF()` writes at most `size`
430 // bytes of the formatted output to `output`, including a NUL-terminator, and
431 // returns the number of bytes that would have been written if truncation did
432 // not occur. In the event of an error, a negative value is returned and `errno`
433 // is set.
434 //
435 // Example:
436 //
437 // std::string_view s = "Ulaanbaatar";
438 // char output[128];
439 // absl::SNPrintF(output, sizeof(output),
440 // "The capital of Mongolia is %s", s);
441 //
442 // Post-condition: output == "The capital of Mongolia is Ulaanbaatar"
443 //
444 template <typename... Args>
SNPrintF(char * output,std::size_t size,const FormatSpec<Args...> & format,const Args &...args)445 int SNPrintF(char* output, std::size_t size, const FormatSpec<Args...>& format,
446 const Args&... args) {
447 return str_format_internal::SnprintF(
448 output, size, str_format_internal::UntypedFormatSpecImpl::Extract(format),
449 {str_format_internal::FormatArgImpl(args)...});
450 }
451
452 // -----------------------------------------------------------------------------
453 // Custom Output Formatting Functions
454 // -----------------------------------------------------------------------------
455
456 // FormatRawSink
457 //
458 // FormatRawSink is a type erased wrapper around arbitrary sink objects
459 // specifically used as an argument to `Format()`.
460 //
461 // All the object has to do define an overload of `AbslFormatFlush()` for the
462 // sink, usually by adding a ADL-based free function in the same namespace as
463 // the sink:
464 //
465 // void AbslFormatFlush(MySink* dest, absl::string_view part);
466 //
467 // where `dest` is the pointer passed to `absl::Format()`. The function should
468 // append `part` to `dest`.
469 //
470 // FormatRawSink does not own the passed sink object. The passed object must
471 // outlive the FormatRawSink.
472 class FormatRawSink {
473 public:
474 // Implicitly convert from any type that provides the hook function as
475 // described above.
476 template <typename T,
477 typename = typename std::enable_if<std::is_constructible<
478 str_format_internal::FormatRawSinkImpl, T*>::value>::type>
FormatRawSink(T * raw)479 FormatRawSink(T* raw) // NOLINT
480 : sink_(raw) {}
481
482 private:
483 friend str_format_internal::FormatRawSinkImpl;
484 str_format_internal::FormatRawSinkImpl sink_;
485 };
486
487 // Format()
488 //
489 // Writes a formatted string to an arbitrary sink object (implementing the
490 // `absl::FormatRawSink` interface), using a format string and zero or more
491 // additional arguments.
492 //
493 // By default, `std::string`, `std::ostream`, and `absl::Cord` are supported as
494 // destination objects. If a `std::string` is used the formatted string is
495 // appended to it.
496 //
497 // `absl::Format()` is a generic version of `absl::StrAppendFormat()`, for
498 // custom sinks. The format string, like format strings for `StrFormat()`, is
499 // checked at compile-time.
500 //
501 // On failure, this function returns `false` and the state of the sink is
502 // unspecified.
503 template <typename... Args>
Format(FormatRawSink raw_sink,const FormatSpec<Args...> & format,const Args &...args)504 bool Format(FormatRawSink raw_sink, const FormatSpec<Args...>& format,
505 const Args&... args) {
506 return str_format_internal::FormatUntyped(
507 str_format_internal::FormatRawSinkImpl::Extract(raw_sink),
508 str_format_internal::UntypedFormatSpecImpl::Extract(format),
509 {str_format_internal::FormatArgImpl(args)...});
510 }
511
512 // FormatArg
513 //
514 // A type-erased handle to a format argument specifically used as an argument to
515 // `FormatUntyped()`. You may construct `FormatArg` by passing
516 // reference-to-const of any printable type. `FormatArg` is both copyable and
517 // assignable. The source data must outlive the `FormatArg` instance. See
518 // example below.
519 //
520 using FormatArg = str_format_internal::FormatArgImpl;
521
522 // FormatUntyped()
523 //
524 // Writes a formatted string to an arbitrary sink object (implementing the
525 // `absl::FormatRawSink` interface), using an `UntypedFormatSpec` and zero or
526 // more additional arguments.
527 //
528 // This function acts as the most generic formatting function in the
529 // `str_format` library. The caller provides a raw sink, an unchecked format
530 // string, and (usually) a runtime specified list of arguments; no compile-time
531 // checking of formatting is performed within this function. As a result, a
532 // caller should check the return value to verify that no error occurred.
533 // On failure, this function returns `false` and the state of the sink is
534 // unspecified.
535 //
536 // The arguments are provided in an `absl::Span<const absl::FormatArg>`.
537 // Each `absl::FormatArg` object binds to a single argument and keeps a
538 // reference to it. The values used to create the `FormatArg` objects must
539 // outlive this function call. (See `str_format_arg.h` for information on
540 // the `FormatArg` class.)_
541 //
542 // Example:
543 //
544 // std::optional<std::string> FormatDynamic(
545 // const std::string& in_format,
546 // const vector<std::string>& in_args) {
547 // std::string out;
548 // std::vector<absl::FormatArg> args;
549 // for (const auto& v : in_args) {
550 // // It is important that 'v' is a reference to the objects in in_args.
551 // // The values we pass to FormatArg must outlive the call to
552 // // FormatUntyped.
553 // args.emplace_back(v);
554 // }
555 // absl::UntypedFormatSpec format(in_format);
556 // if (!absl::FormatUntyped(&out, format, args)) {
557 // return std::nullopt;
558 // }
559 // return std::move(out);
560 // }
561 //
FormatUntyped(FormatRawSink raw_sink,const UntypedFormatSpec & format,absl::Span<const FormatArg> args)562 ABSL_MUST_USE_RESULT inline bool FormatUntyped(
563 FormatRawSink raw_sink, const UntypedFormatSpec& format,
564 absl::Span<const FormatArg> args) {
565 return str_format_internal::FormatUntyped(
566 str_format_internal::FormatRawSinkImpl::Extract(raw_sink),
567 str_format_internal::UntypedFormatSpecImpl::Extract(format), args);
568 }
569
570 //------------------------------------------------------------------------------
571 // StrFormat Extensions
572 //------------------------------------------------------------------------------
573 //
574 // AbslFormatConvert()
575 //
576 // The StrFormat library provides a customization API for formatting
577 // user-defined types using absl::StrFormat(). The API relies on detecting an
578 // overload in the user-defined type's namespace of a free (non-member)
579 // `AbslFormatConvert()` function, usually as a friend definition with the
580 // following signature:
581 //
582 // absl::FormatConvertResult<...> AbslFormatConvert(
583 // const X& value,
584 // const absl::FormatConversionSpec& spec,
585 // absl::FormatSink *sink);
586 //
587 // An `AbslFormatConvert()` overload for a type should only be declared in the
588 // same file and namespace as said type.
589 //
590 // The abstractions within this definition include:
591 //
592 // * An `absl::FormatConversionSpec` to specify the fields to pull from a
593 // user-defined type's format string
594 // * An `absl::FormatSink` to hold the converted string data during the
595 // conversion process.
596 // * An `absl::FormatConvertResult` to hold the status of the returned
597 // formatting operation
598 //
599 // The return type encodes all the conversion characters that your
600 // AbslFormatConvert() routine accepts. The return value should be {true}.
601 // A return value of {false} will result in `StrFormat()` returning
602 // an empty string. This result will be propagated to the result of
603 // `FormatUntyped`.
604 //
605 // Example:
606 //
607 // struct Point {
608 // // To add formatting support to `Point`, we simply need to add a free
609 // // (non-member) function `AbslFormatConvert()`. This method interprets
610 // // `spec` to print in the request format. The allowed conversion characters
611 // // can be restricted via the type of the result, in this example
612 // // string and integral formatting are allowed (but not, for instance
613 // // floating point characters like "%f"). You can add such a free function
614 // // using a friend declaration within the body of the class:
615 // friend absl::FormatConvertResult<absl::FormatConversionCharSet::kString |
616 // absl::FormatConversionCharSet::kIntegral>
617 // AbslFormatConvert(const Point& p, const absl::FormatConversionSpec& spec,
618 // absl::FormatSink* s) {
619 // if (spec.conversion_char() == absl::FormatConversionChar::s) {
620 // s->Append(absl::StrCat("x=", p.x, " y=", p.y));
621 // } else {
622 // s->Append(absl::StrCat(p.x, ",", p.y));
623 // }
624 // return {true};
625 // }
626 //
627 // int x;
628 // int y;
629 // };
630
631 // clang-format off
632
633 // FormatConversionChar
634 //
635 // Specifies the formatting character provided in the format string
636 // passed to `StrFormat()`.
637 enum class FormatConversionChar : uint8_t {
638 c, s, // text
639 d, i, o, u, x, X, // int
640 f, F, e, E, g, G, a, A, // float
641 n, p // misc
642 };
643 // clang-format on
644
645 // FormatConversionSpec
646 //
647 // Specifies modifications to the conversion of the format string, through use
648 // of one or more format flags in the source format string.
649 class FormatConversionSpec {
650 public:
651 // FormatConversionSpec::is_basic()
652 //
653 // Indicates that width and precision are not specified, and no additional
654 // flags are set for this conversion character in the format string.
is_basic()655 bool is_basic() const { return impl_.is_basic(); }
656
657 // FormatConversionSpec::has_left_flag()
658 //
659 // Indicates whether the result should be left justified for this conversion
660 // character in the format string. This flag is set through use of a '-'
661 // character in the format string. E.g. "%-s"
has_left_flag()662 bool has_left_flag() const { return impl_.has_left_flag(); }
663
664 // FormatConversionSpec::has_show_pos_flag()
665 //
666 // Indicates whether a sign column is prepended to the result for this
667 // conversion character in the format string, even if the result is positive.
668 // This flag is set through use of a '+' character in the format string.
669 // E.g. "%+d"
has_show_pos_flag()670 bool has_show_pos_flag() const { return impl_.has_show_pos_flag(); }
671
672 // FormatConversionSpec::has_sign_col_flag()
673 //
674 // Indicates whether a mandatory sign column is added to the result for this
675 // conversion character. This flag is set through use of a space character
676 // (' ') in the format string. E.g. "% i"
has_sign_col_flag()677 bool has_sign_col_flag() const { return impl_.has_sign_col_flag(); }
678
679 // FormatConversionSpec::has_alt_flag()
680 //
681 // Indicates whether an "alternate" format is applied to the result for this
682 // conversion character. Alternative forms depend on the type of conversion
683 // character, and unallowed alternatives are undefined. This flag is set
684 // through use of a '#' character in the format string. E.g. "%#h"
has_alt_flag()685 bool has_alt_flag() const { return impl_.has_alt_flag(); }
686
687 // FormatConversionSpec::has_zero_flag()
688 //
689 // Indicates whether zeroes should be prepended to the result for this
690 // conversion character instead of spaces. This flag is set through use of the
691 // '0' character in the format string. E.g. "%0f"
has_zero_flag()692 bool has_zero_flag() const { return impl_.has_zero_flag(); }
693
694 // FormatConversionSpec::conversion_char()
695 //
696 // Returns the underlying conversion character.
conversion_char()697 FormatConversionChar conversion_char() const {
698 return impl_.conversion_char();
699 }
700
701 // FormatConversionSpec::width()
702 //
703 // Returns the specified width (indicated through use of a non-zero integer
704 // value or '*' character) of the conversion character. If width is
705 // unspecified, it returns a negative value.
width()706 int width() const { return impl_.width(); }
707
708 // FormatConversionSpec::precision()
709 //
710 // Returns the specified precision (through use of the '.' character followed
711 // by a non-zero integer value or '*' character) of the conversion character.
712 // If precision is unspecified, it returns a negative value.
precision()713 int precision() const { return impl_.precision(); }
714
715 private:
FormatConversionSpec(str_format_internal::FormatConversionSpecImpl impl)716 explicit FormatConversionSpec(
717 str_format_internal::FormatConversionSpecImpl impl)
718 : impl_(impl) {}
719
720 friend str_format_internal::FormatConversionSpecImpl;
721
722 absl::str_format_internal::FormatConversionSpecImpl impl_;
723 };
724
725 // Type safe OR operator for FormatConversionCharSet to allow accepting multiple
726 // conversion chars in custom format converters.
727 constexpr FormatConversionCharSet operator|(FormatConversionCharSet a,
728 FormatConversionCharSet b) {
729 return static_cast<FormatConversionCharSet>(static_cast<uint64_t>(a) |
730 static_cast<uint64_t>(b));
731 }
732
733 // FormatConversionCharSet
734 //
735 // Specifies the _accepted_ conversion types as a template parameter to
736 // FormatConvertResult for custom implementations of `AbslFormatConvert`.
737 // Note the helper predefined alias definitions (kIntegral, etc.) below.
738 enum class FormatConversionCharSet : uint64_t {
739 // text
740 c = str_format_internal::FormatConversionCharToConvInt('c'),
741 s = str_format_internal::FormatConversionCharToConvInt('s'),
742 // integer
743 d = str_format_internal::FormatConversionCharToConvInt('d'),
744 i = str_format_internal::FormatConversionCharToConvInt('i'),
745 o = str_format_internal::FormatConversionCharToConvInt('o'),
746 u = str_format_internal::FormatConversionCharToConvInt('u'),
747 x = str_format_internal::FormatConversionCharToConvInt('x'),
748 X = str_format_internal::FormatConversionCharToConvInt('X'),
749 // Float
750 f = str_format_internal::FormatConversionCharToConvInt('f'),
751 F = str_format_internal::FormatConversionCharToConvInt('F'),
752 e = str_format_internal::FormatConversionCharToConvInt('e'),
753 E = str_format_internal::FormatConversionCharToConvInt('E'),
754 g = str_format_internal::FormatConversionCharToConvInt('g'),
755 G = str_format_internal::FormatConversionCharToConvInt('G'),
756 a = str_format_internal::FormatConversionCharToConvInt('a'),
757 A = str_format_internal::FormatConversionCharToConvInt('A'),
758 // misc
759 n = str_format_internal::FormatConversionCharToConvInt('n'),
760 p = str_format_internal::FormatConversionCharToConvInt('p'),
761
762 // Used for width/precision '*' specification.
763 kStar = static_cast<uint64_t>(
764 absl::str_format_internal::FormatConversionCharSetInternal::kStar),
765 // Some predefined values:
766 kIntegral = d | i | u | o | x | X,
767 kFloating = a | e | f | g | A | E | F | G,
768 kNumeric = kIntegral | kFloating,
769 kString = s,
770 kPointer = p,
771 };
772
773 // FormatSink
774 //
775 // An abstraction to which conversions write their string data.
776 //
777 class FormatSink {
778 public:
779 // Appends `count` copies of `ch`.
Append(size_t count,char ch)780 void Append(size_t count, char ch) { sink_->Append(count, ch); }
781
Append(string_view v)782 void Append(string_view v) { sink_->Append(v); }
783
784 // Appends the first `precision` bytes of `v`. If this is less than
785 // `width`, spaces will be appended first (if `left` is false), or
786 // after (if `left` is true) to ensure the total amount appended is
787 // at least `width`.
PutPaddedString(string_view v,int width,int precision,bool left)788 bool PutPaddedString(string_view v, int width, int precision, bool left) {
789 return sink_->PutPaddedString(v, width, precision, left);
790 }
791
792 private:
793 friend str_format_internal::FormatSinkImpl;
FormatSink(str_format_internal::FormatSinkImpl * s)794 explicit FormatSink(str_format_internal::FormatSinkImpl* s) : sink_(s) {}
795 str_format_internal::FormatSinkImpl* sink_;
796 };
797
798 // FormatConvertResult
799 //
800 // Indicates whether a call to AbslFormatConvert() was successful.
801 // This return type informs the StrFormat extension framework (through
802 // ADL but using the return type) of what conversion characters are supported.
803 // It is strongly discouraged to return {false}, as this will result in an
804 // empty string in StrFormat.
805 template <FormatConversionCharSet C>
806 struct FormatConvertResult {
807 bool value;
808 };
809
810 ABSL_NAMESPACE_END
811 } // namespace absl
812
813 #endif // ABSL_STRINGS_STR_FORMAT_H_
814