package org.xtext.example.mydsl.jvmmodel

import com.google.inject.Inject
import org.eclipse.xtext.xbase.jvmmodel.AbstractModelInferrer
import org.eclipse.xtext.xbase.jvmmodel.IJvmDeclaredTypeAcceptor
import org.eclipse.xtext.xbase.jvmmodel.JvmTypesBuilder
import org.xtext.example.mydsl.myDsl.Function
import org.xtext.example.mydsl.myDsl.Model

class MyDslJvmModelInferrer extends AbstractModelInferrer {

	@Inject extension JvmTypesBuilder

	def dispatch void infer(Model element, IJvmDeclaredTypeAcceptor acceptor, boolean isPreIndexingPhase) {
		acceptor.accept(
			element.toClass("DontCare")
		).initializeLater [
			for (f : element.functs) {
				val operation = f.toMethod(f.name, f.type ?: inferredType) [
					documentation = f.documentation
					for (p : f.params) {
						parameters += p.toParameter(p.name, p.parameterType)
					}
					body = f.body
				]
				members += operation
			}
		]
	}
}

package org.xtext.example.mydsl

import javax.inject.Inject
import org.eclipse.emf.ecore.EObject
import org.eclipse.emf.ecore.resource.Resource
import org.eclipse.xtext.common.types.JvmOperation
import org.eclipse.xtext.generator.IFileSystemAccess
import org.eclipse.xtext.generator.IGenerator
import org.eclipse.xtext.generator.trace.ITraceURIConverter
import org.eclipse.xtext.resource.ILocationInFileProvider
import org.eclipse.xtext.xbase.compiler.GeneratorConfig
import org.eclipse.xtext.xbase.compiler.ImportManager
import org.eclipse.xtext.xbase.compiler.JvmModelGenerator
import org.eclipse.xtext.xbase.compiler.output.ITreeAppendable
import org.eclipse.xtext.xbase.compiler.output.TreeAppendable
import org.eclipse.xtext.xbase.jvmmodel.IJvmModelAssociations
import org.xtext.example.mydsl.myDsl.Function
import org.xtext.example.mydsl.myDsl.Model

class DummyGen implements IGenerator {

	@Inject JvmModelGenerator jmg

	@Inject extension IJvmModelAssociations jma

	@Inject
	ILocationInFileProvider locationProvider

	@Inject
	ITraceURIConverter converter

	override doGenerate(Resource input, IFileSystemAccess fsa) {
		for (o : input.contents) {
			o.compile(fsa)
		}
	}

	def dispatch void compile(EObject o, IFileSystemAccess fsa) {
	}

	def dispatch void compile(Model o, IFileSystemAccess fsa) {
		val im = new ImportManager()
		val app = new TreeAppendable(im, converter, locationProvider, jma, o, "", "\n")
		for (f : o.functs) {
			f.compileFunction(app)
		}
		fsa.generateFile("test/Test.java",
			'''
				package test;
				«im.imports.map["import " + it + ";"].join(("\n"))»
				public class Test {
					
					«app.toString»
					
				}
			''')
	}

	def compileFunction(Function f, ITreeAppendable app) {
		val cfg = new GeneratorConfig
		val operation = f.jvmElements.get(0) as JvmOperation
		jmg.generateMember(operation, app, cfg)
	}

}