Allow a custom postprocessor for FilterDocs

Author: wilzbach

source/ddox/main.d

@@ -153,7 +153,16 @@ int setupGeneratorInput(ref string[] args, out GeneratorSettings gensettings, ou
 	return 0;
 }
 
-int cmdFilterDocs(string[] args)
+struct FilterConfig
+{
+	/**
+	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;
+}
+
+int cmdFilterDocs(string[] args, FilterConfig filterConfig = FilterConfig.init)
 {
 	string[] excluded, included;
 	Protection minprot = Protection.Private;
@@ -225,6 +234,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 (filterConfig.postUnittestSourceCode !is null)
+							source = filterConfig.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) {