コード整理

tfandkusu · Aug 12, 2024 · b44627a · b44627a
1 parent 28ebb73
commit b44627a
Showing 1 changed file with 23 additions and 21 deletions.
diff --git a/src/main/kotlin/com/tfandkusu/ga913yaml/KotlinGenerator.kt b/src/main/kotlin/com/tfandkusu/ga913yaml/KotlinGenerator.kt
@@ -21,31 +21,33 @@ object KotlinGenerator {
  .addFileComment(
  "https://github.com/tfandkusu/ga913-yaml/ による自動生成コードです。編集しないでください。",
  ).addType(
- TypeSpec
- .objectBuilder(ROOT_CLASS)
- .addType(
- TypeSpec
- .classBuilder(SCREEN_CLASS)
- .addModifiers(KModifier.SEALED)
- .apply {
- addProperty(
- PropertySpec
- .builder(
- EVENT_NAME_PROPERTY,
- String::class,
- ).addModifiers(KModifier.ABSTRACT)
- .build(),
- )
- screens.forEach { screen ->
- addType(generateScreenClass(screen))
- }
- }.build(),
- ).build(),
+ generateAnalyticsEventClass(screens),
  ).build()
-
  fileSpec.writeTo(System.out)
  }
 
+ private fun generateAnalyticsEventClass(screens: List<Screen>): TypeSpec =
+ TypeSpec
+ .objectBuilder(ROOT_CLASS)
+ .addType(
+ TypeSpec
+ .classBuilder(SCREEN_CLASS)
+ .addModifiers(KModifier.SEALED)
+ .apply {
+ addProperty(
+ PropertySpec
+ .builder(
+ EVENT_NAME_PROPERTY,
+ String::class,
+ ).addModifiers(KModifier.ABSTRACT)
+ .build(),
+ )
+ screens.forEach { screen ->
+ addType(generateScreenClass(screen))
+ }
+ }.build(),
+ ).build()
+
  private fun generateScreenClass(screen: Screen): TypeSpec =
  TypeSpec
  .objectBuilder(screen.className)