private LuaExpressionSyntax BuildQueryBody(LuaExpressionSyntax collection, QueryBodySyntax node, IQueryRangeVariable rangeVariable)
{
foreach (var clause in node.Clauses)
{
switch (clause.Kind())
{
case SyntaxKind.FromClause: {
bool isOver;
collection = BuildFromClause(collection, (FromClauseSyntax)clause, ref rangeVariable, out isOver);
if (isOver)
{
goto Continuation;
}
break;
}
case SyntaxKind.LetClause: {
collection = BuildLetClause(collection, (LetClauseSyntax)clause, ref rangeVariable);
break;
}
case SyntaxKind.JoinClause: {
bool isOver;
collection = BuildJoinClause(collection, (JoinClauseSyntax)clause, ref rangeVariable, out isOver);
if (isOver)
{
goto Continuation;
}
break;
}
case SyntaxKind.WhereClause: {
collection = BuildQueryWhere(collection, (WhereClauseSyntax)clause, rangeVariable);
break;
}
case SyntaxKind.OrderByClause: {
collection = BuildQueryOrderBy(collection, (OrderByClauseSyntax)clause, rangeVariable);
break;
}
default: {
throw new NotSupportedException();
}
}
}
if (node.SelectOrGroup.IsKind(SyntaxKind.SelectClause))
{
var selectClause = (SelectClauseSyntax)node.SelectOrGroup;
collection = BuildQuerySelect(collection, selectClause, rangeVariable);
}
else
{
var groupClause = (GroupClauseSyntax)node.SelectOrGroup;
collection = BuildGroupClause(collection, groupClause, rangeVariable);
}
Continuation:
if (node.Continuation != null)
{
collection = BuildQueryContinuation(collection, node.Continuation);
}
return(collection);
}
public QueryPackVariable(IQueryRangeVariable x1, IQueryRangeVariable x2)
{
x1_ = x1;
x2_ = x2;
AddPackCount();
}
private bool BuildQueryJoin(JoinClauseSyntax node, out LuaExpressionSyntax resultSelectorExpression, out LuaExpressionSyntax resultSelectorType, ref IQueryRangeVariable rangeVariable, QueryIdentifier queryIdentifier)
{
var parentNode = (QueryBodySyntax)node.Parent;
if (IsSpecialQueryNode(parentNode))
{
var selectClause = (SelectClauseSyntax)parentNode.SelectOrGroup;
resultSelectorExpression = (LuaExpressionSyntax)selectClause.Expression.Accept(this);
var typeSymbol = semanticModel_.GetTypeInfo(selectClause.Expression).Type;
resultSelectorType = GetTypeName(typeSymbol);
return(true);
}
else
{
resultSelectorExpression = CreateQueryAnonymousType(rangeVariable.Name, rangeVariable.Name, queryIdentifier.Name, queryIdentifier.Name);
resultSelectorType = LuaIdentifierNameSyntax.AnonymousType;
rangeVariable = new QueryPackVariable(rangeVariable, queryIdentifier);
return(false);
}
}
private LuaExpressionSyntax BuildJoinClause(LuaExpressionSyntax collection, JoinClauseSyntax node, ref IQueryRangeVariable rangeVariable, out bool isOver)
{
isOver = false;
var rangeVariable2 = AddRangeIdentifier(node.Identifier);
var inner = (LuaExpressionSyntax)node.InExpression.Accept(this);
var left = (LuaExpressionSyntax)node.LeftExpression.Accept(this);
var outerKeySelector = new LuaFunctionExpressionSyntax();
outerKeySelector.AddParameter(rangeVariable.Name);
outerKeySelector.AddStatement(new LuaReturnStatementSyntax(left));
var keyTypeSymbol = semanticModel_.GetTypeInfo(node.LeftExpression).Type;
var keyType = GetTypeName(keyTypeSymbol);
var right = (LuaExpressionSyntax)node.RightExpression.Accept(this);
var innerKeySelector = new LuaFunctionExpressionSyntax();
innerKeySelector.AddStatement(rangeVariable2.Name);
innerKeySelector.AddStatement(new LuaReturnStatementSyntax(right));
LuaFunctionExpressionSyntax resultSelector = new LuaFunctionExpressionSyntax();
LuaExpressionSyntax resultSelectorExpression;
LuaExpressionSyntax resultSelectorType;
LuaIdentifierNameSyntax methodName;
var parentNode = (QueryBodySyntax)node.Parent;
if (node.Into == null)
{
methodName = LuaIdentifierNameSyntax.LinqJoin;
resultSelector.AddParameter(rangeVariable.Name);
resultSelector.AddParameter(rangeVariable2.Name);
isOver = BuildQueryJoin(node, out resultSelectorExpression, out resultSelectorType, ref rangeVariable, rangeVariable2);
}
else
{
methodName = LuaIdentifierNameSyntax.LinqGroupJoin;
var rangeVariableOfInto = AddRangeIdentifier(node.Into.Identifier);
resultSelector.AddParameter(rangeVariable.Name);
resultSelector.AddParameter(rangeVariableOfInto.Name);
isOver = BuildQueryJoin(node, out resultSelectorExpression, out resultSelectorType, ref rangeVariable, rangeVariableOfInto);
}
return(new LuaInvocationExpressionSyntax(methodName, collection, inner, outerKeySelector, innerKeySelector, resultSelector, LuaIdentifierLiteralExpressionSyntax.Nil, keyType, resultSelectorType));
}
private LuaExpressionSyntax BuildFromClause(LuaExpressionSyntax collection, FromClauseSyntax node, ref IQueryRangeVariable rangeVariable, out bool isOver)
{
var collectionSelector = new LuaFunctionExpressionSyntax();
collectionSelector.AddParameter(rangeVariable.Name);
var expression = (LuaExpressionSyntax)node.Expression.Accept(this);
collectionSelector.AddStatement(new LuaReturnStatementSyntax(expression));
var rangeVariable2 = AddRangeIdentifier(node.Identifier);
var resultSelector = new LuaFunctionExpressionSyntax();
resultSelector.AddParameter(rangeVariable.Name);
resultSelector.AddParameter(rangeVariable2.Name);
LuaExpressionSyntax resultSelectorExpression;
LuaExpressionSyntax resultSelectorType;
var parentNode = (QueryBodySyntax)node.Parent;
if (IsSpecialQueryNode(parentNode))
{
var selectClause = (SelectClauseSyntax)parentNode.SelectOrGroup;
resultSelectorExpression = (LuaExpressionSyntax)selectClause.Expression.Accept(this);
var type = semanticModel_.GetTypeInfo(selectClause.Expression).Type;
resultSelectorType = GetTypeName(type);
isOver = true;
}
else
{
resultSelectorExpression = CreateQueryAnonymousType(rangeVariable.Name, rangeVariable.Name, rangeVariable2.Name, rangeVariable2.Name);
resultSelectorType = LuaIdentifierNameSyntax.AnonymousType;
rangeVariable = new QueryPackVariable(rangeVariable, rangeVariable2);
isOver = false;
}
resultSelector.AddStatement(new LuaReturnStatementSyntax(resultSelectorExpression));
return(new LuaInvocationExpressionSyntax(LuaIdentifierNameSyntax.LinqSelectMany, collection, collectionSelector, resultSelector, resultSelectorType));
}
private LuaExpressionSyntax BuildLetClause(LuaExpressionSyntax collection, LetClauseSyntax node, ref IQueryRangeVariable rangeVariable)
{
var letExpression = (LuaExpressionSyntax)node.Expression.Accept(this);
var letRangeVariable = AddRangeIdentifier(node.Identifier);
var anonymousType = CreateQueryAnonymousType(rangeVariable.Name, rangeVariable.Name, letRangeVariable.Name, letExpression);
var selectFunction = new LuaFunctionExpressionSyntax();
selectFunction.AddParameter(rangeVariable.Name);
selectFunction.AddStatement(new LuaReturnStatementSyntax(anonymousType));
rangeVariable = new QueryPackVariable(rangeVariable, letRangeVariable);
return(new LuaInvocationExpressionSyntax(LuaIdentifierNameSyntax.LinqSelect, collection, selectFunction, LuaIdentifierNameSyntax.AnonymousType));
}
private LuaExpressionSyntax BuildGroupClause(LuaExpressionSyntax collection, GroupClauseSyntax node, IQueryRangeVariable rangeVariable)
{
var keyType = semanticModel_.GetTypeInfo(node.ByExpression).Type;
var keyTypeName = GetTypeName(keyType);
var byExpression = (LuaExpressionSyntax)node.ByExpression.Accept(this);
var keySelector = new LuaFunctionExpressionSyntax();
keySelector.AddParameter(rangeVariable.Name);
keySelector.AddStatement(new LuaReturnStatementSyntax(byExpression));
var groupExpression = (LuaExpressionSyntax)node.GroupExpression.Accept(this);
if (node.GroupExpression.IsKind(SyntaxKind.IdentifierName))
{
var groupIdentifierName = groupExpression as LuaIdentifierNameSyntax;
if (groupIdentifierName != null && groupIdentifierName.ValueText == rangeVariable.Name.ValueText)
{
return(new LuaInvocationExpressionSyntax(LuaIdentifierNameSyntax.LinqGroupBy, collection, keySelector, keyTypeName));
}
}
var elementType = semanticModel_.GetTypeInfo(node.GroupExpression).Type;
var elementTypeName = GetTypeName(elementType);
var elementSelector = new LuaFunctionExpressionSyntax();
elementSelector.AddParameter(rangeVariable.Name);
elementSelector.AddStatement(new LuaReturnStatementSyntax(groupExpression));
return(new LuaInvocationExpressionSyntax(LuaIdentifierNameSyntax.LinqGroupBy, collection, keySelector, elementSelector, keyTypeName, elementTypeName));
}
private LuaExpressionSyntax BuildQuerySelect(LuaExpressionSyntax collection, SelectClauseSyntax node, IQueryRangeVariable rangeVariable)
{
var expression = (LuaExpressionSyntax)node.Expression.Accept(this);
if (node.Expression.IsKind(SyntaxKind.IdentifierName))
{
var identifierName = expression as LuaIdentifierNameSyntax;
if (identifierName != null && identifierName.ValueText == rangeVariable.Name.ValueText)
{
return(collection);
}
}
var selectFunction = new LuaFunctionExpressionSyntax();
selectFunction.AddParameter(rangeVariable.Name);
selectFunction.AddStatement(new LuaReturnStatementSyntax(expression));
var type = semanticModel_.GetTypeInfo(node.Expression).Type;
var typeExpression = GetTypeName(type);
return(new LuaInvocationExpressionSyntax(LuaIdentifierNameSyntax.LinqSelect, collection, selectFunction, typeExpression));
}
private LuaExpressionSyntax BuildQueryOrderBy(LuaExpressionSyntax collection, OrderByClauseSyntax node, IQueryRangeVariable rangeVariable)
{
foreach (var ordering in node.Orderings)
{
bool isDescending = ordering.AscendingOrDescendingKeyword.IsKind(SyntaxKind.DescendingKeyword);
if (ordering == node.Orderings.First())
{
var methodName = isDescending ? LuaIdentifierNameSyntax.LinqOrderByDescending : LuaIdentifierNameSyntax.LinqOrderBy;
collection = BuildOrdering(methodName, collection, ordering, rangeVariable);
}
else
{
var methodName = isDescending ? LuaIdentifierNameSyntax.LinqThenByDescending : LuaIdentifierNameSyntax.LinqThenBy;
collection = BuildOrdering(methodName, collection, ordering, rangeVariable);
}
}
return(collection);
}
private LuaExpressionSyntax BuildOrdering(LuaIdentifierNameSyntax methodName, LuaExpressionSyntax collection, OrderingSyntax node, IQueryRangeVariable rangeVariable)
{
var type = semanticModel_.GetTypeInfo(node.Expression).Type;
var typeName = GetTypeName(type);
var expression = (LuaExpressionSyntax)node.Expression.Accept(this);
var keySelector = new LuaFunctionExpressionSyntax();
keySelector.AddParameter(rangeVariable.Name);
keySelector.AddStatement(new LuaReturnStatementSyntax(expression));
return(new LuaInvocationExpressionSyntax(methodName, collection, keySelector, LuaIdentifierNameSyntax.Nil, typeName));
}
private LuaExpressionSyntax BuildQueryWhere(LuaExpressionSyntax collection, WhereClauseSyntax node, IQueryRangeVariable rangeVariable)
{
var condition = (LuaExpressionSyntax)node.Condition.Accept(this);
if (condition == LuaIdentifierLiteralExpressionSyntax.True)
{
return(collection);
}
var whereFunction = new LuaFunctionExpressionSyntax();
whereFunction.AddParameter(rangeVariable.Name);
whereFunction.AddStatement(new LuaReturnStatementSyntax(condition));
return(new LuaInvocationExpressionSyntax(LuaIdentifierNameSyntax.LinqWhere, collection, whereFunction));
}
