diff options
author | Stewart Stewart <stewinsalot@gmail.com> | 2016-09-23 11:26:43 -0400 |
---|---|---|
committer | Stewart Stewart <stewinsalot@gmail.com> | 2016-09-23 11:34:17 -0400 |
commit | ff50d1e7a8f57dcf36858ae36f864fb02a6187f1 (patch) | |
tree | 85d198330faba3013dbe53e6027e32b97bf2be14 /src | |
parent | ca9042e1f402aa71e8836149134f30df6db140d8 (diff) | |
download | slick-codegen-plugin-ff50d1e7a8f57dcf36858ae36f864fb02a6187f1.tar.gz slick-codegen-plugin-ff50d1e7a8f57dcf36858ae36f864fb02a6187f1.tar.bz2 slick-codegen-plugin-ff50d1e7a8f57dcf36858ae36f864fb02a6187f1.zip |
reuse built in file output helpers
Diffstat (limited to 'src')
-rw-r--r-- | src/main/scala/NamespacedCodegen.scala | 21 |
1 files changed, 5 insertions, 16 deletions
diff --git a/src/main/scala/NamespacedCodegen.scala b/src/main/scala/NamespacedCodegen.scala index f079b3a..908d832 100644 --- a/src/main/scala/NamespacedCodegen.scala +++ b/src/main/scala/NamespacedCodegen.scala @@ -21,8 +21,6 @@ object Generator { val parsedSchemas: Map[String, List[String]] = SchemaParser.parse(schemaNames) val dbModel: Model = Await.result(dc.db.run(SchemaParser.createModel(dc.driver, parsedSchemas)), Duration.Inf) - FileHelpers.createOutputPath(outputPath) - val generator = new Generator(uri, pkg, dbModel, outputPath, manualForeignKeys) val generatedCode = generator.code parsedSchemas.keys.map(schemaName => FileHelpers.schemaOutputPath(outputPath, schemaName)) @@ -79,9 +77,11 @@ class Generator(uri: URI, pkg: String, dbModel: Model, outputPath: String, manua |} """.stripMargin - FileHelpers.write( - FileHelpers.schemaOutputPath(outputPath, schemaName), - allImports + generatedSchema + writeStringToFile( + outputPath, + allImports + generatedSchema, + pkg, + schemaName ) generatedSchema @@ -220,17 +220,6 @@ object TypeGenerator extends StringGeneratorHelpers { object FileHelpers { - def schemaOutputPath(path: String, schemaName: String): String = Paths.get(path, s"${schemaName}.scala").toAbsolutePath().toString() - - def createOutputPath(path: String): Boolean = (new File(path)).mkdirs() - - def write(filename: String, content: String) = { - val file = new File(filename) - val bw = new BufferedWriter(new FileWriter(file)) - bw.write(content) - bw.close() - } - } |