packages/wyatt_ui_components/lib/src/features/component_theme_data.dart

@@ -18,6 +18,7 @@ import 'package:copy_with_extension/copy_with_extension.dart';
 import 'package:flutter/material.dart';
 import 'package:wyatt_ui_components/src/core/utils/component_registry.dart';
 import 'package:wyatt_ui_components/src/domain/entities/entities.dart';
+
 part 'component_theme_data.g.dart';
 
 /// {@template component_theme_data}
@@ -128,17 +129,35 @@ class ComponentThemeData {
     return returned;
   }
 
+  R? _getOrNull<T extends Component, R>(T? component, R? returned) {
+    if (component == null) {
+      return null;
+    }
+
+    return returned;
+  }
+
   // Bars
   final ComponentRegistry<TopAppBarComponent>? topAppBars;
   $TopAppBarComponentCWProxy topAppBarComponent([String? id]) =>
       _get(topAppBars?.call(id), topAppBars?.call(id)?.copyWith);
 
+  $TopAppBarComponentCWProxy? topAppBarComponentOrNull([String? id]) =>
+      _getOrNull(topAppBars?.call(id), topAppBars?.call(id)?.copyWith);
+
   final ComponentRegistry<TopNavigationBarComponent>? topNavigationBars;
   $TopNavigationBarComponentCWProxy topNavigationBarComponent([String? id]) =>
       _get(
         topNavigationBars?.call(id),
         topNavigationBars?.call(id)?.copyWith,
       );
