Support color printing strings with zero bytes

orivej · orivej · commit 4dc471febd0f · 2019-10-10T22:51:19.000Z
This is a clean up after #1351.
diff --git a/include/fmt/color.h b/include/fmt/color.h
@@ -526,8 +526,7 @@ void vprint(std::FILE* f, const text_style& ts, const S& format,
   if (has_style) {
     internal::reset_color<Char>(buf);
   }
-  buf.push_back(Char(0));
-  internal::fputs(buf.data(), f);
+  std::fwrite(buf.data(), 1, buf.size(), f);
 }
 
 /**
diff --git a/test/color-test.cc b/test/color-test.cc
@@ -45,6 +45,8 @@ TEST(ColorsTest, ColorsPrint) {
   EXPECT_WRITE(stdout,
                fmt::print(bg(fmt::terminal_color::bright_magenta), "tbmagenta"),
                "\x1b[105mtbmagenta\x1b[0m");
+  EXPECT_WRITE(stdout, fmt::print(fmt::emphasis::bold, "{}", std::string("a\0b", 3)),
+               std::string("\x1b[1ma\0b\x1b[0m", 11));
 }
 
 TEST(ColorsTest, Format) {

Original file line number	Diff line number	Diff line change
`@@ -526,8 +526,7 @@ void vprint(std::FILE* f, const text_style& ts, const S& format,`
`526`	`526`	`if (has_style) {`
`527`	`527`	`internal::reset_color<Char>(buf);`
`528`	`528`	`}`
`529`		`- buf.push_back(Char(0));`
`530`		`- internal::fputs(buf.data(), f);`
	`529`	`+ std::fwrite(buf.data(), 1, buf.size(), f);`
`531`	`530`	`}`
`532`	`531`
`533`	`532`	`/**`
Original file line number	Diff line number	Diff line change
`@@ -45,6 +45,8 @@ TEST(ColorsTest, ColorsPrint) {`
`45`	`45`	`EXPECT_WRITE(stdout,`
`46`	`46`	`fmt::print(bg(fmt::terminal_color::bright_magenta), "tbmagenta"),`
`47`	`47`	`"\x1b[105mtbmagenta\x1b[0m");`
	`48`	`+ EXPECT_WRITE(stdout, fmt::print(fmt::emphasis::bold, "{}", std::string("a\0b", 3)),`
	`49`	`+ std::string("\x1b[1ma\0b\x1b[0m", 11));`
`48`	`50`	`}`
`49`	`51`
`50`	`52`	`TEST(ColorsTest, Format) {`