implemented helpers for StatefulShellRoute

packages/go_router/CHANGELOG.md

@@ -1,3 +1,7 @@
+## 8.0.3 
+
+- Adds helpers for go_router_builder for StatefulShellRoute support
+
 ## 8.0.2 
 
 - Fixes a bug in `debugLogDiagnostics` to support StatefulShellRoute.

packages/go_router/lib/go_router.dart

@@ -25,9 +25,13 @@ export 'src/route_data.dart'
         RouteData,
         GoRouteData,
         ShellRouteData,
+        StatefulShellBranchData,
+        StatefulShellRouteData,
         TypedRoute,
         TypedGoRoute,
-        TypedShellRoute;
+        TypedShellRoute,
+        TypedStatefulShellBranch,
+        TypedStatefulShellRoute;
 export 'src/router.dart';
 export 'src/typedefs.dart'
     show

packages/go_router/lib/src/route.dart

@@ -655,8 +655,8 @@ class StatefulShellRoute extends ShellRouteBase {
     required this.navigatorContainerBuilder,
     this.restorationScopeId,
   })  : assert(branches.isNotEmpty),
-        assert((pageBuilder != null) ^ (builder != null),
-            'One of builder or pageBuilder must be provided, but not both'),
+        assert((pageBuilder != null) || (builder != null),
+            'One of builder or pageBuilder must be provided'),
         assert(_debugUniqueNavigatorKeys(branches).length == branches.length,
             'Navigator keys must be unique'),
         assert(_debugValidateParentNavigatorKeys(branches)),

packages/go_router/lib/src/route_data.dart

@@ -137,7 +137,7 @@ abstract class ShellRouteData extends RouteData {
   ) =>
       const NoOpPage();
 
