package com.zioinfo.mes.auth;

import com.zioinfo.mes.auth.dto.AuthHelperResult;
import com.zioinfo.mes.auth.dto.FindIdRequest;
import com.zioinfo.mes.auth.dto.FindIdResponse;
import com.zioinfo.mes.auth.dto.ResetPasswordRequest;
import com.zioinfo.mes.auth.dto.SignupRequest;
import com.zioinfo.mes.auth.mapper.UserMapper;
import com.zioinfo.mes.uiws.auth.TwoFactorService;
import com.zioinfo.mes.uiws.common.UiwsApiException;
import com.zioinfo.mes.uiws.common.UiwsErrorCode;
import com.zioinfo.mes.uiws.common.mail.MailSender;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.security.crypto.password.PasswordEncoder;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import java.security.SecureRandom;
import java.util.HashMap;
import java.util.Map;

/**
 * MES 인증 서비스.
 * - 기존 단일 JWT 로그인 보존(2FA off 또는 미설정 시 회귀 0).
 * - UIWS 2FA 레이어(이메일 코드) 추가: 2FA on 이면 1차 통과 후 verify-token + 이메일코드 발급.
 *   실패 누적 max-login-fail 회 시 계정 잠금.
 */
@Slf4j
@Service
@RequiredArgsConstructor
public class AuthService {

    private static final SecureRandom RANDOM = new SecureRandom();
    /** 임시 비밀번호 문자셋(혼동 문자 0/O/1/l/I 제외). */
    private static final String TMP_PW_CHARS = "ABCDEFGHJKLMNPQRSTUVWXYZabcdefghijkmnpqrstuvwxyz23456789@#$%";

    private final UserMapper userMapper;
    private final PasswordEncoder passwordEncoder;
    private final JwtUtil jwtUtil;
    private final TwoFactorService twoFactorService;
    private final MailSender mailSender;

    /**
     * 1차 로그인. 2FA 활성 시 verify-token + 이메일코드 흐름으로 분기,
     * 비활성 시 기존처럼 access 토큰 즉시 발급.
     *
     * @return 2FA off: { token, type, twofa:"false" }
     *         2FA on : { verifyToken, step:"EMAIL", maskedEmail, twofa:"true" }
     */
    public Map<String, String> login(String username, String password) {
        MesUser user = userMapper.findByUsername(username);

        // 잠금 우선 차단(존재하는 계정에 한해 잠금 메시지 — 존재 여부 누설 최소화)
        if (user != null && twoFactorService.isLocked(user)) {
            throw new UiwsApiException(UiwsErrorCode.ACCOUNT_LOCKED);
        }
        if (user == null || !user.isActive()) {
            throw new RuntimeException("ERR-AUTH-001: 존재하지 않거나 비활성 계정");
        }
        // 회원가입 승인 게이트(로그인 보조 이식): 미승인 계정은 비번 일치 전에 차단.
        if (Boolean.FALSE.equals(user.getApproved())) {
            throw new RuntimeException("ERR-AUTH-003: 승인 대기 중인 계정입니다. 관리자 승인 후 이용 가능합니다.");
        }
        if (!passwordEncoder.matches(password, user.getPasswordHash())) {
            // 2FA 활성 시 실패 누적/잠금. 비활성 시 기존 동작(메시지만) 유지.
            if (twoFactorService.isEnabled()) {
                twoFactorService.recordLoginFailure(username);
                MesUser after = userMapper.findByUsername(username);
                if (after != null && Boolean.TRUE.equals(after.getLocked())) {
                    throw new UiwsApiException(UiwsErrorCode.ACCOUNT_LOCKED);
                }
            }
            throw new RuntimeException("ERR-AUTH-002: 비밀번호 불일치");
        }

        // 비밀번호 검증 통과
        if (twoFactorService.isEnabled()) {
            Map<String, String> step1 = twoFactorService.beginTwoFactor(user);
            // verifyToken/step/maskedEmail + twofa 플래그
            return Map.of(
                    "twofa", "true",
                    "verifyToken", step1.get("verifyToken"),
                    "step", step1.get("step"),
                    "maskedEmail", step1.getOrDefault("maskedEmail", ""));
        }

        // 2FA 비활성 — 기존 단일 로그인 흐름(회귀 0)
        userMapper.resetLoginFail(username);
        String token = jwtUtil.generate(username, user.getRole());
        return Map.of("twofa", "false", "token", token, "type", "Bearer");
    }

    public Map<String, Object> me(String token) {
        String username = jwtUtil.getUsername(token);
        String role = jwtUtil.getRole(token);
        MesUser u = userMapper.findByUsername(username);
        Map<String, Object> m = new HashMap<>();
        m.put("username", username);
        m.put("role", role);
        m.put("displayName", u != null ? u.getDisplayName() : username);
        return m;
    }

