diff options
Diffstat (limited to 'src')
-rw-r--r-- | src/main/scala/NamespacedCodegen.scala | 17 |
1 files changed, 14 insertions, 3 deletions
diff --git a/src/main/scala/NamespacedCodegen.scala b/src/main/scala/NamespacedCodegen.scala index 2febded..651094b 100644 --- a/src/main/scala/NamespacedCodegen.scala +++ b/src/main/scala/NamespacedCodegen.scala @@ -28,11 +28,21 @@ object Generator { } +class PackageNameGenerator(pkg: String, dbModel: Model) extends SourceCodeGenerator(dbModel) { + override def code: String = + s""" + |// format: OFF + |// scalastyle:off + |package ${pkg} + | + |""".stripMargin +} + + class ImportGenerator(dbModel: Model) extends SourceCodeGenerator(dbModel) { val baseImports: String = s""" - | - |import com.drivergrp.core._ + |import com.drivergrp.core._ |import com.drivergrp.core.database._ | |""".stripMargin @@ -60,6 +70,7 @@ class ImportGenerator(dbModel: Model) extends SourceCodeGenerator(dbModel) { class Generator(uri: URI, pkg: String, dbModel: Model, outputPath: String, manualForeignKeys: Map[(String, String), (String, String)]) extends SourceCodeGenerator(dbModel) with OutputHelpers { + val packageName = new PackageNameGenerator(pkg, dbModel).code val allImports: String = new ImportGenerator(dbModel).code override def code: String = { @@ -83,7 +94,7 @@ class Generator(uri: URI, pkg: String, dbModel: Model, outputPath: String, manua """.stripMargin writeStringToFile( - allImports + generatedSchema, + packageName + allImports + generatedSchema, outputPath, pkg, s"${schemaName}.scala" |