diff options
author | Stewart Stewart <stewinsalot@gmail.com> | 2017-09-22 18:22:58 -0700 |
---|---|---|
committer | Stewart Stewart <stewinsalot@gmail.com> | 2017-09-22 18:34:11 -0700 |
commit | fbb85ad4ba79c74544d6c804bbbf0a7ef350b113 (patch) | |
tree | d196723bc77eec582bcb7cb203041c12cf1c70ac /src/main/scala/OutputHelpers.scala | |
parent | 94e2a082467c5fbf70490c33ebe25ca6d2318b8e (diff) | |
download | slick-codegen-plugin-fbb85ad4ba79c74544d6c804bbbf0a7ef350b113.tar.gz slick-codegen-plugin-fbb85ad4ba79c74544d6c804bbbf0a7ef350b113.tar.bz2 slick-codegen-plugin-fbb85ad4ba79c74544d6c804bbbf0a7ef350b113.zip |
use Driver linting/formatting/style settings
Diffstat (limited to 'src/main/scala/OutputHelpers.scala')
-rw-r--r-- | src/main/scala/OutputHelpers.scala | 35 |
1 files changed, 12 insertions, 23 deletions
diff --git a/src/main/scala/OutputHelpers.scala b/src/main/scala/OutputHelpers.scala index 97fb4e9..17c2019 100644 --- a/src/main/scala/OutputHelpers.scala +++ b/src/main/scala/OutputHelpers.scala @@ -1,7 +1,6 @@ import slick.codegen.{SourceCodeGenerator, OutputHelpers} -trait TableOutputHelpers extends TableFileGenerator with OutputHelpers { - self: SourceCodeGenerator => +trait TableOutputHelpers extends TableFileGenerator with OutputHelpers { self: SourceCodeGenerator => def headerComment: String def schemaName: String @@ -31,30 +30,21 @@ trait TableOutputHelpers extends TableFileGenerator with OutputHelpers { |} |""".stripMargin.trim() - def writeTablesToFile(profile: String, - folder: String, - pkg: String, - fileName: String): Unit = { - writeStringToFile( - content = - packageTableCode(headerComment, pkg, schemaName, imports, profile), - folder = folder, - pkg = s"$pkg.$schemaName", - fileName = fileName) + def writeTablesToFile(profile: String, folder: String, pkg: String, fileName: String): Unit = { + writeStringToFile(content = packageTableCode(headerComment, pkg, schemaName, imports, profile), + folder = folder, + pkg = s"$pkg.$schemaName", + fileName = fileName) } } -trait RowOutputHelpers extends RowFileGenerator with OutputHelpers { - self: SourceCodeGenerator => +trait RowOutputHelpers extends RowFileGenerator with OutputHelpers { self: SourceCodeGenerator => def headerComment: String def schemaName: String def imports: String - def packageRowCode(headerComment: String, - schemaName: String, - pkg: String, - imports: String): String = + def packageRowCode(headerComment: String, schemaName: String, pkg: String, imports: String): String = s"""|${headerComment.trim().lines.map("// " + _).mkString("\n")} |/** Definitions for table rows types of database schema $schemaName */ |package $pkg @@ -67,10 +57,9 @@ trait RowOutputHelpers extends RowFileGenerator with OutputHelpers { def writeRowsToFile(folder: String, pkg: String, fileName: String): Unit = { - writeStringToFile( - content = packageRowCode(headerComment, schemaName, pkg, imports), - folder = folder, - pkg = s"$pkg.$schemaName", - fileName = fileName) + writeStringToFile(content = packageRowCode(headerComment, schemaName, pkg, imports), + folder = folder, + pkg = s"$pkg.$schemaName", + fileName = fileName) } } |