package store

import (
	"database/sql"
	"errors"
	"strconv"
)

func fmtSscanInt(s string, out *int64) (int, error) {
	v, err := strconv.ParseInt(s, 10, 64)
	if err != nil {
		return 0, err
	}
	*out = v
	return 1, nil
}

// Session 会话行（token 明文不入库，只存 SHA-256）。
type Session struct {
	TokenHash string
	CSRFToken string
	CreatedAt int64
	ExpiresAt int64
}

// CreateSession 插入会话行。
func (s *Store) CreateSession(tokenHash, csrfToken string, createdAt, expiresAt int64) error {
	_, err := s.db.Exec(
		`INSERT INTO sessions (token_hash, csrf_token, created_at, expires_at) VALUES (?, ?, ?, ?)`,
		tokenHash, csrfToken, createdAt, expiresAt)
	return err
}

// RotateSession 重建会话行：新 token、**csrf_token 保持不变**（§7.3-5）、新过期时间。
// 原子替换，避免窗口期内两行并存。
func (s *Store) RotateSession(oldTokenHash, newTokenHash, csrfToken string, createdAt, expiresAt int64) error {
	tx, err := s.db.Begin()
	if err != nil {
		return err
	}
	defer tx.Rollback()
	if _, err := tx.Exec(`DELETE FROM sessions WHERE token_hash=?`, oldTokenHash); err != nil {
		return err
	}
	if _, err := tx.Exec(
		`INSERT INTO sessions (token_hash, csrf_token, created_at, expires_at) VALUES (?, ?, ?, ?)`,
		newTokenHash, csrfToken, createdAt, expiresAt); err != nil {
		return err
	}
	return tx.Commit()
}

// GetSession 按 token 摘要取会话（含已过期行；过期判定在调用方）。
func (s *Store) GetSession(tokenHash string) (*Session, error) {
	row := s.db.QueryRow(
		`SELECT token_hash, csrf_token, created_at, expires_at FROM sessions WHERE token_hash=?`, tokenHash)
	var sess Session
	err := row.Scan(&sess.TokenHash, &sess.CSRFToken, &sess.CreatedAt, &sess.ExpiresAt)
	if errors.Is(err, sql.ErrNoRows) {
		return nil, ErrNotFound
	}
	return &sess, err
}

// DeleteSession 删除会话行（登出/轮换）。
func (s *Store) DeleteSession(tokenHash string) error {
	_, err := s.db.Exec(`DELETE FROM sessions WHERE token_hash=?`, tokenHash)
	return err
}

// DeleteExpiredSessions 清理过期会话，返回删除行数。
func (s *Store) DeleteExpiredSessions(now int64) (int64, error) {
	res, err := s.db.Exec(`DELETE FROM sessions WHERE expires_at < ?`, now)
	if err != nil {
		return 0, err
	}
	return res.RowsAffected()
}

// DeleteAllSessions 吊销全部会话（口令重置后强制所有端重新登录）。
func (s *Store) DeleteAllSessions() (int64, error) {
	res, err := s.db.Exec(`DELETE FROM sessions`)
	if err != nil {
		return 0, err
	}
	return res.RowsAffected()
}