From 79479db14b8b86c349d3bce0bbc8bba830081841 Mon Sep 17 00:00:00 2001 From: Zach Smith Date: Thu, 1 Feb 2018 09:55:46 -0800 Subject: Bump sbt-settings and run scalafmt --- src/main/scala/Main.scala | 52 +++++++++++++++++++++++++---------------------- 1 file changed, 28 insertions(+), 24 deletions(-) (limited to 'src/main/scala/Main.scala') diff --git a/src/main/scala/Main.scala b/src/main/scala/Main.scala index 39100eb..6fe0417 100644 --- a/src/main/scala/Main.scala +++ b/src/main/scala/Main.scala @@ -16,28 +16,30 @@ trait RowFileGenerator { self: SourceCodeGenerator => object Generator { - private def outputSchemaCode(schemaName: String, - folder: String, - pkg: String, - tableGen: TableFileGenerator, - rowGen: RowFileGenerator): Unit = { + private def outputSchemaCode( + schemaName: String, + folder: String, + pkg: String, + tableGen: TableFileGenerator, + rowGen: RowFileGenerator): Unit = { val camelSchemaName = schemaName.split('_').map(_.capitalize).mkString("") tableGen.writeTablesToFile(folder: String, pkg: String, fileName = s"${camelSchemaName}Tables.scala") rowGen.writeRowsToFile(folder: String, pkg: String, fileName = s"${camelSchemaName}Rows.scala") } - def run(uri: URI, - pkg: String, - schemaNames: Option[List[String]], - outputPath: String, - manualForeignKeys: Map[(String, String), (String, String)], - parentType: Option[String], - idType: Option[String], - header: String, - tablesFileImports: List[String], - rowsFileImports: List[String], - typeReplacements: Map[String, String]) = { + def run( + uri: URI, + pkg: String, + schemaNames: Option[List[String]], + outputPath: String, + manualForeignKeys: Map[(String, String), (String, String)], + parentType: Option[String], + idType: Option[String], + header: String, + tablesFileImports: List[String], + rowsFileImports: List[String], + typeReplacements: Map[String, String]) = { val dc: DatabaseConfig[JdbcProfile] = DatabaseConfig.forURI[JdbcProfile](uri) val parsedSchemasOpt: Option[Map[String, List[String]]] = @@ -51,9 +53,10 @@ object Generator { parsedSchemasOpt.getOrElse(Map.empty).foreach { case (schemaName, tables) => - val schemaOnlyModel = Await.result(dc.db.run(ModelTransformation - .createModel(dc.profile, Some(Map(schemaName -> tables)))), - Duration.Inf) + val schemaOnlyModel = Await.result( + dc.db.run(ModelTransformation + .createModel(dc.profile, Some(Map(schemaName -> tables)))), + Duration.Inf) val rowGenerator = new RowSourceCodeGenerator( model = schemaOnlyModel, @@ -80,11 +83,12 @@ object Generator { typeReplacements ) - outputSchemaCode(schemaName = schemaName, - folder = outputPath, - pkg = pkg, - tableGen = tableGenerator, - rowGen = rowGenerator) + outputSchemaCode( + schemaName = schemaName, + folder = outputPath, + pkg = pkg, + tableGen = tableGenerator, + rowGen = rowGenerator) } } finally { dc.db.close() -- cgit v1.2.3