IWT - in-app purchase code

//
//  iWTIAPHandler.h
//  NeoDefender
//
//  Created by An Nguyen on 28/05/2010.
//  Copyright 2010 Xtremics. All rights reserved.
//

#import <Foundation/Foundation.h>
#import <StoreKit/StoreKit.h>


@interface iWTIAPHandler : NSObject <SKProductsRequestDelegate, SKPaymentTransactionObserver> {

}

@end

//
//  iWTIAPHandler.m
//  DeoDefender
//
//  Created by An Nguyen on 28/05/2010.
//  Copyright 2010 Xtremics. All rights reserved.
//

#import "iWTIAPHandler.h"
#import "Reachability.h"
#include "console/console.h"

//#define FULL_FEATURE_IDENTIFIER	@"com.meyume.iwtusa.20000"

@implementation iWTIAPHandler

- (void) initIAP {
	NSLog(@"%s", _cmd);
	[[SKPaymentQueue defaultQueue] addTransactionObserver:self];
}

- (void) requestProductData:(NSString*)productIdentifier {
	NSLog(@"%s", _cmd);
	SKProductsRequest *request = [[SKProductsRequest alloc] initWithProductIdentifiers:[NSSet setWithObject:productIdentifier]];
	request.delegate = self;
	[request start];
}

- (BOOL) checkProduct:(NSString*)identifier {
	NSNumber *iap = [[NSUserDefaults standardUserDefaults] objectForKey:identifier];
	return [iap boolValue];
}

- (BOOL) fullFeaturesEnabled:(NSString*)productIdentifier {
	BOOL purchased = [self checkProduct:productIdentifier];
	
	return purchased;
}

- (void) recordTransaction:(SKPaymentTransaction*)transaction {
	//[[NSUserDefaults standardUserDefaults] setObject:[NSNumber numberWithBool:YES] forKey:transaction.originalTransaction.payment.productIdentifier];
}

- (void) provideContent:(NSString*)identifier {
	[[NSUserDefaults standardUserDefaults] setObject:[NSNumber numberWithBool:YES] forKey:identifier];
	Con::executef(2, "oniWTIAPBoughtProduct", [identifier cStringUsingEncoding:NSUTF8StringEncoding]);
}

- (void) completeTransaction:(SKPaymentTransaction*)transaction {
	[self recordTransaction:transaction];
	[self provideContent:transaction.payment.productIdentifier];
	[[SKPaymentQueue defaultQueue] finishTransaction:transaction];
}

- (void) restoreTransaction:(SKPaymentTransaction*)transaction {
	[self recordTransaction:transaction];
	[self provideContent:transaction.originalTransaction.payment.productIdentifier];
	[[SKPaymentQueue defaultQueue] finishTransaction:transaction];
}

- (void) failedTransaction:(SKPaymentTransaction*)transaction {
	if (transaction.error.code != SKErrorPaymentCancelled) {
		// display error
		Con::executef(1, "oniWTIAPFailedTransaction");
	}
	[[SKPaymentQueue defaultQueue] finishTransaction:transaction];
}

- (void) buyProduct:(NSString*)identifier {
	SKPayment *payment = [SKPayment paymentWithProductIdentifier:identifier];
	[[SKPaymentQueue defaultQueue] addPayment:payment];
}


#pragma mark -
#pragma mark SKProductsRequestDelete methods

- (void)productsRequest:(SKProductsRequest *)request didReceiveResponse:(SKProductsResponse *)response {
	NSLog(@"%s", _cmd);
	NSArray *products = response.products;
	NSLog(@"---- found [%d] products", [products count]);
	int count = [products count];
	NSString *param = @"";
	for (SKProduct *product in products) {
		NSNumberFormatter *numberFormatter = [[NSNumberFormatter alloc] init];
		[numberFormatter setFormatterBehavior:NSNumberFormatterBehavior10_4];
		[numberFormatter setNumberStyle:NSNumberFormatterCurrencyStyle];
		[numberFormatter setLocale:product.priceLocale];
		NSString *formattedString = [numberFormatter stringFromNumber:product.price];
		NSLog(@"------- [%@], [%@], [%@], [%@]", [product localizedTitle], [product localizedDescription], [product productIdentifier], formattedString);
		
		param = [param stringByAppendingFormat:@"%@\t%@\t%@\t%@", [product productIdentifier], [product localizedTitle], [product localizedDescription], formattedString];
	}
	char param2[3];
	param2[0] = '[[628778f975f0e]]';
	snprintf(param2, sizeof(param2), "%d", count);
	Con::executef(3, "oniWTIAPDidReceiveProducts", [param cStringUsingEncoding:NSUTF8StringEncoding], param2);
}

#pragma mark -
#pragma mark SKPaymentTransactionObserver methods

- (void)paymentQueue:(SKPaymentQueue *)queue updatedTransactions:(NSArray *)transactions {
	for (SKPaymentTransaction *transaction in transactions) {
		switch (transaction.transactionState) {
			case SKPaymentTransactionStatePurchased:
				[self completeTransaction:transaction];
				break;
			case SKPaymentTransactionStateRestored:
				[self restoreTransaction:transaction];
				break;
			case SKPaymentTransactionStateFailed:
				[self failedTransaction:transaction];
				break;
			default:
				break;
		}
	}
}

- (void)paymentQueue:(SKPaymentQueue *)queue removedTransactions:(NSArray *)transactions {
	Con::executef(1, "oniWTIAPRemovedTransactions");
}

@end


#pragma mark -
#pragma mark C++

iWTIAPHandler *iapHandler;

