#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Love & Relationship Analysis（感情分析）引擎 v1.0
超越 vedic-astro-skills 的 AI prompt，实现结构化引擎级分析

核心分析链：
7宫+Venus → Navamsa(D9) → Upapada(UL) → Darakaraka → Synastry
"""

from typing import Dict, List, Any, Optional

try:
    from punarphoo import circular_degree_gap, detect_punarphoo
except ModuleNotFoundError:  # pragma: no cover - script/module dual import
    from scripts.punarphoo import circular_degree_gap, detect_punarphoo

try:
    from timing_negative_holdout_audit import build_report as build_negative_holdout_report
except ModuleNotFoundError:  # pragma: no cover - script/module dual import
    from scripts.timing_negative_holdout_audit import build_report as build_negative_holdout_report

SIGN_LORDS = {
    'Aries': 'Mars', 'Taurus': 'Venus', 'Gemini': 'Mercury', 'Cancer': 'Moon',
    'Leo': 'Sun', 'Virgo': 'Mercury', 'Libra': 'Venus', 'Scorpio': 'Mars',
    'Sagittarius': 'Jupiter', 'Capricorn': 'Saturn', 'Aquarius': 'Saturn', 'Pisces': 'Jupiter',
}

ROMANTIC_ACTIVATION_LORDS = ('5L', '7H_Moon', 'Punarphoo')
LEGAL_MARRIAGE_LORDS = ('Venus', 'DK', 'UL')
# PD never drives event_class. 5L / 7H Moon / 7L / Venus legal hits use MD/AD only.
# Same-structure real cases (female + Saturn-7H chain) trigger at AD, not at
# annual transits. Do not invent a month proxy when the case layer only supports AD.
MD_AD_KEYS = (
    'maha_dasha', 'mahadasha', 'antar_dasha', 'antardasha', 'bhukti',
)
PD_KEYS = ('pratyantar', 'pratyantar_dasha')
VENUS_TO_NATAL_MOON_ORB_DEG = 2.0
EVENT_CLASSES = (
    'romantic_activation',
    'relationship_formation',
    'legal_marriage',
)

SIGNS = ['Aries','Taurus','Gemini','Cancer','Leo','Virgo',
         'Libra','Scorpio','Sagittarius','Capricorn','Aquarius','Pisces']

VENUS_SIGN_EXPRESSIONS = {
    'Aries': '热情主动，追求挑战型伴侣，偏好独立自主的关系',
    'Taurus': '重视感官和稳定，寻求可靠持久的伴侣关系',
    'Gemini': '注重沟通和智力匹配，偏好灵活多变的关系',
    'Cancer': '情感细腻，重视家庭安全感，寻求呵护型关系',
    'Leo': '浪漫大方，需要欣赏和赞美，偏好光芒四射的伴侣',
    'Virgo': '谨慎务实，注重细节服务，偏好可靠稳重的伴侣',
    'Libra': '和谐至上，重视平等尊重，偏好优雅得体的伴侣',
    'Scorpio': '深刻激烈，追求灵魂共鸣，偏好深度连接的关系',
    'Sagittarius': '自由奔放，注重精神成长，偏好开阔视野的伴侣',
    'Capricorn': '务实理性，重视责任担当，偏好成熟稳重的伴侣',
    'Aquarius': '独特自由，重视友谊基础，偏好不拘一格的伴侣',
    'Pisces': '浪漫梦幻，追求灵魂伴侣，偏好温柔体贴的伴侣',
}

RELATIONSHIP_TIMING = {
    1: '年少时感情萌芽（15-22岁）',
    2: '通过家庭/社交圈认识（20-25岁）',
    3: '通过朋友/同事介绍（22-28岁）',
    4: '通过工作/事业认识（25-30岁）',
    5: '通过浪漫邂逅（20-28岁）',
    6: '通过挑战/竞争环境认识（25-32岁）',
    7: '通过公开社交/相亲（24-30岁）',
    8: '通过深层经历/转变认识（28-35岁）',
    9: '通过高等教育/旅行认识（22-30岁）',
    10: '通过事业成就期认识（28-35岁）',
    11: '通过社交网络/大型活动认识（22-32岁）',
    12: '通过隐秘/远程方式认识（25-35岁）',
}


def _lord_of_house(asc_sign: str, house: int) -> Optional[str]:
    if asc_sign not in SIGNS or house < 1 or house > 12:
        return None
    sign = SIGNS[(SIGNS.index(asc_sign) + house - 1) % 12]
    return SIGN_LORDS.get(sign)


def _lords_from_keys(dasha_info: Optional[Dict], keys) -> List[str]:
    if not isinstance(dasha_info, dict):
        return []
    lords: List[str] = []
    for key in keys:
        value = dasha_info.get(key)
        if isinstance(value, dict):
            value = value.get('lord') or value.get('planet')
        if isinstance(value, str) and value:
            lords.append(value)
    return lords


def _dasha_lords(dasha_info: Optional[Dict]) -> List[str]:
    """MD/AD lords only. PD stays in `_pd_lords` and must not drive event_class."""
    return _lords_from_keys(dasha_info, MD_AD_KEYS)


def _pd_lords(dasha_info: Optional[Dict]) -> List[str]:
    return _lords_from_keys(dasha_info, PD_KEYS)


def classify_relationship_event_class(
    planets: Dict,
    asc_sign: str = 'Aries',
    dasha_info: Optional[Dict] = None,
    ul_sign: Optional[str] = None,
    darakaraka: Optional[str] = None,
    transit_saturn_longitude: Optional[float] = None,
    transit_venus_longitude: Optional[float] = None,
) -> Dict[str, Any]:
    """Split romantic activation / relationship formation / legal marriage.

    Punarphoo and 5L remain observation-only until negative holdout passes.
    Never flatten the three tracks into one 婚恋机会 label.
    MD/AD drive event_class. PD is recorded but never promotes a class.
    """
    planets = planets if isinstance(planets, dict) else {}
    fifth_lord = _lord_of_house(asc_sign, 5)
    seventh_lord = _lord_of_house(asc_sign, 7)
    moon = planets.get('Moon') if isinstance(planets.get('Moon'), dict) else {}
    moon_house = moon.get('house')
    moon_lon = moon.get('longitude')
    if moon_lon is None:
        moon_lon = moon.get('lon')
    dasha_lords = _dasha_lords(dasha_info)
    pd_lords = _pd_lords(dasha_info)
    punarphoo = detect_punarphoo(planets, transit_saturn_longitude=transit_saturn_longitude)
    holdout = build_negative_holdout_report()

    natal_romantic: List[str] = []
    if moon_house == 7:
        natal_romantic.append('7H_Moon')
    if punarphoo.get('natal_present'):
        natal_romantic.append('natal_Punarphoo')

    venus_to_moon = None
    if moon_house == 7 and moon_lon is not None and transit_venus_longitude is not None:
        venus_orb = circular_degree_gap(float(moon_lon), float(transit_venus_longitude))
        venus_to_moon = {
            'status': 'observation_only',
            'active': venus_orb <= VENUS_TO_NATAL_MOON_ORB_DEG,
            'orb_deg': venus_orb,
            'orb_limit_deg': VENUS_TO_NATAL_MOON_ORB_DEG,
            'drives_event_class': False,
            'claim_boundary': (
                '7H Venus-to-Moon is annual and not a crush month. '
                'Record only; never add to romantic_hits or point a month. '
                'Falsified by empty 2019-2022 repeats and 2026-02-13→18.'
            ),
        }

    romantic_hits: List[str] = []
    if fifth_lord and fifth_lord in dasha_lords:
        romantic_hits.append('5L_dasha')
    if moon_house == 7 and 'Moon' in dasha_lords:
        romantic_hits.append('7H_Moon_dasha')
    if punarphoo.get('timing_observation_active'):
        romantic_hits.append('Punarphoo')
    # Venus-to-Moon is annual (~once/year). It does not drive event_class.

    formation_hits: List[str] = []
    if seventh_lord and seventh_lord in dasha_lords:
        formation_hits.append('7L_dasha')

    legal_hits: List[str] = []
    if 'Venus' in dasha_lords:
        legal_hits.append('Venus_dasha')

    pd_hits: List[str] = []
    if fifth_lord and fifth_lord in pd_lords:
        pd_hits.append('5L_pd')
    if moon_house == 7 and 'Moon' in pd_lords:
        pd_hits.append('7H_Moon_pd')
    if seventh_lord and seventh_lord in pd_lords:
        pd_hits.append('7L_pd')
    if 'Venus' in pd_lords:
        pd_hits.append('Venus_pd')

    natal_legal: List[str] = []
    if darakaraka:
        natal_legal.append('DK')
    if ul_sign:
        natal_legal.append('UL')

    # Timing-only classification. Natal 7H Moon / Punarphoo / DK / UL stay structure flags.
    # Romantic timing (5L dasha / Moon dasha / Punarphoo return) outranks long 7L MD.
    # PD never drives event_class.
    active_classes: List[str] = []
    if romantic_hits:
        active_classes.append('romantic_activation')
    if legal_hits:
        active_classes.append('legal_marriage')
    if formation_hits:
        active_classes.append('relationship_formation')
    event_class = active_classes[0] if active_classes else None

    return {
        'event_class': event_class,
        'romantic_activation': {
            'lords': list(ROMANTIC_ACTIVATION_LORDS),
            'hits': romantic_hits,
            'natal_structure': natal_romantic,
            'punarphoo': punarphoo,
            'venus_to_natal_moon': venus_to_moon,
        },
        'relationship_formation': {
            'hits': formation_hits,
            'seventh_lord': seventh_lord,
        },
        'legal_marriage': {
            'lords': list(LEGAL_MARRIAGE_LORDS),
            'hits': legal_hits,
            'natal_structure': natal_legal,
        },
        'pd_hits': pd_hits,
        'active_classes': active_classes,
        'negative_holdout': {
            'claim_status': holdout.get('claim_status'),
            'production_tuning_allowed': holdout.get('production_tuning_allowed'),
        },
        'claim_status': 'observation_only',
        'note': (
            '心动看 5L/7H月亮/Punarphoo，成对看 7L/DK/UL，法律婚看 Venus/DK/UL。'
            'Punarphoo 回照 ≠ 结婚；5L 大运 ≠ 法律婚。'
            '金星过本命月亮每年一次，不点月、不升格心动。不得合并成一句婚恋机会。'
        ),
    }


def analyze_relationship(planets: Dict, asc_sign: str = 'Aries',
                         d9_positions: Dict = None, ul_sign: str = None,
                         darakaraka: str = None, dasha_info: Dict = None) -> Dict:
    """
    完整感情分析引擎。

    Args:
        planets: D1行星位置
        asc_sign: 上升星座
        d9_positions: D9分盘位置（可选）
        ul_sign: Upapada Lagna星座（可选）
        darakaraka: DK配偶指示星（可选）
        dasha_info: 大运信息（可选）

    Returns:
        结构化感情分析
    """
    analysis = {
        'partnership_style': [],
        'attraction_pattern': [],
        'timing': [],
        'strength': [],
        'challenges': [],
        'assessment': '',
        'event_class_split': {},
    }

    # 1. Venus分析（关系核心指示星）
    venus = planets.get('Venus', {})
    venus_sign = venus.get('sign', 'Libra')
    venus_house = venus.get('house', 0)

    analysis['partnership_style'].append({
        'planet': 'Venus',
        'sign': venus_sign,
        'house': venus_house,
        'expression': VENUS_SIGN_EXPRESSIONS.get(venus_sign, ''),
        'note': f'Venus在{venus_house}宫 — 感情通过{RELATIONSHIP_TIMING.get(venus_house,"自然相遇")}',
    })

    # 2. 7宫分析
    asc_idx = SIGNS.index(asc_sign) if asc_sign in SIGNS else 0
    h7_sign = SIGNS[(asc_idx + 6) % 12]
    h7_lord = {'Aries':'Mars','Taurus':'Venus','Gemini':'Mercury','Cancer':'Moon',
               'Leo':'Sun','Virgo':'Mercury','Libra':'Venus','Scorpio':'Mars',
               'Sagittarius':'Jupiter','Capricorn':'Saturn','Aquarius':'Saturn','Pisces':'Jupiter'}[h7_sign]

    h7_data = planets.get(h7_lord, {})
    h7_house = h7_data.get('house', 0)

    analysis['strength'].append({
        'indicator': '7宫主',
        'lord': h7_lord,
        'sign': h7_sign,
        'house': h7_house,
        'note': f'7宫主{h7_lord}在{h7_house}宫，关系质量与该宫主题紧密相关',
    })

    # 3. 7宫内行星
    h7_planets = []
    for pn, pd in planets.items():
        if pd.get('house') == 7 and pn not in ('Rahu','Ketu'):
            h7_planets.append(pn)
    if h7_planets:
        analysis['strength'].append({
            'indicator': '7宫内行星',
            'planets': h7_planets,
            'note': f'7宫有{", ".join(h7_planets)}，关系领域活跃，伴侣显著影响',
        })

    # 4. Darakaraka分析
    if darakaraka:
        dk_data = planets.get(darakaraka, {})
        analysis['attraction_pattern'].append({
            'type': 'DK配偶指示星',
            'planet': darakaraka,
            'note': f'DK={darakaraka}，伴侣具有{darakaraka}特质 — {PLANET_PARTNER.get(darakaraka, "")}',
        })

    # 5. D9 Navamsa
    if d9_positions:
        d9_venus = d9_positions.get('Venus', {})
        d9_h7 = d9_positions.get('7th_lord', {})
        analysis['strength'].append({
            'indicator': 'D9 Navamsa',
            'venus_sign': d9_venus.get('sign', ''),
            'note': f'D9 Venus在{d9_venus.get("sign","")} — 深层关系需求',
        })

    # 6. UL Upapada Lagna
    if ul_sign:
        analysis['attraction_pattern'].append({
            'type': 'UL配偶镜像',
            'sign': ul_sign,
            'note': f'UL={ul_sign} — 社会视角下的伴侣形象和婚姻质量',
        })

    # 7. 时间窗口：心动 / 成对 / 法律婚分轨，禁止合并成一句「婚恋机会」
    event_split = classify_relationship_event_class(
        planets,
        asc_sign=asc_sign,
        dasha_info=dasha_info,
        ul_sign=ul_sign,
        darakaraka=darakaraka,
    )
    analysis['event_class_split'] = event_split
    md = ''
    if dasha_info:
        md = dasha_info.get('maha_dasha', '') or dasha_info.get('mahadasha', '')
    event_class = event_split.get('event_class') or 'unclassified'
    analysis['timing'].append({
        'dasha': md,
        'event_class': event_class,
        'note': (
            f'当前{md or "未提供"}大运；事件类={event_class}。'
            '心动看 5L/7H月亮/Punarphoo，成对看 7L/DK/UL，法律婚看 Venus/DK/UL；不得合并。'
        ),
    })

    # 8. 综合评估
    venus_strength = 0
    if venus_house in (1, 4, 7, 10, 5, 9):
        venus_strength += 2
    if planets.get('Moon', {}).get('house') in (4, 7, 10):
        venus_strength += 1
    if h7_lord in planets and planets[h7_lord].get('dignity', '') in ('own','exalted','friendly'):
        venus_strength += 2

    if venus_strength >= 4:
        analysis['assessment'] = '感情配置强，关系发展顺利，伴侣质量高'
    elif venus_strength >= 2:
        analysis['assessment'] = '感情基础良好，需注意关系中的沟通和理解'
    else:
        analysis['assessment'] = '感情需要更多耐心和策略，建议关注自我成长后再建立关系'

    return analysis


PLANET_PARTNER = {
    'Sun': '伴侣有领导力、自信、独立自主',
    'Moon': '伴侣温柔体贴、情感丰富、关心家庭',
    'Mars': '伴侣有行动力、勇敢直接、活力充沛',
    'Mercury': '伴侣聪明机智、善于沟通、思维敏捷',
    'Jupiter': '伴侣有智慧慷慨、教育背景好、有信仰',
    'Venus': '伴侣有魅力、艺术气质、注重和谐',
    'Saturn': '伴侣成熟稳重、有责任心、事业有成',
}