/*
* All or portions of this file Copyright (c) Amazon.com, Inc. or its affiliates or
* its licensors.
*
* For complete copyright and license terms please see the LICENSE at the root of this
* distribution (the "License"). All use of this software is governed by the License,
* or, if provided, by the license below or the license accompanying this file. Do not
* remove or modify any license notices. This file is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*
*/

#include "InAppPurchasesDelegate.h"
#include "InAppPurchasesApple.h"
#include <InAppPurchases/InAppPurchasesInterface.h>
#include <InAppPurchases/InAppPurchasesResponseBus.h>

#include <CommonCrypto/CommonCrypto.h>

#include <AzCore/std/string/conversions.h>

@implementation InAppPurchasesDelegate

-(NSString*) convertPriceToString:(NSDecimalNumber*) price forLocale:(NSLocale*) locale
{
    NSNumberFormatter* numberFormatter = [[NSNumberFormatter alloc] init];
    [numberFormatter setFormatterBehavior:NSNumberFormatterBehavior10_4];
    [numberFormatter setNumberStyle:NSNumberFormatterCurrencyStyle];
    [numberFormatter setLocale:locale];
    NSString* formattedPrice = [numberFormatter stringFromNumber:price];
    [numberFormatter release];
    return formattedPrice;
}

-(InAppPurchases::PurchasedProductDetailsApple*) parseTransactionDetails:(SKPaymentTransaction*) transaction isRestored:(bool) restored
{
    InAppPurchases::PurchasedProductDetailsApple* purchasedProductDetails = new InAppPurchases::PurchasedProductDetailsApple();
    SKPayment* payment = transaction.payment;
    purchasedProductDetails->SetProductId([payment.productIdentifier UTF8String]);
    if ([payment.applicationUsername length] > 0)
    {
        purchasedProductDetails->SetDeveloperPayload([payment.applicationUsername UTF8String]);
    }
    else
    {
        purchasedProductDetails->SetDeveloperPayload("");
    }
    
    if (restored)
    {
        purchasedProductDetails->SetRestoredOrderId([transaction.transactionIdentifier cStringUsingEncoding:NSASCIIStringEncoding]);
        purchasedProductDetails->SetOrderId([transaction.originalTransaction.transactionIdentifier cStringUsingEncoding:NSASCIIStringEncoding]);
        purchasedProductDetails->SetPurchaseTime([transaction.originalTransaction.transactionDate timeIntervalSince1970]);
        purchasedProductDetails->SetRestoredPurchaseTime([transaction.transactionDate timeIntervalSince1970]);
    }
    else
    {
        purchasedProductDetails->SetOrderId([transaction.transactionIdentifier cStringUsingEncoding:NSASCIIStringEncoding]);
        purchasedProductDetails->SetPurchaseTime([transaction.transactionDate timeIntervalSince1970]);
        purchasedProductDetails->SetRestoredOrderId("");
        purchasedProductDetails->SetRestoredPurchaseTime(0);
    }

    purchasedProductDetails->SetHasDownloads((transaction.downloads != nil) ? true:false);
    
    return purchasedProductDetails;
}

-(NSString*) hashUserName:(NSString*) userName
{
    const int HASH_SIZE = 32;
    unsigned char hashedChars[HASH_SIZE];
    const char* userNameString = [userName UTF8String];
    size_t userNameLength = strlen(userNameString);
    
    if (userNameLength > UINT32_MAX)
    {
        AZ_TracePrintf("LumberyardInAppPurchases", "Username too long to hash:%s", [userName cStringUsingEncoding:NSASCIIStringEncoding]);
        return nil;
    }
    
    CC_SHA256(userNameString, (CC_LONG)userNameLength, hashedChars);
    
    NSMutableString* userNameHash = [[NSMutableString alloc] init];
    
    for (int i =0; i < HASH_SIZE; i++)
    {
        if (i != 0 && i % 4 == 0)
        {
            [userNameHash appendString:@"-"];
        }
        
        [userNameHash appendFormat:@"%02x", hashedChars[i]];
    }
    
    return userNameHash;
}

-(void) initialize
{
    self.m_products = [[NSMutableArray alloc] init];
    self.m_unfinishedTransactions = [[NSMutableArray alloc] init];
    self.m_unfinishedDownloads = [[NSMutableArray alloc] init];
    [[SKPaymentQueue defaultQueue] addTransactionObserver:self];
}

