refactor: remove small single use snippets from templates

simolus3 · May 3, 2024 · da788c0 · da788c0
1 parent 8ca18b9
commit da788c0
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 18 deletions.
diff --git a/drift_dev/lib/src/writer/manager/database_manager_writer.dart b/drift_dev/lib/src/writer/manager/database_manager_writer.dart
@@ -46,8 +46,11 @@ class DatabaseManagerWriter {
     }
   }
 
+  /// The code for the database manager getter which will be added to the main database class
+  ///
+  /// E.g. `AppDatabase get managers => AppDatabaseManager(this);`
   String get databaseManagerGetter =>
-      _templates.databaseManagerGetter(_dbClassName);
+      '${_templates.databaseManagerName(_dbClassName)} get managers => ${_templates.databaseManagerName(_dbClassName)}(this);';
 
   /// Write the database manager class
   void writeDatabaseManager() {
@@ -66,7 +69,7 @@ class DatabaseManagerWriter {
 
       /// Write the getter for the table manager
       leaf.writeln(
-          _templates.rootTableManagerGetter(table, rootTableManagerClass));
+          '$rootTableManagerClass get ${table.dbGetterName} => $rootTableManagerClass(_db, _db.${table.dbGetterName});');
     }
     leaf.writeln('}');
   }

diff --git a/drift_dev/lib/src/writer/manager/manager_templates.dart b/drift_dev/lib/src/writer/manager/manager_templates.dart
@@ -33,21 +33,6 @@ class _ManagerCodeTemplates {
     };
   }
 
-  /// The code for the database manager getter which will be added to the main database class
-  ///
-  /// E.g. `AppDatabase get managers => AppDatabaseManager(this);`
-  String databaseManagerGetter(String dbClassName) {
-    return '${databaseManagerName(dbClassName)} get managers => ${databaseManagerName(dbClassName)}(this);';
-  }
-
-  /// The code for a root table manager getter which will be added to the database manager class
-  ///
-  /// E.g. `UserTableManager get todos => UserTableManager(_db,_db.todos);`
-  String rootTableManagerGetter(
-      DriftTable table, String rootTableManagerClass) {
-    return '$rootTableManagerClass get ${table.dbGetterName} => $rootTableManagerClass(_db, _db.${table.dbGetterName});';
-  }
-
   /// Returns the name of the root manager class for a table
   ///
   /// One of these classes is generated for each table in the database
@@ -203,7 +188,6 @@ class _ManagerCodeTemplates {
     ${filterComposerNameWithPrefix(table, leaf)},
     ${orderingComposerNameWithPrefix(table, leaf)},
     ${processedTableManagerName(table)},
-
     ${insertCompanionBuilderTypeDef(table)},
     ${updateCompanionBuilderTypeDefName(table)}>""";
   }