package jnpf.service.impl;

import jnpf.base.ActionResult;
import jnpf.base.LogApi;
import jnpf.base.UserInfo;
import jnpf.base.model.logmodel.WriteLogModel;
import jnpf.constant.MsgCode;
import jnpf.exception.LoginException;
import jnpf.exception.TenantDatabaseException;
import jnpf.exception.TenantInvalidException;
import jnpf.granter.TokenGranter;
import jnpf.granter.TokenGranterBuilder;
import jnpf.model.LoginVO;
import jnpf.model.logout.LogoutResultModel;
import jnpf.service.AuthService;
import jnpf.util.RedisUtil;
import jnpf.util.StringUtil;
import jnpf.util.TenantProvider;
import jnpf.util.UserProvider;
import jnpf.utils.LoginHolder;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.util.Map;

/**
 * 登录与退出服务 其他服务调用
 */
@Service
@Slf4j
public class AuthServiceImpl implements AuthService {

    @Autowired
    private TokenGranterBuilder tokenGranterBuilder;
    @Autowired
    private LogApi logApi;
    @Autowired
    private RedisUtil redisUtil;

    /**
     * 登录
     * @param parameters {grant_type}
     * @return
     * @throws LoginException
     */
    @Override
    public ActionResult<LoginVO> login(Map<String, String> parameters) throws LoginException{
        long millis = System.currentTimeMillis();
        TokenGranter tokenGranter = tokenGranterBuilder.getGranter(parameters.getOrDefault("grant_type", ""));
        ActionResult<LoginVO> result;
        UserInfo userInfo = new UserInfo();
        try {
            String account = parameters.get("account");
            userInfo.setUserAccount(account);
            UserProvider.setLocalLoginUser(userInfo);
            result = tokenGranter.granter(parameters);
            //写入日志
            if (userInfo.getUserId() != null && !UserProvider.isTempUser(userInfo)) {
                logApi.writeLogAsync(WriteLogModel.builder()
                        .userId(userInfo.getUserId())
                        .userInfo(userInfo)
                        .userName(userInfo.getUserName() + "/" + userInfo.getUserAccount())
                        .abstracts(MsgCode.OA015.get())
                        .loginMark(1)
                        .requestDuration(System.currentTimeMillis() - millis).build());
            }
        } catch (TenantDatabaseException tdex){
            throw tdex;
        } catch (Exception e){
            if(!(e instanceof LoginException || e instanceof TenantInvalidException)){
                String msg = e.getMessage();
                if(msg == null){
                    msg = MsgCode.OA007.get();
                }
                log.error(MsgCode.OA007.get()+", Account: {}, Error: {}", parameters.getOrDefault("account", ""), e.getMessage(), e);
                throw new LoginException(msg);
            }
            String userName = StringUtil.isNotEmpty(userInfo.getUserName()) ? userInfo.getUserName()+"/"+userInfo.getUserAccount() : userInfo.getUserAccount();
            logApi.writeLogAsync(WriteLogModel.builder()
                    .userId(userInfo.getUserId())
                    .userName(userName)
                    .abstracts(e.getMessage())
                    .userInfo(userInfo)
                    .loginMark(0)
                    .requestDuration(System.currentTimeMillis()-millis).build());
            throw e;
        }finally{
            LoginHolder.clearUserEntity();
            TenantProvider.clearBaseSystemIfo();
            // 请求之后就删除验证码 不论结果
            String imgCode = parameters.get("timestamp");
            if(StringUtil.isNotEmpty(imgCode)) {
                redisUtil.remove(imgCode);
            }
        }
        return result;
    }


    /**
     * 踢出用户, 用户将收到Websocket下线通知
     * 执行流程：认证服务退出用户->用户踢出监听->消息服务发送Websocket推送退出消息
     * @param tokens
     */
    @Override
    public ActionResult kickoutByToken(String... tokens){
        UserProvider.kickoutByToken(tokens);
        return ActionResult.success();
    }

    /**
     * 踢出用户, 用户将收到Websocket下线通知
     * 执行流程：认证服务退出用户->用户踢出监听->消息服务发送Websocket推送退出消息
     * @param userId
     * @param tenantId
     */
    @Override
    public ActionResult kickoutByUserId(String userId, String tenantId){
        UserProvider.kickoutByUserId(userId, tenantId);
        return ActionResult.success();
    }

    /**
     * 退出登录
     * @param grandtype
     * @return
     */
    @Override
    public ActionResult<LogoutResultModel> logout(String grandtype) {
        long millis = System.currentTimeMillis();
        TokenGranter tokenGranter = tokenGranterBuilder.getGranterByLogin(grandtype);
        if(tokenGranter != null){
            UserInfo userInfo = UserProvider.getUser();
            logApi.writeLogAsync(WriteLogModel.builder()
                    .userId(userInfo.getUserId())
                    .userName(userInfo.getUserName() + "/" + userInfo.getUserAccount())
                    .abstracts("退出登录")
                    .userInfo(userInfo)
                    .loginMark(1)
                    .loginType(1)
                    .requestDuration(System.currentTimeMillis()-millis).build());
            return tokenGranter.logout();
        }
        return ActionResult.success();
    }
}