import java.util.HashSet
import org.eclipse.emf.common.util.URI
import org.eclipse.emf.ecore.resource.Resource
import org.eclipse.xtext.generator.trace.AbsoluteURI
import org.eclipse.xtext.generator.trace.DefaultTraceURIConverter
import org.eclipse.xtext.generator.trace.ITraceURIConverter
import org.eclipse.xtext.generator.trace.SourceRelativeURI
import org.eclipse.xtext.workspace.IProjectConfig

class DomainTraceUriConverter extends DefaultTraceURIConverter implements ITraceURIConverter {
	
	override getURIForTrace(IProjectConfig projectConfig, AbsoluteURI absoluteURI) {
		super.getURIForTrace(projectConfig, absoluteURI).convertToSourceUri
	}
	
	override getURIForTrace(Resource resource) {
		super.getURIForTrace(resource).convertToSourceUri
	}
	
	def SourceRelativeURI convertToSourceUri (SourceRelativeURI uri)
	{
		val segementList = uri.URI.segmentsList
		val HashSet<Integer> mainIndexes = newHashSet()
		val HashSet<Integer> srcIndexes = newHashSet()
		val HashSet<Integer> javaIndexes = newHashSet()
		for (index: 0 ..< segementList.size)
		{
			if ("main".equals(segementList.get(index)))
			{
				mainIndexes.add (index)
			} else if ("src".equals(segementList.get(index)))
			{
				srcIndexes.add (index)
			} else if ("java".equals(segementList.get(index)))
			{
				javaIndexes.add (index)
			}
		}
		val convertedUri = javaIndexes.filter[javaIndex|
			mainIndexes.contains(javaIndex-1)
			srcIndexes.contains(javaIndex-2)
		].map[javaIndex|
			val srcUri = segementList.subList(javaIndex+1, segementList.size)
			val String[] strArray = newArrayOfSize(srcUri.size)
			val segementsArray = srcUri.toArray(strArray)
			new SourceRelativeURI(URI.createHierarchicalURI(segementsArray, null, null))
		].head
		
		if (convertedUri !== null)
		{
			println('''convertUri «uri.toString» -> «convertedUri.toString»''')
			convertedUri
		} else
		{
			println('''not converted Uri «uri.toString»''')
			uri
		}
	}

	def Class<? extends ITraceURIConverter> bindITraceURIConverter(){
//		println("bind ITraceURIConverter to "+DomainTraceUriConverter.name)
		DomainTraceUriConverter
	}