Generate code for triggers

moor/example/example.g.dart

@@ -878,10 +878,9 @@ abstract class _$Database extends GeneratedDatabase {
   }
 
   @override
-  List<TableInfo> get allTables =>
-      [categories, recipes, ingredients, ingredientInRecipes];
+  Iterable<TableInfo> get allTables => allSchemaEntities.whereType<TableInfo>();
   @override
-  List<DatabaseSchemaEntity> get allEntities =>
+  List<DatabaseSchemaEntity> get allSchemaEntities =>
       [categories, recipes, ingredients, ingredientInRecipes];
 }
 

moor/lib/src/runtime/api/db_base.dart

@@ -29,13 +29,13 @@ abstract class GeneratedDatabase extends DatabaseConnectionUser
   MigrationStrategy get _resolvedMigration => _cachedMigration ??= migration;
 
   /// A list of tables specified in this database.
-  List<TableInfo> get allTables;
+  Iterable<TableInfo> get allTables;
 
   /// A list of all [DatabaseSchemaEntity] that are specified in this database.
   ///
   /// This contains [allTables], but also advanced entities like triggers.
   // return allTables for backwards compatibility
-  List<DatabaseSchemaEntity> get allSchemaEntities => allTables;
+  Iterable<DatabaseSchemaEntity> get allSchemaEntities => allTables;
 
   /// A [Type] can't be sent across isolates. Instances of this class shouldn't
   /// be sent over isolates either, so let's keep a reference to a [Type] that

moor/test/data/tables/custom_tables.g.dart

@@ -7,6 +7,173 @@ part of 'custom_tables.dart';
 // **************************************************************************
 
 // ignore_for_file: unnecessary_brace_in_string_interps, unnecessary_this
