diff options
author | Zach Smith <zach@driver.xyz> | 2018-02-01 09:55:46 -0800 |
---|---|---|
committer | Zach Smith <zach@driver.xyz> | 2018-02-01 11:26:54 -0800 |
commit | 79479db14b8b86c349d3bce0bbc8bba830081841 (patch) | |
tree | f187de8fd268c093d30be7544719ce81a473a9dd /src/main/scala/Generators.scala | |
parent | 0ef6befb187e2302209793fa9a9c207cea0118d3 (diff) | |
download | slick-codegen-plugin-79479db14b8b86c349d3bce0bbc8bba830081841.tar.gz slick-codegen-plugin-79479db14b8b86c349d3bce0bbc8bba830081841.tar.bz2 slick-codegen-plugin-79479db14b8b86c349d3bce0bbc8bba830081841.zip |
Bump sbt-settings and run scalafmt
Diffstat (limited to 'src/main/scala/Generators.scala')
-rw-r--r-- | src/main/scala/Generators.scala | 52 |
1 files changed, 27 insertions, 25 deletions
diff --git a/src/main/scala/Generators.scala b/src/main/scala/Generators.scala index e02cbd6..8f81620 100644 --- a/src/main/scala/Generators.scala +++ b/src/main/scala/Generators.scala @@ -1,14 +1,14 @@ import slick.{model => m} class RowSourceCodeGenerator( - model: m.Model, - override val headerComment: String, - override val imports: String, - override val schemaName: String, - fullDatabaseModel: m.Model, - idType: Option[String], - manualForeignKeys: Map[(String, String), (String, String)], - typeReplacements: Map[String, String] + model: m.Model, + override val headerComment: String, + override val imports: String, + override val schemaName: String, + fullDatabaseModel: m.Model, + idType: Option[String], + manualForeignKeys: Map[(String, String), (String, String)], + typeReplacements: Map[String, String] ) extends TypedIdSourceCodeGenerator( singleSchemaModel = model, databaseModel = fullDatabaseModel, @@ -35,20 +35,22 @@ class RowSourceCodeGenerator( override def code = tables.map(_.code.mkString("\n")).mkString("\n\n") } -class TableSourceCodeGenerator(schemaOnlyModel: m.Model, - override val headerComment: String, - override val imports: String, - override val schemaName: String, - fullDatabaseModel: m.Model, - pkg: String, - manualForeignKeys: Map[(String, String), (String, String)], - override val parentType: Option[String], - idType: Option[String], - typeReplacements: Map[String, String]) - extends TypedIdSourceCodeGenerator(singleSchemaModel = schemaOnlyModel, - databaseModel = fullDatabaseModel, - idType, - manualForeignKeys) with TableOutputHelpers { +class TableSourceCodeGenerator( + schemaOnlyModel: m.Model, + override val headerComment: String, + override val imports: String, + override val schemaName: String, + fullDatabaseModel: m.Model, + pkg: String, + manualForeignKeys: Map[(String, String), (String, String)], + override val parentType: Option[String], + idType: Option[String], + typeReplacements: Map[String, String]) + extends TypedIdSourceCodeGenerator( + singleSchemaModel = schemaOnlyModel, + databaseModel = fullDatabaseModel, + idType, + manualForeignKeys) with TableOutputHelpers { val defaultIdImplementation = """|final case class Id[T](v: Int) @@ -68,8 +70,8 @@ class TableSourceCodeGenerator(schemaOnlyModel: m.Model, if (ddlEnabled && schemaName != "public") { // TODO: "public" is postgres-specific // TODO: make generated DDL overrideable upstream try { - val (before, ddl +: after) = tableCode.lines.toVector.span(! _.contains("val schema")) - val Array(identifier, tablesDDL) = ddl.split('=').map(_.trim) + val (before, ddl +: after) = tableCode.lines.toVector.span(!_.contains("val schema")) + val Array(_, tablesDDL) = ddl.split('=').map(_.trim) val schemaDDL: Vector[String] = s"""|lazy val schema: profile.SchemaDescription = { | val schemaDDL = profile.DDL( @@ -77,7 +79,7 @@ class TableSourceCodeGenerator(schemaOnlyModel: m.Model, | drop2 = "drop schema " + profile.quoteIdentifier("$schemaName")) | schemaDDL ++ ${tablesDDL.trim} |}""".stripMargin.lines.toVector - (before ++ schemaDDL ++ after).mkString("\n") + (before ++ schemaDDL ++ after).mkString("\n") } catch { case _: MatchError => throw new Exception("failed to modify schemaddl line") } |