-  /// [pageBuilder] is used to build the page
+  /// [builder] is used to build the widget
   Widget builder(
     BuildContext context,
     GoRouterState state,
@@ -159,7 +159,7 @@ abstract class ShellRouteData extends RouteData {
       final Object? extra = state.extra;
 
       // If the "extra" value is of type `T` then we know it's the source
-      // instance of `GoRouteData`, so it doesn't need to be recreated.
+      // instance of `ShellRouteData`, so it doesn't need to be recreated.
       if (extra is T) {
         return extra;
       }
@@ -205,6 +205,131 @@ abstract class ShellRouteData extends RouteData {
   );
 }
 
+
+
+/// Base class for supporting
+/// [StatefulShellRoute](https://pub.dev/documentation/go_router/latest/go_router/StatefulShellRoute-class.html)
+abstract class StatefulShellRouteData extends RouteData {
+  /// Default const constructor
+  const StatefulShellRouteData();
+
+
+  /// [pageBuilder] is used to build the page
+  Page<void> pageBuilder(
+    BuildContext context,
+    GoRouterState state,
+    Widget navigator,
+  ) =>
+      const NoOpPage();
+
+  /// [builder] is used to build the widget
+  Widget builder(
+    BuildContext context,
+    GoRouterState state,
+    StatefulNavigationShell navigationShell,
+  ) =>
+      throw UnimplementedError(
+        'One of `builder` or `pageBuilder` must be implemented.',
+      );
+
+  /// A helper function used by generated code.
+  ///
+  /// Should not be used directly.
+  static StatefulShellRoute $route<T extends StatefulShellRouteData>({
+    required T Function(GoRouterState) factory,
+    required List<StatefulShellBranch> branches,
+    String? restorationScopeId,
+  }) {
+    T factoryImpl(GoRouterState state) {
+      final Object? extra = state.extra;
+
+      // If the "extra" value is of type `T` then we know it's the source
+      // instance of `StatefulShellRouteData`, so it doesn't need to be recreated.
+      if (extra is T) {
+        return extra;
+      }
+
+      return (_stateObjectExpando[state] ??= factory(state)) as T;
+    }
+
+    Widget builder(
+      BuildContext context,
+      GoRouterState state,
+      StatefulNavigationShell navigationShell,
+    ) =>
+        factoryImpl(state).builder(
+          context,
+          state,
+          navigationShell,
+        );
+
+    Page<void> pageBuilder(
+      BuildContext context,
+      GoRouterState state,
+      StatefulNavigationShell navigationShell,
+    ) =>
+        factoryImpl(state).pageBuilder(
+          context,
+          state,
+          navigationShell,
+        );
+
+    return StatefulShellRoute.indexedStack(
+      branches: branches,
+      builder: builder,
+      pageBuilder: pageBuilder,
+      restorationScopeId: restorationScopeId,
+    );
+  }
+
+  /// Used to cache [StatefulShellRouteData] that corresponds to a given [GoRouterState]
+  /// to minimize the number of times it has to be deserialized.
+  static final Expando<StatefulShellRouteData> _stateObjectExpando =
+      Expando<StatefulShellRouteData>(
+    'GoRouteState to StatefulShellRouteData expando',
+  );
+}
+
+/// Base class for supporting
+/// [StatefulShellRoute](https://pub.dev/documentation/go_router/latest/go_router/StatefulShellRoute-class.html)
+abstract class StatefulShellBranchData extends RouteData {
+  /// Default const constructor
+  const StatefulShellBranchData();
+
+  /// A helper function used by generated code.
+  ///
+  /// Should not be used directly.
+  static StatefulShellBranch $route<T extends StatefulShellBranchData>({
+    required T Function(GoRouterState) factory,
+    GlobalKey<NavigatorState>? navigatorKey,
+    List<RouteBase> routes = const <RouteBase>[],
+  }) {
+    T factoryImpl(GoRouterState state) {
+      final Object? extra = state.extra;
+
+      // If the "extra" value is of type `T` then we know it's the source
+      // instance of `StatefulShellBranchData`, so it doesn't need to be recreated.
+      if (extra is T) {
+        return extra;
+      }
+
+      return (_stateObjectExpando[state] ??= factory(state)) as T;
+    }
+
+    return StatefulShellBranch(
+      routes: routes,
+      navigatorKey: navigatorKey,
+    );
+  }
+
+  /// Used to cache [StatefulShellBranchData] that corresponds to a given [GoRouterState]
+  /// to minimize the number of times it has to be deserialized.
+  static final Expando<StatefulShellBranchData> _stateObjectExpando =
+      Expando<StatefulShellBranchData>(
+    'GoRouteState to StatefulShellBranchData expando',
+  );
+}
+
 /// A superclass for each typed route descendant
 class TypedRoute<T extends RouteData> {
   /// Default const constructor
@@ -256,6 +381,36 @@ class TypedShellRoute<T extends ShellRouteData> extends TypedRoute<T> {
   final List<TypedRoute<RouteData>> routes;
 }
 
+/// A superclass for each typed shell route descendant
+@Target(<TargetKind>{TargetKind.library, TargetKind.classType})
+class TypedStatefulShellRoute<T extends StatefulShellRouteData>
+    extends TypedRoute<T> {
+  /// Default const constructor
+  const TypedStatefulShellRoute({
+    this.routes = const <TypedRoute<RouteData>>[],
+  });
+
+  /// Child route definitions.
+  ///
+  /// See [RouteBase.routes].
+  final List<TypedRoute<RouteData>> routes;
+}
+
+/// A superclass for each typed shell route descendant
+@Target(<TargetKind>{TargetKind.library, TargetKind.classType})
+class TypedStatefulShellBranch<T extends StatefulShellBranchData>
+    extends TypedRoute<T> {
+  /// Default const constructor
+  const TypedStatefulShellBranch({
+    this.routes = const <TypedRoute<RouteData>>[],
+  });
+
+  /// Child route definitions.
+  ///
+  /// See [RouteBase.routes].
+  final List<TypedRoute<RouteData>> routes;
+}
+
 /// Internal class used to signal that the default page behavior should be used.
 @internal
 class NoOpPage extends Page<void> {

packages/go_router/pubspec.yaml

@@ -1,7 +1,7 @@
 name: go_router
 description: A declarative router for Flutter based on Navigation 2 supporting
   deep linking, data-driven routes and more
-version: 8.0.2
+version: 8.0.3
 repository: https://github.com/flutter/packages/tree/main/packages/go_router
 issue_tracker: https://github.com/flutter/flutter/issues?q=is%3Aissue+is%3Aopen+label%3A%22p%3A+go_router%22
 

packages/go_router/test/route_data_test.dart

@@ -44,7 +44,6 @@ final ShellRoute _shellRouteDataBuilder = ShellRouteData.$route(
     ),
   ],
 );
-
 class _GoRouteDataBuildPage extends GoRouteData {
   const _GoRouteDataBuildPage();
   @override
@@ -86,6 +85,71 @@ final ShellRoute _shellRouteDataPageBuilder = ShellRouteData.$route(
   ],
 );
 
+class _StatefulShellBranchDataBuilder extends StatefulShellBranchData {
+  const _StatefulShellBranchDataBuilder();
+}
+class _StatefulShellRouteDataBuilder extends StatefulShellRouteData {
+  const _StatefulShellRouteDataBuilder();
+
+  @override
+  Widget builder(
+    BuildContext context,
+    GoRouterState state,
+    Widget navigator,
+  ) =>
+      SizedBox(
+        key: const Key('builder'),
+        child: navigator,
+      );
+}
+final StatefulShellRoute _statefulShellRouteDataBuilder =
+    StatefulShellRouteData.$route(
+  factory: (GoRouterState state) => const _StatefulShellRouteDataBuilder(),
+  branches: <StatefulShellBranch>[
+    StatefulShellBranchData.$route(
+      factory: (GoRouterState state) => const _StatefulShellBranchDataBuilder(),
+      routes: <RouteBase>[
+        GoRouteData.$route(
+          path: '/child',
+          factory: (GoRouterState state) => const _GoRouteDataBuild(),
+        ),
+      ],
+    ),
+  ],
+);
+
+class _StatefulShellRouteDataPageBuilder extends StatefulShellRouteData {
+  const _StatefulShellRouteDataPageBuilder();
+
+  @override
+  Page<void> pageBuilder(
+    BuildContext context,
+    GoRouterState state,
+    Widget navigator,
+  ) =>
+      MaterialPage<void>(
+        child: SizedBox(
+          key: const Key('page-builder'),
+          child: navigator,
+        ),
+      );
+}
+
+final StatefulShellRoute _statefulShellRouteDataPageBuilder = StatefulShellRouteData.$route(
+  factory: (GoRouterState state) => const _StatefulShellRouteDataPageBuilder(),
+  branches: <StatefulShellBranch>[
+    StatefulShellBranchData.$route(
+      factory: (GoRouterState state) => const _StatefulShellBranchDataBuilder(),
+      routes: <RouteBase>[
+        GoRouteData.$route(
+          path: '/child',
+          factory: (GoRouterState state) => const _GoRouteDataBuild(),
+        ),
+      ],
+    ),
+  ],
+);
+
 class _GoRouteDataRedirectPage extends GoRouteData {
   const _GoRouteDataRedirectPage();
   @override
@@ -181,6 +245,46 @@ void main() {
     );
   });
 
