com.regnosys.rosetta.ide.inlayhints.RosettaInlayHintsService Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of com.regnosys.rosetta.ide Show documentation

Responsibilities: adding support for developing Rosetta in an IDE, including - language server features (semantic highlighting, inlay hints, etc) - syntax highlighting

The newest version!

/*
 * Copyright 2024 REGnosys
 *
 * 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 com.regnosys.rosetta.ide.inlayhints;

import javax.inject.Inject;

import org.eclipse.emf.ecore.EObject;
import org.eclipse.lsp4j.InlayHint;
import org.eclipse.xtext.Keyword;
import org.eclipse.xtext.nodemodel.ICompositeNode;
import org.eclipse.xtext.nodemodel.INode;
import org.eclipse.xtext.nodemodel.util.NodeModelUtils;

import com.regnosys.rosetta.RosettaEcoreUtil;
import com.regnosys.rosetta.types.CardinalityProvider;
import com.regnosys.rosetta.rosetta.expression.InlineFunction;
import com.regnosys.rosetta.rosetta.expression.MapOperation;
import com.regnosys.rosetta.rosetta.expression.ReduceOperation;
import com.regnosys.rosetta.rosetta.expression.RosettaFunctionalOperation;
import com.regnosys.rosetta.services.RosettaGrammarAccess;
import com.regnosys.rosetta.types.RType;
import com.regnosys.rosetta.types.RosettaTypeProvider;

public class RosettaInlayHintsService extends AbstractInlayHintsService {
	@Inject
	private RosettaEcoreUtil extensions;
	@Inject
	private RosettaTypeProvider types;
	@Inject
	private CardinalityProvider card;
	@Inject
	private RosettaGrammarAccess grammar;
	
	private String typeInfo(RType type, boolean isMulti) {
		if (isMulti) {
			return type.getName() + " (0..*)";
		} else {
			return type.getName() + " (0..1)";
		}
	}
	
	@InlayHintCheck
	public InlayHint checkFunctionalOperation(RosettaFunctionalOperation op) {
		if (op.getFunction() != null && operationHasBrackets(op.getFunction())) {
			if (op instanceof ReduceOperation || op instanceof MapOperation) {
				if (extensions.isResolved(op.getFunction())) {
					RType outputType = types.getRMetaAnnotatedType(op).getRType();
					boolean outputMulti = card.isMulti(op);
		
					if (outputType != null) {
						return inlayHintAfter(op, typeInfo(outputType, outputMulti), null);
					}
				}
			}
		}
		return null;
	}
	
	private boolean operationHasBrackets(InlineFunction op) {
		Keyword keyword = grammar.getInlineFunctionAccess().getLeftSquareBracketKeyword_0_0_1();
		ICompositeNode node = NodeModelUtils.findActualNodeFor(op);

        for (INode n : node.getChildren()) {
            EObject ge = n.getGrammarElement();
            if (ge instanceof Keyword && ge == keyword) {
                return true;
            }
        }
        return false;
	}
}