Support iOS 17 setBadgeCount method (starting from iOS 16) (g123k#87)

Author: g-30

ios/Classes/FlutterAppBadgerPlugin.m

@@ -1,4 +1,5 @@
 #import "FlutterAppBadgerPlugin.h"
+#import <UserNotifications/UserNotifications.h>
 
 @implementation FlutterAppBadgerPlugin
 + (void)registerWithRegistrar:(NSObject<FlutterPluginRegistrar>*)registrar {
@@ -21,20 +22,28 @@ - (void)enableNotifications {
 
 - (void)handleMethodCall:(FlutterMethodCall*)call result:(FlutterResult)result {
     [self enableNotifications];
-    
-  if ([@"updateBadgeCount" isEqualToString:call.method]) {
-      NSDictionary *args = call.arguments;
-      NSNumber *count = [args objectForKey:@"count"];
-      [UIApplication sharedApplication].applicationIconBadgeNumber = count.integerValue;
-    result(nil);
-  } else if ([@"removeBadge" isEqualToString:call.method]) {
-      [UIApplication sharedApplication].applicationIconBadgeNumber = 0;
-      result(nil);
-  } else if ([@"isAppBadgeSupported" isEqualToString:call.method]) {
-      result(@YES);
-  } else {
-    result(FlutterMethodNotImplemented);
-  }
+        
+    if ([@"updateBadgeCount" isEqualToString:call.method]) {
+        NSDictionary *args = call.arguments;
+        NSNumber *count = [args objectForKey:@"count"];
+        if (@available(iOS 16, *)) {
+            [[UNUserNotificationCenter currentNotificationCenter] setBadgeCount:count.integerValue withCompletionHandler:^(NSError * _Nullable error) {}];
+        } else {
+            [UIApplication sharedApplication].applicationIconBadgeNumber = count.integerValue;
+        }
+        result(nil);
+    } else if ([@"removeBadge" isEqualToString:call.method]) {
+        if (@available(iOS 16, *)) {
+            [[UNUserNotificationCenter currentNotificationCenter] setBadgeCount:0 withCompletionHandler:^(NSError * _Nullable error) {}];
+        } else {
+            [UIApplication sharedApplication].applicationIconBadgeNumber = 0;
+        }
+        result(nil);
+    } else if ([@"isAppBadgeSupported" isEqualToString:call.method]) {
+        result(@YES);
+    } else {
+        result(FlutterMethodNotImplemented);
+    }
 }
 
 @end