public TSqlObfuscatingFormatterTests()
{
_tokenizer = new TSqlStandardTokenizer();
_parser = new TSqlStandardParser();
_standardFormatter = new TSqlStandardFormatter("\t", 4, 999, true, true, false, true, true, true, false, true, false, true);
_obfuscatingFormatter = new TSqlObfuscatingFormatter();
}
public TSqlObfuscatingFormatterTests()
{
_tokenizer = new TSqlStandardTokenizer();
_parser = new TSqlStandardParser();
_standardFormatter = new TSqlStandardFormatter(new TSqlStandardFormatterOptions
{
TrailingCommas = true,
KeywordStandardization = true
});
_obfuscatingFormatter = new TSqlObfuscatingFormatter();
}
private void ProcessSqlNode(XmlElement contentElement, TSqlStandardFormattingState state)
{
int initialIndent = state.IndentLevel;
if (contentElement.GetAttribute(SqlXmlConstants.ANAME_HASERROR) == "1")
state.OpenClass(SqlHtmlConstants.CLASS_ERRORHIGHLIGHT);
switch (contentElement.Name)
{
case SqlXmlConstants.ENAME_SQL_STATEMENT:
WhiteSpace_SeparateStatements(contentElement, state);
state.ResetKeywords();
ProcessSqlNodeList(contentElement.SelectNodes("*"), state);
state.StatementBreakExpected = true;
break;
case SqlXmlConstants.ENAME_SQL_CLAUSE:
state.UnIndentInitialBreak = true;
ProcessSqlNodeList(contentElement.SelectNodes("*"), state.IncrementIndent());
state.DecrementIndent();
state.BreakExpected = true;
break;
case SqlXmlConstants.ENAME_SET_OPERATOR_CLAUSE:
state.DecrementIndent();
state.WhiteSpace_BreakToNextLine(); //this is the one already recommended by the start of the clause
state.WhiteSpace_BreakToNextLine(); //this is the one we additionally want to apply
ProcessSqlNodeList(contentElement.SelectNodes("*"), state.IncrementIndent());
state.BreakExpected = true;
state.AdditionalBreakExpected = true;
break;
case SqlXmlConstants.ENAME_BATCH_SEPARATOR:
//newline regardless of whether previous element recommended a break or not.
state.WhiteSpace_BreakToNextLine();
ProcessSqlNodeList(contentElement.SelectNodes("*"), state);
state.BreakExpected = true;
break;
case SqlXmlConstants.ENAME_DDL_PROCEDURAL_BLOCK:
case SqlXmlConstants.ENAME_DDL_OTHER_BLOCK:
case SqlXmlConstants.ENAME_DDL_DECLARE_BLOCK:
case SqlXmlConstants.ENAME_CURSOR_DECLARATION:
case SqlXmlConstants.ENAME_BEGIN_TRANSACTION:
case SqlXmlConstants.ENAME_SAVE_TRANSACTION:
case SqlXmlConstants.ENAME_COMMIT_TRANSACTION:
case SqlXmlConstants.ENAME_ROLLBACK_TRANSACTION:
case SqlXmlConstants.ENAME_CONTAINER_OPEN:
case SqlXmlConstants.ENAME_CONTAINER_CLOSE:
case SqlXmlConstants.ENAME_WHILE_LOOP:
case SqlXmlConstants.ENAME_IF_STATEMENT:
case SqlXmlConstants.ENAME_SELECTIONTARGET:
case SqlXmlConstants.ENAME_CONTAINER_GENERALCONTENT:
case SqlXmlConstants.ENAME_CTE_WITH_CLAUSE:
case SqlXmlConstants.ENAME_PERMISSIONS_BLOCK:
case SqlXmlConstants.ENAME_PERMISSIONS_DETAIL:
case SqlXmlConstants.ENAME_MERGE_CLAUSE:
case SqlXmlConstants.ENAME_MERGE_TARGET:
ProcessSqlNodeList(contentElement.SelectNodes("*"), state);
break;
case SqlXmlConstants.ENAME_CASE_INPUT:
case SqlXmlConstants.ENAME_BOOLEAN_EXPRESSION:
case SqlXmlConstants.ENAME_BETWEEN_LOWERBOUND:
case SqlXmlConstants.ENAME_BETWEEN_UPPERBOUND:
WhiteSpace_SeparateWords(state);
ProcessSqlNodeList(contentElement.SelectNodes("*"), state);
break;
case SqlXmlConstants.ENAME_CONTAINER_SINGLESTATEMENT:
case SqlXmlConstants.ENAME_CONTAINER_MULTISTATEMENT:
case SqlXmlConstants.ENAME_MERGE_ACTION:
state.BreakExpected = true;
ProcessSqlNodeList(contentElement.SelectNodes("*"), state);
state.StatementBreakExpected = false; //the responsibility for breaking will be with the OUTER statement; there should be no consequence propagating out from statements in this container;
state.UnIndentInitialBreak = false; //if there was no word spacing after the last content statement's clause starter, doesn't mean the unIndent should propagate to the following content!
break;
case SqlXmlConstants.ENAME_PERMISSIONS_TARGET:
case SqlXmlConstants.ENAME_PERMISSIONS_RECIPIENT:
case SqlXmlConstants.ENAME_DDL_WITH_CLAUSE:
case SqlXmlConstants.ENAME_MERGE_CONDITION:
case SqlXmlConstants.ENAME_MERGE_THEN:
state.BreakExpected = true;
state.UnIndentInitialBreak = true;
ProcessSqlNodeList(contentElement.SelectNodes("*"), state.IncrementIndent());
state.DecrementIndent();
break;
case SqlXmlConstants.ENAME_JOIN_ON_SECTION:
if (BreakJoinOnSections)
state.BreakExpected = true;
ProcessSqlNodeList(contentElement.SelectNodes(SqlXmlConstants.ENAME_CONTAINER_OPEN), state);
if (BreakJoinOnSections)
state.IncrementIndent();
ProcessSqlNodeList(contentElement.SelectNodes(SqlXmlConstants.ENAME_CONTAINER_GENERALCONTENT), state);
if (BreakJoinOnSections)
state.DecrementIndent();
break;
case SqlXmlConstants.ENAME_CTE_ALIAS:
state.UnIndentInitialBreak = true;
ProcessSqlNodeList(contentElement.SelectNodes("*"), state);
break;
case SqlXmlConstants.ENAME_ELSE_CLAUSE:
ProcessSqlNodeList(contentElement.SelectNodes(SqlXmlConstants.ENAME_CONTAINER_OPEN), state.DecrementIndent());
ProcessSqlNodeList(contentElement.SelectNodes(SqlXmlConstants.ENAME_CONTAINER_SINGLESTATEMENT), state.IncrementIndent());
break;
case SqlXmlConstants.ENAME_DDL_AS_BLOCK:
case SqlXmlConstants.ENAME_CURSOR_FOR_BLOCK:
state.BreakExpected = true;
ProcessSqlNodeList(contentElement.SelectNodes(SqlXmlConstants.ENAME_CONTAINER_OPEN), state.DecrementIndent());
state.BreakExpected = true;
ProcessSqlNodeList(contentElement.SelectNodes(SqlXmlConstants.ENAME_CONTAINER_GENERALCONTENT), state);
state.IncrementIndent();
break;
case SqlXmlConstants.ENAME_TRIGGER_CONDITION:
state.DecrementIndent();
state.WhiteSpace_BreakToNextLine();
ProcessSqlNodeList(contentElement.SelectNodes("*"), state.IncrementIndent());
break;
case SqlXmlConstants.ENAME_CURSOR_FOR_OPTIONS:
case SqlXmlConstants.ENAME_CTE_AS_BLOCK:
state.BreakExpected = true;
ProcessSqlNodeList(contentElement.SelectNodes(SqlXmlConstants.ENAME_CONTAINER_OPEN), state.DecrementIndent());
ProcessSqlNodeList(contentElement.SelectNodes(SqlXmlConstants.ENAME_CONTAINER_GENERALCONTENT), state.IncrementIndent());
break;
case SqlXmlConstants.ENAME_DDL_RETURNS:
case SqlXmlConstants.ENAME_MERGE_USING:
case SqlXmlConstants.ENAME_MERGE_WHEN:
state.BreakExpected = true;
state.UnIndentInitialBreak = true;
ProcessSqlNodeList(contentElement.SelectNodes("*"), state);
break;
case SqlXmlConstants.ENAME_BETWEEN_CONDITION:
ProcessSqlNodeList(contentElement.SelectNodes(SqlXmlConstants.ENAME_CONTAINER_OPEN), state);
state.IncrementIndent();
ProcessSqlNodeList(contentElement.SelectNodes(SqlXmlConstants.ENAME_BETWEEN_LOWERBOUND), state.IncrementIndent());
if (ExpandBetweenConditions)
state.BreakExpected = true;
ProcessSqlNodeList(contentElement.SelectNodes(SqlXmlConstants.ENAME_CONTAINER_CLOSE), state.DecrementIndent());
ProcessSqlNodeList(contentElement.SelectNodes(SqlXmlConstants.ENAME_BETWEEN_UPPERBOUND), state.IncrementIndent());
state.DecrementIndent();
state.DecrementIndent();
break;
case SqlXmlConstants.ENAME_DDLDETAIL_PARENS:
case SqlXmlConstants.ENAME_FUNCTION_PARENS:
//simply process sub-nodes - don't add space or expect any linebreaks (but respect linebreaks if necessary)
state.WordSeparatorExpected = false;
WhiteSpace_BreakAsExpected(state);
state.AddOutputContent(FormatOperator("("), Interfaces.SqlHtmlConstants.CLASS_OPERATOR);
ProcessSqlNodeList(contentElement.SelectNodes("*"), state.IncrementIndent());
state.DecrementIndent();
WhiteSpace_BreakAsExpected(state);
state.AddOutputContent(FormatOperator(")"), Interfaces.SqlHtmlConstants.CLASS_OPERATOR);
state.WordSeparatorExpected = true;
break;
case SqlXmlConstants.ENAME_DDL_PARENS:
case SqlXmlConstants.ENAME_EXPRESSION_PARENS:
case SqlXmlConstants.ENAME_SELECTIONTARGET_PARENS:
WhiteSpace_SeparateWords(state);
if (contentElement.Name.Equals(SqlXmlConstants.ENAME_EXPRESSION_PARENS))
state.IncrementIndent();
state.AddOutputContent(FormatOperator("("), Interfaces.SqlHtmlConstants.CLASS_OPERATOR);
TSqlStandardFormattingState innerState = new TSqlStandardFormattingState(state);
ProcessSqlNodeList(contentElement.SelectNodes("*"), innerState);
//if there was a linebreak in the parens content, or if it wanted one to follow, then put linebreaks before and after.
if (innerState.BreakExpected || innerState.OutputContainsLineBreak)
{
if (!innerState.StartsWithBreak)
state.WhiteSpace_BreakToNextLine();
state.Assimilate(innerState);
state.WhiteSpace_BreakToNextLine();
}
else
{
state.Assimilate(innerState);
}
state.AddOutputContent(FormatOperator(")"), Interfaces.SqlHtmlConstants.CLASS_OPERATOR);
if (contentElement.Name.Equals(SqlXmlConstants.ENAME_EXPRESSION_PARENS))
state.DecrementIndent();
state.WordSeparatorExpected = true;
break;
case SqlXmlConstants.ENAME_BEGIN_END_BLOCK:
case SqlXmlConstants.ENAME_TRY_BLOCK:
case SqlXmlConstants.ENAME_CATCH_BLOCK:
if (contentElement.ParentNode.Name.Equals(SqlXmlConstants.ENAME_SQL_CLAUSE)
&& contentElement.ParentNode.ParentNode.Name.Equals(SqlXmlConstants.ENAME_SQL_STATEMENT)
&& contentElement.ParentNode.ParentNode.ParentNode.Name.Equals(SqlXmlConstants.ENAME_CONTAINER_SINGLESTATEMENT)
)
state.DecrementIndent();
ProcessSqlNodeList(contentElement.SelectNodes(SqlXmlConstants.ENAME_CONTAINER_OPEN), state);
ProcessSqlNodeList(contentElement.SelectNodes(SqlXmlConstants.ENAME_CONTAINER_MULTISTATEMENT), state);
state.DecrementIndent();
state.BreakExpected = true;
ProcessSqlNodeList(contentElement.SelectNodes(SqlXmlConstants.ENAME_CONTAINER_CLOSE), state);
state.IncrementIndent();
if (contentElement.ParentNode.Name.Equals(SqlXmlConstants.ENAME_SQL_CLAUSE)
&& contentElement.ParentNode.ParentNode.Name.Equals(SqlXmlConstants.ENAME_SQL_STATEMENT)
&& contentElement.ParentNode.ParentNode.ParentNode.Name.Equals(SqlXmlConstants.ENAME_CONTAINER_SINGLESTATEMENT)
)
state.IncrementIndent();
break;
case SqlXmlConstants.ENAME_CASE_STATEMENT:
ProcessSqlNodeList(contentElement.SelectNodes(SqlXmlConstants.ENAME_CONTAINER_OPEN), state);
state.IncrementIndent();
ProcessSqlNodeList(contentElement.SelectNodes(SqlXmlConstants.ENAME_CASE_INPUT), state);
ProcessSqlNodeList(contentElement.SelectNodes(SqlXmlConstants.ENAME_CASE_WHEN), state);
ProcessSqlNodeList(contentElement.SelectNodes(SqlXmlConstants.ENAME_CASE_ELSE), state);
if (ExpandCaseStatements)
state.BreakExpected = true;
ProcessSqlNodeList(contentElement.SelectNodes(SqlXmlConstants.ENAME_CONTAINER_CLOSE), state);
state.DecrementIndent();
break;
case SqlXmlConstants.ENAME_CASE_WHEN:
case SqlXmlConstants.ENAME_CASE_THEN:
case SqlXmlConstants.ENAME_CASE_ELSE:
if (ExpandCaseStatements)
state.BreakExpected = true;
ProcessSqlNodeList(contentElement.SelectNodes(SqlXmlConstants.ENAME_CONTAINER_OPEN), state);
ProcessSqlNodeList(contentElement.SelectNodes(SqlXmlConstants.ENAME_CONTAINER_GENERALCONTENT), state.IncrementIndent());
ProcessSqlNodeList(contentElement.SelectNodes(SqlXmlConstants.ENAME_CASE_THEN), state);
state.DecrementIndent();
break;
case SqlXmlConstants.ENAME_AND_OPERATOR:
case SqlXmlConstants.ENAME_OR_OPERATOR:
if (ExpandBooleanExpressions)
state.BreakExpected = true;
ProcessSqlNodeList(contentElement.SelectNodes("*"), state);
break;
case SqlXmlConstants.ENAME_COMMENT_MULTILINE:
if (state.SpecialRegionActive == SpecialRegionType.NoFormat && contentElement.InnerXml.ToUpperInvariant().Contains("[/NOFORMAT]"))
{
XmlNode skippedXml = ExtractXmlBetween(state.RegionStartNode, contentElement);
TSqlIdentityFormatter tempFormatter = new TSqlIdentityFormatter(HTMLColoring);
state.AddOutputContentRaw(tempFormatter.FormatSQLTree(skippedXml));
state.SpecialRegionActive = null;
state.RegionStartNode = null;
}
else if (state.SpecialRegionActive == SpecialRegionType.Minify && contentElement.InnerXml.ToUpperInvariant().Contains("[/MINIFY]"))
{
XmlNode skippedXml = ExtractXmlBetween(state.RegionStartNode, contentElement);
TSqlObfuscatingFormatter tempFormatter = new TSqlObfuscatingFormatter();
if (HTMLFormatted)
state.AddOutputContentRaw(Utils.HtmlEncode(tempFormatter.FormatSQLTree(skippedXml)));
else
state.AddOutputContentRaw(tempFormatter.FormatSQLTree(skippedXml));
state.SpecialRegionActive = null;
state.RegionStartNode = null;
}
WhiteSpace_SeparateComment(contentElement, state);
state.AddOutputContent("/*" + contentElement.InnerText + "*/", Interfaces.SqlHtmlConstants.CLASS_COMMENT);
if (contentElement.ParentNode.Name.Equals(SqlXmlConstants.ENAME_SQL_STATEMENT)
|| (contentElement.NextSibling != null
&& contentElement.NextSibling.Name.Equals(SqlXmlConstants.ENAME_WHITESPACE)
&& Regex.IsMatch(contentElement.NextSibling.InnerText, @"(\r|\n)+")
)
)
//if this block comment is at the start or end of a statement, or if it was followed by a
// linebreak before any following content, then break here.
state.BreakExpected = true;
else
{
state.WordSeparatorExpected = true;
}
if (state.SpecialRegionActive == null && contentElement.InnerXml.ToUpperInvariant().Contains("[NOFORMAT]"))
{
state.AddOutputLineBreak();
state.SpecialRegionActive = SpecialRegionType.NoFormat;
state.RegionStartNode = contentElement;
}
else if (state.SpecialRegionActive == null && contentElement.InnerXml.ToUpperInvariant().Contains("[MINIFY]"))
{
state.AddOutputLineBreak();
state.SpecialRegionActive = SpecialRegionType.Minify;
state.RegionStartNode = contentElement;
}
break;
case SqlXmlConstants.ENAME_COMMENT_SINGLELINE:
if (state.SpecialRegionActive == SpecialRegionType.NoFormat && contentElement.InnerXml.ToUpperInvariant().Contains("[/NOFORMAT]"))
{
XmlNode skippedXml = ExtractXmlBetween(state.RegionStartNode, contentElement);
TSqlIdentityFormatter tempFormatter = new TSqlIdentityFormatter(HTMLColoring);
state.AddOutputContentRaw(tempFormatter.FormatSQLTree(skippedXml));
state.SpecialRegionActive = null;
state.RegionStartNode = null;
}
else if (state.SpecialRegionActive == SpecialRegionType.Minify && contentElement.InnerXml.ToUpperInvariant().Contains("[/MINIFY]"))
{
XmlNode skippedXml = ExtractXmlBetween(state.RegionStartNode, contentElement);
TSqlObfuscatingFormatter tempFormatter = new TSqlObfuscatingFormatter();
if (HTMLFormatted)
state.AddOutputContentRaw(Utils.HtmlEncode(tempFormatter.FormatSQLTree(skippedXml)));
else
state.AddOutputContentRaw(tempFormatter.FormatSQLTree(skippedXml));
state.SpecialRegionActive = null;
state.RegionStartNode = null;
}
WhiteSpace_SeparateComment(contentElement, state);
state.AddOutputContent("--" + contentElement.InnerText.Replace("\r", "").Replace("\n", ""), Interfaces.SqlHtmlConstants.CLASS_COMMENT);
state.BreakExpected = true;
state.SourceBreakPending = true;
if (state.SpecialRegionActive == null && contentElement.InnerXml.ToUpperInvariant().Contains("[NOFORMAT]"))
{
state.AddOutputLineBreak();
state.SpecialRegionActive = SpecialRegionType.NoFormat;
state.RegionStartNode = contentElement;
}
else if (state.SpecialRegionActive == null && contentElement.InnerXml.ToUpperInvariant().Contains("[MINIFY]"))
{
state.AddOutputLineBreak();
state.SpecialRegionActive = SpecialRegionType.Minify;
state.RegionStartNode = contentElement;
}
break;
case SqlXmlConstants.ENAME_STRING:
case SqlXmlConstants.ENAME_NSTRING:
WhiteSpace_SeparateWords(state);
string outValue = null;
if (contentElement.Name.Equals(SqlXmlConstants.ENAME_NSTRING))
outValue = "N'" + contentElement.InnerText.Replace("'", "''") + "'";
else
outValue = "'" + contentElement.InnerText.Replace("'", "''") + "'";
state.AddOutputContent(outValue, Interfaces.SqlHtmlConstants.CLASS_STRING);
state.WordSeparatorExpected = true;
break;
case SqlXmlConstants.ENAME_BRACKET_QUOTED_NAME:
WhiteSpace_SeparateWords(state);
state.AddOutputContent("[" + contentElement.InnerText.Replace("]", "]]") + "]");
state.WordSeparatorExpected = true;
break;
case SqlXmlConstants.ENAME_QUOTED_STRING:
WhiteSpace_SeparateWords(state);
state.AddOutputContent("\"" + contentElement.InnerText.Replace("\"", "\"\"") + "\"");
state.WordSeparatorExpected = true;
break;
case SqlXmlConstants.ENAME_COMMA:
//comma always ignores requested word spacing
if (TrailingCommas)
{
WhiteSpace_BreakAsExpected(state);
state.AddOutputContent(FormatOperator(","), Interfaces.SqlHtmlConstants.CLASS_OPERATOR);
if (ExpandCommaLists
&& !(contentElement.ParentNode.Name.Equals(SqlXmlConstants.ENAME_DDLDETAIL_PARENS)
|| contentElement.ParentNode.Name.Equals(SqlXmlConstants.ENAME_FUNCTION_PARENS)
)
)
state.BreakExpected = true;
else
state.WordSeparatorExpected = true;
}
else
{
if (ExpandCommaLists
&& !(contentElement.ParentNode.Name.Equals(SqlXmlConstants.ENAME_DDLDETAIL_PARENS)
|| contentElement.ParentNode.Name.Equals(SqlXmlConstants.ENAME_FUNCTION_PARENS)
)
)
{
state.WhiteSpace_BreakToNextLine();
state.AddOutputContent(FormatOperator(","), Interfaces.SqlHtmlConstants.CLASS_OPERATOR);
if (SpaceAfterExpandedComma)
state.WordSeparatorExpected = true;
}
else
{
WhiteSpace_BreakAsExpected(state);
state.AddOutputContent(FormatOperator(","), Interfaces.SqlHtmlConstants.CLASS_OPERATOR);
state.WordSeparatorExpected = true;
}
}
break;
case SqlXmlConstants.ENAME_PERIOD:
case SqlXmlConstants.ENAME_SEMICOLON:
case SqlXmlConstants.ENAME_SCOPERESOLUTIONOPERATOR:
//always ignores requested word spacing, and doesn't request a following space either.
state.WordSeparatorExpected = false;
WhiteSpace_BreakAsExpected(state);
state.AddOutputContent(FormatOperator(contentElement.InnerText), Interfaces.SqlHtmlConstants.CLASS_OPERATOR);
break;
case SqlXmlConstants.ENAME_ASTERISK:
case SqlXmlConstants.ENAME_EQUALSSIGN:
case SqlXmlConstants.ENAME_ALPHAOPERATOR:
case SqlXmlConstants.ENAME_OTHEROPERATOR:
WhiteSpace_SeparateWords(state);
state.AddOutputContent(FormatOperator(contentElement.InnerText), Interfaces.SqlHtmlConstants.CLASS_OPERATOR);
state.WordSeparatorExpected = true;
break;
case SqlXmlConstants.ENAME_COMPOUNDKEYWORD:
WhiteSpace_SeparateWords(state);
state.SetRecentKeyword(contentElement.Attributes[SqlXmlConstants.ANAME_SIMPLETEXT].Value);
state.AddOutputContent(FormatKeyword(contentElement.Attributes[SqlXmlConstants.ANAME_SIMPLETEXT].Value), Interfaces.SqlHtmlConstants.CLASS_KEYWORD);
state.WordSeparatorExpected = true;
ProcessSqlNodeList(contentElement.SelectNodes(SqlXmlConstants.ENAME_COMMENT_MULTILINE + " | " + SqlXmlConstants.ENAME_COMMENT_SINGLELINE), state.IncrementIndent());
state.DecrementIndent();
state.WordSeparatorExpected = true;
break;
case SqlXmlConstants.ENAME_OTHERKEYWORD:
case SqlXmlConstants.ENAME_DATATYPE_KEYWORD:
WhiteSpace_SeparateWords(state);
state.SetRecentKeyword(contentElement.InnerText);
state.AddOutputContent(FormatKeyword(contentElement.InnerText), Interfaces.SqlHtmlConstants.CLASS_KEYWORD);
state.WordSeparatorExpected = true;
break;
case SqlXmlConstants.ENAME_PSEUDONAME:
WhiteSpace_SeparateWords(state);
state.AddOutputContent(FormatKeyword(contentElement.InnerText), Interfaces.SqlHtmlConstants.CLASS_KEYWORD);
state.WordSeparatorExpected = true;
break;
case SqlXmlConstants.ENAME_FUNCTION_KEYWORD:
WhiteSpace_SeparateWords(state);
state.SetRecentKeyword(contentElement.InnerText);
state.AddOutputContent(contentElement.InnerText, Interfaces.SqlHtmlConstants.CLASS_FUNCTION);
state.WordSeparatorExpected = true;
break;
case SqlXmlConstants.ENAME_OTHERNODE:
case SqlXmlConstants.ENAME_MONETARY_VALUE:
case SqlXmlConstants.ENAME_LABEL:
WhiteSpace_SeparateWords(state);
state.AddOutputContent(contentElement.InnerText);
state.WordSeparatorExpected = true;
break;
case SqlXmlConstants.ENAME_NUMBER_VALUE:
WhiteSpace_SeparateWords(state);
state.AddOutputContent(contentElement.InnerText.ToLowerInvariant());
state.WordSeparatorExpected = true;
break;
case SqlXmlConstants.ENAME_BINARY_VALUE:
WhiteSpace_SeparateWords(state);
state.AddOutputContent("0x");
state.AddOutputContent(contentElement.InnerText.Substring(2).ToUpperInvariant());
state.WordSeparatorExpected = true;
break;
case SqlXmlConstants.ENAME_WHITESPACE:
//take note if it's a line-breaking space, but don't DO anything here
if (Regex.IsMatch(contentElement.InnerText, @"(\r|\n)+"))
state.SourceBreakPending = true;
break;
default:
throw new Exception("Unrecognized element in SQL Xml!");
}
if (contentElement.GetAttribute(SqlXmlConstants.ANAME_HASERROR) == "1")
state.CloseClass();
if (initialIndent != state.IndentLevel)
throw new Exception("Messed up the indenting!! Check code/stack or panic!");
}
public string FormatSQLTree(XmlDocument sqlTreeDoc)
{
//thread-safe - each call to FormatSQLTree() gets its own independent state object
TSqlStandardFormattingState state = new TSqlStandardFormattingState(HTMLColoring, IndentString, SpacesPerTab, MaxLineWidth, 0);
if (sqlTreeDoc.SelectSingleNode(string.Format("/{0}/@{1}[.=1]", SqlXmlConstants.ENAME_SQL_ROOT, SqlXmlConstants.ANAME_ERRORFOUND)) != null)
state.AddOutputContent(ErrorOutputPrefix);
XmlNodeList rootList = sqlTreeDoc.SelectNodes(string.Format("/{0}/*", SqlXmlConstants.ENAME_SQL_ROOT));
ProcessSqlNodeList(rootList, state);
WhiteSpace_BreakAsExpected(state);
if (state.SpecialRegionActive == SpecialRegionType.NoFormat)
{
XmlNode skippedXml = ExtractXmlBetween(state.RegionStartNode, sqlTreeDoc.DocumentElement);
TSqlIdentityFormatter tempFormatter = new TSqlIdentityFormatter(HTMLColoring);
state.AddOutputContentRaw(tempFormatter.FormatSQLTree(skippedXml));
}
else if (state.SpecialRegionActive == SpecialRegionType.Minify)
{
XmlNode skippedXml = ExtractXmlBetween(state.RegionStartNode, sqlTreeDoc.DocumentElement);
TSqlObfuscatingFormatter tempFormatter = new TSqlObfuscatingFormatter();
if (HTMLFormatted)
state.AddOutputContentRaw(Utils.HtmlEncode(tempFormatter.FormatSQLTree(skippedXml)));
else
state.AddOutputContentRaw(tempFormatter.FormatSQLTree(skippedXml));
}
return state.DumpOutput();
}
