nixos/nix
1
1
Fork 0
mirror of https://github.com/NixOS/nix.git synced 2025-11-15 06:52:43 +01:00
Code Activity

Implement operator<< for Suggestions

Browse source 
That way there’s no need to explicitely convert it to a string when
printing it
This commit is contained in:
regnat 2022-03-07 10:04:57 +01:00
parent fd45d85b41
commit 313bbc07a8
3 changed files with 21 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

20
src/libutil/suggestions.cc
View file

@ -65,27 +65,27 @@ Suggestions Suggestions::trim(int limit, int maxDistance) const
return Suggestions{res};
}
std::string Suggestion::pretty_print() const
std::string Suggestion::to_string() const
{
return ANSI_WARNING + filterANSIEscapes(suggestion) + ANSI_NORMAL;
}
std::string Suggestions::pretty_print() const
std::string Suggestions::to_string() const
{
switch (suggestions.size()) {
case 0:
return "";
case 1:
return suggestions.begin()->pretty_print();
return suggestions.begin()->to_string();
default: {
std::string res = "one of ";
auto iter = suggestions.begin();
res += iter->pretty_print(); // Iter can’t be end() because the container isn’t null
res += iter->to_string(); // Iter can’t be end() because the container isn’t null
iter++;
auto last = suggestions.end(); last--;
for ( ; iter != suggestions.end() ; iter++) {
res += (iter == last) ? " or " : ", ";
res += iter->pretty_print();
res += iter->to_string();
}
return res;
}
@ -101,4 +101,14 @@ Suggestions & Suggestions::operator+=(const Suggestions & other)
return *this;
}
std::ostream & operator<<(std::ostream & str, const Suggestion & suggestion)
{
return str << suggestion.to_string();
}
std::ostream & operator<<(std::ostream & str, const Suggestions & suggestions)
{
return str << suggestions.to_string();
}
}