feat: added override parameter to enable/disable override depending u…

…pon that param
StacDev · Jan 5, 2025 · ab2da74 · ab2da74
1 parent f546665
commit ab2da74
Show file tree

Hide file tree

Showing 2 changed files with 35 additions and 12 deletions.
diff --git a/packages/mirai/lib/src/framework/mirai.dart b/packages/mirai/lib/src/framework/mirai.dart
@@ -98,11 +98,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,32 +14,54 @@ class MiraiRegistry {
 
   static final _miraiActionParsers = <String, MiraiActionParser>{};
 
-  bool register(MiraiParser parser) {
+  bool register(MiraiParser parser, [bool override = false]) {
     final String type = parser.type;
-    _miraiParsers[type] = parser;
-    return true;
+    if (_miraiParsers.containsKey(type)) {
+      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;
-    _miraiActionParsers[type] = parser;
-    return true;
+    if (_miraiActionParsers.containsKey(type)) {
+      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);
       },
     );
   }