-(void) deinitialize
{
    [self.m_products removeAllObjects];
    [self.m_products release];
    
    [self.m_unfinishedTransactions removeAllObjects];
    [self.m_unfinishedTransactions release];
    
    [self.m_unfinishedDownloads removeAllObjects];
    [self.m_unfinishedDownloads release];
    
    if (self.m_productsRequest != nil)
    {
        [self.m_productsRequest release];
    }
    
    if (self.m_receiptRefreshRequest != nil)
    {
        [self.m_receiptRefreshRequest release];
    }
}

-(void) requestProducts:(NSMutableArray*) productIds
{
    self.m_productsRequest = [[SKProductsRequest alloc] initWithProductIdentifiers:[NSSet setWithArray:productIds]];
    self.m_productsRequest.delegate = self;
    [self.m_productsRequest start];
}

-(void) productsRequest:(SKProductsRequest*) request didReceiveResponse:(SKProductsResponse*) response
{
    for (NSString* invalidId in response.invalidProductIdentifiers)
    {
        AZ_TracePrintf("LumberyardInAppPurchases:", "Invalid product ID:", [invalidId cStringUsingEncoding:NSASCIIStringEncoding]);
    }
    
    InAppPurchases::InAppPurchasesInterface::GetInstance()->GetCache()->ClearCachedProductDetails();
    [self.m_products removeAllObjects];
    
    for (SKProduct* product in response.products)
    {
        InAppPurchases::ProductDetailsApple* productDetails = new InAppPurchases::ProductDetailsApple;
        
        productDetails->SetProductId([product.productIdentifier UTF8String]);
        productDetails->SetProductTitle([product.localizedTitle UTF8String]);
        productDetails->SetProductDescription([product.localizedDescription UTF8String]);
        productDetails->SetProductPrice([[self convertPriceToString:product.price forLocale:product.priceLocale] UTF8String]);
        productDetails->SetProductCurrencyCode([[product.priceLocale objectForKey:NSLocaleCurrencyCode] UTF8String]);
        AZStd::string priceMicro = [[NSString stringWithFormat:@"%@", product.price] UTF8String];
        productDetails->SetProductPriceMicro((AZStd::stof(priceMicro) * 1000000));
        
        InAppPurchases::InAppPurchasesInterface::GetInstance()->GetCache()->AddProductDetailsToCache(productDetails);
        [self.m_products addObject:product];
    }
    
    EBUS_EVENT(InAppPurchases::InAppPurchasesResponseBus, ProductInfoRetrieved, InAppPurchases::InAppPurchasesInterface::GetInstance()->GetCache()->GetCachedProductDetails());
    
    [self.m_productsRequest release];
    self.m_productsRequest = nil;
}

-(void) purchaseProduct:(NSString *) productId withUserName:(NSString *) userName
{
    SKProduct* productToPurchase = nil;
    for (SKProduct* product in self.m_products)
    {
        if ([product.productIdentifier isEqualToString:productId])
        {
            productToPurchase = product;
            break;
        }
    }
    
    if (productToPurchase != nil)
    {
        SKMutablePayment* payment = [SKMutablePayment paymentWithProduct:productToPurchase];
        payment.quantity = 1;
        
        if ([userName length] > 0)
        {
            NSString* userNameHash = [self hashUserName:userName];
            payment.applicationUsername = userNameHash;
        }
        
        [[SKPaymentQueue defaultQueue] addPayment:payment];
    }
    else
    {
        AZ_TracePrintf("LumberyardInAppPurchases", "Invalid product ID:%s", [productId cStringUsingEncoding:NSASCIIStringEncoding]);
    }
}

