`jeEdZddlmZddlZddlZddlZddlmZddlm Z m Z m Z m Z m Z mZddlmZmZmZmZmZmZmZddlmZmZmZddlmZmZGd d ZdS) a FreeExile Skill & Gem Database Service (PoE2 Multi-Tier Tag Architecture). Manages SQLite persistence, tag validation, PoE2 combo synergies, and build archetype matching. Follows Clean Architecture, high-density queries, and zero-leak connection pooling. ) annotationsN)contextmanager)AnyDictListOptionalSetTuple)SkillTag TagCategoryAttributeAffinity ComboRole FiveElements MartialSectWeaponCategory)get_canonical_skill_tagsget_canonical_build_archetypesget_canonical_passive_nodes)create_standard_active_skillscreate_standard_augment_sigilsceZdZd,d-dZedZd.dZd.d Zd/d0d Zd1dZ d2d3dZ d4d Z d5d$Z d6d(Z d7d+ZdS)8SkillDatabaseServiceNdb_path Optional[str]c&|stjtjtjt ddd}tj|dtj|d}||_|jdkrJtj d|_ tj |j _ |j dnd|_ |dS)Nz..dataT)exist_okzmartial_database.dbz:memory:PRAGMA foreign_keys = ON)ospathabspathjoindirname__file__makedirsrsqlite3connect _memory_connRow row_factoryexecute_init_database)selfrbase_dirs z.s444AG444r1zGINSERT INTO skills VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)cg|] }|j SrErFrHs r/rKz.===AG===r1 excluded_tagscg|] }|j SrErFrHs r/rKz.s<<.s:::QW:::r1zBINSERT INTO build_archetypes VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)cg|] }|j SrErFrHs r/rKz.rMr1z9INSERT INTO passive_nodes VALUES (?, ?, ?, ?, ?, ?, ?, ?))9r9cursorr+fetchonervaluestagrGcategoryname_viname_en descriptionjsondumpsaffected_modifiersrtagsskill_id skill_nameelementsectrPweapon_requirement base_damage energy_cost cooldown_ms cast_time_msanimation_lock_ms combo_role upside_desc downside_descr required_tagshasattrrNsigil_id sigil_name augment_typedamage_multiplierenergy_cost_multiplierprojectile_bonusaoe_radius_multiplierr core_tags archetype_idprimary_attributeplaystyle_summaryrecommended_skillsrecommended_gems combo_guider affected_tagsnode_idcrit_chance_bonusattack_speed_bonusarea_radius_multiplierspecial_mechanic)r-r@r8curtag_defstags_strgreq_tagsex_tagsarvpaff_tagss r/seed_canonical_dataz(SkillDatabaseService.seed_canonical_datas  ! ! # #? t++--C  DEEE<<>>%(1,, ? ? ? ? ? ? ? ?  KK0 1 1 1 KK, - - - KK2 3 3 3 KK6 7 7 7 KK3 4 4 4455<<>>   F[&(8(>_g&94:gF`;a;ac344;;==  44QV444 ]Zqy )/1E1K]AM1=!.(!,*>  ==Q_===@G?@[@[c<R3S3SdQ)//YdZ))4:g+>+>@S-q/A1CZ]AO 5455<<>>  ::ak:::  X^QY 1;N;TY_a&94:i;P;PZ 455tz!BT7U7UWXWdf122  ==Q_=== OY 4:h+?+?AT(!*>@XZ[Zln s? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? sA Q-O QQ Q r`intOptional[Dict[str, Any]]c,|5}|d|f}|s ddddSt|}t j|d|d<|cdddS#1swxYwYdS)u'Lấy thông tin chiêu thức theo ID.z'SELECT * FROM skills WHERE skill_id = ?N tags_jsonr_)r9r+rUdictr\loads)r-r`r8rowress r/ get_skillzSkillDatabaseService.get_skills  ! ! # # t,,H8+VV__aaC         s))C*S%566CK                   s-B -B  B B r_Optional[Set[SkillTag]]rbOptional[FiveElements] attributeOptional[AttributeAffinity]weaponOptional[WeaponCategory]rjOptional[ComboRole]List[Dict[str, Any]]cd}g}|r|dz }||j|r|dz }||j|r/|tjkr|dz }||j|r|dz }||j|5}|||} g} | D]z} t| } tj | d| d<|r5d|D} | t| dse| | {| cd d d S#1swxYwYd S) uFTruy vấn tìm kiếm chiêu thức theo ma trận lọc linh hoạt.zSELECT * FROM skills WHERE 1=1z AND element = ?z AND attribute_affinity = ?z; AND (weapon_requirement = ? OR weapon_requirement = 'ANY')z AND combo_role = ?rr_ch|] }|j SrErFrHs r/ z5SkillDatabaseService.search_skills..s666A666r1N) appendrGrANYr9r+fetchallrr\rissubsetset)r-r_rbrrrjqueryparamsr8rowsresultsritemtag_valss r/ search_skillsz"SkillDatabaseService.search_skillss1  ) ' 'E MM'- ( ( (  + 2 2E MM)/ * * *  (f 222 R RE MM&, ' ' '  , * *E MM** + + +  ! ! # # t<<v..7799D,.G % %Aww#z${*;<<V !66666H#,,Sf->->??! t$$$$                  s-B)E##E'*E'roTuple[bool, str]c ||}|sdd|fS|5}|d|f}|sdd|fcdddSt |}t t j|d}t t j|d}t |d} |r3|| zr.t|| z} dd |d d | fcdddS|rC| | s.t|| z } dd |d d | fcdddSd d|d dfcdddS#1swxYwYdS)uNKiểm tra tính hợp lệ khi liên kết Gem Hỗ Trợ vào Chiêu Thức.Fu!Không tìm thấy chiêu thức z-SELECT * FROM support_gems WHERE sigil_id = ?u(Không tìm thấy Dị Biến Tà Ấn Nrequired_tags_jsonexcluded_tags_jsonr_uDị Biến Tà Ấn 'rpu' xung đột với tag: u' thiếu tag bắt buộc: Tu"Liên kết Dị Biến Tà Ấn 'u' thành công) rr9r+rUrrr\rlistr) r-r`roskillr8rgemrr skill_tagsconflictmissings r/validate_gem_linkz&SkillDatabaseService.validate_gem_linksx(( IHhHHH H  ! ! # # `t,,NQYP[\\eeggC TSSSS ` ` ` ` ` ` ` `s))C4:c*>&?@@AAH$*S)=%>??@@GU6]++J nGj0 n* 455ms<7Hmmckmmm ` ` ` ` ` ` ` ` p 1 1* = = px*455os<7Hoofmooo ` ` ` ` ` ` ` `_c,>O____ ` ` ` ` ` ` ` ` ` ` ` ` ` ` ` ` ` `s$2E82BE88E8 E88E<?E< starter_id finisher_idDict[str, Any]c||}||}|r|sdddSt|d}t|d}g}d}tjj|vp|dt jjk} tjj|vp|dt jjk} | r| r|dz }|dtjj|vr-tj j|vr|d z }|d tj j|vr@tj j|vr-tj j|vr|d z }|d tj j|vrStjj|vs&tjj|vstjj|vr|d z }|dt#|dk|||d|ddS)uNĐánh giá hiệp đồng Combo PoE2 giữa đòn Starter và đòn Finisher.Fu Chiêu thức không tồn tại)synergy_activereasonr_?rjg?uhChuỗi Combo PoE2: Khai Chiêu Mồi Bẫy -> Kích Nổ Tuyệt Sát (+150% Sát thương bộc phát)g?uqHiệp đồng Băng Phách & Trọng Kích: Đập vỡ mục tiêu đóng băng gây nổ vụn băng (Shatter)g333333?unHiệp đồng Hủ Độc & Nổ Xác: Kích nổ dịch độc lây lan toàn bản đồ (Toxin Detonation)g?umHiệp đồng Phá Giáp & Cương Kình: Xé toạc da thịt gây Xuất Huyết gấp đôi (Armor Shred)rra)rcombo_multiplier synergies starter_name finisher_name)rrr COMBO_STARTERrGrCOMBO_FINISHERrCOLDSLAMPOISONAOECORPSE ARMOR_BREAKPHYSICALPHYSIQUE_ALIGNEDMELEElen) r-rrstarterfinishers_tagsf_tagsr multiplier has_starter has_finishers r/evaluate_combo_synergyz+SkillDatabaseService.evaluate_combo_synergysF..,,>>+.. [h [&+7YZZ ZWV_%%Xf%&&!  ,2f<v @UYbYpYv@v .4>z(z=SkillDatabaseService.match_build_archetypes..as q)>'?r1T)keyreverseN)rrupdater9r+rrr\rrrroundrsort)r-rr combined_tagssidrr8rrrrrv intersectionscores r/match_build_archetypesz+SkillDatabaseService.match_build_archetypesAs #&%%  0 0Cs##A 0$$QvY///  ! ! # # t<< @AAJJLLDG  Aww 40@+A B BCC ,y8 >GPL))C NN::SD==NN(,^(<#' ?#' ?#' ?-12E-F/4US[!/D/D(,\(:(:-12E-F'+M': $ $    LL??L N N N+                  sDE::E>E>node_idsDict[str, float]c||}|r|sdddddSt|d}d}d}d}d}|5} ddt |z} | d| d|} | D]Z} ttj| d } | |zr,|| d z}|| d z }|| d z }|| d z}[ dddn #1swxYwYt|dt|dt|dt|ddS)ugTính toán tổng chỉ số khuếch đại từ các điểm Huyết Cốt Ma Đồ dựa trên Tag.rr)rrrrrr_,?z.SELECT * FROM passive_nodes WHERE node_id IN ()affected_tags_jsonrrrrrN) rrr9r"rr+rr\rr)r-r`rrrdmg_mult spd_bonus crit_bonus area_multr8 placeholdersrrp_tagss r/calculate_passive_bonusz,SkillDatabaseService.calculate_passive_bonusdsx(( BH B),C^a~ABB BU6]##    ! ! # # =t88C#h--$788L<< `Q] ` ` `bjkkttvvD = =TZ*>(?@@AAF?="5 66H#7!88I!$7"88J#;!<r sp #""""" %%%%%%8888888888888888  ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ r1