X-Git-Url: http://nitlanguage.org

diff --git a/src/test_parser.nit b/src/test_parser.nit
index 25cda14..fa95f13 100644
--- a/src/test_parser.nit
+++ b/src/test_parser.nit
@@ -18,32 +18,15 @@
 module test_parser
 
 import parser
-
-class PrintTreeVisitor
-	super Visitor
-	private var rank: Int
-	redef fun visit(n)
-	do
-		if n isa Token then
-			printn("  " * rank, n.class_name, " \"", n.text.escape_to_c, "\" ", n.location, "\n")
-		else
-			printn("  " * rank, n.class_name, " ", n.location, "\n")
-		end
-		rank = rank + 1
-		n.visit_all(self)
-		rank = rank - 1
-	end
-
-	init
-	do
-		rank = 0
-	end
-end
+import parser_util
+import astutil
 
 var no_print = false
 var only_lexer = false
 var need_help = false
 var no_file = false
+var interactive = false
+var xml = false
 
 while not args.is_empty and args.first.chars.first == '-' do
 	if args.first == "-n" then
@@ -52,8 +35,12 @@ while not args.is_empty and args.first.chars.first == '-' do
 		only_lexer = true
 	else if args.first == "-p" then
 		only_lexer = false
+	else if args.first == "-x" then
+		xml = true
 	else if args.first == "-e" then
 		no_file = true
+	else if args.first == "-i" then
+		interactive = true
 	else if args.first == "-h" or args.first == "-?" then
 		need_help = true
 	else
@@ -63,22 +50,61 @@ while not args.is_empty and args.first.chars.first == '-' do
 	args.shift
 end
 
-if args.is_empty or need_help then
+if (args.is_empty and not interactive) or need_help then
 	print("usage:")
 	print("  test_parser [options]... <filename.nit>...")
+	print("  test_parser -e [options]... <text>...")
+	print("  test_parser -i [options]...")
 	print("options:")
 	print("  -n	do not print anything")
 	print("  -l	only lexer")
 	print("  -p	lexer and parser (default)")
+	print("  -x	instead of a ascii tree, output a XML document")
 	print("  -e	instead on files, each argument is a content to parse")
+	print("  -i	tree to parse are read interactively")
 	print("  -h	print this help")
+else if interactive then
+	if only_lexer then
+		print "Error: -l and -i are incompatible"
+		exit 1
+	else if no_file then
+		print "Error: -e and -i are incompatible"
+		exit 1
+	else if not args.is_empty then
+		print "Error: -i works without arguments"
+		exit 1
+	end
+
+	var tc = new ToolContext
+
+	loop
+		var n = tc.interactive_parse("-->")
+		if n isa TString then
+			var s = n.text
+			if s == ":q" then
+				break
+			else
+				print "`:q` to quit"
+			end
+			continue
+		end
+
+		if n isa AError then
+			print "{n.location.colored_line("0;31")}: {n.message}"
+			continue
+		end
+
+		if not no_print then
+			n.dump_tree
+		end
+	end
 else
 	for a in args do
 		var source
 		if no_file then
 			source = new SourceFile.from_string("", a)
 		else
-			var f = new IFStream.open(a)
+			var f = new FileReader.open(a)
 			source = new SourceFile(a, f)
 			f.close
 		end
@@ -101,8 +127,11 @@ else
 				return
 			end
 
-			if not no_print then
-				(new PrintTreeVisitor).enter_visit(tree)
+			if xml then
+				tree.parentize_tokens
+				tree.to_xml.write_to(stdout)
+			else if not no_print then
+				tree.dump_tree
 			end
 		end
 	end