+class Config extends DataClass implements Insertable<Config> {
+  final String configKey;
+  final String configValue;
+  Config({@required this.configKey, this.configValue});
+  factory Config.fromData(Map<String, dynamic> data, GeneratedDatabase db,
+      {String prefix}) {
+    final effectivePrefix = prefix ?? '';
+    final stringType = db.typeSystem.forDartType<String>();
+    return Config(
+      configKey: stringType
+          .mapFromDatabaseResponse(data['${effectivePrefix}config_key']),
+      configValue: stringType
+          .mapFromDatabaseResponse(data['${effectivePrefix}config_value']),
+    );
+  }
+  factory Config.fromJson(Map<String, dynamic> json,
+      {ValueSerializer serializer}) {
+    serializer ??= moorRuntimeOptions.defaultSerializer;
+    return Config(
+      configKey: serializer.fromJson<String>(json['config_key']),
+      configValue: serializer.fromJson<String>(json['config_value']),
+    );
+  }
+  factory Config.fromJsonString(String encodedJson,
+          {ValueSerializer serializer}) =>
+      Config.fromJson(DataClass.parseJson(encodedJson) as Map<String, dynamic>,
+          serializer: serializer);
+  @override
+  Map<String, dynamic> toJson({ValueSerializer serializer}) {
+    serializer ??= moorRuntimeOptions.defaultSerializer;
+    return <String, dynamic>{
+      'config_key': serializer.toJson<String>(configKey),
+      'config_value': serializer.toJson<String>(configValue),
+    };
+  }
+
+  @override
+  ConfigCompanion createCompanion(bool nullToAbsent) {
+    return ConfigCompanion(
+      configKey: configKey == null && nullToAbsent
+          ? const Value.absent()
+          : Value(configKey),
+      configValue: configValue == null && nullToAbsent
+          ? const Value.absent()
+          : Value(configValue),
+    );
+  }
+
+  Config copyWith({String configKey, String configValue}) => Config(
+        configKey: configKey ?? this.configKey,
+        configValue: configValue ?? this.configValue,
+      );
+  @override
+  String toString() {
+    return (StringBuffer('Config(')
+          ..write('configKey: $configKey, ')
+          ..write('configValue: $configValue')
+          ..write(')'))
+        .toString();
+  }
+
+  @override
+  int get hashCode => $mrjf($mrjc(configKey.hashCode, configValue.hashCode));
+  @override
+  bool operator ==(dynamic other) =>
+      identical(this, other) ||
+      (other is Config &&
+          other.configKey == this.configKey &&
+          other.configValue == this.configValue);
+}
+
+class ConfigCompanion extends UpdateCompanion<Config> {
+  final Value<String> configKey;
+  final Value<String> configValue;
+  const ConfigCompanion({
+    this.configKey = const Value.absent(),
+    this.configValue = const Value.absent(),
+  });
+  ConfigCompanion.insert({
+    @required String configKey,
+    this.configValue = const Value.absent(),
+  }) : configKey = Value(configKey);
+  ConfigCompanion copyWith(
+      {Value<String> configKey, Value<String> configValue}) {
+    return ConfigCompanion(
+      configKey: configKey ?? this.configKey,
+      configValue: configValue ?? this.configValue,
+    );
+  }
+}
+
+class ConfigTable extends Table with TableInfo<ConfigTable, Config> {
+  final GeneratedDatabase _db;
+  final String _alias;
+  ConfigTable(this._db, [this._alias]);
+  final VerificationMeta _configKeyMeta = const VerificationMeta('configKey');
+  GeneratedTextColumn _configKey;
+  GeneratedTextColumn get configKey => _configKey ??= _constructConfigKey();
+  GeneratedTextColumn _constructConfigKey() {
+    return GeneratedTextColumn('config_key', $tableName, false,
+        $customConstraints: 'not null primary key');
+  }
+
+  final VerificationMeta _configValueMeta =
+      const VerificationMeta('configValue');
+  GeneratedTextColumn _configValue;
+  GeneratedTextColumn get configValue =>
+      _configValue ??= _constructConfigValue();
+  GeneratedTextColumn _constructConfigValue() {
+    return GeneratedTextColumn('config_value', $tableName, true,
+        $customConstraints: '');
+  }
+
+  @override
+  List<GeneratedColumn> get $columns => [configKey, configValue];
+  @override
+  ConfigTable get asDslTable => this;
+  @override
+  String get $tableName => _alias ?? 'config';
+  @override
+  final String actualTableName = 'config';
+  @override
+  VerificationContext validateIntegrity(ConfigCompanion d,
+      {bool isInserting = false}) {
+    final context = VerificationContext();
+    if (d.configKey.present) {
+      context.handle(_configKeyMeta,
+          configKey.isAcceptableValue(d.configKey.value, _configKeyMeta));
+    } else if (isInserting) {
+      context.missing(_configKeyMeta);
+    }
+    if (d.configValue.present) {
+      context.handle(_configValueMeta,
+          configValue.isAcceptableValue(d.configValue.value, _configValueMeta));
+    }
+    return context;
+  }
+
+  @override
+  Set<GeneratedColumn> get $primaryKey => {configKey};
+  @override
+  Config map(Map<String, dynamic> data, {String tablePrefix}) {
+    final effectivePrefix = tablePrefix != null ? '$tablePrefix.' : null;
+    return Config.fromData(data, _db, prefix: effectivePrefix);
+  }
+
+  @override
+  Map<String, Variable> entityToSql(ConfigCompanion d) {
+    final map = <String, Variable>{};
+    if (d.configKey.present) {
+      map['config_key'] = Variable<String, StringType>(d.configKey.value);
+    }
+    if (d.configValue.present) {
+      map['config_value'] = Variable<String, StringType>(d.configValue.value);
+    }
+    return map;
+  }
+
+  @override
+  ConfigTable createAlias(String alias) {
+    return ConfigTable(_db, alias);
+  }
+
+  @override
+  bool get dontWriteConstraints => true;
+}
+
 class NoId extends DataClass implements Insertable<NoId> {
   final Uint8List payload;
   NoId({@required this.payload});
@@ -485,173 +652,6 @@ class WithConstraints extends Table
   bool get dontWriteConstraints => true;
 }
 
