package cn.rongcloud.im.viewmodel;

import android.app.Application;
import android.os.CountDownTimer;
import androidx.annotation.NonNull;
import androidx.arch.core.util.Function;
import androidx.lifecycle.AndroidViewModel;
import androidx.lifecycle.LiveData;
import androidx.lifecycle.MediatorLiveData;
import androidx.lifecycle.MutableLiveData;
import androidx.lifecycle.Observer;
import cn.rongcloud.im.common.ErrorCode;
import cn.rongcloud.im.model.ImageCodeResult;
import cn.rongcloud.im.model.RegisterResult;
import cn.rongcloud.im.model.Resource;
import cn.rongcloud.im.model.Status;
import cn.rongcloud.im.model.UserCacheInfo;
import cn.rongcloud.im.task.AppTask;
import cn.rongcloud.im.task.UserTask;
import cn.rongcloud.im.utils.DataCenter;
import cn.rongcloud.im.utils.SingleSourceLiveData;
import cn.rongcloud.im.utils.SingleSourceMapLiveData;

public class LoginViewModel extends AndroidViewModel {
    private final AppTask mAppTask;
    private SingleSourceLiveData<Resource<String>> loginResult = new SingleSourceLiveData<>();
    // TODO 示例代码，当需要转换结果类型时参考
    private SingleSourceMapLiveData<Resource<String>, String> loginResultNoResource;

    private MediatorLiveData<Resource> loadingState = new MediatorLiveData<>();

    //    private SingleSourceLiveData<Resource<String>> sendCodeResult = new
    // SingleSourceLiveData<>();

    private SingleSourceMapLiveData<Resource<String>, Resource<String>> sendCodeState;

    private MutableLiveData<Integer> codeCountDown = new MutableLiveData<>();

    private SingleSourceMapLiveData<Resource<RegisterResult>, Resource<RegisterResult>>
            registerResult;

    private SingleSourceLiveData<Resource<Boolean>> checkPhoneResult = new SingleSourceLiveData<>();

    private SingleSourceMapLiveData<Resource<String>, Resource<String>> resetPasswordResult;

    private MediatorLiveData<Resource<String>> checkPhoneAndSendCodeResult =
            new MediatorLiveData<>();

    private MutableLiveData<UserCacheInfo> lastLoginUserCache = new MutableLiveData<>();

    private MutableLiveData<DataCenter> dataCenterLiveData = new MediatorLiveData<>();

    private SingleSourceMapLiveData<Resource<ImageCodeResult>, Resource<ImageCodeResult>>
            imageCodeResult;

    private UserTask userTask;
    private CountDownTimer countDownTimer =
            new CountDownTimer(60 * 1000, 1000) {

                @Override
                public void onTick(long millisUntilFinished) {
                    int count = Math.round(millisUntilFinished / 1000);
                    codeCountDown.postValue(count);
                }

                @Override
                public void onFinish() {
                    codeCountDown.postValue(0);
                }
            };
    ;

    public LoginViewModel(@NonNull Application application) {
        super(application);

        userTask = new UserTask(application);
        mAppTask = new AppTask(application);
        loadingState.addSource(loginResult, resource -> loadingState.setValue(resource));
        loginResultNoResource = new SingleSourceMapLiveData<>(input -> input.data);

        // 返送验证码请求的结果
        sendCodeState =
                new SingleSourceMapLiveData<>(
                        new Function<Resource<String>, Resource<String>>() {
                            @Override
                            public Resource<String> apply(Resource<String> input) {
                                if (input.status == Status.SUCCESS) {
                                    // 开始计时
                                    startCodeCountDown();
                                }
                                return input;
                            }
                        });

        // 注册结果
        registerResult =
                new SingleSourceMapLiveData<>(
                        new Function<Resource<RegisterResult>, Resource<RegisterResult>>() {
                            @Override
                            public Resource<RegisterResult> apply(Resource<RegisterResult> input) {
                                if (input.status != Status.LOADING
                                        && input.code
                                                != ErrorCode.CHECK_VERIFY_CODE_FAILED.getCode()) {
                                    // 停止计时. 除了验证码错误
                                    stopCodeCountDown();
                                }
                                return input;
                            }
                        });

        // 重设密码
        resetPasswordResult =
                new SingleSourceMapLiveData<>(
                        new Function<Resource<String>, Resource<String>>() {
                            @Override
                            public Resource<String> apply(Resource<String> input) {
                                if (input.status != Status.LOADING
                                        && input.code
                                                != ErrorCode.CHECK_VERIFY_CODE_FAILED.getCode()) {
                                    // 停止计时. 除了验证码错误
                                    stopCodeCountDown();
                                }
                                return input;
                            }
                        });

        imageCodeResult = new SingleSourceMapLiveData<>(input -> input);

        UserCacheInfo userCache = userTask.getUserCache();
        if (userCache != null) {
            lastLoginUserCache.setValue(userCache);
        }
        changeDataCenter(mAppTask.getCurrentDataCenter());
        getImageCode();
    }

    public void login(String region, String phone, String pwd) {
        loginResult.setSource(userTask.login(region, phone, pwd));
        // TODO 示例代码，当需要转换类型时参考
        // loginResultNoResource.setSource(userTask.login(region, phone, pwd));
    }

    public void registerAndLogin(String region, String phone, String code) {
        loginResult.setSource(userTask.registerAndLogin(region, phone, code));
    }

    public LiveData<Resource<String>> getLoginResult() {
        return loginResult;
    }

    public LiveData<String> getLoginResultNoResource() {
        return loginResultNoResource;
    }

    public LiveData<Resource> getLoadingState() {
        return loadingState;
    }

