whose_ios/NIMDemo/NIMDemo/RechargeViewController.m

//
//  RechargeViewController.m
//  NIMDemo
//
//  Created by Fenix Wang on 2017/6/13.
//  Copyright © 2017年 Netease. All rights reserved.
//

#import "RechargeViewController.h"
#import "RechargeItem.h"
#import "UIView+NIM.h"
#import "User.h"
#import "PayOptionViewController.h"
#import <StoreKit/StoreKit.h>
#import <Foundation/Foundation.h>
#import "SVProgressHUD.h"
#import "HttpRequest.h"
#import "UIView+Toast.h"

@interface RechargeViewController ()<RechargeItemTouchDelegate>

@property (nonatomic, strong) NSMutableArray *datas;
@property (nonatomic, strong) NSDictionary *currentData;
@property (nonatomic, strong) NSMutableArray *itemViews;

@property (nonatomic, strong) NSArray<SKProduct *> *products;
@property (nonatomic, strong) NSString *readyToPayCode;

@end

@implementation RechargeViewController

- (void)viewDidLoad {
    [super viewDidLoad];
    
    self.view.backgroundColor = UIColor.whiteColor;
    
    _datas = [[NSMutableArray alloc] init];
    [_datas addObject:@{@"id":@"1", @"title":@"28元", @"desc":@"28说币", @"num":@"28", @"code":@"com.shuishuo.app.cny28"}];
    [_datas addObject:@{@"id":@"2", @"title":@"68元", @"desc":@"68说币", @"num":@"68", @"code":@"com.shuishuo.app.cny68"}];
    [_datas addObject:@{@"id":@"3", @"title":@"98元", @"desc":@"98说币", @"num":@"98", @"code":@"com.shuishuo.app.cny98"}];
    [_datas addObject:@{@"id":@"4", @"title":@"188元", @"desc":@"188说币", @"num":@"188", @"code":@"com.shuishuo.app.cny188"}];
    [_datas addObject:@{@"id":@"5", @"title":@"288元", @"desc":@"288说币", @"num":@"288", @"code":@"com.shuishuo.app.cny288"}];
    [_datas addObject:@{@"id":@"6", @"title":@"388元", @"desc":@"388说币", @"num":@"388", @"code":@"com.shuishuo.app.cny388"}];
    
    
    _itemViews = [[NSMutableArray alloc] init];
    CGFloat startX = 10;
    CGFloat startY = 20;
    CGFloat padding = 10;
    CGFloat width = (self.view.nim_width - padding * 4) / 3;
    CGFloat height = 80;
    int columns = 3;
    for(int i=0; i<_datas.count; i++)
    {
        NSDictionary *data = [_datas objectAtIndex:i];
        startX = (i % columns) * (width + padding) + padding;
        startY = (i / columns) * (height + padding) + padding + 74;
        RechargeItem *item = [RechargeItem itemWithData:data size:CGSizeMake(width, height) delegate:self];
        item.frame = CGRectMake(startX, startY, width, height);
        [_itemViews addObject:item];
        [self.view addSubview:item];
    }
    
    UIButton *rechargeBtn = [UIButton buttonWithType:UIButtonTypeCustom];
    rechargeBtn.frame = CGRectMake(10, 300, self.view.nim_width-20, 50);
    rechargeBtn.backgroundColor = [UIColor colorWithRed:(255.0 / 255.0) green:(171.0 / 255.0) blue:(8.0 / 255.0) alpha:1.0];
    rechargeBtn.layer.cornerRadius = 4.0;
    [rechargeBtn setTitle:@"立即充值" forState:UIControlStateNormal];
    [rechargeBtn addTarget:self action:@selector(doRecharge:) forControlEvents:UIControlEventTouchUpInside];
    [self.view addSubview:rechargeBtn];
    
    [self selectedItem:[_datas objectAtIndex:0]];
    
    
    [[SKPaymentQueue defaultQueue] addTransactionObserver:(id)self];
    [self fetchProducts];
}


-(void)fetchProducts{
    
    NSMutableArray *idArr = [[NSMutableArray alloc] init];
    for(NSDictionary *item in _datas)
    {
        NSString *code = [item objectForKey:@"code"];
        [idArr addObject:code];
    }
    
    SKProductsRequest *request = [[SKProductsRequest alloc]initWithProductIdentifiers:[NSSet setWithArray:idArr]];
    request.delegate = (id)self;
    [request start];
    
}


