com.prezi.spaghetti.kotlin.AbstractKotlinMethodGeneratorVisitor.groovy Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of spaghetti-kotlin-support Show documentation

Spaghetti

The newest version!

package com.prezi.spaghetti.kotlin

import com.prezi.spaghetti.ast.MethodNode
import com.prezi.spaghetti.ast.MethodParameterNode

abstract class AbstractKotlinMethodGeneratorVisitor extends AbstractKotlinGeneratorVisitor {

	@Override
	String visitMethodNode(MethodNode node) {
		def returnType = node.returnType.accept(this)
		returnType = wrapNullableTypeReference(returnType, node)
		def typeParams = node.typeParameters ? "<" + node.typeParameters*.name.join(", ") + "> " : ""
		def params = node.parameters*.accept(this).join(", ")

		return \
"""	fun ${typeParams}${node.name}(${params}):${returnType}
"""
	}

	@Override
	String visitMethodParameterNode(MethodParameterNode node) {
		def type = node.type.accept(this)
		type = wrapNullableTypeReference(type, node)
		def result = node.name + ':' + type
		if (node.isOptional()) {
			result = result + "? = null"
		}
		return result
	}
}