    public LiveData<DataCenter> getDataCenterLiveData() {
        return dataCenterLiveData;
    }

    //    /**
    //     *验证请求发送结果
    //     * @return
    //     */
    //    public LiveData<Resource<String>> getSendCodeResult(){
    //        return sendCodeResult;
    //    }

    /**
     * 验证请求发送结果
     *
     * @return
     */
    public LiveData<Resource<String>> getSendCodeState() {
        return sendCodeState;
    }

    public LiveData<Resource<ImageCodeResult>> getImageCodeResult() {
        return imageCodeResult;
    }

    /**
     * 验证码接受倒计时
     *
     * @return
     */
    public LiveData<Integer> getCodeCountDown() {
        return codeCountDown;
    }

    /**
     * 检测手机号并发送验证码
     *
     * @return
     */
    public LiveData<Resource<String>> getCheckPhoneAndSendCode() {
        return checkPhoneAndSendCodeResult;
    }

    /**
     * 最后一次的用户信息
     *
     * @return
     */
    public LiveData<UserCacheInfo> getLastLoginUserCache() {
        return lastLoginUserCache;
    }

    /**
     * 获取注册结果
     *
     * @return
     */
    public LiveData<Resource<RegisterResult>> getRegisterResult() {
        return registerResult;
    }

    /**
     * 重置密码
     *
     * @return
     */
    public LiveData<Resource<String>> getResetPasswordResult() {
        return resetPasswordResult;
    }

    // 操作方法

    /**
     * 发送验证码
     *
     * @param phoneCode 国家区域手机区号
     * @param phoneNumber 手机号
     */
    public void sendCode(String phoneCode, String phoneNumber, String picCode, String picCodeId) {
        sendCodeState.setSource(userTask.sendCode(phoneCode, phoneNumber, picCode, picCodeId));
    }

    public void getImageCode() {
        imageCodeResult.setSource(userTask.getImageCode());
    }

    /**
     * 注册操作
     *
     * @param phoneCode 国家区域手机区号
     * @param phoneNumber 手机号
     * @param shortMsgCode 短信验证码
     * @param nickName 昵称
     * @param password 密码
     */
    public void register(
            String phoneCode,
            String phoneNumber,
            String shortMsgCode,
            String nickName,
            String password) {
        registerResult.setSource(
                userTask.register(phoneCode, phoneNumber, shortMsgCode, nickName, password));
    }

    /**
     * 检测手机是否注册
     *
     * @param phoneCode 国家区域手机区号
     * @param phoneNumber 手机号
     */
    public void checkPhoneAvailable(String phoneCode, String phoneNumber) {
        checkPhoneResult.setSource(userTask.checkPhoneAvailable(phoneCode, phoneNumber));
    }

    /**
     * 重设置密码
     *
     * @param countryCode
     * @param phoneNumber
     * @param shortMsgCode
     * @param password
     */
    public void resetPassword(
            String countryCode, String phoneNumber, String shortMsgCode, String password) {
        resetPasswordResult.setSource(
                userTask.resetPassword(countryCode, phoneNumber, shortMsgCode, password));
    }

    /**
     * 检验密码并发送验证码
     *
     * @param phoneCode
     * @param phoneNumber
     */
    public void checkPhoneAndSendCode(
            String phoneCode, String phoneNumber, String picCode, String picCodeId) {
        checkPhoneAvailable(phoneCode, phoneNumber);
        checkPhoneAndSendCodeResult.removeSource(checkPhoneResult);
        checkPhoneAndSendCodeResult.addSource(
                checkPhoneResult,
                new Observer<Resource<Boolean>>() {
                    @Override
                    public void onChanged(Resource<Boolean> resource) {
                        if (resource.status == Status.SUCCESS) {
                            checkPhoneAndSendCodeResult.removeSource(checkPhoneResult);
                            sendCode(phoneCode, phoneNumber, picCode, picCodeId);
                            checkPhoneAndSendCodeResult.removeSource(sendCodeState);
                            checkPhoneAndSendCodeResult.addSource(
                                    sendCodeState,
                                    new Observer<Resource<String>>() {
                                        @Override
                                        public void onChanged(Resource<String> resource) {
                                            if (resource.status == Status.SUCCESS) {
                                                checkPhoneAndSendCodeResult.removeSource(
                                                        sendCodeState);
                                                checkPhoneAndSendCodeResult.removeSource(
                                                        checkPhoneAndSendCodeResult);
                                                checkPhoneAndSendCodeResult.postValue(resource);
                                            } else if (resource.status == Status.ERROR) {
                                                checkPhoneAndSendCodeResult.removeSource(
                                                        sendCodeState);
                                                checkPhoneAndSendCodeResult.postValue(resource);
                                            }
                                        }
                                    });
                        } else if (resource.status == Status.ERROR) {
                            checkPhoneAndSendCodeResult.removeSource(checkPhoneResult);
                            Resource<String> resourceTmp =
                                    new Resource<>(resource.status, null, resource.code);
                            checkPhoneAndSendCodeResult.postValue(resourceTmp);
                        }
                    }
                });
    }

    private void startCodeCountDown() {
        countDownTimer.cancel();
        countDownTimer.start();
    }

    public void stopCodeCountDown() {
        countDownTimer.cancel();
    }

    @Override
    protected void onCleared() {
        super.onCleared();
    }

    public void changeDataCenter(DataCenter center) {
        mAppTask.changeDataCenter(center);
        dataCenterLiveData.postValue(center);
    }

    public void saveDataCenter(DataCenter dataCenter) {
        mAppTask.saveDataCenter(dataCenter);
    }

    /** 是否打开了不展示登录图片验证码 */
    public boolean isHidePicCode() {
        return userTask.isHidePicCode();
    }
}
