LoginPresenter.java

/*
 * This program is part of the OpenLMIS logistics management information
 * system platform software.
 *
 * Copyright © 2015 ThoughtWorks, Inc.
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as published
 * by the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version. This program is distributed in the
 * hope that it will be useful, but WITHOUT ANY WARRANTY; without even the
 * implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
 * See the GNU Affero General Public License for more details. You should
 * have received a copy of the GNU Affero General Public License along with
 * this program. If not, see http://www.gnu.org/licenses. For additional
 * information contact info@OpenLMIS.org
 */

package org.openlmis.core.presenter;


import android.util.Log;

import com.google.inject.Inject;

import org.apache.commons.lang3.StringUtils;
import org.openlmis.core.LMISApp;
import org.openlmis.core.R;
import org.openlmis.core.exceptions.LMISException;
import org.openlmis.core.exceptions.NetWorkException;
import org.openlmis.core.manager.SharedPreferenceMgr;
import org.openlmis.core.manager.UserInfoMgr;
import org.openlmis.core.model.Program;
import org.openlmis.core.model.User;
import org.openlmis.core.model.repository.LotRepository;
import org.openlmis.core.model.repository.ProgramRepository;
import org.openlmis.core.model.repository.RnrFormRepository;
import org.openlmis.core.model.repository.StockRepository;
import org.openlmis.core.model.repository.UserRepository;
import org.openlmis.core.network.model.UserResponse;
import org.openlmis.core.service.SyncDownManager;
import org.openlmis.core.service.SyncDownManager.SyncProgress;
import org.openlmis.core.service.SyncService;
import org.openlmis.core.training.TrainingEnvironmentHelper;
import org.openlmis.core.utils.ToastUtil;
import org.openlmis.core.view.BaseView;

import retrofit.Callback;
import retrofit.RetrofitError;
import retrofit.client.Response;
import rx.Subscriber;

public class LoginPresenter extends Presenter {

    LoginView view;

    @Inject
    UserRepository userRepository;

    @Inject
    LotRepository lotRepository;

    @Inject
    StockRepository stockRepository;

    @Inject
    RnrFormRepository rnrFormRepository;

    @Inject
    SyncService syncService;

    @Inject
    SyncDownManager syncDownManager;

    private boolean hasGoneToNextPage;

    @Inject
    private ProgramRepository programRepository;

    @Override
    public void attachView(BaseView v) {
        this.view = (LoginView) v;
    }

    public void startLogin(String userName, String password) {
        hasGoneToNextPage = false;
        if (StringUtils.EMPTY.equals(userName.trim())) {
            view.showUserNameEmpty();
            return;
        }
        if (StringUtils.EMPTY.equals(password)) {
            view.showPasswordEmpty();
            return;
        }
        view.loading();

        User user = new User(userName.trim(), password);
        if (LMISApp.getInstance().isConnectionAvailable() && !LMISApp.getInstance().getFeatureToggleFor(R.bool.feature_training)) {
            authorizeAndLoginUserRemote(user);
        } else {
            authorizeAndLoginUserLocal(user);
        }
    }

    private void authorizeAndLoginUserLocal(User user) {
        if (LMISApp.getInstance().getFeatureToggleFor(R.bool.feature_training)) {
            if (userRepository.getLocalUser() == null) {
                TrainingEnvironmentHelper.getInstance().setUpData();
            }
        }
        User localUser = userRepository.mapUserFromLocal(user);

        if (localUser == null) {
            onLoginFailed();
            return;
        }

        UserInfoMgr.getInstance().setUser(localUser);

        if (SharedPreferenceMgr.getInstance().getLastSyncProductTime() == null) {
            view.loaded();
            ToastUtil.show(R.string.msg_sync_products_list_failed);
            return;
        }

        if (!SharedPreferenceMgr.getInstance().isLastMonthStockDataSynced()) {
            view.loaded();
            ToastUtil.show(R.string.msg_sync_stock_movement_failed);
            return;
        }
        if (!SharedPreferenceMgr.getInstance().isRequisitionDataSynced()) {
            view.loaded();
            ToastUtil.show(R.string.msg_sync_requisition_failed);
            return;
        }

        goToNextPage();
    }

