ModuleLevelVariableDeclarationAnalysisTask.java

/*
 * Copyright (c) 2022, 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 io.ballerina.stdlib.graphql.compiler;

import io.ballerina.compiler.api.SemanticModel;
import io.ballerina.compiler.api.symbols.VariableSymbol;
import io.ballerina.compiler.syntax.tree.ExpressionNode;
import io.ballerina.compiler.syntax.tree.ModuleVariableDeclarationNode;
import io.ballerina.compiler.syntax.tree.ObjectConstructorExpressionNode;
import io.ballerina.compiler.syntax.tree.SyntaxKind;
import io.ballerina.projects.DocumentId;
import io.ballerina.projects.plugins.SyntaxNodeAnalysisContext;
import io.ballerina.stdlib.graphql.commons.types.Schema;
import io.ballerina.stdlib.graphql.compiler.schema.generator.GeneratorUtils;
import io.ballerina.stdlib.graphql.compiler.service.InterfaceEntityFinder;
import io.ballerina.stdlib.graphql.compiler.service.validator.ServiceValidator;

import java.util.Map;

import static io.ballerina.stdlib.graphql.commons.utils.Utils.isGraphQLServiceObjectDeclaration;
import static io.ballerina.stdlib.graphql.compiler.Utils.hasCompilationErrors;

/**
 * Validates a Ballerina GraphQL Service variable declaration.
 */
public class ModuleLevelVariableDeclarationAnalysisTask extends ServiceAnalysisTask {

    public ModuleLevelVariableDeclarationAnalysisTask(Map<String, Object> nodeMap) {
        super(nodeMap);
    }

    @Override
    public void perform(SyntaxNodeAnalysisContext context) {
        if (hasCompilationErrors(context)) {
            return;
        }
        if (context.node().kind() != SyntaxKind.MODULE_VAR_DECL) {
            return;
        }
        ModuleVariableDeclarationNode moduleVariableDeclarationNode = (ModuleVariableDeclarationNode) context.node();
        if (!isGraphQLServiceObjectDeclaration(moduleVariableDeclarationNode)) {
            return;
        }
        if (moduleVariableDeclarationNode.initializer().isEmpty()) {
            return;
        }
        ExpressionNode expressionNode = moduleVariableDeclarationNode.initializer().get();
        if (expressionNode.kind() != SyntaxKind.OBJECT_CONSTRUCTOR) {
            return;
        }
        ObjectConstructorExpressionNode graphqlServiceObjectNode = (ObjectConstructorExpressionNode) expressionNode;
        InterfaceEntityFinder interfaceEntityFinder = getInterfaceEntityFinder(context.semanticModel());
        ServiceValidator serviceObjectValidator = getServiceValidator(context, graphqlServiceObjectNode,
                                                        interfaceEntityFinder, new CacheConfigContext(false));
        if (serviceObjectValidator.isErrorOccurred()) {
            return;
        }
        String description = getDescription(context.semanticModel(), moduleVariableDeclarationNode);
        Schema schema = generateSchema(context, interfaceEntityFinder, graphqlServiceObjectNode, description);
        DocumentId documentId = context.documentId();
        addToModifierContextMap(documentId, moduleVariableDeclarationNode, schema);
        addToModifierContextMap(documentId, moduleVariableDeclarationNode,
                serviceObjectValidator.getCacheConfigContext());
    }

    public static String getDescription(SemanticModel semanticModel,
                          ModuleVariableDeclarationNode moduleVariableDeclarationNode) {
        if (semanticModel.symbol(moduleVariableDeclarationNode).isEmpty()) {
            return null;
        }
        VariableSymbol serviceVariableSymbol = (VariableSymbol) semanticModel
                .symbol(moduleVariableDeclarationNode).get();
        return GeneratorUtils.getDescription(serviceVariableSymbol);
    }
}