11package io.github.cdsap.generator
22
33import io.github.cdsap.generator.files.*
4- import io.github.cdsap.generator.model.LanguageAttributes
5- import io.github.cdsap.generator.model.ProjectGraph
6- import io.github.cdsap.generator.model.Versions
4+ import io.github.cdsap.generator.model.*
75import io.github.cdsap.generator.writer.ConventionPluginWriter
86import io.github.cdsap.generator.writer.ModulesWriter
97import java.io.File
@@ -12,11 +10,12 @@ class ProjectWriter(
1210 private val nodes : List <ProjectGraph >,
1311 private val languages : List <LanguageAttributes >,
1412 private val numberOfClassPerModule : Int ,
15- private val versions : Versions
13+ private val versions : Versions ,
14+ private val typeOfProjectRequested : TypeProjectRequested
1615) {
1716 fun write () {
1817 println (" Creating Convention Plugin files" )
19- ConventionPluginWriter (languages, versions).write()
18+ ConventionPluginWriter (languages, versions, typeOfProjectRequested ).write()
2019 println (" Creating Modules files" )
2120 ModulesWriter (nodes, languages, numberOfClassPerModule).write()
2221
@@ -36,8 +35,11 @@ class ProjectWriter(
3635 }
3736 }
3837
39- private fun createProjectBuildGradle (languages : List <LanguageAttributes >) {
40- val plugins = BuildGradle ().get(versions)
38+ private fun createProjectBuildGradle (
39+ languages : List <LanguageAttributes >
40+ ) {
41+ val plugins = if (typeOfProjectRequested == TypeProjectRequested .JVM ) BuildGradle ().getJvm(versions)
42+ else BuildGradle ().get(versions)
4143 languages.forEach {
4244 File (" ${it.projectName} /build.${it.extension} " ).createNewFile()
4345 File (" ${it.projectName} /build.${it.extension} " ).writeText(plugins)
0 commit comments