-(void) paymentQueue:(SKPaymentQueue*) queue updatedTransactions:(nonnull NSArray*) transactions
{
    bool isRestoredTransaction = false;
    
    for (SKPaymentTransaction* transaction in transactions)
    {
        switch (transaction.transactionState)
        {
            case SKPaymentTransactionStatePurchasing:
            {
                AZ_TracePrintf("LumberyardInAppPurchases", "Transaction in progress");
                break;
            }
                
            case SKPaymentTransactionStateDeferred:
            {
                AZ_TracePrintf("LumberyardInAppPurchases", "Transaction deferred");
                break;
            }
                
            case SKPaymentTransactionStateFailed:
            {
                if ([self.m_unfinishedTransactions containsObject:transaction] == false)
                {
                    AZ_TracePrintf("LumberyardInAppPurchases", "Transaction failed! Error: %s", [[transaction.error localizedDescription] cStringUsingEncoding:NSASCIIStringEncoding]);
                    InAppPurchases::PurchasedProductDetailsApple* productDetails = [self parseTransactionDetails:transaction isRestored:false];
                    productDetails->SetPurchaseState(InAppPurchases::PurchaseState::FAILED);
                    [self.m_unfinishedTransactions addObject:transaction];
                    EBUS_EVENT(InAppPurchases::InAppPurchasesResponseBus, PurchaseFailed, productDetails);
                    delete productDetails;
                }
            }
                break;
                
            case SKPaymentTransactionStatePurchased:
            {
                if ([self.m_unfinishedTransactions containsObject:transaction] == false)
                {
                    AZ_TracePrintf("LumberyardInAppPurchases", "Transaction succeeded");
                    InAppPurchases::PurchasedProductDetailsApple* productDetails = [self parseTransactionDetails:transaction isRestored:false];
                    productDetails->SetPurchaseState(InAppPurchases::PurchaseState::PURCHASED);
                    InAppPurchases::InAppPurchasesInterface::GetInstance()->GetCache()->AddPurchasedProductDetailsToCache(productDetails);
                    [self.m_unfinishedTransactions addObject:transaction];
                    for (SKDownload* download in transaction.downloads)
                    {
                        if ([self.m_unfinishedDownloads containsObject:download.contentIdentifier] == false)
                        {
                            [self.m_unfinishedDownloads addObject:download.contentIdentifier];
                        }
                    }
                    EBUS_EVENT(InAppPurchases::InAppPurchasesResponseBus, NewProductPurchased, productDetails);
                }
            }
                break;
                
            case SKPaymentTransactionStateRestored:
            {
                if ([self.m_unfinishedTransactions containsObject:transaction] == false)
                {
                    AZ_TracePrintf("LumberyardInAppPurchases", "Transaction restored");
                    InAppPurchases::PurchasedProductDetailsApple* productDetails = [self parseTransactionDetails:transaction isRestored:true];
                    productDetails->SetPurchaseState(InAppPurchases::PurchaseState::RESTORED);
                    InAppPurchases::InAppPurchasesInterface::GetInstance()->GetCache()->AddPurchasedProductDetailsToCache(productDetails);
                    [self.m_unfinishedTransactions addObject:transaction];
                    for (SKDownload* download in transaction.downloads)
                    {
                        if ([self.m_unfinishedDownloads containsObject:download.contentIdentifier] == false)
                        {
                            [self.m_unfinishedDownloads addObject:download.contentIdentifier];
                        }
                    }
                    isRestoredTransaction = true;
                }
            }
                break;
        }
    }
    
    if (isRestoredTransaction)
    {
        EBUS_EVENT(InAppPurchases::InAppPurchasesResponseBus, PurchasedProductsRestored, InAppPurchases::InAppPurchasesInterface::GetInstance()->GetCache()->GetCachedPurchasedProductDetails());
    }
}

-(void) finishTransaction:(NSString*) transactionId
{
    for (SKPaymentTransaction* transaction in self.m_unfinishedTransactions)
    {
        if ([transaction.transactionIdentifier isEqualToString:transactionId])
        {
            [[SKPaymentQueue defaultQueue] finishTransaction:transaction];
            [self.m_unfinishedTransactions removeObject:transaction];
            return;
        }
    }
    
    AZ_TracePrintf("LumberyardInAppPurchases", "No unfinished transaction found with ID: %s", [transactionId cStringUsingEncoding:NSASCIIStringEncoding]);
}

