From be54827b4e2b2bc7035763ea4e1cd981ccbc459e Mon Sep 17 00:00:00 2001
From: Hackerpilot <briancschott@gmail.com>
Date: Tue, 10 Sep 2013 22:47:58 -0700
Subject: [PATCH] Added outline output

---
 main.d     |  15 ++++-
 outliner.d | 163 +++++++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 176 insertions(+), 2 deletions(-)
 create mode 100644 outliner.d

diff --git a/main.d b/main.d
index 2e0abfb..0354c4d 100644
--- a/main.d
+++ b/main.d
@@ -23,6 +23,7 @@ import stats;
 import ctags;
 import astprinter;
 import imports;
+import outliner;
 
 int main(string[] args)
 {
@@ -37,13 +38,15 @@ int main(string[] args)
 	bool ast;
 	bool imports;
 	bool muffin;
+	bool outline;
 
 	try
 	{
 		getopt(args, "sloc|l", &sloc, "highlight", &highlight,
 			"ctags|c", &ctags, "recursive|r|R", &recursive, "help|h", &help,
 			"tokenCount|t", &tokenCount, "syntaxCheck|s", &syntaxCheck,
-			"ast|xml", &ast, "imports|i", &imports, "muffinButton", &muffin);
+			"ast|xml", &ast, "imports|i", &imports, "outline|o", &outline,
+			"muffinButton", &muffin);
 	}
 	catch (Exception e)
 	{
@@ -72,7 +75,7 @@ int main(string[] args)
 	}
 
 	auto optionCount = count!"a"([sloc, highlight, ctags, tokenCount,
-		syntaxCheck, ast, imports]);
+		syntaxCheck, ast, imports, outline]);
 	if (optionCount > 1)
 	{
 		stderr.writeln("Too many options specified");
@@ -150,6 +153,14 @@ int main(string[] args)
 			printer.output = stdout;
 			printer.visit(mod);
 		}
+		else if (outline)
+		{
+			auto tokens = byToken(usingStdin ? readStdin() : readFile(args[1]),
+				config);
+			auto mod = parseModule(tokens.array(), config.fileName);
+			auto outliner = new Outliner(stdout);
+			outliner.visit(mod);
+		}
 	}
 	return 0;
 }
diff --git a/outliner.d b/outliner.d
new file mode 100644
index 0000000..46647a1
--- /dev/null
+++ b/outliner.d
@@ -0,0 +1,163 @@
+//          Copyright Brian Schott (Sir Alaran) 2012.
+// Distributed under the Boost Software License, Version 1.0.
+//    (See accompanying file LICENSE_1_0.txt or copy at
+//          http://www.boost.org/LICENSE_1_0.txt)
+
+import stdx.d.lexer;
+import stdx.d.ast;
+import std.stdio;
+import std.string;
+import std.array;
+import std.conv;
+import formatter;
+
+class Outliner : ASTVisitor
+{
+	this(File output)
+	{
+		this.output = output;
+	}
+
+	override void visit(ClassDeclaration classDec)
+	{
+		printIndentation();
+		output.writeln("class ", classDec.name.value, " : ", classDec.name.line);
+		indent();
+		classDec.accept(this);
+		outdent();
+		finish();
+	}
+
+	override void visit(EnumDeclaration enumDec)
+	{
+		printIndentation();
+		output.writeln("enum ", enumDec.name.value, " : ", enumDec.name.line);
+		indent();
+		enumDec.accept(this);
+		outdent();
+		finish();
+	}
+
+	override void visit(EnumMember enumMem)
+	{
+		printIndentation();
+		output.writeln(enumMem.name.value, " : ", enumMem.name.line);
+		finish();
+	}
+
+	override void visit(FunctionDeclaration functionDec)
+	{
+		printIndentation();
+		if (functionDec.hasAuto)
+			output.write("auto ");
+		if (functionDec.hasRef)
+			output.write("ref ");
+		auto app = appender!(char[])();
+		auto f = new Formatter!(typeof(app))(app);
+		if (functionDec.returnType !is null)
+			f.format(functionDec.returnType);
+		app.put(" ");
+		app.put(functionDec.name.value);
+		f.format(functionDec.parameters);
+		app.put(" : ");
+		app.put(to!string(functionDec.name.line));
+		output.writeln(app.data);
+		finish();
+	}
+
+	override void visit(InterfaceDeclaration interfaceDec)
+	{
+		printIndentation();
+		output.writeln("interface ", interfaceDec.name.value, " : ",
+			interfaceDec.name.line);
+		indent();
+		interfaceDec.accept(this);
+		outdent();
+		finish();
+	}
+
+	override void visit(StructDeclaration structDec)
+	{
+		printIndentation();
+		output.writeln("struct ", structDec.name.value, " : ",
+			structDec.name.line);
+		indent();
+		structDec.accept(this);
+		outdent();
+		finish();
+	}
+
+	override void visit(TemplateDeclaration templateDeclaration)
+	{
+		printIndentation();
+		output.writeln("template", templateDeclaration.name.value, " : ",
+			templateDeclaration.name.line);
+		finish();
+	}
+
+	override void visit(StaticConstructor s) {}
+	override void visit(StaticDestructor s) {}
+	override void visit(SharedStaticConstructor s) {}
+	override void visit(SharedStaticDestructor s) {}
+	override void visit(Constructor c) {}
+	override void visit(Unittest u) {}
+
+	override void visit(UnionDeclaration unionDeclaration)
+	{
+		printIndentation();
+		output.writeln("union ", unionDeclaration.name.value, " : ",
+			unionDeclaration.name.line);
+		indent();
+		unionDeclaration.accept(this);
+		outdent();
+		finish();
+	}
+
+	override void visit(VariableDeclaration variableDeclaration)
+	{
+		foreach (Declarator d; variableDeclaration.declarators)
+		{
+			printIndentation();
+			auto app = appender!(char[])();
+			if (variableDeclaration.type !is null)
+			{
+				auto f = new Formatter!(typeof(app))(app);
+				f.format(variableDeclaration.type);
+			}
+			app.put(" ");
+			app.put(d.name.value);
+			app.put(" : ");
+			app.put(to!string(d.name.line));
+			output.writeln(app.data);
+		}
+		finish();
+	}
+
+	void finish()
+	{
+		if (indentLevel == 0)
+			output.writeln();
+	}
+
+	void printIndentation()
+	{
+		foreach (i; 0 .. indentLevel)
+			output.write("    ");
+	}
+
+	void indent()
+	{
+		indentLevel++;
+	}
+
+	void outdent()
+	{
+		indentLevel--;
+	}
+
+	int indentLevel;
+
+	alias ASTVisitor.visit visit;
+
+	File output;
+}