Handle printing of extended tokens

1 files changed, 27 insertions, 22 deletions

diff --git a/build2/token.cxx b/build2/token.cxx
index 5a47eb7..bf2249f 100644
--- a/build2/token.cxx
+++ b/build2/token.cxx
@@ -8,34 +8,39 @@ using namespace std;
 
 namespace build2
 {
-  ostream&
-  operator<< (ostream& os, const token& t)
+  void
+  token_printer (ostream& os, const token& t, bool d)
   {
+    // Only quote non-name tokens for diagnostics.
+    //
+    const char* q (d ? "'" : "");
+
     switch (t.type)
     {
-    case token_type::eos:            os << "<end-of-file>"; break;
+    case token_type::eos:            os << "<end of file>"; break;
     case token_type::newline:        os << "<newline>"; break;
     case token_type::pair_separator: os << "<pair separator>"; break;
-    case token_type::colon:          os << "':'"; break;
-    case token_type::lcbrace:        os << "'{'"; break;
-    case token_type::rcbrace:        os << "'}'"; break;
-    case token_type::lsbrace:        os << "'['"; break;
-    case token_type::rsbrace:        os << "']'"; break;
-    case token_type::assign:         os << "'='"; break;
-    case token_type::prepend:        os << "'=+'"; break;
-    case token_type::append:         os << "'+='"; break;
-    case token_type::equal:          os << "'=='"; break;
-    case token_type::not_equal:      os << "'!='"; break;
-    case token_type::less:           os << "'<'";  break;
-    case token_type::greater:        os << "'>'";  break;
-    case token_type::less_equal:     os << "'<='"; break;
-    case token_type::greater_equal:  os << "'>='"; break;
-    case token_type::dollar:         os << "'$'"; break;
-    case token_type::lparen:         os << "'('"; break;
-    case token_type::rparen:         os << "')'"; break;
     case token_type::name:           os << '\'' << t.value << '\''; break;
-    }
 
-    return os;
+    case token_type::colon:          os << q << ':'  << q; break;
+    case token_type::lcbrace:        os << q << '{'  << q; break;
+    case token_type::rcbrace:        os << q << '}'  << q; break;
+    case token_type::lsbrace:        os << q << '['  << q; break;
+    case token_type::rsbrace:        os << q << ']'  << q; break;
+    case token_type::assign:         os << q << '='  << q; break;
+    case token_type::prepend:        os << q << "=+" << q; break;
+    case token_type::append:         os << q << "+=" << q; break;
+    case token_type::equal:          os << q << "==" << q; break;
+    case token_type::not_equal:      os << q << "!=" << q; break;
+    case token_type::less:           os << q << '<'  << q; break;
+    case token_type::greater:        os << q << '>'  << q; break;
+    case token_type::less_equal:     os << q << "<=" << q; break;
+    case token_type::greater_equal:  os << q << ">=" << q; break;
+    case token_type::dollar:         os << q << '$'  << q; break;
+    case token_type::lparen:         os << q << '('  << q; break;
+    case token_type::rparen:         os << q << ')'  << q; break;
+
+    default: assert (false); // Unhandled extended token.
+    }
   }
 }