tasks/src/googleplay/java/org/tasks/billing/PurchaseHelper.java

package org.tasks.billing;

import android.app.Activity;
import android.content.Context;
import android.content.DialogInterface;
import android.content.Intent;
import android.widget.Toast;

import com.android.vending.billing.IabHelper;
import com.android.vending.billing.IabResult;
import com.android.vending.billing.Purchase;
import com.google.common.base.Strings;

import org.tasks.Broadcaster;
import org.tasks.BuildConfig;
import org.tasks.R;
import org.tasks.analytics.Tracker;
import org.tasks.dialogs.DialogBuilder;
import org.tasks.injection.ForApplication;
import org.tasks.preferences.Preferences;

import java.util.ArrayList;
import java.util.List;

import javax.inject.Inject;
import javax.inject.Singleton;

import timber.log.Timber;

import static com.todoroo.andlib.utility.AndroidUtilities.isAppInstalled;

@Singleton
public class PurchaseHelper implements IabHelper.OnIabSetupFinishedListener {

    private final Context context;
    private final Preferences preferences;
    private final Tracker tracker;
    private final Broadcaster broadcaster;
    private final InventoryHelper inventory;

    private PurchaseHelperCallback activityResultCallback;
    private IabHelper iabHelper;

    @Inject
    public PurchaseHelper(@ForApplication Context context, Preferences preferences, Tracker tracker,
                          Broadcaster broadcaster, InventoryHelper inventory) {
        Timber.d("Injecting new PurchaseHelper");
        this.context = context;
        this.preferences = preferences;
        this.tracker = tracker;
        this.broadcaster = broadcaster;
        this.inventory = inventory;
    }

    @Override
    public void onIabSetupFinished(IabResult result) {
        if (result.isFailure()) {
            Timber.e("in-app billing setup failed: %s", result.getMessage());
        }
    }

    public boolean purchase(DialogBuilder dialogBuilder, final Activity activity, final String sku, final String pref, final int requestCode, final PurchaseHelperCallback callback) {
        if (activity.getString(R.string.sku_tasker).equals(sku) && isAppInstalled(activity, "org.tasks.locale")) {
            dialogBuilder.newMessageDialog(R.string.tasker_message)
                    .setCancelable(false)
                    .setPositiveButton(R.string.buy, new DialogInterface.OnClickListener() {
                        @Override
                        public void onClick(DialogInterface dialog, int which) {
                            launchPurchaseFlow(activity, sku, pref, requestCode, callback);
                        }
                    })
                    .setNegativeButton(android.R.string.cancel, new DialogInterface.OnClickListener() {
                        @Override
                        public void onClick(DialogInterface dialog, int which) {
                            callback.purchaseCompleted(false, sku);
                        }
                    })
                    .show();
            return false;
        } else {
            launchPurchaseFlow(activity, sku, pref, requestCode, callback);
            return true;
        }
    }