+  group('StatefulShellRouteData', () {
+    testWidgets(
+      'It should build the page from the overridden build method',
+      (WidgetTester tester) async {
+        final GoRouter goRouter = GoRouter(
+          initialLocation: '/child',
+          routes: <RouteBase>[
+            _statefulShellRouteDataBuilder,
+          ],
+        );
+        await tester.pumpWidget(MaterialApp.router(
+          routeInformationProvider: goRouter.routeInformationProvider,
+          routeInformationParser: goRouter.routeInformationParser,
+          routerDelegate: goRouter.routerDelegate,
+        ));
+        expect(find.byKey(const Key('builder')), findsOneWidget);
+        expect(find.byKey(const Key('page-builder')), findsNothing);
+      },
+    );
+
+        testWidgets(
+      'It should build the page from the overridden buildPage method',
+      (WidgetTester tester) async {
+        final GoRouter goRouter = GoRouter(
+          initialLocation: '/child',
+          routes: <RouteBase>[
+            _statefulShellRouteDataPageBuilder,
+          ],
+        );
+        await tester.pumpWidget(MaterialApp.router(
+          routeInformationProvider: goRouter.routeInformationProvider,
+          routeInformationParser: goRouter.routeInformationParser,
+          routerDelegate: goRouter.routerDelegate,
+        ));
+        expect(find.byKey(const Key('builder')), findsNothing);
+        expect(find.byKey(const Key('page-builder')), findsOneWidget);
+      },
+    );
+  });
+
   testWidgets(
     'It should redirect using the overridden redirect method',
     (WidgetTester tester) async {