diff --git a/keyBoard/Class/Pay/VC/KBJfPay.m b/keyBoard/Class/Pay/VC/KBJfPay.m index 3241ad0..616d92a 100644 --- a/keyBoard/Class/Pay/VC/KBJfPay.m +++ b/keyBoard/Class/Pay/VC/KBJfPay.m @@ -9,6 +9,7 @@ #import "PayVM.h" #import "KBPayProductModel.h" #import "KBBizCode.h" +#import "KBShopVM.h" static NSString * const kKBJfPayCellId = @"kKBJfPayCellId"; @interface KBJfPay () @@ -40,6 +41,7 @@ static NSString * const kKBJfPayCellId = @"kKBJfPayCellId"; @property (nonatomic, strong) FGIAPProductsFilter *filter; @property (nonatomic, strong) PayVM *payVM; +@property (nonatomic, strong) KBShopVM *shopVM; @end @@ -49,6 +51,7 @@ static NSString * const kKBJfPayCellId = @"kKBJfPayCellId"; [super viewDidLoad]; self.filter = [[FGIAPProductsFilter alloc] init]; self.payVM = [PayVM new]; + self.shopVM = [KBShopVM new]; self.data = @[]; self.selectedIndex = NSNotFound; self.bgImageView = [[UIImageView alloc] initWithImage:[UIImage imageNamed:@"my_bg_icon"]]; @@ -138,6 +141,11 @@ static NSString * const kKBJfPayCellId = @"kKBJfPayCellId"; [self fetchInAppProductList]; } +- (void)viewWillAppear:(BOOL)animated { + [super viewWillAppear:animated]; + [self fetchWalletBalance]; +} + - (void)viewDidAppear:(BOOL)animated { [super viewDidAppear:animated]; [self selectItemAtCurrentIndexAnimated:NO]; @@ -217,6 +225,23 @@ static NSString * const kKBJfPayCellId = @"kKBJfPayCellId"; } #pragma mark - Data +- (void)fetchWalletBalance { + __weak typeof(self) weakSelf = self; + [self.shopVM fetchWalletBalanceWithCompletion:^(NSString * _Nullable balance, NSError * _Nullable error) { + dispatch_async(dispatch_get_main_queue(), ^{ + __strong typeof(weakSelf) self = weakSelf; + if (!self) { return; } + if (error) { + NSString *msg = error.localizedDescription ?: KBLocalized(@"Network error"); + [KBHUD showInfo:msg]; + return; + } + NSString *displayValue = balance.length ? balance : @"0"; + self.pointsLabel.text = displayValue; + }); + }]; +} + - (void)fetchInAppProductList { __weak typeof(self) weakSelf = self; [self.payVM fetchInAppProductsNeedShow:YES completion:^(NSInteger sta, NSString * _Nullable msg, NSArray * _Nullable products) { @@ -320,7 +345,7 @@ static NSString * const kKBJfPayCellId = @"kKBJfPayCellId"; - (UILabel *)pointsLabel { if (!_pointsLabel) { _pointsLabel = [UILabel new]; - _pointsLabel.text = @"4230"; // 示例值 + _pointsLabel.text = @"--"; _pointsLabel.font = [KBFont bold:30]; _pointsLabel.textColor = [UIColor colorWithHex:KBColorValue]; }