Translate syntax 3

rosetta-ide/src/main/java/com/regnosys/rosetta/ide/RosettaIdeModule.xtend

@@ -30,6 +30,10 @@ import com.regnosys.rosetta.ide.contentassist.cancellable.CancellableRosettaPars
 import com.regnosys.rosetta.ide.contentassist.cancellable.CancellableContentAssistService
 import com.regnosys.rosetta.ide.contentassist.cancellable.RosettaOperationCanceledManager
 import com.regnosys.rosetta.ide.semantictokens.RosettaSemanticTokenModifiersProvider
+import org.eclipse.xtext.ide.server.symbol.DocumentSymbolService
+import com.regnosys.rosetta.ide.symbol.RosettaDocumentSymbolService
+import org.eclipse.xtext.findReferences.TargetURICollector
+import com.regnosys.rosetta.ide.symbol.RosettaTargetURICollector
 
 /**
  * Use this class to register ide components.
@@ -95,4 +99,12 @@ class RosettaIdeModule extends AbstractRosettaIdeModule {
 	def Class<? extends OperationCanceledManager> bindOperationCanceledManager() {
 		RosettaOperationCanceledManager
 	}
+
+	def Class<? extends DocumentSymbolService> bindDocumentSymbolService() {
+		RosettaDocumentSymbolService
+	}
+
+	def Class<? extends TargetURICollector> bindTargetURICollector() {
+		RosettaTargetURICollector
+	}
 }

rosetta-ide/src/main/java/com/regnosys/rosetta/ide/build/RosettaSerializableResourceDescription.java

@@ -0,0 +1,111 @@
+package com.regnosys.rosetta.ide.build;
+
+import static com.google.common.collect.Iterables.transform;
+import static org.eclipse.xtext.resource.persistence.SerializationExtensions.readCastedObject;
+
+import java.io.IOException;
+import java.io.ObjectInput;
+import java.io.ObjectOutput;
+import java.util.ArrayList;
+import java.util.Collections;
+import java.util.HashMap;
+import java.util.List;
+
+import org.eclipse.emf.common.util.URI;
+import org.eclipse.xtext.resource.IEObjectDescription;
+import org.eclipse.xtext.resource.IReferenceDescription;
+import org.eclipse.xtext.resource.persistence.SerializableEObjectDescription;
+import org.eclipse.xtext.resource.persistence.SerializableEObjectDescriptionProvider;
+import org.eclipse.xtext.resource.persistence.SerializableReferenceDescription;
+import org.eclipse.xtext.resource.persistence.SerializableResourceDescription;
+
+import com.google.common.collect.Lists;
+import com.regnosys.rosetta.resource.IImplicitReferenceDescription;
+import com.regnosys.rosetta.resource.IRosettaResourceDescription;
+
+public class RosettaSerializableResourceDescription extends SerializableResourceDescription implements IRosettaResourceDescription {
+
+	private List<SerializableImplicitReferenceDescription> implicitReferences = Collections.emptyList();
+
+	@Override
+	public void updateResourceURI(URI uri) {
+		for (SerializableImplicitReferenceDescription implRef : implicitReferences) {
+			implRef.updateResourceURI(uri, getURI());
+		}
+		super.updateResourceURI(uri);
+	}
+
+	@Override
+	@SuppressWarnings("unchecked")
+	public Iterable<IImplicitReferenceDescription> getImplicitReferenceDescriptions() {
+		return ((Iterable<IImplicitReferenceDescription>) ((Iterable<?>) implicitReferences));
+	}
+
+	@Override
+	public void readExternal(ObjectInput in) throws IOException, ClassNotFoundException {
+		super.readExternal(in);
+		int implicitReferencesSize = in.readInt();
+		implicitReferences = new ArrayList<>(implicitReferencesSize);
+		for (int i = 0; i < implicitReferencesSize; i++)
+			implicitReferences.add(readCastedObject(in));
+	}
+
+	@Override
+	public void writeExternal(ObjectOutput out) throws IOException {
+		super.writeExternal(out);
+		out.writeInt(implicitReferences.size());
+		for (SerializableImplicitReferenceDescription implRef : implicitReferences)
+			out.writeObject(implRef);
+	}
+
+	public List<SerializableImplicitReferenceDescription> getImplicitReferences() {
+		return implicitReferences;
+	}
+
+	public void setImplicitReferences(List<SerializableImplicitReferenceDescription> implicitReferences) {
+		this.implicitReferences = implicitReferences;
+	}
+
+	public static RosettaSerializableResourceDescription createCopy(IRosettaResourceDescription desc) {
+		RosettaSerializableResourceDescription description = new RosettaSerializableResourceDescription();
+		description.setURI(desc.getURI());
+		description.setDescriptions(Lists.newArrayList(
+				transform(desc.getExportedObjects(), RosettaSerializableResourceDescription::createCopy)));
+		description.setReferences(Lists.newArrayList(
+				transform(desc.getReferenceDescriptions(), RosettaSerializableResourceDescription::createCopy)));
+		description.setImportedNames(Lists.newArrayList(desc.getImportedNames()));
+		description.setImplicitReferences(Lists.newArrayList(
+				transform(desc.getImplicitReferenceDescriptions(), RosettaSerializableResourceDescription::createCopy)));
+		return description;
+	}
+
+	private static SerializableEObjectDescription createCopy(IEObjectDescription desc) {
+		if (desc instanceof SerializableEObjectDescriptionProvider)
+			return ((SerializableEObjectDescriptionProvider) desc).toSerializableEObjectDescription();
+		SerializableEObjectDescription result = new SerializableEObjectDescription();
+		result.setEClass(desc.getEClass());
+		result.setEObjectURI(desc.getEObjectURI());
+		result.setQualifiedName(desc.getQualifiedName());
+		result.setUserData(new HashMap<String, String>(desc.getUserDataKeys().length));
+		for (String key : desc.getUserDataKeys())
+			result.getUserData().put(key, desc.getUserData(key));
+		return result;
+	}
+
+	private static SerializableReferenceDescription createCopy(IReferenceDescription desc) {
+		SerializableReferenceDescription result = new SerializableReferenceDescription();
+		result.setSourceEObjectUri(desc.getSourceEObjectUri());
+		result.setTargetEObjectUri(desc.getTargetEObjectUri());
+		result.setEReference(desc.getEReference());
+		result.setIndexInList(desc.getIndexInList());
+		result.setContainerEObjectURI(desc.getContainerEObjectURI());
+		return result;
+	}
+
+	private static SerializableImplicitReferenceDescription createCopy(IImplicitReferenceDescription desc) {
+		SerializableImplicitReferenceDescription result = new SerializableImplicitReferenceDescription();
+		result.setSourceEObjectUri(desc.getSourceEObjectUri());
+		result.setTargetEObjectUri(desc.getTargetEObjectUri());
+		return result;
+	}
+}

rosetta-ide/src/main/java/com/regnosys/rosetta/ide/build/RosettaStatefulIncrementalBuilder.java

@@ -0,0 +1,16 @@
+package com.regnosys.rosetta.ide.build;
+
+import org.eclipse.xtext.build.IncrementalBuilder;
+import org.eclipse.xtext.resource.IResourceDescription;
+
+import com.regnosys.rosetta.resource.IRosettaResourceDescription;
+
+public class RosettaStatefulIncrementalBuilder extends IncrementalBuilder.InternalStatefulIncrementalBuilder {
+	@Override
+	protected IResourceDescription getSerializableResourceDescription(IResourceDescription description) {
+		if (description instanceof IRosettaResourceDescription) {
+			return RosettaSerializableResourceDescription.createCopy((IRosettaResourceDescription)description);
+		}
+		return super.getSerializableResourceDescription(description);
+	}
+}

rosetta-ide/src/main/java/com/regnosys/rosetta/ide/build/SerializableImplicitReferenceDescription.java

@@ -0,0 +1,53 @@
+package com.regnosys.rosetta.ide.build;
+
+import static org.eclipse.xtext.resource.persistence.SerializationExtensions.readURI;
+import static org.eclipse.xtext.resource.persistence.SerializationExtensions.writeURI;
+
+import java.io.Externalizable;
+import java.io.IOException;
+import java.io.ObjectInput;
+import java.io.ObjectOutput;
+
+import org.eclipse.emf.common.util.URI;
+
+import com.regnosys.rosetta.resource.IImplicitReferenceDescription;
+
+public class SerializableImplicitReferenceDescription implements IImplicitReferenceDescription, Externalizable {
+	private URI sourceEObjectUri;
+
+	private URI targetEObjectUri;
+
+	@Override
+	public void readExternal(ObjectInput in) throws IOException, ClassNotFoundException {
+		sourceEObjectUri = readURI(in);
+		targetEObjectUri = readURI(in);
+	}
+
+	@Override
+	public void writeExternal(ObjectOutput out) throws IOException {
+		writeURI(out, sourceEObjectUri);
+		writeURI(out, targetEObjectUri);
+	}
+
+	public void updateResourceURI(URI newURI, URI oldURI) {
+		sourceEObjectUri = newURI.appendFragment(sourceEObjectUri.fragment());
+		if (targetEObjectUri.trimFragment().equals(oldURI))
+			targetEObjectUri = newURI.appendFragment(targetEObjectUri.fragment());
+	}
+
+	public URI getSourceEObjectUri() {
+		return sourceEObjectUri;
+	}
+
+	public void setSourceEObjectUri(URI sourceEObjectUri) {
+		this.sourceEObjectUri = sourceEObjectUri;
+	}
+
+	public URI getTargetEObjectUri() {
+		return targetEObjectUri;
+	}
+
+	public void setTargetEObjectUri(URI targetEObjectUri) {
+		this.targetEObjectUri = targetEObjectUri;
+	}
+}

rosetta-ide/src/main/java/com/regnosys/rosetta/ide/semantictokens/RosettaSemanticTokensService.java

@@ -37,10 +37,12 @@
 import com.regnosys.rosetta.rosetta.RosettaSymbol;
 import com.regnosys.rosetta.rosetta.RosettaType;
 import com.regnosys.rosetta.rosetta.RosettaTypeAlias;
+import com.regnosys.rosetta.rosetta.translate.TranslationParameter;
 import com.regnosys.rosetta.rosetta.TypeCall;
 import com.regnosys.rosetta.rosetta.TypeParameter;
 import com.regnosys.rosetta.rosetta.expression.ClosureParameter;
 import com.regnosys.rosetta.rosetta.expression.ConstructorKeyValuePair;
+import com.regnosys.rosetta.rosetta.expression.RosettaExpression;
 import com.regnosys.rosetta.rosetta.expression.RosettaFeatureCall;
 import com.regnosys.rosetta.rosetta.expression.RosettaImplicitVariable;
 import com.regnosys.rosetta.rosetta.expression.RosettaSymbolReference;
@@ -62,6 +64,7 @@
 import static com.regnosys.rosetta.rosetta.simple.SimplePackage.Literals.*;
 
 import java.util.ArrayList;
+import java.util.HashSet;
 import java.util.List;
 import java.util.Optional;
 import java.util.Set;
@@ -76,7 +79,7 @@ public class RosettaSemanticTokensService extends AbstractSemanticTokensService
 	private CardinalityProvider cardinalityProvider;
 	@Inject
 	private RosettaTypeProvider typeProvider;
-	
+
 	@Inject
 	public RosettaSemanticTokensService(ISemanticTokenTypesProvider tokenTypesProvider,
 			ISemanticTokenModifiersProvider tokenModifiersProvider) {
@@ -174,7 +177,7 @@ private SemanticToken markAttribute(EObject objectToMark, EStructuralFeature fea
 		} else if (t == AttributeType.OUTPUT) {
 			return createSemanticToken(objectToMark, featureToMark, OUTPUT, getCardinalityModifier(attribute));
 		}
-		
+
 		RosettaSemanticTokenTypesEnum tokenType = null;
 		EReference containmentFeature = attribute.eContainmentFeature();
 		if (containmentFeature.equals(FUNCTION__INPUTS)) {
@@ -283,6 +286,8 @@ private SemanticToken markSymbol(EObject objectToMark, EStructuralFeature featur
 			return markAlias(objectToMark, featureToMark, (ShortcutDeclaration)symbol);
 		} else if (symbol instanceof TypeParameter) {
 			return createSemanticToken(objectToMark, featureToMark, PARAMETER);
+		} else if (symbol instanceof TranslationParameter) {
+			return createSemanticToken(objectToMark, featureToMark, PARAMETER);
 		}
 		return null;
 	}

rosetta-ide/src/main/java/com/regnosys/rosetta/ide/server/RosettaServerModule.java

@@ -20,13 +20,15 @@
 import java.lang.reflect.InvocationTargetException;
 
 import org.eclipse.lsp4j.services.LanguageServer;
+import org.eclipse.xtext.build.IncrementalBuilder.InternalStatefulIncrementalBuilder;
 import org.eclipse.xtext.ide.server.ILanguageServerShutdownAndExitHandler;
 import org.eclipse.xtext.ide.server.LanguageServerImpl;
 import org.eclipse.xtext.ide.server.ServerModule;
 import org.eclipse.xtext.ide.server.concurrent.RequestManager;
 import org.eclipse.xtext.service.AbstractGenericModule;
 
 import com.google.inject.util.Modules;
+import com.regnosys.rosetta.ide.build.RosettaStatefulIncrementalBuilder;
 import com.google.inject.Module;
 
 public class RosettaServerModule extends AbstractGenericModule {
@@ -78,4 +80,8 @@ public Class<? extends RequestManager> bindRequestManager() {
 	public Class<? extends ILanguageServerShutdownAndExitHandler> bindILanguageServerShutdownAndExitHandler() {
 		return RosettaLanguageServerShutdownAndExitHandler.class;
 	}
+
+	public Class<? extends InternalStatefulIncrementalBuilder> bindInternalStatefulIncrementalBuilder() {
+		return RosettaStatefulIncrementalBuilder.class;
+	}
 }