+j 8UdZddlZddlmZmZddlmZGddeZddd d d iZe e efe d <d e de fdZ e dddZe e efe d<de de defdZde e effdZdejdzdzdddiidejdZe e efe d<deDZieeZe e efe d< d'd e e e e effd!e d"e d#edef d$Zd e e e e effd!e d"e de fd%Zde e effd&ZdS)(z Feature flags module for ComfyUI WebSocket protocol negotiation. This module handles capability negotiation between frontend and backend, allowing graceful protocol evolution while maintaining backward compatibility. N)Any TypedDict)argsc.eZdZUeed<eed<eed<dS)FeatureFlagInfotypedefault descriptionN)__name__ __module__ __qualname__str__annotations__r7/home/wildlama/comfy/ComfyUI/comfy_api/feature_flags.pyrrs0 III LLLrrshow_signin_buttonboolFz?Show the sign-in button in the frontend even when not signed in)rr r CLI_FEATURE_FLAG_REGISTRYvreturncn|}|dkrdS|dkrdStd|)zStrict bool coercion: only 'true'/'false' (case-insensitive). Anything else raises ValueError so the caller can warn and drop the flag, rather than silently treating typos like 'ture' or 'yes' as False. trueTfalseFz expected 'true' or 'false', got )lower ValueError)rrs r _coerce_boolrsF GGIIE t u === > >>rc t|SN)intrs rr"/s SVVrc t|Sr)floatr!s rr"r"0s uQxxr)rr r$ _COERCE_FNSkey raw_valuect|}||St|d}||S||S)a9Coerce a raw string value using the registry type, or keep as string. Returns the raw string if the key is unregistered or the type is unknown. Raises ValueError/TypeError if the key is registered with a known type but the value cannot be coerced; callers are expected to warn and drop the flag. Nr)rgetr%)r&r'infocoerces r_coerce_flag_valuer,4sR % ( ( - -D | __T&\ * *F ~ 6)  rc i}ttdgD]}|d\}}}|}|s2|sd} t ||||<]#t t f$rc}t|i}tj d|||dd|Yd}~d}~wwxYw|S)aMParse --feature-flag key=value pairs from CLI args into a dict. Items without '=' default to the value 'true' (bare flag form). Flags whose value cannot be coerced to the registered type are dropped with a warning, so a typo like '--feature-flag some_bool=ture' does not silently take effect as the wrong value. feature_flag=rz@Could not coerce --feature-flag %s=%r to %s (%s); dropping flag.r?N) getattrr partitionstripr,r TypeErrorrr)loggingwarning)resultitemr&sepr'er*s r_parse_cli_feature_flagsr;Ds Fnb11"nnS11S)iikk   I ,S)//2C2CDDF3KKI&   ,00b99D ORY__&&(=(=q          Ms%A66C*AC%%C*Timanager supports_v4)supports_preview_metadatamax_upload_size extensionnode_replacementsassets_CORE_FEATURE_FLAGSc,i|]\}}|tv||Sr)rC).0krs r rGis) b b btq!QNaEaEaaEaEaEarSERVER_FEATURE_FLAGSsockets_metadatasid feature_namer cn||vr|S||di||S)aM Get a feature flag value for a specific connection. Args: sockets_metadata: Dictionary of socket metadata sid: Session ID of the connection feature_name: Name of the feature to check default: Default value if feature not found Returns: Feature value or default if not found feature_flags)r))rIrJrKr s rget_connection_featurerNns@$ """ C $ $_b 9 9 = =lG T TTrc*t|||dduS)a Check if a connection supports a specific feature. Args: sockets_metadata: Dictionary of socket metadata sid: Session ID of the connection feature_name: Name of the feature to check Returns: Boolean indicating if feature is supported FT)rN)rIrJrKs rsupports_featurerPs ""2Cu M MQU UUrc4tS)zb Get the server's feature flags. Returns: Dictionary of server feature flags )rHcopyrrrget_server_featuresrSs $ $ & &&r)F)__doc__r5typingrrcomfy.cli_argsrrrdictrrrrr%r,r;r? enable_assetsrCitems _cli_flagsrHrNrPrSrrrr[s!!!!!!!!iX94_ 45 ?C ?D ? ? ? ?      T#s(^ C C C     $sCx.:"&+d2T9mT23 ''T#s(^c b7799??AA b b b 'L*='L'Ld38nLLL UU3S#X./U UU U  UUUU0V3S#X./V VV VVVV&'T#s(^''''''r