    // ── 로그인 보조 이식: 회원가입 / 아이디찾기 / 비밀번호 초기화 ───────────────────

    /**
     * 운영자 회원가입(승인 대기). username/email 중복 검사 후 approved=false·role=VIEWER 로 INSERT.
     * 비밀번호는 BCrypt 저장. 활성/승인 전까지 로그인 차단(AuthService.login 의 승인 게이트).
     */
    @Transactional
    public AuthHelperResult signup(SignupRequest req) {
        if (req.username() == null || req.username().isBlank()
                || req.password() == null || req.password().length() < 4
                || req.email() == null || req.email().isBlank()) {
            return new AuthHelperResult(false, "아이디·비밀번호(4자 이상)·이메일은 필수입니다.");
        }
        if (userMapper.countByUsername(req.username()) > 0) {
            return new AuthHelperResult(false, "이미 사용 중인 아이디입니다.");
        }
        if (userMapper.countByEmail(req.email()) > 0) {
            return new AuthHelperResult(false, "이미 등록된 이메일입니다.");
        }
        MesUser u = new MesUser();
        u.setUsername(req.username());
        u.setPasswordHash(passwordEncoder.encode(req.password()));
        u.setDisplayName(req.displayName() != null && !req.displayName().isBlank()
                ? req.displayName() : req.username());
        u.setEmail(req.email());
        userMapper.signup(u);
        log.info("[auth-helper] signup pending approval: username={}", req.username());
        return new AuthHelperResult(true, "가입 신청이 접수되었습니다. 관리자 승인 후 로그인할 수 있습니다.");
    }

    /**
     * 아이디 찾기: 표시명+이메일 동시 일치 운영자 1건 조회. username 은 부분 마스킹 후 반환.
     * 미발견 시 found=false(원문 username 절대 미노출).
     */
    public FindIdResponse findId(FindIdRequest req) {
        if (req.displayName() == null || req.displayName().isBlank()
                || req.email() == null || req.email().isBlank()) {
            return new FindIdResponse(false, "");
        }
        MesUser u = userMapper.findByDisplayNameAndEmail(req.displayName(), req.email());
        if (u == null) {
            return new FindIdResponse(false, "");
        }
        return new FindIdResponse(true, maskUsername(u.getUsername()));
    }

    /**
     * 비밀번호 초기화: username+email 일치 검증 → 임시비번 생성·BCrypt 저장·잠금/실패카운트 해제.
     * 임시비번은 메일(미설정 시 LogMailSender 로그)로만 전달. API 응답·로그 메시지에 비번 미노출.
     * 대상 미존재여도 success=true(계정 열거 방지).
     */
    @Transactional
    public AuthHelperResult resetPassword(ResetPasswordRequest req) {
        final String okMsg = "임시 비밀번호를 등록된 이메일로 발송했습니다. 메일을 확인하세요.";
        if (req.username() == null || req.username().isBlank()
                || req.email() == null || req.email().isBlank()) {
            return new AuthHelperResult(false, "아이디와 이메일을 모두 입력하세요.");
        }
        MesUser u = userMapper.findByUsernameAndEmail(req.username(), req.email());
        if (u == null) {
            // 존재 여부 누설 방지 — 동일 성공 메시지 반환(실제 발송 없음).
            log.info("[auth-helper] reset-password no match (suppressed): username={}", req.username());
            return new AuthHelperResult(true, okMsg);
        }
        String tempPw = generateTempPassword();
        userMapper.updatePasswordHash(req.username(), passwordEncoder.encode(tempPw));

        String subject = "[GUARDiA MES] 임시 비밀번호 안내";
        String body = String.format(
                "안녕하세요 %s 님,\n임시 비밀번호는 [%s] 입니다.\n로그인 후 즉시 비밀번호를 변경하세요.",
                u.getDisplayName() != null ? u.getDisplayName() : u.getUsername(), tempPw);
        // 메일 본문에만 임시비번 포함. mailSender 미설정 환경은 LogMailSender 폴백(서버 로그).
        mailSender.send(u.getEmail(), subject, body);
        log.info("[auth-helper] reset-password issued temp pw (sent via mail/log): username={}", req.username());
        return new AuthHelperResult(true, okMsg);
    }

    private static String generateTempPassword() {
        StringBuilder sb = new StringBuilder(10);
        for (int i = 0; i < 10; i++) {
            sb.append(TMP_PW_CHARS.charAt(RANDOM.nextInt(TMP_PW_CHARS.length())));
        }
        return sb.toString();
    }

    /** username 부분 마스킹: 앞 2자만 노출(예: admin → ad***). 2자 이하는 첫 글자만. */
    private static String maskUsername(String username) {
        if (username == null || username.isBlank()) {
            return "";
        }
        if (username.length() <= 2) {
            return username.charAt(0) + "*";
        }
        return username.substring(0, 2) + "*".repeat(Math.max(1, username.length() - 2));
    }
}
