chore(ast): Add version field to DatabaseSchema

Author: dnys1

apps/cli/lib/src/analyzer/celest_analysis_helpers.dart

@@ -1,8 +1,12 @@
 import 'package:analyzer/dart/analysis/analysis_context.dart';
 import 'package:analyzer/dart/analysis/results.dart';
+import 'package:analyzer/dart/ast/ast.dart';
 import 'package:analyzer/dart/constant/value.dart';
 import 'package:analyzer/dart/element/element.dart';
+import 'package:analyzer/dart/element/element2.dart';
 import 'package:analyzer/dart/element/type.dart';
+import 'package:analyzer/src/dart/analysis/session_helper.dart';
+import 'package:analyzer/src/dart/ast/ast.dart';
 import 'package:analyzer/src/dart/resolver/scope.dart';
 import 'package:celest_cli/src/analyzer/analysis_error.dart';
 import 'package:celest_cli/src/analyzer/resolver/project_resolver.dart';
@@ -53,6 +57,11 @@ mixin CelestAnalysisHelpers implements CelestErrorReporter {
   Set<InterfaceElement> get customExceptionTypes;
   Set<InterfaceElement> get customModelTypes;
 
+  static final Expando<AnalysisSessionHelper> _sessionHelpers = Expando();
+  AnalysisSessionHelper get helper =>
+      _sessionHelpers[context.currentSession] ??=
+          AnalysisSessionHelper(context.currentSession);
+
   final pendingEdits = <String, Set<SourceEdit>>{};
 
   static final Logger _logger = Logger('CelestAnalyzer');
@@ -292,6 +301,84 @@ mixin CelestAnalysisHelpers implements CelestErrorReporter {
       );
     }
   }
+
+  /// Resolves the schema version of a Drift database class.
+  Future<int> resolveSchemaVersion(ClassElement2 databaseClass) async {
+    final schemaVersionMethod =
+        databaseClass.getGetter2('schemaVersion')?.variable3;
+    if (schemaVersionMethod == null) {
+      reportError(
+        'Invalid database class: No `schemaVersion` getter.',
+        severity: AnalysisErrorSeverity.error,
+        location: databaseClass.sourceLocation,
+      );
+      return -1;
+    }
+    final declaration = await helper.getElementDeclaration(
+      schemaVersionMethod.isSynthetic
+          ? schemaVersionMethod.getter2!.firstFragment
+          : schemaVersionMethod.firstFragment,
+    );
+
+    int? resolveExpression(Expression? expression) {
+      switch (expression) {
+        // int get schemaVersion => 1;
+        case IntegerLiteral(value: final version?):
+          return version;
+
+        // int get schemaVersion => someConstVariable;
+        case Identifier(
+            staticElement: PropertyAccessorElement(:final variable2?)
+          ):
+          if (variable2.computeConstantValue()?.toIntValue()
+              case final version?) {
+            return version;
+          }
+      }
+
+      return null;
+    }
+
+    switch (declaration?.node) {
+      case VariableDeclaration(:final initializer):
+        if (resolveExpression(initializer) case final version?) {
+          return version;
+        }
+
+        // Could not determine schema version.
+        reportError(
+          'Invalid `schemaVersion` declaration: Must resolve to an integer.',
+          severity: AnalysisErrorSeverity.error,
+          location: schemaVersionMethod.sourceLocation,
+        );
+        return -1;
+
+      case MethodDeclaration(:final body):
+        switch (body) {
+          case ExpressionFunctionBody(:final expression):
+            if (resolveExpression(expression) case final version?) {
+              return version;
+            }
+          case BlockFunctionBody():
+          case EmptyFunctionBody():
+          case NativeFunctionBody():
+        }
+
+        // Could not determine schema version.
+        reportError(
+          'Invalid `schemaVersion` declaration: Must use => syntax.',
+          severity: AnalysisErrorSeverity.error,
+          location: schemaVersionMethod.sourceLocation,
+        );
+        return -1;
+
+      case final unknown:
+        throw StateError(
+          'Failed to resolve method declaration for $schemaVersionMethod: '
+          'Got $unknown (${unknown.runtimeType})',
+        );
+    }
+  }
 }
 
 extension TopLevelConstants on LibraryElement {

apps/cli/lib/src/utils/analyzer.dart

@@ -4,6 +4,7 @@ import 'dart:math';
 import 'package:analyzer/dart/ast/ast.dart';
 import 'package:analyzer/dart/constant/value.dart';
 import 'package:analyzer/dart/element/element.dart';
+import 'package:analyzer/dart/element/element2.dart';
 import 'package:analyzer/dart/element/nullability_suffix.dart';
 import 'package:analyzer/dart/element/type.dart';
 import 'package:analyzer/source/source.dart';
@@ -109,6 +110,12 @@ extension DartTypeHelper on DartType {
   bool get isCelestSecret =>
       element == typeHelper.coreTypes.celestSecretElement;
 
+  bool get isCloudAuthDatabaseMixin => switch (element) {
+        MixinElement(:final name, :final library) =>
+          name == 'CloudAuthDatabaseMixin' && library.isCelestSdk,
+        _ => false,
+      };
+
   bool get isAuth => switch (element) {
         ClassElement(:final name, :final library) =>
           name == 'Auth' && library.isPackageCelest,
@@ -660,6 +667,21 @@ extension ElementSourceLocation on Element {
       };
 }
 
+extension Element2SourceLocation on Element2 {
+  FileSpan? get sourceLocation {
+    final source = library2?.firstFragment.source;
+    if (source == null || !source.exists()) {
+      return null;
+    }
+    final nameOffset = firstFragment.nameOffset2 ?? -1;
+    final nameLength = firstFragment.name2?.length ?? 0;
+    if (nameOffset < 0) {
+      return null;
+    }
+    return source.toSpan(nameOffset, nameOffset + nameLength);
+  }
+}
+
 extension ElementAnnotationSourceLocation on ElementAnnotation {
   FileSpan sourceLocation(Source source) {
     final impl = this as ElementAnnotationImpl;

packages/celest_ast/CHANGELOG.md

@@ -1,5 +1,6 @@
-## 0.1.6
+## 0.1.6-wip
 
+- chore: Add `version` field to `DatabaseSchema`
 - chore: Update license
 
 ## 0.1.5

packages/celest_ast/lib/src/ast.dart

@@ -1065,6 +1065,9 @@ sealed class DatabaseSchema implements AstNode {
   /// The type of database schema.
   DatabaseSchemaType get type;
 
+  /// The version of the schema.
+  int get version;
+
   /// The Dart type declaring the database schema.
   TypeReference get declaration;
 }
@@ -1088,10 +1091,12 @@ abstract class DriftDatabaseSchema
         DatabaseSchema {
   factory DriftDatabaseSchema({
     required TypeReference declaration,
+    int version = 1,
     required FileSpan location,
   }) {
     return _$DriftDatabaseSchema._(
       declaration: declaration,
+      version: version,
       location: location,
     );
   }