protected string gatherHeadings(string src, int index, int prevIndex, HashSet<string> usingNamespaces, WcfCleanerOptions options)
{
var prevLines = src.Substring(prevIndex, index - prevIndex).Split('\n');
var emptyLineRegex = new Regex(options.DiscardEmptyRemarks ?
@"^\s*(\{|\/\/\/ \<remarks\/\>)?$" :
@"^\s*\{?$"
);
var attrNamespaceRegex = new Regex(@"(\s+\[)([\w_\.`\$]+)\.([^\]]+)(.+)");
// Search backwards through preceding lines looking for an empty line or one with just an open curly brace
int i = prevLines.Length - 2;
for (; i >= 0; i--)
{
if (emptyLineRegex.IsMatch(prevLines[i]))
break;
if (options.AddUsingStatements)
{
string line = prevLines[i];
var matches = RegexHelper.MatchesGeneric(attrNamespaceRegex, line);
if (matches.Count() == 1)
{
var groups = matches.First().Groups.Cast<Group>().ToArray();
string attrNamespace = groups[2].Value;
string attrStatement = groups[3].Value;
if (attrStatement.EndsWith("()"))
attrStatement = attrStatement.Substring(0, attrStatement.Length - 2);
prevLines[i] = groups[1].Value + attrStatement + groups[4].Value;
if (!usingNamespaces.Contains(attrNamespace))
usingNamespaces.Add(attrNamespace);
}
}
}
string headings = String.Join("\n", prevLines.Skip(i + 1).Take(prevLines.Length - i - 2));
return headings;
}
public void Clean(string sourcePath, string outputPath, WcfCleanerOptions options)
{
string src = File.ReadAllText(sourcePath).Replace("\r", ""); // Normalize to just \n for newlines
// 1) Gather any existing using statements, set aside
int namespaceIndex = src.IndexOf("namespace");
var usings = src.Substring(0, namespaceIndex);
var usingNamespaces = new HashSet<string>(RegexHelper.ListGroups(@"using ([^;]+);", usings));
// 2) Gather namespace, set aside
var fileNamespace = RegexHelper.FirstGroup(@"namespace ([\w_`\$\.]+)", src, namespaceIndex);
// 3) Begin class loop
int index = src.IndexOf('\n', namespaceIndex + 1) + 1;
var codeSections = parseSections(src, index, usingNamespaces, options);
// 8) Write out
using (var text = new FileStreamWriter(outputPath))
{
writeFile(text, usingNamespaces, fileNamespace, codeSections, options);
}
}
protected KeyValuePair<string, CodeBlock> writeProperty(FileStreamWriter text, WcfCleanerOptions options, CodeClass codeClass, KeyValuePair<string, CodeBlock> kv)
{
var prop = (CodeProperty)kv.Value;
if (options.AutoSetSpecified && codeClass.Properties.ContainsKey(kv.Key + "Specified"))
{
text.WriteLine("\t\tprotected ", prop.Type, " _", kv.Key, ";")
.WriteLine()
.WriteLine(prop.Headings)
.WriteLine("\t\tpublic ", prop.Type, " ", kv.Key)
.WriteLine("\t\t{")
.WriteLine("\t\t\tget { return _", kv.Key, "; }")
.WriteLine("\t\t\tset {")
.WriteLine("\t\t\t\t_", kv.Key, " = value;")
.WriteLine("\t\t\t\t", kv.Key, "Specified = true;")
.WriteLine("\t\t\t}")
.WriteLine("\t\t}");
}
else
{
text.WriteLine(prop.Headings)
.WriteLine("\t\tpublic ", prop.Type, " ", kv.Key, " { get; set; }");
}
return kv;
}
protected string parseChunk(int index, int prevIndex, string src, HashSet<string> usingNamespaces, WcfCleanerOptions options)
{
string headers = gatherHeadings(src, index, prevIndex, usingNamespaces, options);
int endOfLine = src.IndexOf('\n', index);
// if ends with semicolon it's a one-liner like an event declaration
if (src[endOfLine - 1] == ';')
return headers + src.Substring(index, endOfLine - index);
int startChunk = 0;
int endChunk = 0;
countCodeBraces(src, index, out startChunk, out endChunk);
return headers + src.Substring(index, endChunk - index + 1);
}
protected void writeFile(FileStreamWriter text, HashSet<string> usingNamespaces, string fileNamespace,
List<CodeSection> codeSections, WcfCleanerOptions options)
{
foreach (string usingNamespace in usingNamespaces)
{
text.WriteLine("using ", usingNamespace, ";");
}
text.WriteLine();
text.WriteLine("namespace ", fileNamespace)
.WriteLine("{");
var codeEnums = codeSections.Where(c => c is CodeEnum).Cast<CodeEnum>();
codeEnums.FencePostBefore(
codeEnum =>
{
text.WriteLine();
},
codeEnum =>
{
writeEnum(text, options, codeEnum);
}
);
var codeInterfaces = codeSections.Where(c => c is CodeInterface).Cast<CodeInterface>();
var interfaceAction = new Action<CodeInterface>(codeInterface => writeInterface(text, options, codeInterface));
if (codeEnums.Count() == 0)
codeInterfaces.FencePostBefore(c => text.WriteLine(), interfaceAction);
else
codeInterfaces.ForEach(interfaceAction);
var codeClasses = codeSections.Where(c => c is CodeClass).Cast<CodeClass>();
var classAction = new Action<CodeClass>(codeClass => writeClass(text, options, codeClass));
if (codeEnums.Count() == 0 && codeInterfaces.Count() == 0)
codeClasses.FencePostBefore(c => text.WriteLine(), classAction);
else
codeClasses.ForEach(classAction);
text.WriteLine("}");
}
protected void writeInterface(FileStreamWriter text, WcfCleanerOptions options, CodeInterface codeInterface)
{
text.WriteLine(codeInterface.Headings)
.WriteLine(codeInterface.Text);
}
protected void writeClass(FileStreamWriter text, WcfCleanerOptions options, CodeClass codeClass)
{
text.WriteLine(codeClass.Headings)
.WriteLine("\tpublic class ", codeClass.Name, " ", codeClass.Inheritance)
.WriteLine("\t{");
int written = 0;
codeClass.Chunks.FencePostBefore(
chunk =>
{
text.WriteLine();
},
chunk =>
{
written++;
writeCodeChunk(text, chunk);
}
);
if (written > 0)
{
foreach (var kv in codeClass.Properties)
{
text.WriteLine();
writeProperty(text, options, codeClass, kv);
}
}
else
{
codeClass.Properties.FencePostBefore(
kv =>
{
text.WriteLine();
},
kv =>
{
writeProperty(text, options, codeClass, kv);
}
);
}
text.WriteLine("\t}");
}
protected void writeEnum(FileStreamWriter text, WcfCleanerOptions options, CodeEnum codeEnum)
{
text.WriteLine(codeEnum.Headings)
.WriteLine("\tpublic enum ", codeEnum.Name)
.WriteLine("\t{");
codeEnum.Values.FencePostBefore(
enumValue =>
{
text.WriteLine();
},
enumValue =>
{
writeEnumValue(text, enumValue);
}
);
text.WriteLine("\t}");
}
protected CodeProperty parseProperty(string type, string name, int index, int prevIndex, string src, HashSet<string> usingNamespaces, WcfCleanerOptions options)
{
string headings = gatherHeadings(src, index, prevIndex, usingNamespaces, options);
var prop = new CodeProperty { Headings = headings, Type = type };
return prop;
}
protected List<CodeSection> parseSections(string src, int index, HashSet<string> usingNamespaces, WcfCleanerOptions options)
{
var enums = parseEnums(src, index, usingNamespaces, options);
var interfaces = parseInterfaces(src, index, usingNamespaces, options);
var classes = parseClasses(src, index, usingNamespaces, options);
var merged = enums.Union<CodeSection>(interfaces).Union<CodeSection>(classes).ToList();
return merged;
}
protected List<CodeInterface> parseInterfaces(string src, int namespaceIndex, HashSet<string> usingNamespaces, WcfCleanerOptions options)
{
var interfaces = new List<CodeInterface>();
var interfaceMatches = RegexHelper.MatchesGeneric(@"\s+public interface ([^ ]+) {", src.Substring(namespaceIndex));
int index = namespaceIndex;
foreach(var interfaceMatch in interfaceMatches)
{
int prevIndex = index;
index = interfaceMatch.Index + namespaceIndex;
string headings = gatherHeadings(src, index + 1, prevIndex, usingNamespaces, options);
// Count open/closing curly braces until end of interface
int startInterface = 0;
int endInterface = 0;
countCodeBraces(src, index, out startInterface, out endInterface);
string interfaceText = src.Substring(index + 1, endInterface - index);
var codeInterface = new CodeInterface
{
Headings = headings,
Text = interfaceText
};
interfaces.Add(codeInterface);
index = endInterface + 1;
}
return interfaces;
}
protected List<CodeClass> parseClasses(string src, int namespaceIndex, HashSet<string> usingNamespaces, WcfCleanerOptions options)
{
var classes = new List<CodeClass>();
var classMatches = RegexHelper.MatchesGeneric("public partial class ([^ ]+) ([^{]+)? ?{", src.Substring(namespaceIndex));
int index = namespaceIndex;
foreach(var classMatch in classMatches)
{
int prevIndex = index;
// 4) Parse forward until finding public partial class ____
index = classMatch.Index + namespaceIndex;
// 5) Gather preceding remarks and attributes
string headings = gatherHeadings(src, index, prevIndex, usingNamespaces, options);
// 6) Save class name, inheritance
var groups = classMatch.Groups;
var className = groups[1].Value;
var inheritance = groups[2].Value;
// TODO: Cleanup namespaces in inheritance, get rid of unnecessary : object
var codeClass = new CodeClass
{
Headings = headings,
Name = className,
Inheritance = inheritance
};
// Count open/closing curly braces until end of class
int startClass = 0;
int endClass = 0;
countCodeBraces(src, index, out startClass, out endClass);
string classBody = src.Substring(startClass, endClass - startClass);
// 7) Begin property/method loop
parseCodeBlocks(classBody, codeClass, usingNamespaces, options);
index = endClass + 1;
classes.Add(codeClass);
}
return classes;
}
protected CodeEnumValue parseEnumValue(string name, int index, int prevIndex, string enumBody, HashSet<string> usingNamespaces, WcfCleanerOptions options)
{
// Need to push index ahead a few so our skip back by 2 lines doesn't miss the line with attributes on them, if any.
string headings = gatherHeadings(enumBody, index + 3, prevIndex, usingNamespaces, options);
return new CodeEnumValue { Headings = headings, Name = name };
}
protected List<CodeEnum> parseEnums(string src, int namespaceIndex, HashSet<string> usingNamespaces, WcfCleanerOptions options)
{
var enums = new List<CodeEnum>();
var enumMatches = RegexHelper.MatchesGeneric("public enum ([^ ]+) {", src.Substring(namespaceIndex));
int index = namespaceIndex;
foreach (var enumMatch in enumMatches)
{
int prevIndex = index;
index = enumMatch.Index + namespaceIndex;
// 5) Gather preceding remarks and attributes
string headings = gatherHeadings(src, index, prevIndex, usingNamespaces, options);
// 6) Save enum name
var groups = enumMatch.Groups;
var enumName = groups[1].Value;
var codeEnum = new CodeEnum
{
Headings = headings,
Name = enumName
};
// Count open/closing curly braces until end of enum
int startEnum = 0;
int endEnum = 0;
countCodeBraces(src, index, out startEnum, out endEnum);
string enumBody = src.Substring(startEnum, endEnum - startEnum);
// 7) Begin property/method loop
parseEnumItems(enumBody, codeEnum, usingNamespaces, options);
index = endEnum + 1;
enums.Add(codeEnum);
}
return enums;
}
protected void parseEnumItems(string enumBody, CodeEnum codeEnum, HashSet<string> usingNamespaces, WcfCleanerOptions options)
{
var enumItems = RegexHelper.MatchesGeneric(@"\n\s+([\w_\$]+),\n", enumBody);
int index = 0;
foreach (var enumItemMatch in enumItems)
{
int prevIndex = index;
index = enumItemMatch.Index;
var groups = enumItemMatch.Groups.Cast<Group>().ToArray();
string name = groups[1].Value;
var enumValue = parseEnumValue(name, index, prevIndex, enumBody, usingNamespaces, options);
codeEnum.Values.Add(enumValue);
}
}
public void Clean(string sourcePath, WcfCleanerOptions options)
{
Clean(sourcePath, sourcePath, options);
}
protected void parseCodeBlocks(string src, CodeClass codeClass, HashSet<string> usingNamespaces, WcfCleanerOptions options)
{
//var chunks = RegexHelper.MatchesGeneric(@"(public|protected)? ?(event)? ?([\w_\.\[\]\<\>]+) ([\w_\$\<\>]+)(\()?", src);
/*
TODO: If method is private, generated code contains no protection statement - not public/protected/private, just
omitting it, meaning the remaining pattern is entirely contextual - we can't just search the broad way below,
we have to count curly braces throughout the entire class to identify these.
Semicolon and curly brace counting could do to find miscellaneous unhandled extras in classes as a catchall...
*/
var chunks = RegexHelper.MatchesGeneric(@"(public|protected) (event)? ?([\w_\.\[\]\<\>]+) ([\w_\$\<\>]+)(\()?", src);
int index = 0;
foreach (var chunkMatch in chunks)
{
int prevIndex = index;
index = chunkMatch.Index;
var groups = chunkMatch.Groups.Cast<Group>().ToArray();
// event
if (groups[2].Value == "event")
{
string chunk = parseChunk(index, prevIndex, src, usingNamespaces, options);
codeClass.Chunks.Add(chunk);
continue;
}
if (groups[5].Value == "(")
{
string chunk = parseChunk(index, prevIndex, src, usingNamespaces, options);
codeClass.Chunks.Add(chunk);
continue;
}
string name = groups[4].Value;
var prop = parseProperty(groups[3].Value, name, index, prevIndex, src, usingNamespaces, options);
codeClass.Properties.Add(name, prop);
}
}
