static void ObfuscateStat(toSQLParse::statement &stat, QString &ret)
{
if (ret.length() > 0 &&
stat.String.length() > 0 &&
toIsIdent(ret.at(ret.length() - 1)) &&
toIsIdent(stat.String.at(0)))
{
if (CountLine(ret) < 60)
ret += QString::fromLatin1(" ");
else
ret += QString::fromLatin1("\n");
}
ret += stat.String;
if (!stat.Comment.isEmpty())
{
ret += stat.Comment;
ret += QString::fromLatin1("\n");
}
for (std::list<toSQLParse::statement>::iterator i = stat.subTokens().begin();
i != stat.subTokens().end();
i++)
{
ObfuscateStat(*i, ret);
}
}
QString toSQLParse::indentStatement(statement &stat, int level/*SQLITEMAN, toSyntaxAnalyzer &syntax*/)
{
QString ret;
switch (stat.Type)
{
default:
{
QMessageBox::warning(QApplication::activeWindow(), "Sqliteman",
"toSQLparse: Internal error in toSQLParse, should never get here");
}
case statement::Block:
{
ret = IndentComment(level, 0, stat.Comment, false);
int exc = 0;
for (std::list<toSQLParse::statement>::iterator i = stat.subTokens().begin();
i != stat.subTokens().end();
i++)
{
int add
= 0;
std::list<toSQLParse::statement>::iterator j = i;
j++;
if (i != stat.subTokens().begin() &&
j != stat.subTokens().end())
add
= Settings.IndentLevel;
else
exc = 0;
QString t;
if ((*i).subTokens().begin() != (*i).subTokens().
end())
t = (*(*i).subTokens().begin()).String.toUpper();
if (t == ("BEGIN") || t == ("WHEN") || t == ("ELSE") || t == ("ELSIF"))
add
= 0;
if ((*i).Type == statement::List)
ret += indentString(level + add
+ exc);
ret += indentStatement(*i, level + add
+ exc/*SQLITEMAN , syntax*/);
if ((*i).Type == statement::List)
{
int i;
for (i = ret.length() - 1;i >= 0 && ret[i].isSpace();i--)
;
ret = ret.mid(0, std::max(i + 1, 0));
ret += ("\n");
ret += indentString(level + exc);
}
if (t == ("EXCEPTION"))
exc = Settings.IndentLevel * 2;
}
if (Settings.EndBlockNewline && level != 0)
ret += ("\n");
}
break;
case statement::List:
case statement::Statement:
int maxlev = 0;
int maxlevorig = 0;
bool useMaxLev = false;
bool any = true;
int current;
bool first;
bool noKeyBreak = false;
bool lineList = false;
QString comment;
if (stat.Type == statement::Statement)
{
ret = IndentComment(level, 0, stat.Comment, false);
useMaxLev = true;
first = true;
current = 0;
}
else
{
for (std::list<toSQLParse::statement>::iterator i = stat.subTokens().begin();
i != stat.subTokens().end();)
{
if ((*i).Type != statement::Keyword)
noKeyBreak = true;
else
useMaxLev = true;
break;
}
current = level;
first = true;
}
if (useMaxLev)
{
int count = 0;
for (std::list<toSQLParse::statement>::iterator i = stat.subTokens().begin();
i != stat.subTokens().end();
i++)
{
if (any)
{
QString upp = (*i).String.toUpper();
if ((*i).Type == statement::Keyword &&
upp != ("LOOP") &&
upp != ("DO") &&
upp != ("THEN") &&
upp != ("AS") &&
upp != ("IS"))
{
if (int((*i).String.length()) + 1 > maxlev)
maxlev = (*i).String.length() + 1;
count++;
any = false;
}
else if (i == stat.subTokens().begin())
{
noKeyBreak = true;
break;
}
}
else if ((*i).Type == statement::Token)
any = true;
if ((*i).Type == statement::List)
count++;
}
if (count <= 1 && maxlev > 0)
maxlev--;
maxlevorig = maxlev;
any = true;
}
for (std::list<toSQLParse::statement>::iterator i = stat.subTokens().begin();
i != stat.subTokens().end();
i++)
{
comment = AddComment(comment, (*i).Comment);
QString upp = (*i).String.toUpper();
#ifdef TOPARSE_DEBUG
printf("%s\n", (const char*)(*i).String.toUtf8());
#endif
if ((*i).Type == statement::List)
{
if (Settings.OperatorSpace)
{
ret += (" ");
current++;
}
QString t = indentStatement(*i, current/*SQLITEMAN, syntax*/);
if (t.indexOf(("\n")) >= 0)
current = CurrentColumn(t);
else
current += CurrentColumn(t);
ret += t;
any = true;
}
else if ((*i).String == (","))
{
if (Settings.CommaBefore)
{
ret += IndentComment(Settings.CommentColumn, current, comment, true);
comment = QString::null;
ret += indentString(level + maxlev - (Settings.OperatorSpace ? 2 : 1));
ret += (",");
}
else
{
ret += (",");
ret += IndentComment(Settings.CommentColumn, current + 1, comment, true);
comment = QString::null;
ret += indentString(level + maxlev);
}
current = level + maxlev;
any = false;
lineList = true;
}
else if ((*i).Type == statement::Keyword && (upp == ("LOOP") ||
upp == ("DO") ||
upp == ("THEN") ||
upp == ("AS") ||
upp == ("IS")))
{
if (!Settings.BlockOpenLine)
{
if (ret.length() > 0)
{
if (toIsIdent(ret.at(ret.length() - 1)) ||
ret.at(ret.length() - 1) == QUOTE_CHARACTER /*SQLITEMAN syntax.quoteCharacter()*/ ||
ret.at(ret.length() - 1) == '\'' ||
Settings.OperatorSpace)
{
ret += (" ");
current++;
}
}
ret += Settings.KeywordUpper ? (*i).String.toUpper() : (*i).String;
current += (*i).String.length();
}
else
{
ret += IndentComment(Settings.CommentColumn, current, comment, true);
comment = QString::null;
ret += indentString(level);
ret += Settings.KeywordUpper ? (*i).String.toUpper() : (*i).String;
current = level + (*i).String.length();
}
any = false;
}
else if (any && (*i).Type == statement::Keyword && !noKeyBreak)
{
if (first)
first = false;
else
{
ret += IndentComment(Settings.CommentColumn, current, comment, true);
current = 0;
comment = QString::null;
}
if (current == 0)
{
ret += indentString(level);
current = level;
}
else
while (current < level)
{
ret += (" ");
current++;
}
maxlev = maxlevorig;
QString word = Settings.KeywordUpper ? (*i).String.toUpper() : (*i).String;
if (ret.length())
{
ret += QString("%1").arg(word,
Settings.RightSeparator ? maxlev - 1 : 1 - maxlev);
current = level + std::max(int(word.length()), maxlev - 1);
}
else
{
ret += word;
current = level + word.length();
}
any = false;
lineList = false;
}
else
{
QString t = (*i).String;
bool add
= false;
if ((*i).Type == statement::Keyword)
{
if (!lineList &&
!any &&
(*i).Type == statement::Keyword &&
!noKeyBreak &&
upp == ("BY"))
add
= true;
}
else
{
any = true;
}
if (isKeyword(upp) /*SQLITEMAN syntax.reservedWord(upp)*/ && Settings.KeywordUpper)
t = upp;
int extra;
if (first)
{
first = false;
any = false;
extra = 0;
}
else
{
if (ret.length() > 0 &&
!ret.at(ret.length() - 1).isSpace() &&
(Settings.OperatorSpace || ((toIsIdent(t[0]) ||
t[0] == QUOTE_CHARACTER /*SQLITEMAN syntax.quoteCharacter()*/ || t[0] == '\'') &&
(toIsIdent(ret.at(ret.length() - 1)) ||
ret.at(ret.length() - 1) == QUOTE_CHARACTER /*SQLITEMAN syntax.quoteCharacter()*/ ||
ret.at(ret.length() - 1) == '\'')
)
)
)
{
if (t != (";") &&
t != (".") &&
ret.at(ret.length() - 1) != '.' &&
current != 0)
{
current++;
ret += (" ");
}
}
else if (ret.length() > 2 && ret.at(ret.length() - 2) == '*' && ret.at(ret.length() - 1) == '/')
{
current++;
ret += (" ");
}
extra = maxlev;
}
if (current < level + maxlev)
{
if (current == 0)
ret += indentString(level + maxlev);
else
while (current < level + maxlev)
{
ret += (" ");
current++;
}
current = level + maxlev;
}
ret += t;
current += t.length();
if (t.startsWith(("<<")))
{
ret += ("\n");
current = 0;
}
if (add
)
maxlev += t.length() + 1;
}
}
if (stat.Type == statement::Statement)
{
ret += IndentComment(Settings.CommentColumn, current, comment, true);
comment = QString::null;
if (Settings.EndBlockNewline &&
level == 0 &&
stat.subTokens().begin() != stat.subTokens().end() &&
(*stat.subTokens().rbegin()).String == (";"))
ret += ("\n");
}
else if (!comment.isEmpty())
{
ret += IndentComment(Settings.CommentColumn, current, comment, true);
comment = QString::null;
ret += indentString(level - (Settings.OperatorSpace ? 2 : 1));
}
break;
}
return ret;
}
QString toSQLParse::stringTokenizer::getToken(bool forward, bool comments)
{
QChar c;
QChar nc;
QChar endString;
enum
{
space,
any,
identifier,
string,
comment,
label,
bindOpen,
bindClose
} state = space;
QString token;
int inc = forward ? 1 : -1;
while ((forward && Offset < int(String.length())) || (!forward && Offset >= 1))
{
if (!forward)
Offset--;
c = String[Offset];
if (c == '\n')
Line++;
if ((forward && Offset < int(String.length() - 1)) || (!forward && Offset > 0))
nc = String[Offset + inc];
else
nc = '\n';
if (state == space)
{
if (forward && c == '-' && nc == '-')
{
int spos = Offset;
if (forward)
for (Offset++;Offset < int(String.length()) && String[Offset] != '\n';Offset++)
;
if (comments)
return String.mid(spos, Offset - spos);
continue;
}
if (c == '/' && nc == '*')
state = comment;
else if ((forward && c == '<' && nc == '<') ||
(!forward && c == '>' && nc == '>'))
state = label;
else if (!c.isSpace())
state = any;
}
if (forward)
Offset++;
if (state != space)
{
if (forward)
token += c;
else
token.prepend(c);
switch (state)
{
case comment:
if (c == '*' && nc == '/')
{
if (forward)
token += nc;
else
token.prepend(nc);
Offset += inc;
if (comments)
return token;
else
{
state = space;
token = QString::null;
}
}
break;
case label:
if ((forward && c == '>' && nc == '>') ||
(!forward && c == '<' && nc == '<'))
{
if (forward)
token += nc;
else
token.prepend(nc);
Offset += inc;
return token;
}
break;
case space:
break;
case bindOpen:
if (!toIsIdent(nc))
{
if (nc == '<')
state = bindClose;
else
return token;
}
break;
case bindClose:
if (c == '>')
return token;
break;
case any:
if (c == ':' && toIsIdent(nc))
{
state = bindOpen;
}
else if (toIsIdent(c))
{
if (!toIsIdent(nc))
return token;
state = identifier;
}
else if (c == '\'' || c == QUOTE_CHARACTER) // SQLITEMAN c == analyzer().quoteCharacter())
{
endString = c;
state = string;
}
else
{
for (int i = 0;Operators[i];i++)
{
if ((forward && c == Operators[i][0] && nc == Operators[i][1]) ||
(!forward && nc == Operators[i][0] && c == Operators[i][1]))
{
if (forward)
token += nc;
else
token.prepend(nc);
Offset += inc;
break;
}
}
return token;
}
break;
case identifier:
if (!toIsIdent(nc))
return token;
break;
case string:
if (c == endString)
{
if (nc == endString)
{
if (forward)
{
token += nc;
Offset++;
}
else
{
token.prepend(nc);
Offset--;
}
}
else
return token;
}
break;
}
}
}
return token;
}
