ActionStage.kt

package ps.eden.server.game.content.dialogue

/**
 * @author Jire
 */
class ActionStage(id: Int) : Stage(id) {
	
	var defaultHandler: (() -> Any?)? = null
		private set
	val onButtonHandlers = HashMap<Int, () -> Any?>()
	
	operator fun <R> Int.invoke(onButtonID: () -> R) {
		onButtonHandlers[this] = onButtonID
	}
	
	fun <R> or(onDefaultID: () -> R) {
		defaultHandler = onDefaultID
	}
	
	fun or(onDefaultID: Int) = or { onDefaultID }
	
}