Add package, turn off scalastyle/format

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"