From 04feb4499564449924f16d335c95771b57dd66b7 Mon Sep 17 00:00:00 2001
From: Hugo Pointcheval <git@pcl.ovh>
Date: Wed, 15 Feb 2023 15:57:04 +0100
Subject: [PATCH] feat(ui_kit): add theme extension shortcut

---
 .../lib/src/core/extensions/theme_extensions.dart           | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/packages/wyatt_ui_kit/lib/src/core/extensions/theme_extensions.dart b/packages/wyatt_ui_kit/lib/src/core/extensions/theme_extensions.dart
index 22efeb80..021dca70 100644
--- a/packages/wyatt_ui_kit/lib/src/core/extensions/theme_extensions.dart
+++ b/packages/wyatt_ui_kit/lib/src/core/extensions/theme_extensions.dart
@@ -16,8 +16,6 @@
 
 import 'package:flutter/material.dart';
 
-extension ThemeExtension on BuildContext {
-  TextTheme get textTheme => Theme.of(this).textTheme;
-  ColorScheme get colorScheme => Theme.of(this).colorScheme;
-  ButtonThemeData get buttonTheme => Theme.of(this).buttonTheme;
+extension BuildContextThemeExtension on BuildContext {
+  T? themeExtension<T>() => Theme.of(this).extension<T>();
 }