package jnpf.listener;

import cn.dev33.satoken.listener.SaTokenListenerForSimple;
import cn.dev33.satoken.session.SaSession;
import cn.dev33.satoken.stp.StpUtil;
import cn.dev33.satoken.stp.parameter.SaLoginParameter;
import jnpf.base.UserInfo;
import jnpf.base.model.logmodel.LogoutReason;
import jnpf.message.SentMessageApi;
import jnpf.permission.UserApi;
import jnpf.permission.entity.UserEntity;
import jnpf.permission.model.user.UserUpdateModel;
import jnpf.service.LoginSessionLogRecorder;
import jnpf.util.DateUtil;
import jnpf.util.IpUtil;
import jnpf.util.UserProvider;
import jnpf.utils.LoginHolder;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Component;

/**
 * Sa-Token lifecycle listener shared by aggregated and microservice deployments.
 */
@Component
@Slf4j
@RequiredArgsConstructor
public class LoginListener extends SaTokenListenerForSimple {

    private final SentMessageApi messageApi;
    private final UserApi userApi;
    private final LoginSessionLogRecorder loginSessionLogRecorder;

    @Override
    public void doLogin(String loginType, Object loginId, String tokenValue, SaLoginParameter loginModel) {
        println("用户登录：{}, 设备：{}, TOKEN：{}", loginId, loginType, tokenValue);
        UserInfo userInfo = UserProvider.getUser();
        // Temp users participate in session logs, but do not update normal account statistics.
        if (!UserProvider.isTempUser(userInfo)) {
            UserEntity entity = LoginHolder.getUserEntity();
            if (entity == null) {
                return;
            }
            entity.setLogErrorCount(0);
            entity.setUnlockTime(null);
            entity.setEnabledMark(1);
            entity.setDisableTime(null);
            entity.setPrevLogIp(IpUtil.getIpAddr());
            entity.setPrevLogTime(DateUtil.getNowDate());
            entity.setLastLogIp(IpUtil.getIpAddr());
            entity.setLastLogTime(DateUtil.getNowDate());
            entity.setLogSuccessCount(entity.getLogSuccessCount() != null ? entity.getLogSuccessCount() + 1 : 1);
            userApi.updateById(new UserUpdateModel(entity, userInfo.getTenantId()));
        }
    }

    @Override
    public void doLogout(String loginType, Object loginId, String tokenValue) {
        println("用户退出：{}, 设备：{}, TOKEN：{}", loginId, loginType, tokenValue);
        loginSessionLogRecorder.close(tokenValue, getUserInfo(tokenValue), LogoutReason.MANUAL, "主动退出");
    }

    @Override
    public void doKickout(String loginType, Object loginId, String tokenValue) {
        println("用户踢出：{}, 设备：{}, TOKEN：{}", loginId, loginType, tokenValue);
        loginSessionLogRecorder.close(tokenValue, getUserInfo(tokenValue), LogoutReason.KICKOUT, "强制下线");
        messageApi.logoutWebsocketByToken(tokenValue, null);
        SaSession session = safeTokenSession(tokenValue);
        if (session != null) {
            session.logout();
        }
    }

    @Override
    public void doReplaced(String loginType, Object loginId, String tokenValue) {
        println("用户顶替：{}, 设备：{}, TOKEN：{}", loginId, loginType, tokenValue);
        loginSessionLogRecorder.close(tokenValue, getUserInfo(tokenValue), LogoutReason.REPLACED, "登录顶替");
        messageApi.logoutWebsocketByToken(tokenValue, null);
        SaSession session = safeTokenSession(tokenValue);
        if (session != null) {
            session.logout();
        }
    }

    private UserInfo getUserInfo(String tokenValue) {
        SaSession session = safeTokenSession(tokenValue);
        return session == null ? null : (UserInfo) session.get("userInfo");
    }

    private SaSession safeTokenSession(String tokenValue) {
        try {
            return StpUtil.getTokenSessionByToken(tokenValue);
        } catch (Exception e) {
            return null;
        }
    }

    private void println(String text, Object... params) {
        if (log.isDebugEnabled()) {
            log.debug(text, params);
        }
    }
}