Merge pull request #81 from khanjasir90/users/jasir/parsers-action-ov…

…erride-feat feat: added ability to override mirai parsers and action_parsers.
StacDev · Jan 5, 2025 · 7e32dbb · 7e32dbb
2 parents f8f18fd + d101f8a
commit 7e32dbb
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 12 deletions.
diff --git a/packages/mirai/lib/src/framework/mirai.dart b/packages/mirai/lib/src/framework/mirai.dart
@@ -100,11 +100,12 @@ class Mirai {
     List<MiraiParser> parsers = const [],
     List<MiraiActionParser> actionParsers = const [],
     Dio? dio,
+    bool override = false,
   }) async {
     _parsers.addAll(parsers);
     _actionParsers.addAll(actionParsers);
-    MiraiRegistry.instance.registerAll(_parsers);
-    MiraiRegistry.instance.registerAllActions(_actionParsers);
+    MiraiRegistry.instance.registerAll(_parsers, override);
+    MiraiRegistry.instance.registerAllActions(_actionParsers, override);
     MiraiNetworkService.initialize(dio ?? Dio());
   }
 

diff --git a/packages/mirai/lib/src/framework/mirai_registry.dart b/packages/mirai/lib/src/framework/mirai_registry.dart
@@ -14,42 +14,56 @@ class MiraiRegistry {
 
   static final _miraiActionParsers = <String, MiraiActionParser>{};
 
-  bool register(MiraiParser parser) {
+  bool register(MiraiParser parser, [bool override = false]) {
     final String type = parser.type;
     if (_miraiParsers.containsKey(type)) {
-      Log.w('Widget $type is already registered');
-      return false;
+      if (override) {
+        Log.w('Widget $type is being overridden');
+        _miraiParsers[type] = parser;
+        return true;
+      } else {
+        Log.w('Parser $type is already registered');
+        return false;
+      }
     } else {
       _miraiParsers[type] = parser;
       return true;
     }
   }
 
-  bool registerAction(MiraiActionParser parser) {
+  bool registerAction(MiraiActionParser parser, [bool override = false]) {
     final String type = parser.actionType;
     if (_miraiActionParsers.containsKey(type)) {
-      Log.w('Action $type is already registered');
-      return false;
+      if (override) {
+        Log.w('Action $type is being overridden');
+        _miraiActionParsers[type] = parser;
+        return true;
+      } else {
+        Log.w('Action $type is already registered');
+        return false;
+      }
     } else {
       _miraiActionParsers[type] = parser;
       return true;
     }
   }
 
-  Future<dynamic> registerAll(List<MiraiParser> parsers) {
+  Future<dynamic> registerAll(List<MiraiParser> parsers,
+      [bool override = false]) {
     return Future.forEach(
       parsers,
       (MiraiParser parser) {
-        return register(parser);
+        return register(parser, override);
       },
     );
   }
 
-  Future<dynamic> registerAllActions(List<MiraiActionParser> parsers) {
+  Future<dynamic> registerAllActions(List<MiraiActionParser> parsers,
+      [bool override = false]) {
     return Future.forEach(
       parsers,
       (MiraiActionParser parser) {
-        return registerAction(parser);
+        return registerAction(parser, override);
       },
     );
   }