i UdZddlZddlZddlZddlmZddlmZddlmZddl m Z ddl m Z eGdd Z iZeee fed <d e eefd e fd Z dd e eefdeded efdZded eefdZd eded dfdZy)u1MEMORY.md 관리 모듈 - Frozen Snapshot 패턴. Hermes Agent의 memory_tool.py MemoryStore 설계를 참고하여 dev2-team 시스템에 맞게 재설계한 경량 메모리 관리 모듈. 주요 기능: - Frozen Snapshot: 세션 시작 시 파일을 읽어 FrozenMemory 객체로 캐시. 재호출 시 동일 객체 반환(캐시). 파일 변경이 세션 중 반영되지 않음. - 문자 제한: update_memory()에서 max_chars 초과 시 거부. - 인젝션 탐지: update_memory() 호출 시 injection_guard 재사용. - 파일 잠금: fcntl.flock 사용 (Linux 전용). Usage: from utils.memory_manager import load_frozen_memory, update_memory snapshot = load_frozen_memory("/path/to/MEMORY.md") print(snapshot.content) ok = update_memory("/path/to/MEMORY.md", "new content") N) dataclass)datetime)Path)Union) scan_contentc0eZdZUdZeed<eed<eed<y) FrozenMemoryu9세션 시작 시 캡처된 메모리 스냅샷 (불변).content snapshot_time char_countN)__name__ __module__ __qualname____doc__str__annotations__rintE/home/jay/workspace/.worktrees/task-2116-dev1/utils/memory_manager.pyr r !sC LOrr _snapshot_cachepathreturnc`tt|j}|tvr t|St|}|j r |j d}nd}t|tjt|}|t|<|S#t tf$rd}YGwxYw)uVMEMORY.md를 읽어 FrozenMemory 스냅샷을 반환합니다. 동일 경로를 재호출하면 캐시된 동일 객체를 반환합니다. 파일이 없으면 빈 FrozenMemory를 반환합니다. Args: path: MEMORY.md 파일 경로 (str 또는 Path). Returns: FrozenMemory: 캡처된 스냅샷 객체. utf-8encoding)r r r ) rrresolverexists read_textOSErrorIOErrorr rnowlen)rresolved file_pathr snapshots rload_frozen_memoryr)1s4:%%'(H?"x((XI ))7);Gllnw<H !)OH O! G sBB-,B- new_content max_charsct|j}t|}|jsyt ||kDry |j j dd|j|jdz} t|d} tj|tjt||tj|tj|j!y#t$rYywxYw#tj|tj|j!wxYw#tt"t$f$rYywxYw)uMEMORY.md를 새 내용으로 갱신합니다. 인젝션 패턴 탐지와 문자 수 제한을 검사합니다. 파일 잠금(fcntl.flock)으로 동시 쓰기를 방지합니다. Args: path: MEMORY.md 파일 경로 (str 또는 Path). new_content: 저장할 새 내용. max_chars: 최대 허용 문자 수 (기본값 2200). Returns: bool: 성공 시 True, 실패 시 False. FT)parentsexist_okz.lockw)rrris_safer%parentmkdirr" with_suffixsuffixopenfcntlflockLOCK_EX _atomic_writeLOCK_UNcloser# RuntimeError)rr*r+r' scan_result lock_pathlock_fds r update_memoryr@Ts$T ""$I{+K    ;)#td; %%i&6&6&@AI y#&  KK / )[ 1 KK / MMO #  KK / MMO Wl +s<C0> D8 0C?;4D80 C<;C<?6D55D88EEr ct|}|jrgS|jDcgc]}|jc}Scc}w)uX메모리 내용에서 인젝션 패턴을 탐지합니다. injection_guard.scan_content()를 재사용하여 탐지된 패턴 이름 목록을 반환합니다. Args: content: 검사할 문자열. Returns: list[str]: 탐지된 위협의 pattern_name 목록. 안전하면 빈 리스트. )rr0threats pattern_name)r resultthreats rscan_memory_injectionrFs8' "F ~~ .4nn =FF   == =s>ctjt|jdd\}} t j |dd5}|j ||jt j|jdddt j|t|y#1swY)xYw#t$r' t j|#t$rYwxYwwxYw)u1임시 파일 + os.replace()로 원자적 쓰기.z.tmpz.mem_)dirr4prefixr/rrN)tempfilemkstemprr1osfdopenwriteflushfsyncfilenoreplace BaseExceptionunlinkr")rr fdtmp_pathfs rr9r9s##  LB  YYr3 1 !Q GGG  GGI HHQXXZ  ! 8SY'  ! !   IIh      sHCAB5 'C5B>:C C1 C! C1! C-*C1,C--C1)i)rr6rLrJ dataclassesrrpathlibrtypingrutils.injection_guardrr rdictrrr)rboolr@listrFr9rrrr_s* !.  ,.c<'(- U39- , L1 T 111 1h>3>49>0str