From fda5f944690bed8da53fb0a3ad6b3ad0a97077a9 Mon Sep 17 00:00:00 2001
From: Tobe O <thosakwe@gmail.com>
Date: Sat, 14 Jul 2018 18:08:29 -0400
Subject: [PATCH] Update file service generator

---
 .../service_generators/file_service.dart      | 27 +++++++++++++------
 1 file changed, 19 insertions(+), 8 deletions(-)

diff --git a/lib/src/commands/service_generators/file_service.dart b/lib/src/commands/service_generators/file_service.dart
index d536cbc8..7ff54579 100644
--- a/lib/src/commands/service_generators/file_service.dart
+++ b/lib/src/commands/service_generators/file_service.dart
@@ -14,18 +14,29 @@ class FileServiceGenerator extends ServiceGenerator {
   bool get goesFirst => true;
 
   @override
-  void applyToLibrary(LibraryBuilder library, String name, String lower) {
-    library.addMember(new ImportBuilder('dart:io'));
-    library.addMember(new ImportBuilder(
-        'package:angel_file_service/angel_file_service.dart'));
+  void applyToConfigureServer(MethodBuilder configureServer, BlockBuilder block,
+      String name, String lower) {
+    configureServer.requiredParameters.add(new Parameter((b) => b
+      ..name = 'dbDirectory'
+      ..type = refer('Directory')));
   }
 
   @override
-  ExpressionBuilder createInstance(
+  void applyToLibrary(LibraryBuilder library, String name, String lower) {
+    library.directives.addAll([
+      new Directive.import(
+          'package:angel_file_service/angel_file_service.dart'),
+      new Directive.import('package:file/file.dart'),
+    ]);
+  }
+
+  @override
+  Expression createInstance(
       MethodBuilder methodBuilder, String name, String lower) {
-    return new TypeBuilder('JsonFileService').newInstance([
-      new TypeBuilder('File')
-          .newInstance([literal(pluralize(lower) + '_db.json')])
+    return refer('JsonFileService').newInstance([
+      refer('dbDirectory')
+          .property('childFile')
+          .call([literal(pluralize(lower) + '_db.json')])
     ]);
   }
 }