package com.zioinfo.mes.auth.mapper;

import com.zioinfo.mes.auth.MesUser;
import org.apache.ibatis.annotations.Mapper;
import org.apache.ibatis.annotations.Param;
import org.apache.ibatis.annotations.Update;

import java.time.LocalDateTime;

@Mapper
public interface UserMapper {

    // findByUsername / insert 는 mapper/UserMapper.xml 에 정의(2FA 컬럼 포함 resultMap).
    // 어노테이션 중복 정의 시 "statement already contains value" 크래시 → XML 유지.
    MesUser findByUsername(@Param("username") String username);

    int insert(MesUser user);

    // ── UIWS 2FA 이식: 실패 카운트/잠금/인증코드 갱신 (멱등 UPDATE) ───────────────

    /** 로그인 성공 시 실패 카운트 초기화. */
    @Update("UPDATE mes_user SET login_fail_count = 0 WHERE username = #{username}")
    int resetLoginFail(@Param("username") String username);

    /** 로그인 실패 누적(+1) 및 임계 도달 시 잠금. */
    @Update("""
            UPDATE mes_user
            SET login_fail_count = COALESCE(login_fail_count, 0) + 1,
                locked = (COALESCE(login_fail_count, 0) + 1 >= #{maxFail})
            WHERE username = #{username}
            """)
    int incrementLoginFail(@Param("username") String username, @Param("maxFail") int maxFail);

    /** 1차 통과 시 이메일 인증코드/만료 저장(verify 단계에서 검증). */
    @Update("""
            UPDATE mes_user
            SET email_verify_code = #{code}, email_verify_expire = #{expire}, login_fail_count = 0
            WHERE username = #{username}
            """)
    int saveEmailCode(@Param("username") String username,
                      @Param("code") String code,
                      @Param("expire") LocalDateTime expire);

    /** 2차 검증 성공 시 코드 폐기. */
    @Update("UPDATE mes_user SET email_verify_code = NULL, email_verify_expire = NULL WHERE username = #{username}")
    int clearEmailCode(@Param("username") String username);

    /** 관리자 잠금 해제(실패 카운트/잠금 초기화). */
    @Update("UPDATE mes_user SET locked = false, login_fail_count = 0 WHERE username = #{username}")
    int unlock(@Param("username") String username);

    // ── TOTP(OTP 2차 인증) 이식 (멱등 UPDATE) ────────────────────────────────────

    /** 보류 시크릿 저장(등록 확정 전). otp_enabled 은 건드리지 않는다(verify/confirm 에서 확정). */
    @Update("UPDATE mes_user SET otp_secret = #{secret} WHERE username = #{username}")
    int updateOtpSecret(@Param("username") String username, @Param("secret") String secret);

    /** 등록 확정: otp_enabled=true (시크릿은 유지). */
    @Update("UPDATE mes_user SET otp_enabled = true WHERE username = #{username}")
    int enableOtp(@Param("username") String username);

    /** 해제/초기화: 시크릿 폐기 + otp_enabled=false. (마이페이지 해제) */
    @Update("UPDATE mes_user SET otp_secret = NULL, otp_enabled = false WHERE username = #{username}")
    int disableOtp(@Param("username") String username);

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

    /** username 존재 여부(회원가입 중복 검사). */
    int countByUsername(@Param("username") String username);

    /** email 존재 여부(회원가입 중복 검사 — 운영자 계정 한정). */
    int countByEmail(@Param("email") String email);

    /**
     * 회원가입(승인 대기). approved=false·is_active=true·role=VIEWER 고정.
     * 관리자 화면에서 활성/승인 전까지 로그인 차단.
     */
    int signup(MesUser user);

    /** 아이디찾기: 표시명(display_name)+이메일 일치 운영자 1건. */
    MesUser findByDisplayNameAndEmail(@Param("displayName") String displayName,
                                      @Param("email") String email);

    /** 비밀번호 초기화 대상 검증: username+email 동시 일치 운영자 1건. */
    MesUser findByUsernameAndEmail(@Param("username") String username,
                                   @Param("email") String email);

    /** 임시 비밀번호 적용 + 잠금/실패카운트 해제(초기화 시). */
    int updatePasswordHash(@Param("username") String username,
                           @Param("passwordHash") String passwordHash);
}