- (void)didReceiveMemoryWarning {
    [super didReceiveMemoryWarning];
    // Dispose of any resources that can be recreated.
}

- (void)viewWillAppear:(BOOL)animated{
    [super viewWillAppear:animated];
    [self.navigationItem setTitle:@"充值"];
}

- (void)selectedItem:(NSDictionary *)data
{
    _currentData = data;
    for(int i=0; i<_itemViews.count; i++){
        RechargeItem *item = [_itemViews objectAtIndex:i];
        BOOL selected = [[item.data objectForKey:@"id"] isEqual:[data objectForKey:@"id"]];
        [item setSelected:selected];
    }
}

- (void)doRecharge:(int)sender
{
//    UIStoryboard *board = [UIStoryboard storyboardWithName:@"Main" bundle:nil];
//    PayOptionViewController *payOptionVC = [board instantiateViewControllerWithIdentifier:@"PayOption"];
//    int money = [[_currentData objectForKey:@"num"] intValue];
//    [payOptionVC setMoney:money];
//    [self.navigationController pushViewController:payOptionVC animated:YES];
    
    [SVProgressHUD show];
    
    NSString *pid = [_currentData objectForKey:@"code"];
    
    if(_products && _products.count > 0)
    {
        for (SKProduct *product in _products)
        {
            NSLog(@"Check product %@ %@", pid, product.productIdentifier);
            if([product.productIdentifier isEqualToString:pid])
            {
                SKPayment *payment = [SKPayment paymentWithProduct:product];
                [[SKPaymentQueue defaultQueue]addPayment:payment];
                NSLog(@"Buy %@", product.productIdentifier);
                break;
            }
        }
        _readyToPayCode = nil;
    }
    else
    {
        _readyToPayCode = pid;
    }
}

-(void)productsRequest:(SKProductsRequest *)request didReceiveResponse:(nonnull SKProductsResponse *)response{
    
    _products = response.products;
    for (SKProduct *product in _products) {
        NSLog(@"Product %@ %@", product.productIdentifier, product.price.stringValue);
    }
    
    if(_readyToPayCode)
    {
        [self doRecharge:nil];
    }
}

-(void)request:(SKRequest *)request didFailWithError:(nonnull NSError *)error{
    
    NSLog(@"%@", error);
    
}

-(void)paymentQueue:(SKPaymentQueue *)queue updatedTransactions:(nonnull NSArray<SKPaymentTransaction *> *)transactions{
    
    for (SKPaymentTransaction *tx in transactions) {
        switch (tx.transactionState) {
            case SKPaymentTransactionStatePurchased:
                [[SKPaymentQueue defaultQueue]finishTransaction:tx];
                [self paymentSuccess:tx];
                break;
                
            case SKPaymentTransactionStateFailed:
                [[SKPaymentQueue defaultQueue]finishTransaction:tx];
                NSLog(@"Error: %@", tx.error);
                [self paymentFailed:tx];
                break;
                
            case SKPaymentTransactionStateRestored:
                [[SKPaymentQueue defaultQueue]finishTransaction:tx];
                NSLog(@"SKPaymentTransactionStateRestored");
                break;
            default:
                break;
        }
    }
    
}



-(void)paymentSuccess:(SKPaymentTransaction *)tx{
    
    [SVProgressHUD dismiss];
    
    NSData *receiptData;
    if (NSFoundationVersionNumber >= NSFoundationVersionNumber_iOS_7_0) {
        NSLog(@"ReceiptData later iOS 7");
        receiptData = [NSData dataWithContentsOfURL:[[NSBundle mainBundle] appStoreReceiptURL]];
    } else {
        NSLog(@"ReceiptData earlier iOS 7");
        receiptData = tx.transactionReceipt;
    }
    
    NSString *itemId = tx.payment.productIdentifier;
    NSString *receiptStr = [receiptData base64EncodedStringWithOptions:0];
    NSLog(@"SKPaymentTransactionStatePurchased %@", receiptStr);
    
    [[HttpRequest shared] iosPayment:itemId receipt:receiptStr success:^{
        [self.view makeToast:@"支付成功" duration:2.0 position:CSToastPositionCenter];
    } failure:^{
        [self.view makeToast:@"支付失败" duration:2.0 position:CSToastPositionCenter];
    }];
    
}

-(void)paymentFailed:(SKPaymentTransaction *)tx{
    
    [SVProgressHUD dismiss];
    [self.view makeToast:@"支付失败" duration:2.0 position:CSToastPositionCenter];
}

@end