From 5ccdde398f0ec19c2297977524841c63d9ce1365 Mon Sep 17 00:00:00 2001
From: Sebastian Wilzbach <seb@wilzba.ch>
Date: Mon, 19 Feb 2018 06:05:50 +0100
Subject: [PATCH] Allow a custom postprocessor for FilterDocs

---
 source/ddox/main.d     | 18 +++++++++---------
 source/ddox/settings.d |  6 ++++++
 2 files changed, 15 insertions(+), 9 deletions(-)

diff --git a/source/ddox/main.d b/source/ddox/main.d
index 5e652378..a4dc7ca9 100644
--- a/source/ddox/main.d
+++ b/source/ddox/main.d
@@ -46,9 +46,8 @@ int ddoxMain(string[] args)
 	return 1;
 }
 
-int cmdGenerateHtml(string[] args)
+int cmdGenerateHtml(string[] args, GeneratorSettings gensettings = new GeneratorSettings())
 {
-	GeneratorSettings gensettings;
 	Package pack;
 	if( auto ret = setupGeneratorInput(args, gensettings, pack) )
 		return ret;
@@ -57,14 +56,13 @@ int cmdGenerateHtml(string[] args)
 	return 0;
 }
 
-int cmdServeHtml(string[] args)
+int cmdServeHtml(string[] args, GeneratorSettings gensettings = new GeneratorSettings())
 {
 	string[] webfiledirs;
 	getopt(args,
 		config.passThrough,
 		"web-file-dir", &webfiledirs);
 
-	GeneratorSettings gensettings;
 	Package pack;
 	if( auto ret = setupGeneratorInput(args, gensettings, pack) )
 		return ret;
@@ -84,11 +82,9 @@ int cmdServeHtml(string[] args)
 	return runEventLoop();
 }
 
-int cmdServeTest(string[] args)
+int cmdServeTest(string[] args, GeneratorSettings gensettings = new GeneratorSettings(), DdoxSettings docsettings = new DdoxSettings())
 {
 	string[] webfiledirs;
-	auto docsettings = new DdoxSettings;
-	auto gensettings = new GeneratorSettings;
 
 	auto pack = parseD(args[2 .. $]);
 
@@ -111,7 +107,9 @@ int cmdServeTest(string[] args)
 
 int setupGeneratorInput(ref string[] args, out GeneratorSettings gensettings, out Package pack)
 {
-	gensettings = new GeneratorSettings;
+	if (gensettings is null)
+		gensettings = new GeneratorSettings;
+
 	auto docsettings = new DdoxSettings;
 
 	string[] macrofiles;
@@ -153,7 +151,7 @@ int setupGeneratorInput(ref string[] args, out GeneratorSettings gensettings, ou
 	return 0;
 }
 
-int cmdFilterDocs(string[] args)
+int cmdFilterDocs(string[] args, GeneratorSettings generatorSettings = new GeneratorSettings())
 {
 	string[] excluded, included;
 	Protection minprot = Protection.Private;
@@ -225,6 +223,8 @@ int cmdFilterDocs(string[] args)
 					if (last_decl["comment"].opt!string.empty) {
 						writefln("Warning: Cannot add documented unit test %s to %s, which is not documented.", name, last_decl["name"].opt!string);
 					} else {
+						if (generatorSettings.postUnittestSourceCode !is null)
+							source = generatorSettings.postUnittestSourceCode(source);
 						last_decl["comment"] ~= format("Example:\n%s$(DDOX_UNITTEST_HEADER %s)\n---\n%s\n---\n$(DDOX_UNITTEST_FOOTER %s)\n", comment.strip, name, source, name);
 					}
 				} catch (Exception e) {
diff --git a/source/ddox/settings.d b/source/ddox/settings.d
index d26d393a..83a4d737 100644
--- a/source/ddox/settings.d
+++ b/source/ddox/settings.d
@@ -61,4 +61,10 @@ class GeneratorSettings {
 	@property bool lowerCase() const { return fileNameStyle == MethodStyle.lowerCase; }
 	deprecated("Use fileNameStyle = MethodStyle.lowerCase instead.")
 	@property void lowerCase(bool v) { fileNameStyle = MethodStyle.lowerCase; }
+
+	/**
+	Custom delegate that is called after the unittest source code has been parsed
+	Can be used to inject custom logic like an assert/writeln transformation.
+	*/
+	string delegate(string) postUnittestSourceCode;
 }