-(void) downloadAppleHostedContentAndFinishTransaction:(NSString*) transactionId
{
    SKPaymentTransaction* requiredTransaction = nil;
    for (SKPaymentTransaction* transaction in self.m_unfinishedTransactions)
    {
        if ([transaction.transactionIdentifier isEqualToString:transactionId])
        {
            requiredTransaction = transaction;
            break;
        }
    }
    
    if (requiredTransaction != nil)
    {
        if (requiredTransaction.downloads != nil)
        {
            [[SKPaymentQueue defaultQueue] startDownloads:requiredTransaction.downloads];
        }
        else
        {
            [self finishTransaction:requiredTransaction.transactionIdentifier];
        }
    }
    else
    {
        AZ_TracePrintf("LumberyardInAppPurchases", "No unfinished transaction found with ID: %s", [transactionId cStringUsingEncoding:NSASCIIStringEncoding]);
    }
}

-(void) paymentQueue:(SKPaymentQueue*) queue updatedDownloads:(NSArray<SKDownload*>*) downloads
{
    for (SKDownload* download in downloads)
    {
        bool wasDownloadHandled = true;
        for (NSString* contentIdentifier in self.m_unfinishedDownloads)
        {
            if ([contentIdentifier isEqualToString:download.contentIdentifier])
            {
                wasDownloadHandled = false;
                break;
            }
        }
        
        if (wasDownloadHandled)
        {
            continue;
        }

#if (defined(TARGET_OS_OSX) && TARGET_OS_OSX) || (defined(__IPHONE_12_0) && __IPHONE_OS_VERSION_MIN_REQUIRED >= __IPHONE_12_0)
        switch(download.state)
#else
        switch(download.downloadState)
#endif	
        {
            case SKDownloadStateFinished:
            {
                AZStd::string pathToDownloadedContent = [download.contentURL.absoluteString cStringUsingEncoding:NSASCIIStringEncoding];
                AZStd::string transactionId = [download.transaction.transactionIdentifier cStringUsingEncoding:NSASCIIStringEncoding];
                EBUS_EVENT(InAppPurchases::InAppPurchasesResponseBus, HostedContentDownloadComplete, transactionId, pathToDownloadedContent);
                [self.m_unfinishedDownloads removeObject:download.contentIdentifier];
            }
                break;
                
            case SKDownloadStateFailed:
            {
                AZ_TracePrintf("LumberyardInAppPurchases", "Download failed with error: %s", [[download.error localizedDescription] cStringUsingEncoding:NSASCIIStringEncoding]);
                AZStd::string transactionId = [download.transaction.transactionIdentifier cStringUsingEncoding:NSASCIIStringEncoding];
                AZStd::string contentId = [download.contentIdentifier UTF8String];
                EBUS_EVENT(InAppPurchases::InAppPurchasesResponseBus, HostedContentDownloadFailed, transactionId, contentId);
                [self.m_unfinishedDownloads removeObject:download.contentIdentifier];
            }
                break;
                
            case SKDownloadStateCancelled:
            {
                AZ_TracePrintf("LumberyardInAppPurchases", "Download cancelled: %s", [download.contentIdentifier cStringUsingEncoding:NSASCIIStringEncoding]);
                [self.m_unfinishedDownloads removeObject:download.contentIdentifier];
            }
                break;
                
            case SKDownloadStateActive:
            case SKDownloadStateWaiting:
            case SKDownloadStatePaused:
                break;
        }
    }
    
    if ([self.m_unfinishedDownloads count] == 0)
    {
        [self finishTransaction:[downloads objectAtIndex:0].transaction.transactionIdentifier];
    }
}

-(void) restorePurchasedProducts
{
    [[SKPaymentQueue defaultQueue] restoreCompletedTransactions];
}

-(void) refreshAppReceipt
{
    self.m_receiptRefreshRequest = [[SKReceiptRefreshRequest alloc] init];
    self.m_receiptRefreshRequest.delegate = self;
    [self.m_receiptRefreshRequest start];
}

-(void) request:(SKRequest*) request didFailWithError:(NSError*) error
{
    AZ_TracePrintf("LumberyardInAppPurchases", "Request failed with error: %s", [[error localizedDescription] cStringUsingEncoding:NSASCIIStringEncoding]);
}

-(void) requestDidFinish:(SKRequest *)request
{
    if ([request isKindOfClass:[SKReceiptRefreshRequest class]])
    {
        [self.m_receiptRefreshRequest release];
        self.m_receiptRefreshRequest = nil;
    }
}

@end