From 63e5eb0a24f0494f182f152fde76f6be1311b3a5 Mon Sep 17 00:00:00 2001
From: Salvatore Giordano <salvatoregiordanoo@gmail.com>
Date: Wed, 3 Nov 2021 13:12:39 +0100
Subject: [PATCH] clean up

---
 packages/stream_chat/lib/version.dart         | 22 -------------------
 .../lib/src/stream_chat.dart                  |  2 +-
 .../lib/src/stream_chat_core.dart             | 14 +++++-------
 3 files changed, 7 insertions(+), 31 deletions(-)

diff --git a/packages/stream_chat/lib/version.dart b/packages/stream_chat/lib/version.dart
index f0ea7c92..c839a929 100644
--- a/packages/stream_chat/lib/version.dart
+++ b/packages/stream_chat/lib/version.dart
@@ -4,25 +4,3 @@ import 'package:stream_chat/src/client/client.dart';
 /// Used in [StreamChatClient] to build the `x-stream-client` header
 // ignore: constant_identifier_names
 const PACKAGE_VERSION = '3.2.0';
-
-/// Possible packages
-enum Package {
-  /// The Low Level Client
-  llc,
-
-  /// The Core package
-  core,
-
-  /// The UI package
-  ui,
-}
-
-///
-extension XPackage on Package {
-  ///
-  String get name => {
-        Package.llc: 'llc',
-        Package.core: 'core',
-        Package.ui: 'ui',
-      }[this]!;
-}
diff --git a/packages/stream_chat_flutter/lib/src/stream_chat.dart b/packages/stream_chat_flutter/lib/src/stream_chat.dart
index db645943..76b0f27a 100644
--- a/packages/stream_chat_flutter/lib/src/stream_chat.dart
+++ b/packages/stream_chat_flutter/lib/src/stream_chat.dart
@@ -115,7 +115,7 @@ class StreamChatState extends State<StreamChat> {
                   builder: (context) {
                     StreamChatClient.additionalHeaders = {
                       'X-Stream-Client':
-                          '${StreamChatClient.defaultUserAgent}-${Package.ui}',
+                          '${StreamChatClient.defaultUserAgent}-ui',
                     };
                     return widget.child ?? const Offstage();
                   },
diff --git a/packages/stream_chat_flutter_core/lib/src/stream_chat_core.dart b/packages/stream_chat_flutter_core/lib/src/stream_chat_core.dart
index d43f3f51..260aa1cd 100644
--- a/packages/stream_chat_flutter_core/lib/src/stream_chat_core.dart
+++ b/packages/stream_chat_flutter_core/lib/src/stream_chat_core.dart
@@ -90,20 +90,18 @@ class StreamChatCore extends StatefulWidget {
 /// State class associated with [StreamChatCore].
 class StreamChatCoreState extends State<StreamChatCore>
     with WidgetsBindingObserver {
-  /// Constructor used for creating a new instance of [StreamChatCoreState].
-  StreamChatCoreState() {
-    StreamChatClient.additionalHeaders = {
-      'X-Stream-Client': '${StreamChatClient.defaultUserAgent}-${Package.ui}',
-    };
-  }
-
   /// Initialized client used throughout the application.
   StreamChatClient get client => widget.client;
 
   Timer? _disconnectTimer;
 
   @override
-  Widget build(BuildContext context) => widget.child;
+  Widget build(BuildContext context) {
+    StreamChatClient.additionalHeaders = {
+      'X-Stream-Client': '${StreamChatClient.defaultUserAgent}-ui',
+    };
+    return widget.child;
+  }
 
   // coverage:ignore-start