/ 0 U d Z ddlmZ ddlZddlZddlZddlmZ dZdZ i a
de d< d a d
e d < d"dZ
d#dZd$dZd%dZd&dZd%dZd'dZd'd Zd(d)d!ZdS )*a
Shared reader for the local feature flags file.
The file is written by:
- Go resident-agent FeatureFlags plugin (IM360 mode)
- Python FeatureFlagsSync plugin (AV mode)
Other subsystems (e.g. message_status_publisher) use this module
to check individual flag values at runtime.
Supported JSON shapes on disk (readers / ``is_enabled``):
- Legacy object ``{"mqtt_tracking": true, ...}`` (preferred on disk after sync).
- JSON array of enabled names ``["mqtt_tracking"]`` (still accepted).
- Legacy wrapper ``{"flags": ["mqtt_tracking", ...]}``.
The sync API checksum is always over a **canonical JSON array** of enabled
names (same as correlation sync). ``sync_checksum_hex_from_flags_file`` derives
that MD5 from whatever shape is on disk.
The sync plugin also writes ``FLAGS_PLAIN_PATH`` (``/var/imunify360/feature_flags``):
plain text, one enabled flag name per line (sorted), for scripts.
) annotationsN)Anyz"/var/imunify360/feature_flags.jsonz /var/imunify360/feature_flagsdict[str, Any]
_cached_flags float
_cached_mtimerawr returnc | i S t | t r#i }| D ] }t |t rd||< |S t | t r;| d }t |t rt | S | S i S )z@Map file JSON to a flat name->value dict for :func:`is_enabled`.NTflags)
isinstanceliststrdictget_normalize_flags_from_file)r
outiteminners \/opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/internals/feature_flags.pyr r ' s
{ #t ! !D$$$
! D
#t eT " " 5-e44 4
I c t j t } n# t $ r i adat
cY S w xY w| t k rt
S t t 5 }t j | }d d d n # 1 swxY w Y t | an # t t j f$ r i aY nw xY w| at
S )Nr ) ospathgetmtime
FLAGS_PATHOSErrorr r openjsonloadr JSONDecodeError)mtimefr
s r _read_flagsr% 9 s ,,
*
)A,,C 2377
T) *
M sB $' ??B% )B
> B%
BB% BB% %C ?C r
list[str]c t | t t f s$t dt | j t
| }t d | D S )a Return sorted enabled flag names for JSON and plain-text sidecar.
Accepts the same shapes as :func:`_normalize_flags_from_file` (array,
flat map, ``{"flags": [...]}``) so checksums and sidecars match Go
``enabledNamesSortedForChecksum`` / :func:`is_enabled`.
z flags must be list or dict, not c 3 $ K | ] \ }}||V d S )N ).0kvs r z,enabled_flag_names_sorted..[ s+ 881a8!888888r ) r r r TypeErrortype__name__r sorteditems)r
normalizeds r enabled_flag_names_sortedr4 O sx edD\ * *
EtE{{/C E E
,E22J 88
0 0 2 2888 8 88r namesbytesc r t | }t j |dd S )zYJSON array bytes used for sync MD5 (matches correlation_api checksum_for_sync_flag_list).T sort_keysindentr1 r dumpsencode)r5 ordereds r canonical_sync_flag_list_bytesr@ ^ s0 UmmG :ga 8 8 8 ? ? A AAr r r c R t | d 5 }t j | }ddd n # 1 swxY w Y n## t t t j f$ r Y dS w xY wt
| }t | }t j |d
S )zOMD5 hex of canonical enabled-name array for ``path``, or "" if missing/invalid.zutf-8)encodingN F)usedforsecurity) r r r! r UnicodeDecodeErrorr" r4 r@ hashlibmd5 hexdigest)r r$ r
r5 payloads r !sync_checksum_hex_from_flags_filerJ d s
$
)
)
) Q)A,,C ')= > rr %c * *E,U33G ;w 6 6 6 @ @ B BBs, A 4 A 8A 8A A A c d t d | D D }t j |dd S )zz2legacy_feature_flags_map_bytes..r s KKKQDKKKr c < h | ]}t |t |S r) )r r )r* xs r z1legacy_feature_flags_map_bytes..r s' !I!I!IjC6H6H!I!!I!I!Ir Tr8 r9 r< )r5 ds r legacy_feature_flags_map_bytesrS p sO KK&!I!IU!I!I!IJJKKKA :a4 2 2 2 9 9 ; ;;r c | t | }|sdS d | dz S )zPBody for ``FLAGS_PLAIN_PATH``: one name per line, trailing newline if non-empty.r
)r4 joinr> )r
r5 s r $plain_text_payload_for_enabled_flagsrW v s? %e , ,E s IIe t # + + - --r c t | t r)t j | dd S t dt
| j )zBSerialize dict flags for writing ``FLAGS_PATH`` (legacy map only).Tr8 r9 zflags must be dict, not )r r r r= r> r. r/ r0 )r
s r $serialize_feature_flags_file_payloadrY ~ sX % Dz%4:::AACCC
EtE{{/CEE
F
FFr F flag_namedefaultboolc n t }| | }||S t | S )zReturn whether *flag_name* is enabled.
If the file is missing, unreadable, or the flag is absent,
*default* is returned. Defaults to False so unknown flags are
treated as disabled unless the caller explicitly opts in.
)r% r r\ )rZ r[ r
values r
is_enabledr_ s4
MME IIi E } ;;r )r
r r r )r r )r
r r r& )r5 r& r r6 )r r r r )r
r r r6 )F)rZ r r[ r\ r r\ )__doc__
__future__r rF r r typingr r FLAGS_PLAIN_PATHr __annotations__r r r% r4 r@ rJ rS rW rY r_ r) r r re sQ . # " " " " "
1
2 "
" " " "
$ , 9 9 9 9 B B B B C C C C< < < < . . . .G G G G r