org.eclipse.osbp.xtext.signal.ui/src/org/eclipse/osbp/xtext/signal/ui/contentassist/SignalDSLProposalProvider.xtend - osbp/org.eclipse.osbp.xtext.signal - Git at Google

 /**
  *
  *  Copyright (c) 2011, 2018 - Loetz GmbH&Co.KG (69115 Heidelberg, Germany)
  *
  *  All rights reserved. This program and the accompanying materials
  *  are made available under the terms of the Eclipse Public License 2.0
  *  which accompanies this distribution, and is available at
  *  https://www.eclipse.org/legal/epl-2.0/
  *
  *  SPDX-License-Identifier: EPL-2.0
  *
  *  Contributors:
  * 	   Christophe Loetz (Loetz GmbH&Co.KG) - initial implementation
  *
  * generated by Xtext 2.11.0
  *
  */

 package org.eclipse.osbp.xtext.signal.ui.contentassist

 import com.google.inject.Inject
 import java.net.MalformedURLException
 import java.net.URL
 import java.nio.file.Paths
 import org.eclipse.emf.common.util.EList
 import org.eclipse.emf.ecore.EObject
 import org.eclipse.jface.viewers.StyledString
 import org.eclipse.osbp.xtext.basic.ui.contentassist.BasicDSLProposalProviderHelper
 import org.eclipse.osbp.xtext.datainterchange.DataInterchange
 import org.eclipse.osbp.xtext.signal.SignalDatainterchange
 import org.eclipse.osbp.xtext.signal.SignalDefinition
 import org.eclipse.osbp.xtext.signal.SignalFunction
 import org.eclipse.osbp.xtext.signal.SignalScheduler
 import org.eclipse.osbp.xtext.signal.SignalTask
 import org.eclipse.osbp.xtext.signal.SignalWatcher
 import org.eclipse.osbp.xtext.signal.ui.SignalDSLDocumentationTranslator
 import org.eclipse.swt.widgets.DirectoryDialog
 import org.eclipse.wb.internal.core.nls.ui.ChooseLocaleDialog
 import org.eclipse.xtext.Keyword
 import org.eclipse.xtext.RuleCall
 import org.eclipse.xtext.common.ui.contentassist.TerminalsProposalProvider
 import org.eclipse.xtext.ui.editor.contentassist.ConfigurableCompletionProposal
 import org.eclipse.xtext.ui.editor.contentassist.ContentAssistContext
 import org.eclipse.xtext.ui.editor.contentassist.ICompletionProposalAcceptor
 import org.eclipse.xtext.ui.editor.contentassist.ReplacementTextApplier

 class LocaleTextApplier extends ReplacementTextApplier {
 	var ContentAssistContext context

 	def setContext(ContentAssistContext context) {
 		this.context = context
 	}

 	override getActualReplacementString(ConfigurableCompletionProposal proposal) {
 		var display = context.getViewer().getTextWidget().getDisplay();
 		var localeDialog = new ChooseLocaleDialog(display.getActiveShell());
 		localeDialog.open();
 		return "\"".concat(localeDialog.selectedLocale.toString).concat("\"");
 	}
 }

 class DirectoryNameTextApplier extends ReplacementTextApplier {
 	var ContentAssistContext context

 	def setContext(ContentAssistContext context) {
 		this.context = context
 	}

 	// this will inject a directory dialog when selecting the dialog picker proposal
 	override getActualReplacementString(ConfigurableCompletionProposal proposal) {
 		var display = context.getViewer().getTextWidget().getDisplay();
 		var dirDialog = new DirectoryDialog(display.getActiveShell());
 		var dirName = dirDialog.open();
 		if (dirName !== null) {
 			dirName = dirName.replace("\\", "/")
 		} else {
 			dirName = "Directory"
 		}
 		return "\"".concat(dirName).concat("\"");
 	}
 }

 class SignalDSLProposalProvider extends AbstractSignalDSLProposalProvider {

 	private val APPLYON = "applyon"

 	@Inject TerminalsProposalProvider provider
 	@Inject BasicDSLProposalProviderHelper providerHelper

 	override protected StyledString getKeywordDisplayString(Keyword keyword) {
 		return BasicDSLProposalProviderHelper.getKeywordDisplayString(keyword,
 			SignalDSLDocumentationTranslator.instance())
 	}

 	/**
 	 * This override will enable 1 length non letter characters as keyword.
 	 */
 	override protected boolean isKeywordWorthyToPropose(Keyword keyword) {
 		return true
 	}

 	override complete_QualifiedName(EObject model, RuleCall ruleCall, ContentAssistContext context,
 		ICompletionProposalAcceptor acceptor) {
 		providerHelper.complete_PackageName(model, ruleCall, context, acceptor, this)
 	}

 	def directoryPickerProposal(EObject model, ContentAssistContext context, ICompletionProposalAcceptor acceptor) {
 		var dirName = createCompletionProposal("Select directory...", context) as ConfigurableCompletionProposal
 		if (dirName !== null) {
 			var applier = new DirectoryNameTextApplier()
 			applier.setContext(context)
 			dirName.setTextApplier = applier
 		}
 		acceptor.accept(dirName)
 	}

 	/**
 	 * This method decided which proposals will be valid.
 	 */
 	override protected boolean isValidProposal(String proposal, String prefix, ContentAssistContext context) {
 		var result = super.isValidProposal(proposal, prefix, context)

 		if (context.currentModel instanceof SignalWatcher ) {
 			result = isValidProposalForSignalWatchers(proposal, prefix, (context.currentModel as SignalWatcher), result)
 		}
 		else if(context.currentModel instanceof SignalDatainterchange){
 			result = isValidProposalSignalDatainterchange(proposal, prefix, (context.currentModel as SignalDatainterchange), result)
 		}
 		return result
 	}

 	def isValidProposalSignalDatainterchange(String proposal, String prefix, SignalDatainterchange interchange, boolean result) {
 		if (proposal.equals(APPLYON)) {
 			var signal = interchange.eContainer

 			if(signal instanceof SignalWatcher){
 				if(signal.defaultInterchange !== null || signal.tasks.basedInterchangeDefined ){
 					return false
 				}

 			}
 			else if(signal instanceof SignalScheduler){
 				return false
 			}
 		}
 		return result
 	}

 	def private boolean isValidProposalForSignalWatchers(String proposal, String prefix, SignalWatcher watcher,	boolean result) {
 		if (watcher !== null && watcher.tasks !== null && !watcher.tasks.empty) {

 			if(proposal.equals(APPLYON)){
 				if( watcher.defaultInterchange !== null ){ // default interchange already defined
 					return false
 				}

 				if( watcher.tasks.basedInterchangeDefined && watcher.defaultInterchange === null){ // base interchange already defined
 					return false
 				}
 			}

 		}
 		return result
 	}

 	def boolean basedInterchangeDefined(EList<SignalTask> tasks){
 		for (task : tasks) {
 			if(task instanceof SignalDatainterchange)
 			if (task.baseinterchange) {
 				return true
 			}
 		}
 		return false
 	}

 	def getDirectoryURL(String directory) {
 	    try {
 			return new URL(directory);
 		} catch (MalformedURLException e1) {
 			if(e1.getMessage().startsWith("unknown protocol") || e1.getMessage().startsWith("no protocol")) {
 				try {
 					return Paths.get(directory).toUri().toURL();
 				} catch (MalformedURLException e2) {
 					return null
 				}
 			}
 		}
 		return null
 	}

 	def DataInterchange baseInterchange(SignalDefinition signal){
 		if(signal instanceof SignalWatcher) {
 			if(signal.defaultInterchange !== null){
 				return signal.defaultInterchange
 			}
 			else if (signal.tasks !== null && !signal.tasks.empty){
 				for(task : signal.tasks){
 					if(task instanceof SignalDatainterchange){
 						if(task.baseinterchange){
 							return task.dataRef
 						}
 					}
 				}
 			}
 		}
 		return null
 	}

 	def int definedBaseInterchangeCount(SignalDefinition signal) {
 		var count = 0
 		for (task : signal.tasks) {
 			if(task instanceof SignalDatainterchange){
 				if(task.baseinterchange){
 					count++
 				}
 			}
 		}
 		return count
 	}

 	def SignalDatainterchange definedBaseInterchange(SignalDefinition signal) {
 		for (task : signal.tasks) {
 			if(task instanceof SignalDatainterchange){
 				if(task.baseinterchange){
 					return task
 				}
 			}
 		}
 		return null
 	}

 	def int taskCount(SignalDefinition signal, boolean interchange) {
 		var count = 0
 		for (task : signal.tasks) {
 			if(interchange){
 				if(task instanceof SignalDatainterchange){
 					count++
 				}
 			}
 			else if(task instanceof SignalFunction){
 				count++
 			}
 		}
 		return count
 	}

 	// ------------------------ delegates to TerminalsProposalProvider -----------------
 	override public void complete_INT(EObject model, RuleCall ruleCall, ContentAssistContext context,
 		ICompletionProposalAcceptor acceptor) {
 		provider.complete_INT(model, ruleCall, context, acceptor)
 	}

 	override public void complete_TRANSLATABLEID(EObject model, RuleCall ruleCall, ContentAssistContext context,
 		ICompletionProposalAcceptor acceptor) {
 		provider.complete_ID(model, ruleCall, context, acceptor)
 	}

 	override public void complete_TRANSLATABLESTRING(EObject model, RuleCall ruleCall, ContentAssistContext context,
 		ICompletionProposalAcceptor acceptor) {
 		provider.complete_STRING(model, ruleCall, context, acceptor)
 	}

 	override public void complete_STRING(EObject model, RuleCall ruleCall, ContentAssistContext context,
 		ICompletionProposalAcceptor acceptor) {
 		provider.complete_STRING(model, ruleCall, context, acceptor)
 	}
 }
	/**
	*
	* Copyright (c) 2011, 2018 - Loetz GmbH&Co.KG (69115 Heidelberg, Germany)
	*
	* All rights reserved. This program and the accompanying materials
	* are made available under the terms of the Eclipse Public License 2.0
	* which accompanies this distribution, and is available at
	* https://www.eclipse.org/legal/epl-2.0/
	*
	* SPDX-License-Identifier: EPL-2.0
	*
	* Contributors:
	* Christophe Loetz (Loetz GmbH&Co.KG) - initial implementation
	*
	* generated by Xtext 2.11.0
	*
	*/

	package org.eclipse.osbp.xtext.signal.ui.contentassist

	import com.google.inject.Inject
	import java.net.MalformedURLException
	import java.net.URL
	import java.nio.file.Paths
	import org.eclipse.emf.common.util.EList
	import org.eclipse.emf.ecore.EObject
	import org.eclipse.jface.viewers.StyledString
	import org.eclipse.osbp.xtext.basic.ui.contentassist.BasicDSLProposalProviderHelper
	import org.eclipse.osbp.xtext.datainterchange.DataInterchange
	import org.eclipse.osbp.xtext.signal.SignalDatainterchange
	import org.eclipse.osbp.xtext.signal.SignalDefinition
	import org.eclipse.osbp.xtext.signal.SignalFunction
	import org.eclipse.osbp.xtext.signal.SignalScheduler
	import org.eclipse.osbp.xtext.signal.SignalTask
	import org.eclipse.osbp.xtext.signal.SignalWatcher
	import org.eclipse.osbp.xtext.signal.ui.SignalDSLDocumentationTranslator
	import org.eclipse.swt.widgets.DirectoryDialog
	import org.eclipse.wb.internal.core.nls.ui.ChooseLocaleDialog
	import org.eclipse.xtext.Keyword
	import org.eclipse.xtext.RuleCall
	import org.eclipse.xtext.common.ui.contentassist.TerminalsProposalProvider
	import org.eclipse.xtext.ui.editor.contentassist.ConfigurableCompletionProposal
	import org.eclipse.xtext.ui.editor.contentassist.ContentAssistContext
	import org.eclipse.xtext.ui.editor.contentassist.ICompletionProposalAcceptor
	import org.eclipse.xtext.ui.editor.contentassist.ReplacementTextApplier

	class LocaleTextApplier extends ReplacementTextApplier {
	var ContentAssistContext context

	def setContext(ContentAssistContext context) {
	this.context = context
	}

	override getActualReplacementString(ConfigurableCompletionProposal proposal) {
	var display = context.getViewer().getTextWidget().getDisplay();
	var localeDialog = new ChooseLocaleDialog(display.getActiveShell());
	localeDialog.open();
	return "\"".concat(localeDialog.selectedLocale.toString).concat("\"");
	}
	}

	class DirectoryNameTextApplier extends ReplacementTextApplier {
	var ContentAssistContext context

	def setContext(ContentAssistContext context) {
	this.context = context
	}

	// this will inject a directory dialog when selecting the dialog picker proposal
	override getActualReplacementString(ConfigurableCompletionProposal proposal) {
	var display = context.getViewer().getTextWidget().getDisplay();
	var dirDialog = new DirectoryDialog(display.getActiveShell());
	var dirName = dirDialog.open();
	if (dirName !== null) {
	dirName = dirName.replace("\\", "/")
	} else {
	dirName = "Directory"
	}
	return "\"".concat(dirName).concat("\"");
	}
	}

	class SignalDSLProposalProvider extends AbstractSignalDSLProposalProvider {

	private val APPLYON = "applyon"

	@Inject TerminalsProposalProvider provider
	@Inject BasicDSLProposalProviderHelper providerHelper

	override protected StyledString getKeywordDisplayString(Keyword keyword) {
	return BasicDSLProposalProviderHelper.getKeywordDisplayString(keyword,
	SignalDSLDocumentationTranslator.instance())
	}

	/**
	* This override will enable 1 length non letter characters as keyword.
	*/
	override protected boolean isKeywordWorthyToPropose(Keyword keyword) {
	return true
	}

	override complete_QualifiedName(EObject model, RuleCall ruleCall, ContentAssistContext context,
	ICompletionProposalAcceptor acceptor) {
	providerHelper.complete_PackageName(model, ruleCall, context, acceptor, this)
	}

	def directoryPickerProposal(EObject model, ContentAssistContext context, ICompletionProposalAcceptor acceptor) {
	var dirName = createCompletionProposal("Select directory...", context) as ConfigurableCompletionProposal
	if (dirName !== null) {
	var applier = new DirectoryNameTextApplier()
	applier.setContext(context)
	dirName.setTextApplier = applier
	}
	acceptor.accept(dirName)
	}

	/**
	* This method decided which proposals will be valid.
	*/
	override protected boolean isValidProposal(String proposal, String prefix, ContentAssistContext context) {
	var result = super.isValidProposal(proposal, prefix, context)

	if (context.currentModel instanceof SignalWatcher ) {
	result = isValidProposalForSignalWatchers(proposal, prefix, (context.currentModel as SignalWatcher), result)
	}
	else if(context.currentModel instanceof SignalDatainterchange){
	result = isValidProposalSignalDatainterchange(proposal, prefix, (context.currentModel as SignalDatainterchange), result)
	}
	return result
	}

	def isValidProposalSignalDatainterchange(String proposal, String prefix, SignalDatainterchange interchange, boolean result) {
	if (proposal.equals(APPLYON)) {
	var signal = interchange.eContainer

	if(signal instanceof SignalWatcher){
	if(signal.defaultInterchange !== null \|\| signal.tasks.basedInterchangeDefined ){
	return false
	}

	}
	else if(signal instanceof SignalScheduler){
	return false
	}
	}
	return result
	}

	def private boolean isValidProposalForSignalWatchers(String proposal, String prefix, SignalWatcher watcher, boolean result) {
	if (watcher !== null && watcher.tasks !== null && !watcher.tasks.empty) {

	if(proposal.equals(APPLYON)){
	if( watcher.defaultInterchange !== null ){ // default interchange already defined
	return false
	}

	if( watcher.tasks.basedInterchangeDefined && watcher.defaultInterchange === null){ // base interchange already defined
	return false
	}
	}

	}
	return result
	}

	def boolean basedInterchangeDefined(EList<SignalTask> tasks){
	for (task : tasks) {
	if(task instanceof SignalDatainterchange)
	if (task.baseinterchange) {
	return true
	}
	}
	return false
	}

	def getDirectoryURL(String directory) {
	try {
	return new URL(directory);
	} catch (MalformedURLException e1) {
	if(e1.getMessage().startsWith("unknown protocol") \|\| e1.getMessage().startsWith("no protocol")) {
	try {
	return Paths.get(directory).toUri().toURL();
	} catch (MalformedURLException e2) {
	return null
	}
	}
	}
	return null
	}

	def DataInterchange baseInterchange(SignalDefinition signal){
	if(signal instanceof SignalWatcher) {
	if(signal.defaultInterchange !== null){
	return signal.defaultInterchange
	}
	else if (signal.tasks !== null && !signal.tasks.empty){
	for(task : signal.tasks){
	if(task instanceof SignalDatainterchange){
	if(task.baseinterchange){
	return task.dataRef
	}
	}
	}
	}
	}
	return null
	}

	def int definedBaseInterchangeCount(SignalDefinition signal) {
	var count = 0
	for (task : signal.tasks) {
	if(task instanceof SignalDatainterchange){
	if(task.baseinterchange){
	count++
	}
	}
	}
	return count
	}

	def SignalDatainterchange definedBaseInterchange(SignalDefinition signal) {
	for (task : signal.tasks) {
	if(task instanceof SignalDatainterchange){
	if(task.baseinterchange){
	return task
	}
	}
	}
	return null
	}

	def int taskCount(SignalDefinition signal, boolean interchange) {
	var count = 0
	for (task : signal.tasks) {
	if(interchange){
	if(task instanceof SignalDatainterchange){
	count++
	}
	}
	else if(task instanceof SignalFunction){
	count++
	}
	}
	return count
	}

	// ------------------------ delegates to TerminalsProposalProvider -----------------
	override public void complete_INT(EObject model, RuleCall ruleCall, ContentAssistContext context,
	ICompletionProposalAcceptor acceptor) {
	provider.complete_INT(model, ruleCall, context, acceptor)
	}

	override public void complete_TRANSLATABLEID(EObject model, RuleCall ruleCall, ContentAssistContext context,
	ICompletionProposalAcceptor acceptor) {
	provider.complete_ID(model, ruleCall, context, acceptor)
	}

	override public void complete_TRANSLATABLESTRING(EObject model, RuleCall ruleCall, ContentAssistContext context,
	ICompletionProposalAcceptor acceptor) {
	provider.complete_STRING(model, ruleCall, context, acceptor)
	}

	override public void complete_STRING(EObject model, RuleCall ruleCall, ContentAssistContext context,
	ICompletionProposalAcceptor acceptor) {
	provider.complete_STRING(model, ruleCall, context, acceptor)
	}
	}