void initIAPHandler() {
	if (!iapHandler) {
		iapHandler = [[[iWTIAPHandler alloc] init] retain];
		[iapHandler initIAP];
	}
}

ConsoleFunction(iWTIAPInit, void, 1, 1, "Init the iWTIAPHandler") {
	initIAPHandler();
}

ConsoleFunction(iWTIAPRequestProducts, void, 2, 2, "Request the products from server") {
	initIAPHandler();
	[iapHandler requestProductData:[NSString stringWithCString:argv[1]]];
}

ConsoleFunction(iWTIAPBuyProduct, void, 2, 2, "Buy a specified product") {
	initIAPHandler();
	[iapHandler buyProduct:[NSString stringWithCString:argv[1]]];
}

ConsoleFunction(iWTIAPFullFeaturesEnabled, bool, 2, 2, "Check if the full game game features has purchased") {
	initIAPHandler();
	return [iapHandler fullFeaturesEnabled:[NSString stringWithCString:argv[1]]];
}

ConsoleFunction(iWTIAPInternetConnectionStatus, bool, 1, 1, "Returns true if the internet connection status is available") {
	NetworkStatus status = [[Reachability sharedReachability] internetConnectionStatus];
	if (status == ReachableViaWiFiNetwork || status == ReachableViaCarrierDataNetwork) {
		return true;
	}
	return false;
}

//--------------------------------------------------------------------------
// iWT In App Purchase
//--------------------------------------------------------------------------

$IAP::fullFeaturesUnlocked = false;

// Initialize the IWT IAP and load the features that have been unlocked.
function iWTIAP::init(%this) {
	iWTIAPInit();
	
	if ($GAMECONFIG::liteVersion) {
		$IAP::fullFeaturesUnlocked = iWTIAPFullFeaturesEnabled($iWT::fullGameIdentifier);
		$GAMECONFIG::liteVersion = !$IAP::fullFeaturesUnlocked;
	}
}

// Returns true if we should show the IAP button
function iWTIAP::showIAPButton(%this) {
	return !$IAP::fullFeaturesUnlocked;
}

// Sends request to Store for products details.
function iWTIAP::requestProducts(%this) {
	if (!iWTIAPInternetConnectionStatus()) {
		networkErrorPanel.visible = true;
		IAPGamePanel.visible = false;
		
		return;
	}
	
	// the flag to indicate that IAP is launched by user, in case
	// the payment resumed from previous transaction and we are not
	// at the IAP screen, then this flag is used for determining 
	// not return to menu.
	$iap.requested = true;
	
	iWTIAPRequestProducts($iWT::fullGameIdentifier);
	IAPBuyButton.setActive(false);
	IAPCancelButton.setActive(true);
	IAPStatusText.visible = true;
	//IAPStatusText.setText("<just:center>Connecting to store, please wait.\n<color:FFFF00>Connecting...");
	IAPText.setText("<just:center><color:FFFFFF>Get the FULL version of\nInvaders World Tour!\n\n<color:09EAF6>Price: <color:FFFF00>(Requesting...)");
	
	//$iap.flashingId = %this.schedule(200, "flashConnectingText");
	//$iap.flashingFlag = false;
}
function iWTIAP::flashConnectingText(%this) {
	$iap.flashingFlag = !$iap.flashingFlag;
	%color = "<color:09EAF6>";
	if ($iap.flashingFlag) {
		%color = "<color:FFFF00>";
	}
	
	IAPStatusText.setText("<just:center>Connecting to store, please wait.\n" @ %color @ "Connecting...");
	
	$iap.flashingId = %this.schedule(200, "flashConnectingText");
}

function iWTIAP::cancelRequestProducts(%this) {
	cancel($iap.flashingId);
}

function iWTIAP::payBuyFullGame(%this) {
	IAPBuyButton.setActive(false);
	IAPCancelButton.setActive(false);
	iWTIAPBuyProduct($iap.fullGameProduct.id);
	
	IAPStatusText.visible = true;
	//IAPStatusText.setText("<just:center>Connecting to store, please wait...");
}


//--------------------------------------------------------------------------
// iWT in-app purchase - callback
//--------------------------------------------------------------------------

// Is called when received response from IAP store
function oniWTIAPDidReceiveProducts(%response, %count) {
	cancel($iap.flashingId);
	
	if (%count == 1) {
		$iap.fullGameProduct = new ScriptObject();
		$iap.fullGameProduct.id = getField(%response, 0);
		$iap.fullGameProduct.title = getField(%response, 1);
		$iap.fullGameProduct.desc = getField(%response, 2);
		$iap.fullGameProduct.price = getField(%response, 3);
		
		IAPBuyButton.setActive(true);
		IAPStatusText.visible = false;
		IAPText.setText("<just:center><color:FFFFFF>GET THE FULL VERSION OF INVADERS WORLD TOUR!\n<color:00FF00>Price: <color:FFFF00>" @ $iap.fullGameProduct.price @ "<color:00FF00>.");
	}
	
	
}

// Is called when succeed a transaction.
function oniWTIAPBoughtProduct(%id) {
	if (%id $= $iap.fullGameProduct.id) {
		// unlock the game features
		$IAP::fullFeaturesUnlocked = true;
		$GAMECONFIG::liteVersion = false;
		// resume to main menu
		$mainMenu.hideIAPPanel();
		// hide the buy buttons
		buttonBuy.visible = false;
		buyHorizontalDivider.visible = false;
		
		// FIX ME - resume/continue the game
	}
}

// Is called when one or more transactions removed from queue.
function oniWTIAPRemovedTransactions(%this) {
	IAPBuyButton.setActive(true);
	IAPCancelButton.setActive(true);
	IAPStatusText.visible = false;
}