/* * Copyright 2013-2017 consulo.io * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ package consulo.csharp.lang.psi.impl.source; import org.jetbrains.annotations.NotNull; import org.jetbrains.annotations.Nullable; import consulo.annotations.RequiredReadAction; import consulo.csharp.lang.psi.CSharpElementVisitor; import consulo.csharp.lang.psi.CSharpSoftTokens; import com.intellij.lang.ASTNode; import com.intellij.psi.PsiElement; import consulo.csharp.lang.psi.impl.CSharpImplicitReturnModel; import consulo.dotnet.psi.DotNetExpression; import consulo.dotnet.resolve.DotNetTypeRef; /** * @author VISTALL * @since 09.09.14 */ public class CSharpAwaitExpressionImpl extends CSharpExpressionImpl implements DotNetExpression { public CSharpAwaitExpressionImpl(@NotNull ASTNode node) { super(node); } @NotNull public PsiElement getAwaitKeywordElement() { return findNotNullChildByType(CSharpSoftTokens.AWAIT_KEYWORD); } @Override public void accept(@NotNull CSharpElementVisitor visitor) { visitor.visitAwaitExpression(this); } @RequiredReadAction @NotNull @Override public DotNetTypeRef toTypeRefImpl(boolean b) { DotNetExpression innerExpression = getInnerExpression(); if(innerExpression == null) { return DotNetTypeRef.ERROR_TYPE; } return CSharpImplicitReturnModel.Async.extractTypeRef(innerExpression.toTypeRef(true), this); } @Nullable public DotNetExpression getInnerExpression() { return findChildByClass(DotNetExpression.class); } }