    private void authorizeAndLoginUserRemote(final User user) {
        LMISApp.getInstance().getRestApi().authorizeUser(user, new Callback<UserResponse>() {
            @Override
            public void success(UserResponse userResponse, Response response) {
                if (userResponse == null || userResponse.getUserInformation() == null) {
                    onLoginFailed();
                } else {
                    userResponse.getUserInformation().setUsername(user.getUsername());
                    userResponse.getUserInformation().setPassword(user.getPassword());

                    onLoginSuccess(userResponse);
                }
            }

            @Override
            public void failure(RetrofitError error) {
                if (error.getCause() instanceof NetWorkException) {
                    authorizeAndLoginUserLocal(user);
                } else {
                    onLoginFailed();
                }
            }
        });
    }

    protected void saveUserDataToLocalDatabase(UserResponse response) throws LMISException {
        userRepository.createOrUpdate(response.getUserInformation());

        if (response.getFacilitySupportedPrograms() != null) {
            for (Program program : response.getFacilitySupportedPrograms()) {
                programRepository.createOrUpdate(program);
            }
        }

    }

    private void onLoginSuccess(UserResponse userResponse) {
        Log.d("Login Presenter", "Log in successful, setting up sync account");
        syncService.createSyncAccount(userResponse.getUserInformation());

        try {
            saveUserDataToLocalDatabase(userResponse);
        } catch (LMISException e) {
            e.reportToFabric();
        }
        UserInfoMgr.getInstance().setUser(userResponse.getUserInformation());
        view.clearErrorAlerts();

        syncDownManager.syncDownServerData(getSyncSubscriber());

        view.sendScreenToGoogleAnalyticsAfterLogin();

        if (LMISApp.getInstance().getFeatureToggleFor(R.bool.feature_archive_old_data)) {
            ArchiveOldData();
        }
    }

    private void ArchiveOldData() {
        if (stockRepository.hasOldDate()) {
            stockRepository.deleteOldData();
            SharedPreferenceMgr.getInstance().setHasDeletedOldStockMovement(true);
        }
        if (rnrFormRepository.hasOldDate()) {
            rnrFormRepository.deleteOldData();
            SharedPreferenceMgr.getInstance().setHasDeletedOldRnr(true);
        }
    }

    public void onLoginFailed() {
        view.loaded();
        view.showInvalidAlert();
        view.clearPassword();
    }

    protected Subscriber<SyncProgress> getSyncSubscriber() {
        return new Subscriber<SyncProgress>() {
            @Override
            public void onCompleted() {
                syncService.kickOff();
                tryGoToNextPage();
            }

            @Override
            public void onError(Throwable e) {
                ToastUtil.show(e.getMessage());
                view.loaded();
            }

            @Override
            public void onNext(SyncProgress progress) {
                switch (progress) {
                    case SyncingProduct:
                    case SyncingStockCardsLastMonth:
                    case SyncingRequisition:
                        view.loading(LMISApp.getInstance().getString(progress.getMessageCode()));
                        break;

                    case ProductSynced:
                    case StockCardsLastMonthSynced:
                        view.loaded();
                        break;

                    case RequisitionSynced:
                        if (!view.needInitInventory()) {
                            ToastUtil.showLongTimeAsOfficialWay(R.string.msg_initial_sync_success);
                        }
                        goToNextPage();
                        break;
                    case SyncingStockCardsLastYear:
                        tryGoToNextPage();
                        break;
                }
            }
        };
    }

    private void tryGoToNextPage() {
        if (!hasGoneToNextPage) {
            goToNextPage();
        }
    }

    private void goToNextPage() {
        view.loaded();

        if (view.needInitInventory()) {
            view.goToInitInventory();
        } else {
            view.goToHomePage();
        }
        hasGoneToNextPage = true;
    }

    public interface LoginView extends BaseView {

        void clearPassword();

        void goToHomePage();

        void goToInitInventory();

        boolean needInitInventory();

        void showInvalidAlert();

        void showUserNameEmpty();

        void showPasswordEmpty();

        void clearErrorAlerts();

        void sendScreenToGoogleAnalyticsAfterLogin();
    }
}