org.ballerinalang.langserver.completions.providers.contextproviders.WorkerDeclarationContextProvider.java Source code

Introduction

Here is the source code for org.ballerinalang.langserver.completions.providers.contextproviders.WorkerDeclarationContextProvider.java

Source

/*
 *  Copyright (c) 2019, WSO2 Inc. (http://www.wso2.org) All Rights Reserved.
 *
 *  WSO2 Inc. licenses this file to you 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 org.ballerinalang.langserver.completions.providers.contextproviders;

import org.antlr.v4.runtime.CommonToken;
import org.antlr.v4.runtime.Token;
import org.ballerinalang.annotation.JavaSPIService;
import org.ballerinalang.langserver.SnippetGenerator;
import org.ballerinalang.langserver.compiler.LSContext;
import org.ballerinalang.langserver.completions.CompletionKeys;
import org.ballerinalang.langserver.completions.SymbolInfo;
import org.ballerinalang.langserver.completions.spi.LSCompletionProvider;
import org.eclipse.lsp4j.CompletionItem;
import org.wso2.ballerinalang.compiler.parser.antlr4.BallerinaParser;

import java.util.ArrayList;
import java.util.List;
import java.util.stream.Collectors;

/**
 * Parser rule based variable definition statement context resolver.
 */
@JavaSPIService("org.ballerinalang.langserver.completions.spi.LSCompletionProvider")
public class WorkerDeclarationContextProvider extends LSCompletionProvider {
    public WorkerDeclarationContextProvider() {
        this.attachmentPoints.add(BallerinaParser.WorkerDeclarationContext.class);
    }

    @Override
    @SuppressWarnings("unchecked")
    public List<CompletionItem> getCompletions(LSContext context) {
        ArrayList<CompletionItem> completionItems = new ArrayList<>();
        List<SymbolInfo> visibleSymbols = context.get(CompletionKeys.VISIBLE_SYMBOLS_KEY);
        Boolean inWorkerReturnCtx = context.get(CompletionKeys.IN_WORKER_RETURN_CONTEXT_KEY);

        if (this.isInvocationOrInteractionOrFieldAccess(context)) {
            List<CommonToken> defaultTokens = context.get(CompletionKeys.LHS_TOKENS_KEY).stream()
                    .filter(commonToken -> commonToken.getChannel() == Token.DEFAULT_CHANNEL)
                    .collect(Collectors.toList());
            List<Integer> defaultTokenTypes = defaultTokens.stream().map(CommonToken::getType)
                    .collect(Collectors.toList());
            int pkgDelimIndex = defaultTokenTypes.indexOf(BallerinaParser.COLON);
            if (pkgDelimIndex > -1) {
                String pkgName = defaultTokens.get(pkgDelimIndex - 1).getText();
                completionItems.addAll(this.getTypesInPackage(visibleSymbols, pkgName, context));
            }
        } else if (inWorkerReturnCtx != null && inWorkerReturnCtx) {
            completionItems.addAll(this.getBasicTypes(visibleSymbols));
            completionItems.addAll(this.getPackagesCompletionItems(context));
        } else {
            completionItems.add(SnippetGenerator.getReturnsKeywordSnippet().build(context));
        }

        return completionItems;
    }
}