Distinguish token quoting type and completeness

1 files changed, 94 insertions, 0 deletions

diff --git a/unit-tests/lexer/driver.cxx b/unit-tests/lexer/driver.cxx
new file mode 100644
index 0000000..326ac8a
--- /dev/null
+++ b/unit-tests/lexer/driver.cxx
@@ -0,0 +1,94 @@
+// file      : unit-tests/lexer/driver.cxx -*- C++ -*-
+// copyright : Copyright (c) 2014-2016 Code Synthesis Ltd
+// license   : MIT; see accompanying LICENSE file
+
+#include <cassert>
+#include <iostream>
+
+#include <build2/types>
+#include <build2/utility>
+
+#include <build2/token>
+#include <build2/lexer>
+
+using namespace std;
+
+namespace build2
+{
+  // Usage: argv[0] [-q] [<lexer-mode>]
+  //
+  int
+  main (int argc, char* argv[])
+  {
+    bool quote (false);
+    lexer_mode m (lexer_mode::normal);
+
+    for (int i (1); i != argc; ++i)
+    {
+      string a (argv[i]);
+
+      if (a == "-q")
+        quote = true;
+      else
+      {
+        if      (a == "normal")    m = lexer_mode::normal;
+        else if (a == "variable")  m = lexer_mode::variable;
+        else if (a == "value")     m = lexer_mode::value;
+        else if (a == "attribute") m = lexer_mode::attribute;
+        else if (a == "eval")      m = lexer_mode::eval;
+        else                       assert (false);
+        break;
+      }
+    }
+
+    try
+    {
+      cin.exceptions (istream::failbit | istream::badbit);
+
+      // Most alternative modes auto-expire so we need something underneath.
+      //
+      lexer l (cin, path ("stdin"));
+
+      if (m != lexer_mode::normal)
+        l.mode (m);
+
+      // No use printing eos since we will either get it or loop forever.
+      //
+      for (token t (l.next ()); t.type != token_type::eos; t = l.next ())
+      {
+        // Print each token on a separate line without quoting operators.
+        //
+        t.printer (cout, t, false);
+
+        if (quote)
+        {
+          char q ('\0');
+          switch (t.qtype)
+          {
+          case quote_type::single:   q = 'S'; break;
+          case quote_type::double_:  q = 'D'; break;
+          case quote_type::mixed:    q = 'M'; break;
+          case quote_type::unquoted:          break;
+          }
+
+          if (q != '\0')
+            cout << " [" << q << (t.qcomp ? "/C" : "/P") << ']';
+        }
+
+        cout << endl;
+      }
+    }
+    catch (const failed&)
+    {
+      return 1;
+    }
+
+    return 0;
+  }
+}
+
+int
+main (int argc, char* argv[])
+{
+  return build2::main (argc, argv);
+}