internal string DoStatement(Statement s)
{
if (s is AssignmentStatement && !(s is AugmentedAssignmentStatement))
{
AssignmentStatement a = s as AssignmentStatement;
StringBuilder sb = new StringBuilder();
if (a.IsLocal)
{
sb.Append("local ");
}
for (int i = 0; i < a.Lhs.Count; i++)
{
sb.Append(DoExpr(a.Lhs[i]));
if (i != a.Lhs.Count - 1)
{
sb.Append(",");
}
}
if (a.Rhs.Count > 0)
{
sb.Append("=");
for (int i = 0; i < a.Rhs.Count; i++)
{
sb.Append(DoExpr(a.Rhs[i]));
if (i != a.Rhs.Count - 1)
{
sb.Append(",");
}
}
}
return(sb.ToString());
}
else if (s is AugmentedAssignmentStatement)
{
AugmentedAssignmentStatement a = s as AugmentedAssignmentStatement;
StringBuilder sb = new StringBuilder();
//sb.Append(DoExpr(a.Lhs[0]));
if (a.IsLocal)
{
sb.Append("local ");
}
Expression assignment = ((BinOpExpr)a.Rhs[0]).Rhs;
sb.Append(DoExpr((((BinOpExpr)a.Rhs[0]).Lhs)));
sb.Append("" + ((BinOpExpr)a.Rhs[0]).Op + "=");
sb.Append(DoExpr(assignment));
return(sb.ToString());
}
else if (s is BreakStatement)
{
// HAHAHA this is incredibly simple...
return("break");
}
else if (s is CallStatement)
{
// Also incredibly simple...
CallStatement c = s as CallStatement;
return(DoExpr(c.Expression));
}
else if (s is DoStatement)
{
DoStatement d = s as DoStatement;
StringBuilder sb = new StringBuilder();
sb.Append("do ");
sb.Append(DoChunk(d.Body));
sb.Append(" end");
return(sb.ToString());
}
else if (s is GenericForStatement)
{
GenericForStatement g = s as GenericForStatement;
StringBuilder sb = new StringBuilder();
sb.Append("for ");
for (int i = 0; i < g.VariableList.Count; i++)
{
sb.Append(g.VariableList[i].Name);
if (i != g.VariableList.Count - 1)
{
sb.Append(",");
}
}
sb.Append(" in ");
for (int i = 0; i < g.Generators.Count; i++)
{
sb.Append(DoExpr(g.Generators[i]));
if (i != g.Generators.Count - 1)
{
sb.Append(",");
}
}
sb.Append(" do ");
sb.Append(DoChunk(g.Body));
sb.Append(" end");
return(sb.ToString());
}
else if (s is NumericForStatement)
{
NumericForStatement n = s as NumericForStatement;
StringBuilder sb = new StringBuilder();
sb.Append("for ");
sb.Append(n.Variable.Name);
sb.Append("=");
sb.Append(DoExpr(n.Start));
sb.Append(",");
sb.Append(DoExpr(n.End));
if (n.Step != null)
{
sb.Append(",");
sb.Append(DoExpr(n.Step));
}
sb.Append(" do ");
sb.Append(DoChunk(n.Body));
sb.Append(" end");
return(sb.ToString());
}
else if (s is FunctionStatement)
{
FunctionStatement f = s as FunctionStatement;
StringBuilder sb = new StringBuilder();
sb.Append("function " + DoExpr(f.Name) + "(");
for (int i = 0; i < f.Arguments.Count; i++)
{
sb.Append(f.Arguments[i].Name);
if (i != f.Arguments.Count - 1 || f.IsVararg)
{
sb.Append(",");
}
}
if (f.IsVararg)
{
sb.Append("...");
}
sb.Append(")");
sb.Append(" " + DoStatement(f.Body[0]));
sb.Append(" end");
return(sb.ToString());
}
else if (s is GotoStatement)
{
GotoStatement g = s as GotoStatement;
return("goto " + g.Label);
}
else if (s is IfStmt)
{
IfStmt i = s as IfStmt;
StringBuilder sb = new StringBuilder();
for (int x = 0; x < i.Clauses.Count; x++)
{
string ss = DoStatement(i.Clauses[x]);
if (x == 0)
{
sb.Append("if ");
sb.Append(ss);
}
else if (i.Clauses[x] is ElseStmt)
{
sb.Append(" else ");
sb.Append(ss);
}
else
{
sb.Append("elseif " + ss);
}
}
sb.Append(" end");
return(sb.ToString());
}
else if (s is LabelStatement)
{
LabelStatement l = s as LabelStatement;
return("::" + l.Label + "::");
}
else if (s is RepeatStatement)
{
RepeatStatement r = s as RepeatStatement;
StringBuilder sb = new StringBuilder();
sb.Append("repeat ");
sb.Append(DoChunk(r.Body));
sb.Append(" until " + DoExpr(r.Condition));
return(sb.ToString());
}
else if (s is ReturnStatement)
{
ReturnStatement r = s as ReturnStatement;
StringBuilder sb = new StringBuilder();
sb.Append("return ");
for (int i = 0; i < r.Arguments.Count; i++)
{
sb.Append(DoExpr(r.Arguments[i]));
if (i != r.Arguments.Count - 1)
{
sb.Append(",");
}
}
return(sb.ToString());
}
else if (s is UsingStatement)
{
UsingStatement u = s as UsingStatement;
StringBuilder sb = new StringBuilder();
sb.Append("using ");
sb.Append(DoStatement(u.Vars));
sb.Append(" do ");
sb.Append(DoChunk(u.Body));
sb.Append(" end");
return(sb.ToString());
}
else if (s is WhileStatement)
{
WhileStatement w = s as WhileStatement;
StringBuilder sb = new StringBuilder();
sb.Append("while ");
sb.Append(DoExpr(w.Condition));
sb.Append(" do ");
sb.Append(DoChunk(w.Body));
sb.Append(" end");
return(sb.ToString());
}
else if (s is ElseIfStmt)
{
ElseIfStmt e = s as ElseIfStmt;
string s2 = DoExpr(e.Condition) + " then ";
s2 += DoChunk(e.Body);
return(s2);
}
else if (s is ElseStmt)
{
return(DoChunk(((ElseStmt)s).Body));
}
throw new NotImplementedException(s.GetType().Name + " is not implemented");
}
static List <CompletionItem> DoStatement(Statement s)
{
List <CompletionItem> ret = new List <CompletionItem>();
if (s is AssignmentStatement && !(s is AugmentedAssignmentStatement))
{
AssignmentStatement a = s as AssignmentStatement;
foreach (Expression e in a.Lhs)
{
ret.AddRange(DoExpr(e));
}
foreach (Expression e in a.Rhs)
{
ret.AddRange(DoExpr(e));
}
}
else if (s is AugmentedAssignmentStatement)
{
AugmentedAssignmentStatement a = s as AugmentedAssignmentStatement;
foreach (Expression e in a.Lhs)
{
ret.AddRange(DoExpr(e));
}
foreach (Expression e in a.Rhs)
{
ret.AddRange(DoExpr(e));
}
}
else if (s is BreakStatement)
{
}
else if (s is CallStatement)
{
// Also incredibly simple...
CallStatement c = s as CallStatement;
return(DoExpr(c.Expression));
}
else if (s is DoStatement)
{
DoStatement d = s as DoStatement;
return(DoChunk(d.Body));
}
else if (s is GenericForStatement)
{
GenericForStatement g = s as GenericForStatement;
foreach (Variable v in g.VariableList)
{
ret.Add(new CompletionItem(v.Name));
}
ret.AddRange(DoChunk(g.Body));
}
else if (s is NumericForStatement)
{
NumericForStatement n = s as NumericForStatement;
ret.Add(new CompletionItem(n.Variable.Name));
ret.AddRange(DoChunk(n.Body));
}
else if (s is FunctionStatement)
{
FunctionStatement f = s as FunctionStatement;
ret.AddRange(DoExpr(f.Name));
foreach (Variable v in f.Arguments)
{
ret.Add(new CompletionItem(v.Name));
}
ret.AddRange(DoChunk(f.Body));
}
else if (s is GotoStatement)
{
}
else if (s is IfStmt)
{
IfStmt i = s as IfStmt;
for (int x = 0; x < i.Clauses.Count; x++)
{
ret.AddRange(DoStatement(i.Clauses[x]));
}
}
else if (s is LabelStatement)
{
}
else if (s is RepeatStatement)
{
RepeatStatement r = s as RepeatStatement;
ret.AddRange(DoChunk(r.Body));
ret.AddRange(DoExpr(r.Condition));
}
else if (s is ReturnStatement)
{
// no variable defined here. hopefully.
}
else if (s is UsingStatement)
{
UsingStatement u = s as UsingStatement;
ret.AddRange(DoStatement(u.Vars));
ret.AddRange(DoChunk(u.Body));
}
else if (s is WhileStatement)
{
WhileStatement w = s as WhileStatement;
ret.AddRange(DoExpr(w.Condition));
ret.AddRange(DoChunk(w.Body));
}
else if (s is ElseIfStmt)
{
ElseIfStmt e = s as ElseIfStmt;
return(DoChunk(e.Body));
}
else if (s is ElseStmt)
{
return(DoChunk(((ElseStmt)s).Body));
}
return(ret);
}
internal string DoStatement(Statement s, ref int i)
{
// If the statement contains a body, we cant just fromTokens it, as it's body might not be
// fully tokenized input. Therefore, we run DoChunk on Body's
if (s.ScannedTokens != null && s.ScannedTokens.Count > 0)
{
if (s is AssignmentStatement && !(s is AugmentedAssignmentStatement))
{
AssignmentStatement a = s as AssignmentStatement;
StringBuilder sb = new StringBuilder();
if (a.IsLocal)
{
sb.Append(fromToken(a.ScannedTokens[i++], a.Scope));
sb.Append(" ");
}
for (int i2 = 0; i2 < a.Lhs.Count; i2++)
{
sb.Append(DoExpr(a.Lhs[i2], s.ScannedTokens, ref i, s.Scope));
if (i2 != a.Lhs.Count - 1)
{
sb.Append(fromToken(a.ScannedTokens[i++], a.Scope));
sb.Append(" ");
}
}
if (a.Rhs.Count > 0)
{
sb.Append(" ");
sb.Append(fromToken(a.ScannedTokens[i++], a.Scope));
sb.Append(" ");
for (int i2 = 0; i2 < a.Rhs.Count; i2++)
{
sb.Append(DoExpr(a.Rhs[i2], s.ScannedTokens, ref i, s.Scope));
if (i2 != a.Rhs.Count - 1)
{
sb.Append(fromToken(a.ScannedTokens[i++], s.Scope));
sb.Append(" ");
}
}
}
return(sb.ToString());
}
else if (s is AugmentedAssignmentStatement)
{
AugmentedAssignmentStatement a = s as AugmentedAssignmentStatement;
StringBuilder sb = new StringBuilder();
//sb.Append(DoExpr(a.Lhs[0]));
if (a.IsLocal)
{
sb.Append(fromToken(a.ScannedTokens[i++], a.Scope));
sb.Append(" ");
}
Expression assignment = ((BinOpExpr)a.Rhs[0]).Rhs;
Expression tmp = ((BinOpExpr)a.Rhs[0]).Lhs;
sb.Append(DoExpr(tmp, s.ScannedTokens, ref i, s.Scope));
sb.Append(" ");
sb.Append(fromToken(s.ScannedTokens[i++], s.Scope));
sb.Append(" ");
sb.Append(DoExpr(assignment, s.ScannedTokens, ref i, s.Scope));
return(sb.ToString());
}
else if (s is BreakStatement)
{
// HAHAHA this is incredibly simple...
return(fromTokens(s.ScannedTokens, s.Scope));
}
else if (s is ContinueStatement)
{
return(fromTokens(s.ScannedTokens, s.Scope));
}
else if (s is CallStatement)
{
// Also incredibly simple...
return(DoExpr(((CallStatement)s).Expression, s.ScannedTokens, ref i, s.Scope));
}
else if (s is DoStatement)
{
DoStatement d = s as DoStatement;
StringBuilder sb = new StringBuilder();
sb.Append(fromToken(d.ScannedTokens[0], s.Scope)); // 'do'
sb.Append(options.EOL);
indent++;
sb.Append(DoChunk(d.Body, ref i));
sb.Append(nldedent());
sb.Append(fromToken(d.ScannedTokens[d.ScannedTokens.Count - 1], s.Scope)); // end
return(sb.ToString());
}
else if (s is GenericForStatement)
{
GenericForStatement g = s as GenericForStatement;
StringBuilder sb = new StringBuilder();
sb.Append(fromToken(g.ScannedTokens[i++], s.Scope)); // 'for'
sb.Append(" ");
for (int x = 0; x < g.VariableList.Count; x++)
{
sb.Append(fromToken(s.ScannedTokens[i++], s.Scope));
if (x != g.VariableList.Count - 1)
{
sb.Append(fromToken(s.ScannedTokens[i++], s.Scope)); // ','
sb.Append(" ");
}
}
sb.Append(" ");
sb.Append(fromToken(g.ScannedTokens[i++], s.Scope)); // 'in'
sb.Append(" ");
for (int x = 0; x < g.Generators.Count; x++)
{
//sb.Append(fromToken(s.ScannedTokens[i++], s.Scope));
DoExpr(g.Generators[x], g.ScannedTokens, ref i, s.Scope);
if (x != g.VariableList.Count - 1)
{
sb.Append(fromToken(s.ScannedTokens[i++], s.Scope)); // ','
sb.Append(" ");
}
}
sb.Append(" ");
sb.Append(fromToken(g.ScannedTokens[i++], s.Scope)); // 'do'
sb.Append(options.EOL);
indent++;
sb.Append(DoChunk(g.Body));
sb.Append(nldedent());
sb.Append(fromToken(s.ScannedTokens[s.ScannedTokens.Count - 1], s.Scope)); // <end>
return(sb.ToString());
}
else if (s is NumericForStatement)
{
NumericForStatement n = s as NumericForStatement;
StringBuilder sb = new StringBuilder();
sb.Append(fromToken(n.ScannedTokens[i++], s.Scope)); // 'for'
sb.Append(" ");
sb.Append(fromToken(n.ScannedTokens[i++], s.Scope)); // <var>
sb.Append(" ");
sb.Append(fromToken(n.ScannedTokens[i++], s.Scope)); // '='
sb.Append(" ");
sb.Append(DoExpr(n.Start, n.ScannedTokens, ref i, s.Scope)); // <start>
//sb.Append(" ");
sb.Append(fromToken(n.ScannedTokens[i++], s.Scope)); // ','
sb.Append(" ");
sb.Append(DoExpr(n.End, n.ScannedTokens, ref i, s.Scope)); // <end>
if (n.Step != null)
{
sb.Append(fromToken(n.ScannedTokens[i++], s.Scope)); // ','
sb.Append(" ");
sb.Append(DoExpr(n.Step, n.ScannedTokens, ref i, s.Scope)); // <step>
sb.Append(" ");
}
else
{
sb.Append(" ");
}
sb.Append(fromToken(n.ScannedTokens[i++], s.Scope)); // 'do'
sb.Append(options.EOL);
indent++;
sb.Append(DoChunk(n.Body));
sb.Append(nldedent());
sb.Append(fromToken(s.ScannedTokens[s.ScannedTokens.Count - 1], s.Scope)); // <end>
return(sb.ToString());
}
else if (s is FunctionStatement)
{
FunctionStatement f = s as FunctionStatement;
StringBuilder sb = new StringBuilder();
sb.Append(fromToken(s.ScannedTokens[i++], s.Scope)); // 'function'
sb.Append(" ");
sb.Append(DoExpr(f.Name, s.ScannedTokens, ref i, s.Scope));
sb.Append(fromToken(s.ScannedTokens[i++], s.Scope)); // '('
for (int i2 = 0; i2 < f.Arguments.Count; i2++)
{
sb.Append(fromToken(s.ScannedTokens[i++], s.Scope));
if (i2 != f.Arguments.Count - 1)// || f.IsVararg)
{
sb.Append(fromToken(s.ScannedTokens[i++], s.Scope));
sb.Append(" ");
}
}
if (f.IsVararg)
{
sb.Append(fromToken(s.ScannedTokens[i++], s.Scope));
sb.Append(" ");
sb.Append(fromToken(s.ScannedTokens[i++], s.Scope));
}
sb.Append(fromToken(s.ScannedTokens[i++], s.Scope)); // ')'
sb.Append(options.EOL);
indent++;
sb.Append(DoChunk(f.Body));
indent--;
sb.Append(options.EOL);
sb.Append(writeIndent());
sb.Append(fromToken(s.ScannedTokens[s.ScannedTokens.Count - 1], s.Scope)); // <end>
sb.Append(options.EOL);
return(sb.ToString());
}
else if (s is GotoStatement)
{
// goto <string label>, so no expr
//GotoStatement g = s as GotoStatement;
return(fromTokens(s.ScannedTokens, s.Scope));
}
else if (s is IfStmt)
{
IfStmt ifs = s as IfStmt;
StringBuilder sb = new StringBuilder();
foreach (SubIfStmt clause in ifs.Clauses)
{
int i3 = 0;
if (clause is ElseIfStmt)
{
ElseIfStmt c = clause as ElseIfStmt;
sb.Append(fromToken(c.ScannedTokens[i3++], s.Scope)); // if/elseif
sb.Append(" ");
sb.Append(DoExpr(c.Condition, c.ScannedTokens, ref i3, c.Scope));
sb.Append(" ");
sb.Append(fromToken(c.ScannedTokens[i3++], s.Scope)); // 'then'
sb.Append(options.EOL);
indent++;
sb.Append(DoChunk(clause.Body));
sb.Append(nldedent());
}
else if (clause is ElseStmt)
{
sb.Append(fromToken(clause.ScannedTokens[i3++], s.Scope)); // if/elseif
sb.Append(options.EOL);
indent++;
sb.Append(DoChunk(clause.Body));
sb.Append(nldedent());
}
else
{
throw new NotImplementedException(clause.GetType().Name);
}
}
sb.Append(fromToken(s.ScannedTokens[s.ScannedTokens.Count - 1], s.Scope)); // 'end'
return(sb.ToString());
}
else if (s is LabelStatement)
{
// ::<string label>::, so no expr
return(fromTokens(s.ScannedTokens, s.Scope));
}
else if (s is RepeatStatement)
{
RepeatStatement r = s as RepeatStatement;
StringBuilder sb = new StringBuilder();
sb.Append(fromToken(r.ScannedTokens[0], s.Scope));
i = 1;
sb.Append(options.EOL);
indent++;
sb.Append(DoChunk(r.Body, ref i));
sb.Append(nldedent());
sb.Append(fromToken(r.ScannedTokens[i++], r.Scope));
sb.Append(" ");
sb.Append(DoExpr(r.Condition, r.ScannedTokens, ref i, r.Scope));
return(sb.ToString());
}
else if (s is ReturnStatement)
{
ReturnStatement rs = s as ReturnStatement;
StringBuilder sb = new StringBuilder();
sb.Append(fromToken(s.ScannedTokens[i++], s.Scope)); // 'return'
sb.Append(" ");
for (int x = 0; x < rs.Arguments.Count; x++)
{
sb.Append(DoExpr(rs.Arguments[x], rs.ScannedTokens, ref i, s.Scope));
if (x != rs.Arguments.Count - 1)
{
sb.Append(fromToken(s.ScannedTokens[i++], s.Scope)); // ','
sb.Append(" ");
}
}
return(sb.ToString());
}
else if (s is UsingStatement)
{
UsingStatement u = s as UsingStatement;
StringBuilder sb = new StringBuilder();
sb.Append(fromToken(s.ScannedTokens[i++], s.Scope)); // 'using'
sb.Append(" ");
AssignmentStatement a = u.Vars;
for (int i2 = 0; i2 < a.Lhs.Count; i2++)
{
sb.Append(DoExpr(a.Lhs[i2], u.ScannedTokens, ref i, s.Scope));
if (i2 != a.Lhs.Count - 1)
{
sb.Append(fromToken(u.ScannedTokens[i++], a.Scope));
sb.Append(" ");
}
}
if (a.Rhs.Count > 0)
{
sb.Append(" ");
sb.Append(fromToken(u.ScannedTokens[i++], a.Scope)); // '='
sb.Append(" ");
for (int i2 = 0; i2 < a.Rhs.Count; i2++)
{
sb.Append(DoExpr(a.Rhs[i2], u.ScannedTokens, ref i, s.Scope));
if (i2 != a.Rhs.Count - 1)
{
sb.Append(fromToken(u.ScannedTokens[i++], s.Scope));
sb.Append(" ");
}
}
}
sb.Append(" ");
sb.Append(fromToken(s.ScannedTokens[i++], s.Scope)); // 'do'
sb.Append(options.EOL);
indent++;
sb.Append(DoChunk(u.Body));
sb.Append(nldedent());
sb.Append(fromToken(s.ScannedTokens[s.ScannedTokens.Count - 1], s.Scope)); // 'end'
return(sb.ToString());
}
else if (s is WhileStatement)
{
WhileStatement w = s as WhileStatement;
StringBuilder sb = new StringBuilder();
sb.Append(fromToken(s.ScannedTokens[i++], s.Scope)); // 'while'
sb.Append(" ");
sb.Append(DoExpr(w.Condition, w.ScannedTokens, ref i, s.Scope));
sb.Append(" ");
sb.Append(fromToken(s.ScannedTokens[i++], s.Scope)); // 'do'
sb.Append(options.EOL);
indent++;
sb.Append(DoChunk(w.Body));
sb.Append(nldedent());
sb.Append(fromToken(s.ScannedTokens[s.ScannedTokens.Count - 1], s.Scope));
return(sb.ToString());
}
}
else // No token stream, beautify
{
return(beautifier.DoStatement(s));
}
throw new NotImplementedException(s.GetType().Name + " is not implemented");
}
