"""账号管理服务层：封装账号导入、查询等业务逻辑，隔离 core/ 依赖。"""

import csv
import re

from sqlalchemy.orm import Session

from ..models import Account, LoginTask


EMAIL_PATTERN = re.compile(r'^[^\s@|]+@[^\s@|]+\.[^\s@|]+$')


def split_account_line(line: str) -> list[str]:
    """拆分一行账号文本，支持 |、tab、逗号、空格分隔。"""
    if '|' in line:
        return line.split('|')
    if '\t' in line:
        return line.split('\t')
    if ',' in line:
        return next(csv.reader([line]))
    return line.split()


def cookie_account_ids_query(db: Session):
    """返回有成功登录记录(cookie非空)的账号ID子查询。"""
    return db.query(LoginTask.account_id).filter(
        LoginTask.status == 'success',
        LoginTask.cookie != '',
        LoginTask.cookie.isnot(None),
    ).distinct()


def parse_and_build_accounts(
    db: Session,
    text: str,
    default_tag: str = "",
) -> tuple[list[Account], int, int]:
    """
    解析批量导入文本，构建 Account ORM 对象列表，并过滤已存在的重复账号。

    Args:
        db: 数据库会话，用于查询已有账号
        text: 批量导入文本，每行格式：用户名|密码|邮箱|邮箱密码|标签(可选)
        default_tag: 统一标签兜底。某行未单独写标签时使用该值，
            行内第5列标签优先。

    Returns:
        (accounts, skipped_count, duplicated_count)
        skipped_count 为格式非法的行数；duplicated_count 为库中已存在或同批重复被跳过的行数。
    """
    from core.douyu.email_verifier import get_email_config_for_account

    default_tag = (default_tag or "").strip()
    # 库中已有用户名（大小写不敏感），用于导入去重
    existing_usernames = {
        row[0].lower()
        for row in db.query(Account.username).filter(Account.username != "").all()
    }
    accounts = []
    skipped = 0
    duplicated = 0
    seen_in_batch: set[str] = set()
    for line in text.strip().split('\n'):
        line = line.strip()
        if not line or line.startswith('#'):
            continue
        parts = split_account_line(line)
        if len(parts) < 4:
            skipped += 1
            continue

        username, password, email, email_password = [p.strip() for p in parts[:4]]
        line_tag = parts[4].strip() if len(parts) > 4 else ""
        # 行内标签优先，未写则用统一兜底标签
        tag = line_tag or default_tag
        if not all([username, password, email, email_password]):
            skipped += 1
            continue
        if not EMAIL_PATTERN.match(email):
            skipped += 1
            continue

        username_key = username.lower()
        if username_key in existing_usernames or username_key in seen_in_batch:
            duplicated += 1
            continue
        seen_in_batch.add(username_key)

        email_cfg = get_email_config_for_account(email)
        accounts.append(Account(
            username=username,
            password=password,
            email=email,
            email_password=email_password,
            email_imap_server=email_cfg['server'],
            email_imap_port=email_cfg.get('port', 993),
            email_imap_ssl=email_cfg.get('ssl', True),
            tag=tag,
        ))
    return accounts, skipped, duplicated