Threema-Android-mirror/app/src/main/java/ch/threema/app/routines/CheckLicenseRoutine.java

package ch.threema.app.routines;

import android.app.Activity;
import android.content.Context;
import android.content.ReceiverCallNotAllowedException;

import org.slf4j.Logger;

import java.lang.ref.WeakReference;

import androidx.annotation.NonNull;
import androidx.localbroadcastmanager.content.LocalBroadcastManager;
import ch.threema.app.BuildFlavor;
import ch.threema.app.licensing.StoreLicenseCheck;
import ch.threema.app.restrictions.AppRestrictions;
import ch.threema.app.services.DeviceService;
import ch.threema.app.services.UserService;
import ch.threema.app.services.license.LicenseService;
import ch.threema.app.services.license.LicenseServiceThreema;
import ch.threema.app.utils.ConfigUtils;
import ch.threema.app.utils.IntentDataUtil;
import static ch.threema.base.utils.LoggingKt.getThreemaLogger;

import ch.threema.domain.protocol.api.APIConnector;
import ch.threema.domain.stores.IdentityStore;

/**
 * Checking the License of current Threema and send a not allowed broadcast
 */
public class CheckLicenseRoutine implements Runnable {
    private static final Logger logger = getThreemaLogger("CheckLicenseRoutine");

    @NonNull
    private final WeakReference<Activity> activityWeakReference;
    @NonNull
    private final APIConnector apiConnector;
    @NonNull
    private final UserService userService;
    @NonNull
    private final DeviceService deviceService;
    @NonNull
    private final LicenseService<?> licenseService;
    @NonNull
    private final IdentityStore identityStore;
    @NonNull
    private final AppRestrictions appRestrictions;

    public CheckLicenseRoutine(
        @NonNull
        Activity activity,
        @NonNull
        APIConnector apiConnector,
        @NonNull
        UserService userService,
        @NonNull
        DeviceService deviceService,
        @NonNull
        LicenseService<?> licenseService,
        @NonNull
        IdentityStore identityStore,
        @NonNull
        AppRestrictions appRestrictions
    ) {
        this.activityWeakReference = new WeakReference<>(activity);
        this.apiConnector = apiConnector;
        this.userService = userService;
        this.deviceService = deviceService;
        this.licenseService = licenseService;
        this.identityStore = identityStore;
        this.appRestrictions = appRestrictions;
    }

    @Override
    public void run() {
        Activity activity = activityWeakReference.get();
        if (activity == null) {
            // TODO(ANDR-4545): Report this to sentry
            logger.error("Could not check license");
            return;
        }
        switch (BuildFlavor.getCurrent().getLicenseType()) {
            case GOOGLE:
            case HMS:
                StoreLicenseCheck.checkLicense(activity, userService);
                break;
            case SERIAL:
            case GOOGLE_WORK:
            case HMS_WORK:
            case ONPREM:
                checkSerial(activity.getApplicationContext());
                break;
        }
    }

    private void checkSerial(@NonNull Context applicationContext) {
        logger.info("Checking serial license");

        String error = licenseService.validate(true);
        if (error != null) {
            invalidLicense(applicationContext, error);
        } else {
            userService.setCredentials(licenseService.loadCredentials());

            if (licenseService instanceof LicenseServiceThreema && BuildFlavor.getCurrent().getMaySelfUpdate()) {
                LicenseServiceThreema<?> licenseServiceThreema = (LicenseServiceThreema<?>) licenseService;
                if (licenseServiceThreema.getUpdateMessage() != null && !licenseServiceThreema.isUpdateMessageShown()) {
                    try {
                        LocalBroadcastManager.getInstance(applicationContext).sendBroadcast(
                            IntentDataUtil.createActionIntentUpdateAvailable(
                                licenseServiceThreema.getUpdateMessage(),
                                licenseServiceThreema.getUpdateUrl()
                            )
                        );
                        licenseServiceThreema.setUpdateMessageShown(true);
                    } catch (ReceiverCallNotAllowedException x) {
                        logger.error("Exception", x);
                    }
                }
            }

            //run update work info route on the work build
            if (ConfigUtils.isWorkBuild()) {
                (new UpdateWorkInfoRoutine(
                    this.apiConnector,
                    this.identityStore,
                    this.deviceService,
                    this.licenseService,
                    this.appRestrictions
                )).run();
            }
        }
    }

    private void invalidLicense(@NonNull Context applicationContext, String message) {
        try {
            LocalBroadcastManager.getInstance(applicationContext).sendBroadcast(IntentDataUtil.createActionIntentLicenseNotAllowed(message));
        } catch (ReceiverCallNotAllowedException x) {
            logger.error("Exception", x);
        }
    }
}