diff --git a/convention/src/main/kotlin/io/gitlab/jfronny/scripts/codegen/ClassGenerator.kt b/convention/src/main/kotlin/io/gitlab/jfronny/scripts/codegen/ClassGenerator.kt index 7a7a50f..5b2db47 100644 --- a/convention/src/main/kotlin/io/gitlab/jfronny/scripts/codegen/ClassGenerator.kt +++ b/convention/src/main/kotlin/io/gitlab/jfronny/scripts/codegen/ClassGenerator.kt @@ -53,10 +53,12 @@ class ClassGenerator(`package`: String, val name: String, val indent: String = " gen.appendLine("$indent$modifiers $type $name = $valueRaw;") } - fun field(name: String, value: Int, modifiers: String = defaultModifiers) = fieldRaw(name, "int", value.toString(), modifiers) fun field(name: String, value: Boolean, modifiers: String = defaultModifiers) = fieldRaw(name, "boolean", value.toString(), modifiers) - fun field(name: String, value: Double, modifiers: String = defaultModifiers) = fieldRaw(name, "double", value.toString(), modifiers) + fun field(name: String, value: Short, modifiers: String = defaultModifiers) = fieldRaw(name, "short", value.toString(), modifiers) + fun field(name: String, value: Int, modifiers: String = defaultModifiers) = fieldRaw(name, "int", value.toString(), modifiers) + fun field(name: String, value: Long, modifiers: String = defaultModifiers) = fieldRaw(name, "long", value.toString(), modifiers) fun field(name: String, value: Float, modifiers: String = defaultModifiers) = fieldRaw(name, "float", value.toString() + "f", modifiers) + fun field(name: String, value: Double, modifiers: String = defaultModifiers) = fieldRaw(name, "double", value.toString(), modifiers) fun field(name: String, value: String, modifiers: String = defaultModifiers) = fieldRaw(name, "String", enquote(value), modifiers) fun procedure(name: String, argument: Map, modifiers: String = defaultModifiers, body: String, throws: String? = null): Unit = ensureBody {