2

keyBoard/Class/Shop/VM/KBShopVM.h

@@ -18,6 +18,8 @@ typedef void(^KBShopStylesCompletion)(NSArray<KBShopStyleModel *> *_Nullable sty
                                       NSError *_Nullable error);
 typedef void(^KBShopThemesCompletion)(NSArray<KBShopThemeModel *> *_Nullable themes,
                                       NSError *_Nullable error);
+typedef void(^KBShopBalanceCompletion)(NSNumber *_Nullable balance,
+                                       NSError *_Nullable error);
 
 @interface KBShopVM : NSObject
 @property (nonatomic, copy, readonly, nullable) NSArray<KBShopStyleModel *> *styles;
@@ -29,6 +31,9 @@ typedef void(^KBShopThemesCompletion)(NSArray<KBShopThemeModel *> *_Nullable the
 - (void)fetchThemesForStyleId:(nullable NSString *)styleId
                    completion:(KBShopThemesCompletion)completion;
 
+/// 查询钱包余额
+- (void)fetchWalletBalanceWithCompletion:(KBShopBalanceCompletion)completion;
+
 @end
 
 NS_ASSUME_NONNULL_END

keyBoard/Class/Shop/VM/KBShopVM.m

@@ -82,4 +82,35 @@
     }];
 }
 
+- (void)fetchWalletBalanceWithCompletion:(KBShopBalanceCompletion)completion {
+    [[KBNetworkManager shared] GET:API_WALLET_BALANCE
+                        parameters:nil
+                           headers:nil
+              autoShowBusinessError:NO
+                        completion:^(NSDictionary * _Nullable json,
+                                     NSURLResponse * _Nullable response,
+                                     NSError * _Nullable error) {
+        if (error) {
+            if (completion) completion(nil, error);
+            return;
+        }
+        id dataObj = json[KBData] ?: json[@"data"];
+        if (![dataObj isKindOfClass:[NSDictionary class]]) {
+            if (completion) completion(nil, [self kb_invalidResponseError]);
+            return;
+        }
+        id balanceValue = dataObj[@"balance"];
+        NSNumber *balanceNumber = nil;
+        if ([balanceValue isKindOfClass:[NSNumber class]]) {
+            balanceNumber = balanceValue;
+        } else if ([balanceValue isKindOfClass:[NSString class]]) {
+            balanceNumber = @([(NSString *)balanceValue doubleValue]);
+        }
+        if (!balanceNumber) {
+            balanceNumber = @(0);
+        }
+        if (completion) completion(balanceNumber, nil);
+    }];
+}
+
 @end