    public void consumePurchases() {
        if (BuildConfig.DEBUG) {
            final List<Purchase> purchases = new ArrayList<>();
            final Purchase tasker = inventory.getPurchase(context.getString(R.string.sku_tasker));
            final Purchase dashclock = inventory.getPurchase(context.getString(R.string.sku_dashclock));
            final Purchase teslaUnread = inventory.getPurchase(context.getString(R.string.sku_tesla_unread));
            if (tasker != null) {
                purchases.add(tasker);
            }
            if (dashclock != null) {
                purchases.add(dashclock);
            }
            if (teslaUnread != null) {
                purchases.add(teslaUnread);
            }
            final IabHelper iabHelper = new IabHelper(context, context.getString(R.string.gp_key));
            iabHelper.enableDebugLogging(true);
            iabHelper.startSetup(new IabHelper.OnIabSetupFinishedListener() {
                @Override
                public void onIabSetupFinished(IabResult result) {
                    if (result.isSuccess()) {
                        iabHelper.consumeAsync(purchases, new IabHelper.OnConsumeMultiFinishedListener() {
                            @Override
                            public void onConsumeMultiFinished(List<Purchase> purchases, List<IabResult> results) {
                                for (int i = 0 ; i < purchases.size() ; i++) {
                                    Purchase purchase = purchases.get(i);
                                    IabResult iabResult = results.get(i);
                                    if (iabResult.isSuccess()) {
                                        if (purchase.equals(tasker)) {
                                            preferences.setBoolean(R.string.p_purchased_tasker, false);
                                        } else if (purchase.equals(dashclock)) {
                                            preferences.setBoolean(R.string.p_purchased_dashclock, false);
                                        } else if (purchase.equals(teslaUnread)) {
                                            preferences.setBoolean(R.string.p_purchased_tesla_unread, false);
                                            preferences.setBoolean(R.string.p_tesla_unread_enabled, false);
                                        } else {
                                            Timber.e("Unhandled consumption for purchase: %s", purchase);
                                        }
                                        inventory.erasePurchase(purchase.getSku());
                                        Timber.d("Consumed %s", purchase);
                                    } else {
                                        Timber.e("Consume failed: %s, %s", purchase, iabResult);
                                    }
                                }
                                iabHelper.dispose();
                            }
                        });
                    } else {
                        Timber.e("setup failed: %s", result.getMessage());
                        iabHelper.dispose();
                    }
                }
            });
        }
    }

    private void launchPurchaseFlow(final Activity activity, final String sku, final String pref, final int requestCode, final PurchaseHelperCallback callback) {
        if (iabHelper != null) {
            Toast.makeText(activity, R.string.billing_service_busy, Toast.LENGTH_LONG).show();
            callback.purchaseCompleted(false, sku);
            return;
        }
        iabHelper = new IabHelper(context, context.getString(R.string.gp_key));
        iabHelper.enableDebugLogging(BuildConfig.DEBUG);
        Timber.d("%s: startSetup", iabHelper);
        iabHelper.startSetup(new IabHelper.OnIabSetupFinishedListener() {
            @Override
            public void onIabSetupFinished(IabResult result) {
                if (result.isSuccess()) {
                    try {
                        Timber.d("%s: launchPurchaseFlow for %s", iabHelper, sku);
                        iabHelper.launchPurchaseFlow(activity, sku, requestCode, new IabHelper.OnIabPurchaseFinishedListener() {
                            @Override
                            public void onIabPurchaseFinished(IabResult result, Purchase info) {
                                Timber.d(result.toString());
                                tracker.reportIabResult(result, info);
                                if (result.isSuccess()) {
                                    if (!Strings.isNullOrEmpty(pref)) {
                                        preferences.setBoolean(pref, true);
                                        broadcaster.refresh();
                                    }
                                    inventory.refreshInventory();
                                } else if (result.getResponse() != IabHelper.BILLING_RESPONSE_RESULT_USER_CANCELED &&
                                        result.getResponse() != IabHelper.IABHELPER_USER_CANCELLED) {
                                    Toast.makeText(activity, result.getMessage(), Toast.LENGTH_LONG).show();
                                }
                                activityResultCallback.purchaseCompleted(result.isSuccess(), sku);
                                disposeIabHelper();
                            }
                        });
                    } catch (IllegalStateException e) {
                        tracker.reportException(e);
                        Toast.makeText(activity, R.string.billing_service_busy, Toast.LENGTH_LONG).show();
                        callback.purchaseCompleted(false, sku);
                        disposeIabHelper();
                    }
                } else {
                    Timber.e(result.toString());
                    Toast.makeText(activity, result.getMessage(), Toast.LENGTH_LONG).show();
                    callback.purchaseCompleted(false, sku);
                    disposeIabHelper();
                }
            }
        });
    }

    public void disposeIabHelper() {
        if (iabHelper != null) {
            Timber.d("%s: dispose", iabHelper);
            iabHelper.dispose();
            iabHelper = null;
        }
    }

    public void handleActivityResult(PurchaseHelperCallback callback, int requestCode, int resultCode, Intent data) {
        this.activityResultCallback = callback;

        if (iabHelper != null) {
            iabHelper.handleActivityResult(requestCode, resultCode, data);
        }
    }
}