-class Config extends DataClass implements Insertable<Config> {
-  final String configKey;
-  final String configValue;
-  Config({@required this.configKey, this.configValue});
-  factory Config.fromData(Map<String, dynamic> data, GeneratedDatabase db,
-      {String prefix}) {
-    final effectivePrefix = prefix ?? '';
-    final stringType = db.typeSystem.forDartType<String>();
-    return Config(
-      configKey: stringType
-          .mapFromDatabaseResponse(data['${effectivePrefix}config_key']),
-      configValue: stringType
-          .mapFromDatabaseResponse(data['${effectivePrefix}config_value']),
-    );
-  }
-  factory Config.fromJson(Map<String, dynamic> json,
-      {ValueSerializer serializer}) {
-    serializer ??= moorRuntimeOptions.defaultSerializer;
-    return Config(
-      configKey: serializer.fromJson<String>(json['config_key']),
-      configValue: serializer.fromJson<String>(json['config_value']),
-    );
-  }
-  factory Config.fromJsonString(String encodedJson,
-          {ValueSerializer serializer}) =>
-      Config.fromJson(DataClass.parseJson(encodedJson) as Map<String, dynamic>,
-          serializer: serializer);
-  @override
-  Map<String, dynamic> toJson({ValueSerializer serializer}) {
-    serializer ??= moorRuntimeOptions.defaultSerializer;
-    return <String, dynamic>{
-      'config_key': serializer.toJson<String>(configKey),
-      'config_value': serializer.toJson<String>(configValue),
-    };
-  }
-
-  @override
-  ConfigCompanion createCompanion(bool nullToAbsent) {
-    return ConfigCompanion(
-      configKey: configKey == null && nullToAbsent
-          ? const Value.absent()
-          : Value(configKey),
-      configValue: configValue == null && nullToAbsent
-          ? const Value.absent()
-          : Value(configValue),
-    );
-  }
-
-  Config copyWith({String configKey, String configValue}) => Config(
-        configKey: configKey ?? this.configKey,
-        configValue: configValue ?? this.configValue,
-      );
-  @override
-  String toString() {
-    return (StringBuffer('Config(')
-          ..write('configKey: $configKey, ')
-          ..write('configValue: $configValue')
-          ..write(')'))
-        .toString();
-  }
-
-  @override
-  int get hashCode => $mrjf($mrjc(configKey.hashCode, configValue.hashCode));
-  @override
-  bool operator ==(dynamic other) =>
-      identical(this, other) ||
-      (other is Config &&
-          other.configKey == this.configKey &&
-          other.configValue == this.configValue);
-}
-
-class ConfigCompanion extends UpdateCompanion<Config> {
-  final Value<String> configKey;
-  final Value<String> configValue;
-  const ConfigCompanion({
-    this.configKey = const Value.absent(),
-    this.configValue = const Value.absent(),
-  });
-  ConfigCompanion.insert({
-    @required String configKey,
-    this.configValue = const Value.absent(),
-  }) : configKey = Value(configKey);
-  ConfigCompanion copyWith(
-      {Value<String> configKey, Value<String> configValue}) {
-    return ConfigCompanion(
-      configKey: configKey ?? this.configKey,
-      configValue: configValue ?? this.configValue,
-    );
-  }
-}
-
-class ConfigTable extends Table with TableInfo<ConfigTable, Config> {
-  final GeneratedDatabase _db;
-  final String _alias;
-  ConfigTable(this._db, [this._alias]);
-  final VerificationMeta _configKeyMeta = const VerificationMeta('configKey');
-  GeneratedTextColumn _configKey;
-  GeneratedTextColumn get configKey => _configKey ??= _constructConfigKey();
-  GeneratedTextColumn _constructConfigKey() {
-    return GeneratedTextColumn('config_key', $tableName, false,
-        $customConstraints: 'not null primary key');
-  }
-
-  final VerificationMeta _configValueMeta =
-      const VerificationMeta('configValue');
-  GeneratedTextColumn _configValue;
-  GeneratedTextColumn get configValue =>
-      _configValue ??= _constructConfigValue();
-  GeneratedTextColumn _constructConfigValue() {
-    return GeneratedTextColumn('config_value', $tableName, true,
-        $customConstraints: '');
-  }
-
-  @override
-  List<GeneratedColumn> get $columns => [configKey, configValue];
-  @override
-  ConfigTable get asDslTable => this;
-  @override
-  String get $tableName => _alias ?? 'config';
-  @override
-  final String actualTableName = 'config';
-  @override
-  VerificationContext validateIntegrity(ConfigCompanion d,
-      {bool isInserting = false}) {
-    final context = VerificationContext();
-    if (d.configKey.present) {
-      context.handle(_configKeyMeta,
-          configKey.isAcceptableValue(d.configKey.value, _configKeyMeta));
-    } else if (isInserting) {
-      context.missing(_configKeyMeta);
-    }
-    if (d.configValue.present) {
-      context.handle(_configValueMeta,
-          configValue.isAcceptableValue(d.configValue.value, _configValueMeta));
-    }
-    return context;
-  }
-
-  @override
-  Set<GeneratedColumn> get $primaryKey => {configKey};
-  @override
-  Config map(Map<String, dynamic> data, {String tablePrefix}) {
-    final effectivePrefix = tablePrefix != null ? '$tablePrefix.' : null;
-    return Config.fromData(data, _db, prefix: effectivePrefix);
-  }
-
-  @override
-  Map<String, Variable> entityToSql(ConfigCompanion d) {
-    final map = <String, Variable>{};
-    if (d.configKey.present) {
-      map['config_key'] = Variable<String, StringType>(d.configKey.value);
-    }
-    if (d.configValue.present) {
-      map['config_value'] = Variable<String, StringType>(d.configValue.value);
-    }
-    return map;
-  }
-
-  @override
-  ConfigTable createAlias(String alias) {
-    return ConfigTable(_db, alias);
-  }
-
-  @override
-  bool get dontWriteConstraints => true;
-}
-
 class MytableData extends DataClass implements Insertable<MytableData> {
   final int someid;
   final String sometext;
@@ -1087,6 +1087,8 @@ class Email extends Table
 abstract class _$CustomTablesDb extends GeneratedDatabase {
   _$CustomTablesDb(QueryExecutor e) : super(SqlTypeSystem.defaultInstance, e);
   _$CustomTablesDb.connect(DatabaseConnection c) : super.connect(c);
+  ConfigTable _config;
+  ConfigTable get config => _config ??= ConfigTable(this);
   NoIds _noIds;
   NoIds get noIds => _noIds ??= NoIds(this);
   WithDefaults _withDefaults;
@@ -1094,8 +1096,6 @@ abstract class _$CustomTablesDb extends GeneratedDatabase {
   WithConstraints _withConstraints;
   WithConstraints get withConstraints =>
       _withConstraints ??= WithConstraints(this);
-  ConfigTable _config;
-  ConfigTable get config => _config ??= ConfigTable(this);
   Mytable _mytable;
   Mytable get mytable => _mytable ??= Mytable(this);
   Email _email;
@@ -1210,8 +1210,19 @@ abstract class _$CustomTablesDb extends GeneratedDatabase {
   }
 
   @override
-  List<TableInfo> get allTables =>
-      [noIds, withDefaults, withConstraints, config, mytable, email];
+  Iterable<TableInfo> get allTables => allSchemaEntities.whereType<TableInfo>();
+  @override
+  List<DatabaseSchemaEntity> get allSchemaEntities => [
+        config,
+        Trigger(
+            'CREATE TRIGGER my_trigger AFTER INSERT ON config BEGIN\n    INSERT INTO with_defaults VALUES (new.config_key, LENGTH(new.config_value));\nEND;',
+            'my_trigger'),
+        noIds,
+        withDefaults,
+        withConstraints,
+        mytable,
+        email
+      ];
 }
 
 class MultipleResult {

moor/test/data/tables/todos.g.dart

@@ -1424,16 +1424,9 @@ abstract class _$TodoDb extends GeneratedDatabase {
   }
 
   @override
-  List<TableInfo> get allTables => [
-        todosTable,
-        categories,
-        users,
-        sharedTodos,
-        tableWithoutPK,
-        pureDefaults
-      ];
+  Iterable<TableInfo> get allTables => allSchemaEntities.whereType<TableInfo>();
   @override
-  List<DatabaseSchemaEntity> get allEntities => [
+  List<DatabaseSchemaEntity> get allSchemaEntities => [
         todosTable,
         categories,
         users,

moor_generator/lib/src/analyzer/runner/steps/analyze_dart.dart

@@ -44,7 +44,7 @@ class AnalyzeDartStep extends AnalyzingStep {
       parser.parse();
 
       accessor
-        ..tables = availableTables
+        ..entities = availableEntities
         ..queries = availableQueries.followedBy(parser.foundQueries).toList();
     }
   }

moor_generator/lib/src/model/database.dart

@@ -1,6 +1,7 @@
 import 'package:analyzer/dart/element/element.dart';
 import 'package:analyzer/dart/element/type.dart';
 import 'package:meta/meta.dart';
+import 'package:moor_generator/moor_generator.dart';
 import 'package:moor_generator/src/analyzer/runner/file_graph.dart';
 import 'package:moor_generator/src/model/sql_query.dart';
 
@@ -18,7 +19,8 @@ abstract class BaseMoorAccessor implements HasDeclaration {
   /// All tables that have been declared on this accessor directly.
   ///
   /// This contains the `tables` field from a `UseMoor` or `UseDao` annotation,
-  /// but not tables that are declared in imported moor files. Use
+  /// but not tables that are declared in imported moor files. Use [tables] for
+  /// that.
   final List<MoorTable> declaredTables;
 
   /// The `includes` field from the `UseMoor` or `UseDao` annotation.
@@ -27,9 +29,13 @@ abstract class BaseMoorAccessor implements HasDeclaration {
   /// All queries declared directly in the `UseMoor` or `UseDao` annotation.
   final List<DeclaredQuery> declaredQueries;
 
+  /// All entities for this database accessor. This contains [declaredTables]
+  /// and all tables, triggers and other entities available through includes.
+  List<MoorSchemaEntity> entities = [];
+
   /// All tables for this database accessor. This contains the [declaredTables]
-  /// and all tables that are reachable through
-  List<MoorTable> tables = [];
+  /// and all tables that are reachable through includes.
+  Iterable<MoorTable> get tables => entities.whereType();
 
   /// All resolved queries.
   ///

moor_generator/lib/src/model/trigger.dart

@@ -8,13 +8,15 @@ class MoorTrigger implements MoorSchemaEntity {
   final String displayName;
 
   @override
-  final Declaration declaration;
+  final TriggerDeclaration declaration;
 
   /// The table on which this trigger operates.
   ///
   /// This field can be null in case the table wasn't resolved.
   MoorTable on;
 
+  String _create;
+
   MoorTrigger(this.displayName, this.declaration, this.on);
 
   factory MoorTrigger.fromMoor(CreateTriggerStatement stmt, FoundFile file) {
@@ -27,4 +29,12 @@ class MoorTrigger implements MoorSchemaEntity {
 
   @override
   Iterable<MoorSchemaEntity> get references => [on];
+
+  /// The `CREATE TRIGGER` statement that can be used to create this trigger.
+  String get create {
+    if (_create != null) return _create;
+
+    final node = (declaration as MoorTriggerDeclaration).node;
+    return node.span.text;
+  }
 }

moor_generator/lib/src/writer/database_writer.dart

@@ -1,4 +1,5 @@
 import 'package:moor_generator/moor_generator.dart';
+import 'package:moor_generator/src/utils/string_escaper.dart';
 import 'package:moor_generator/writer.dart';
 import 'package:recase/recase.dart';
 
@@ -30,10 +31,10 @@ class DatabaseWriter {
           '$className.connect(DatabaseConnection c): super.connect(c); \n');
     }
 
-    final tableGetters = <String>[];
+    final tableGetters = <MoorTable, String>{};
 
     for (final table in db.tables) {
-      tableGetters.add(table.tableFieldName);
+      tableGetters[table] = table.tableFieldName;
       final tableClassName = table.tableInfoName;
 
       writeMemoizedGetter(
@@ -64,10 +65,30 @@ class DatabaseWriter {
       QueryWriter(query, dbScope.child(), writtenMappingMethods).write();
     }
 
-    // Write List of tables, close bracket for class
-    dbScope.leaf()
-      ..write('@override\nList<TableInfo> get allTables => [')
-      ..write(tableGetters.join(','))
-      ..write('];\n}');
+    // Write List of tables
+    final schemaScope = dbScope.leaf();
+    schemaScope
+      ..write('@override\nIterable<TableInfo> get allTables => ')
+      ..write('allSchemaEntities.whereType<TableInfo>();\n')
+      ..write('@override\nList<DatabaseSchemaEntity> get allSchemaEntities ')
+      ..write('=> [');
+
+    var first = true;
+    for (final entity in db.entities) {
+      if (!first) {
+        schemaScope.write(', ');
+      }
+
+      if (entity is MoorTable) {
+        schemaScope.write(tableGetters[entity]);
+      } else if (entity is MoorTrigger) {
+        schemaScope.write('Trigger(${asDartLiteral(entity.create)}, '
+            '${asDartLiteral(entity.displayName)})');
+      }
+      first = false;
+    }
+
+    // finally, close bracket for the allSchemaEntities override and class.
+    schemaScope.write('];\n}');
   }
 }

sqlparser/lib/src/reader/parser/schema.dart

@@ -4,7 +4,7 @@ mixin SchemaParser on ParserBase {
   SchemaStatement _create() {
     if (!_matchOne(TokenType.create)) return null;
 
-    if (_check(TokenType.table)) {
+    if (_check(TokenType.table) || _check(TokenType.virtual)) {
       return _createTable();
     } else if (_check(TokenType.trigger)) {
       return _createTrigger();