org.eclipse.osbp.xtext.datainterchange.ui/src/org/eclipse/osbp/xtext/datainterchange/ui/contentassist/DataDSLProposalProvider.xtend

/**
 *                                                                            
 *  Copyright (c) 2011, 2016 - 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 v1.0       
 *  which accompanies this distribution, and is available at                  
 *  http://www.eclipse.org/legal/epl-v10.html                                 
 *                                                                            
 *  Contributors:                                                      
 * 	   Christophe Loetz (Loetz GmbH&Co.KG) - initial implementation
 * 
 * 
 *  This copyright notice shows up in the generated Java code
 * 
 */

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

import com.google.inject.Inject
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.ui.DataDSLDocumentationTranslator
import org.eclipse.swt.widgets.FileDialog
import org.eclipse.wb.internal.core.nls.ui.ChooseLocaleDialog
import org.eclipse.xtext.Assignment
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
import org.eclipse.osbp.xtext.datainterchange.DataInterchangeBean
import java.util.SortedMap
import java.nio.charset.Charset
import java.util.Set
import java.util.Iterator

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 FileNameTextApplier extends ReplacementTextApplier {
	var ContentAssistContext context
	var String[] extensions

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

	def setExtensions(String[] fileExtensions) {
		extensions = fileExtensions
	}

	// this will inject a file dialog when selecting the file picker proposal 
	override getActualReplacementString(ConfigurableCompletionProposal proposal) {
		var display = context.getViewer().getTextWidget().getDisplay();
		var fileDialog = new FileDialog(display.getActiveShell());
		fileDialog.setFilterExtensions(extensions)
		var fileName = fileDialog.open();
		fileName = fileName.replace("\\", "/")
		return "\"".concat(fileName).concat("\"");
	}
}

class DataDSLProposalProvider extends AbstractDataDSLProposalProvider {

	@Inject TerminalsProposalProvider provider
	@Inject BasicDSLProposalProviderHelper providerHelper

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

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

	override completeDataInterchangeFileXML_FileURL(EObject model, Assignment assignment, ContentAssistContext context,
		ICompletionProposalAcceptor acceptor) {
		filePickerProposal(model, assignment, context, acceptor, "*.xml")
	}

	override completeDataInterchangeFileCSV_FileURL(EObject model, Assignment assignment, ContentAssistContext context,
		ICompletionProposalAcceptor acceptor) {
		filePickerProposal(model, assignment, context, acceptor, "*.csv")
	}

	override completeDataInterchangeFileEDI_FileURL(EObject model, Assignment assignment, ContentAssistContext context,
		ICompletionProposalAcceptor acceptor) {
		filePickerProposal(model, assignment, context, acceptor, "*.edi")
	}

	override completeDataInterchangeFormat_Locale(EObject model, Assignment assignment, ContentAssistContext context,
		ICompletionProposalAcceptor acceptor) {
		chooseLocaleProposal(model, assignment, context, acceptor)
	}

	override completeDataInterchangeFileEDI_MappingModel(EObject model, Assignment assignment,
		ContentAssistContext context, ICompletionProposalAcceptor acceptor) {
		mappingPickerProposal(model, assignment, context, acceptor, "*.xml")
	}

	override completeDataInterchangeFileXML_Encoding(EObject model, Assignment assignment, ContentAssistContext context,
		ICompletionProposalAcceptor acceptor) {
		Charset.availableCharsets.keySet.forEach[acceptor.accept(createCompletionProposal("\"" + it + "\"", it, null, context))];
	}

	override completeDataInterchangeFileCSV_Encoding(EObject model, Assignment assignment, ContentAssistContext context,
		ICompletionProposalAcceptor acceptor) {
		Charset.availableCharsets.keySet.forEach[acceptor.accept(createCompletionProposal("\"" + it + "\"", it, null, context))];
	}

	override completeDataInterchangeFileEDI_Encoding(EObject model, Assignment assignment, ContentAssistContext context,
		ICompletionProposalAcceptor acceptor) {
		Charset.availableCharsets.keySet.forEach[acceptor.accept(createCompletionProposal("\"" + it + "\"", it, null, context))];
	}

	def filePickerProposal(EObject model, Assignment assignment, ContentAssistContext context,
		ICompletionProposalAcceptor acceptor, String fileExtensions) {
		var fileName = createCompletionProposal("Select input file...", context) as ConfigurableCompletionProposal
		if (fileName != null) {
			var applier = new FileNameTextApplier()
			applier.setExtensions(fileExtensions.split(","))
			applier.setContext(context)
			fileName.setTextApplier = applier
		}
		acceptor.accept(fileName)
	}

	def mappingPickerProposal(EObject model, Assignment assignment, ContentAssistContext context,
		ICompletionProposalAcceptor acceptor, String fileExtensions) {
		var fileName = createCompletionProposal("Select mapping file...", context) as ConfigurableCompletionProposal
		if (fileName != null) {
			var applier = new FileNameTextApplier()
			applier.setExtensions(fileExtensions.split(","))
			applier.setContext(context)
			fileName.setTextApplier = applier
		}
		acceptor.accept(fileName)
	}

	def chooseLocaleProposal(EObject model, Assignment assignment, ContentAssistContext context,
		ICompletionProposalAcceptor acceptor) {
		var locale = createCompletionProposal("Select locale...", context) as ConfigurableCompletionProposal
		if (locale != null) {
			var applier = new LocaleTextApplier()
			applier.setContext(context)
			locale.setTextApplier = applier
		}
		acceptor.accept(locale)
	}

	override protected isValidProposal(String proposal, String prefix, ContentAssistContext context) {
		var result = super.isValidProposal(proposal, prefix, context)
		if (context.getCurrentModel() instanceof DataInterchangeBean) {
			return isBeanValidProposal(context.currentModel as DataInterchangeBean, proposal, result);
		}
		return result
	}

	def isBeanValidProposal(DataInterchangeBean bean, String proposal, boolean result) {
		return result
	}

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

	// ------------------------ 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)
	}

}