+  $TopNavigationBarComponentCWProxy? topNavigationBarComponentOrNull([
+    String? id,
+  ]) =>
+      _getOrNull(
+        topNavigationBars?.call(id),
+        topNavigationBars?.call(id)?.copyWith,
+      );
 
   final ComponentRegistry<BottomNavigationBarComponent>? bottomNavigationBars;
   $BottomNavigationBarComponentCWProxy bottomNavigationBarComponent([
@@ -148,46 +167,84 @@ class ComponentThemeData {
         bottomNavigationBars?.call(id),
         bottomNavigationBars?.call(id)?.copyWith,
       );
+  $BottomNavigationBarComponentCWProxy? bottomNavigationBarComponentOrNull([
+    String? id,
+  ]) =>
+      _getOrNull(
+        bottomNavigationBars?.call(id),
+        bottomNavigationBars?.call(id)?.copyWith,
+      );
 
   // CRUD Widgets
   final ComponentRegistry<ErrorComponent>? errors;
   $ErrorComponentCWProxy errorComponent([String? id]) =>
       _get(errors?.call(id), errors?.call(id)?.copyWith);
+  $ErrorComponentCWProxy? errorComponentOrNull([String? id]) =>
+      _getOrNull(errors?.call(id), errors?.call(id)?.copyWith);
 
   final ComponentRegistry<LoaderComponent>? loaders;
   $LoaderComponentCWProxy loaderComponent([String? id]) =>
       _get(loaders?.call(id), loaders?.call(id)?.copyWith);
+  $LoaderComponentCWProxy? loaderComponentOrNull([String? id]) =>
+      _getOrNull(loaders?.call(id), loaders?.call(id)?.copyWith);
 
   // Cards
   final ComponentRegistry<InformationCardComponent>? informationCards;
   $InformationCardComponentCWProxy informationCardComponent([String? id]) =>
       _get(informationCards?.call(id), informationCards?.call(id)?.copyWith);
+  $InformationCardComponentCWProxy? informationCardComponentOrNull([
+    String? id,
+  ]) =>
+      _getOrNull(
+        informationCards?.call(id),
+        informationCards?.call(id)?.copyWith,
+      );
 
   final ComponentRegistry<PortfolioCardComponent>? portfolioCards;
   $PortfolioCardComponentCWProxy portfolioCardComponent([String? id]) =>
       _get(portfolioCards?.call(id), portfolioCards?.call(id)?.copyWith);
+  $PortfolioCardComponentCWProxy? portfolioCardComponentOrNull([String? id]) =>
+      _getOrNull(
+        portfolioCards?.call(id),
+        portfolioCards?.call(id)?.copyWith,
+      );
 
   final ComponentRegistry<QuoteCardComponent>? quoteCards;
   $QuoteCardComponentCWProxy quoteCardComponent([String? id]) =>
       _get(quoteCards?.call(id), quoteCards?.call(id)?.copyWith);
+  $QuoteCardComponentCWProxy? quoteCardComponentOrNull([String? id]) =>
+      _getOrNull(quoteCards?.call(id), quoteCards?.call(id)?.copyWith);
 
   final ComponentRegistry<SkillCardComponent>? skillCards;
   $SkillCardComponentCWProxy skillCardComponent([String? id]) =>
       _get(skillCards?.call(id), skillCards?.call(id)?.copyWith);
+  $SkillCardComponentCWProxy? skillCardComponentOrNull([String? id]) =>
+      _getOrNull(skillCards?.call(id), skillCards?.call(id)?.copyWith);
 
   final ComponentRegistry<PricingCardComponent>? pricingCards;
   $PricingCardComponentCWProxy pricingCardComponent([String? id]) =>
       _get(pricingCards?.call(id), pricingCards?.call(id)?.copyWith);
+  $PricingCardComponentCWProxy? pricingCardComponentOrNull([String? id]) =>
+      _getOrNull(pricingCards?.call(id), pricingCards?.call(id)?.copyWith);
 
   // Rich Text
   final ComponentRegistry<RichTextBuilderComponent>? richTextBuilders;
   $RichTextBuilderComponentCWProxy richTextBuilderComponent([String? id]) =>
       _get(richTextBuilders?.call(id), richTextBuilders?.call(id)?.copyWith);
+  $RichTextBuilderComponentCWProxy? richTextBuilderComponentOrNull([
+    String? id,
+  ]) =>
+      _getOrNull(
+        richTextBuilders?.call(id),
+        richTextBuilders?.call(id)?.copyWith,
+      );
 
   // Text Inputs
   final ComponentRegistry<TextInputComponent>? textInputs;
   $TextInputComponentCWProxy textInputComponent([String? id]) =>
       _get(textInputs?.call(id), textInputs?.call(id)?.copyWith);
+  $TextInputComponentCWProxy? textInputComponentOrNull([String? id]) =>
+      _getOrNull(textInputs?.call(id), textInputs?.call(id)?.copyWith);
 
   // Buttons
   final ComponentRegistry<FileSelectionButtonComponent>? fileSelectionButtons;
@@ -198,10 +255,19 @@ class ComponentThemeData {
         fileSelectionButtons?.call(id),
         fileSelectionButtons?.call(id)?.copyWith,
       );
+  $FileSelectionButtonComponentCWProxy? fileSelectionButtonComponentOrNull([
+    String? id,
+  ]) =>
+      _getOrNull(
+        fileSelectionButtons?.call(id),
+        fileSelectionButtons?.call(id)?.copyWith,
+      );
 
   final ComponentRegistry<FlatButtonComponent>? flatButtons;
   $FlatButtonComponentCWProxy flatButtonComponent([String? id]) =>
       _get(flatButtons?.call(id), flatButtons?.call(id)?.copyWith);
+  $FlatButtonComponentCWProxy? flatButtonComponentOrNull([String? id]) =>
+      _getOrNull(flatButtons?.call(id), flatButtons?.call(id)?.copyWith);
 
   final ComponentRegistry<SimpleIconButtonComponent>? simpleIconButtons;
   $SimpleIconButtonComponentCWProxy simpleIconButtonComponent([String? id]) =>
@@ -209,10 +275,19 @@ class ComponentThemeData {
         simpleIconButtons?.call(id),
         simpleIconButtons?.call(id)?.copyWith,
       );
+  $SimpleIconButtonComponentCWProxy? simpleIconButtonComponentOrNull([
+    String? id,
+  ]) =>
+      _getOrNull(
+        simpleIconButtons?.call(id),
+        simpleIconButtons?.call(id)?.copyWith,
+      );
 
   final ComponentRegistry<SymbolButtonComponent>? symbolButtons;
   $SymbolButtonComponentCWProxy symbolButtonComponent([String? id]) =>
       _get(symbolButtons?.call(id), symbolButtons?.call(id)?.copyWith);
+  $SymbolButtonComponentCWProxy? symbolButtonComponentOrNull([String? id]) =>
+      _getOrNull(symbolButtons?.call(id), symbolButtons?.call(id)?.copyWith);
 
   final ComponentRegistry<FloatingActionButtonComponent>? floatingActionButtons;
   $FloatingActionButtonComponentCWProxy floatingActionButtonComponent([
@@ -222,4 +297,11 @@ class ComponentThemeData {
         floatingActionButtons?.call(id),
         floatingActionButtons?.call(id)?.copyWith,
       );
+  $FloatingActionButtonComponentCWProxy? floatingActionButtonComponentOrNull([
+    String? id,
+  ]) =>
+      _getOrNull(
+        floatingActionButtons?.call(id),
+        floatingActionButtons?.call(id)?.copyWith,
+      );
 }

packages/wyatt_ui_layout/lib/src/presentation/layouts/structural_layouts/bottom_navigation_bar_layout.dart

@@ -57,9 +57,13 @@ class BottomNavigationBarLayout extends StructuralLayout {
   Widget build(BuildContext context) => Scaffold(
         body: body,
         bottomNavigationBar: custom?.call(
-              context.components.bottomNavigationBarComponent(barId).call(),
+              context.components
+                  .bottomNavigationBarComponentOrNull(barId)
+                  ?.call(),
             ) ??
-            context.components.bottomNavigationBarComponent(barId).call(),
+            context.components
+                .bottomNavigationBarComponentOrNull(barId)
+                ?.call(),
         floatingActionButtonLocation:
             scaffoldFieldsWrapper?.floatingActionButtonLocation,
         floatingActionButtonAnimator:

packages/wyatt_ui_layout/lib/src/presentation/layouts/structural_layouts/frame_layout.dart

@@ -76,29 +76,37 @@ class FrameLayout extends StructuralLayout {
 
   final ScaffoldFieldsWrapper? scaffoldFieldsWrapper;
 
-  @override
+  PreferredSize? _buildAppBar(BuildContext context) {
-  Widget build(BuildContext context) => Scaffold(
+    final appBar = customAppBar?.call(
-        appBar: (customAppBar
+          context.components.topAppBarComponentOrNull(appBarId)?.call(),
-                    ?.call(context.components.topAppBarComponent(appBarId)()) !=
+        ) ??
-                null)
+        context.components.topAppBarComponentOrNull(appBarId)?.call();
+
+    return appBar != null
         ? PreferredSize(
             preferredSize: Size.fromHeight(height),
-                child: customAppBar?.call(
+            child: appBar,
-                      context.components.topAppBarComponent(appBarId)(),
-                    ) ??
-                    context.components.topAppBarComponent(appBarId)(),
           )
-            : null,
+        : null;
+  }
+
+  @override
+  Widget build(BuildContext context) => Scaffold(
+        appBar: _buildAppBar(context),
         floatingActionButton: customFloatingActionButton?.call(
               context.components
-                  .floatingActionButtonComponent(floatingActionButtonId)(),
+                  .floatingActionButtonComponentOrNull(floatingActionButtonId)
+                  ?.call(),
             ) ??
             context.components
-                .floatingActionButtonComponent(floatingActionButtonId)(),
+                .floatingActionButtonComponentOrNull(floatingActionButtonId)
+                ?.call(),
         bottomNavigationBar: customBottomNavBar?.call(
               context.components.bottomNavigationBarComponent(bottomNavBarId)(),
             ) ??
-            context.components.bottomNavigationBarComponent(bottomNavBarId)(),
+            context.components
+                .bottomNavigationBarComponentOrNull(bottomNavBarId)
+                ?.call(),
         body: body,
         floatingActionButtonLocation:
             scaffoldFieldsWrapper?.floatingActionButtonLocation,

packages/wyatt_ui_layout/lib/src/presentation/layouts/structural_layouts/top_app_bar_layout.dart

@@ -59,7 +59,7 @@ abstract class TopBarLayout<T extends TopBarComponent>
   final double height;
 
   /// Returns the top bar component for the given [BuildContext].
-  T appBar(BuildContext context, String? barId);
+  T? appBar(BuildContext context, String? barId);
 
   /// The [scaffoldFieldsWrapper] is a final variable that serves as a wrapper
   /// for customizing the scaffold.
@@ -67,18 +67,21 @@ abstract class TopBarLayout<T extends TopBarComponent>
   /// scaffold functionality.
   final ScaffoldFieldsWrapper? scaffoldFieldsWrapper;
 
-  @override
+  PreferredSize? _buildAppBar(BuildContext context) {
-  Widget build(BuildContext context) => Scaffold(
+    final topAppBar =
-        appBar: (custom?.call(
+        custom?.call(appBar(context, barId)) ?? appBar(context, barId);
-                  appBar(context, barId),
+
-                ) !=
+    return topAppBar != null
-                null)
         ? PreferredSize(
             preferredSize: Size.fromHeight(height),
-                child: custom?.call(appBar(context, barId)) ??
+            child: topAppBar,
-                    appBar(context, barId),
           )
-            : null,
+        : null;
+  }
+
+  @override
+  Widget build(BuildContext context) => Scaffold(
+        appBar: _buildAppBar(context),
         body: body,
         floatingActionButtonLocation:
             scaffoldFieldsWrapper?.floatingActionButtonLocation,
@@ -130,9 +133,23 @@ class TopAppBarLayout extends TopBarLayout<TopAppBarComponent> {
     super.key,
   });
 
+  PreferredSize? _buildAppBar(BuildContext context) {
+    final appBar = custom?.call(
+          context.components.topAppBarComponentOrNull(barId)?.call(),
+        ) ??
+        context.components.topAppBarComponentOrNull(barId)?.call();
+
+    return appBar != null
+        ? PreferredSize(
+            preferredSize: Size.fromHeight(height),
+            child: appBar,
+          )
+        : null;
+  }
+
   @override
-  TopAppBarComponent appBar(BuildContext context, String? barId) =>
+  TopAppBarComponent? appBar(BuildContext context, String? barId) =>
-      context.components.topAppBarComponent(barId).call();
+      context.components.topAppBarComponentOrNull(barId)?.call();
 }
 
 /// A concrete implementation of [TopBarLayout] for a navigation bar.
@@ -153,6 +170,6 @@ class TopNavigationBarLayout extends TopBarLayout<TopNavigationBarComponent> {
   });
 
   @override
-  TopNavigationBarComponent appBar(BuildContext context, String? barId) =>
+  TopNavigationBarComponent? appBar(BuildContext context, String? barId) =>
-      context.components.topNavigationBarComponent(barId).call();
+      context.components.topNavigationBarComponentOrNull(barId)?.call();
 }