:j &@dZddlZddlmZddlmZddlmZddlm Z ddl m Z dd l m Z dd lmZed ZGd d e ZddZde fdZde fdZdeefdZdeefdZddedeeeffdZdefdZdefdZddZdeedeeddfdZdeeddfdZ y) a Resource Manager - Thread-safe Resource Management with Dependency Injection This module provides a centralized resource manager that eliminates global state and implements dependency injection pattern for better testability and maintainability. Key Features: - Thread-safe resource management without global variables - Dependency injection pattern for better testability - Singleton pattern for resource sharing - Automatic initialization from configuration - Resource lifecycle management N)Optional) get_config)Config) IAuthProvider)Agents) Credentials) get_loggermanagerceZdZUdZdZeded<ejZ ddee fdZ e ddee ddfdZ e ddZdd Zdefd Zdefd Zdeefd Zdeefd ZddedeeeffdZdefdZy)ResourceManagerz4Thread-safe resource manager using singleton patternN _instanceconfigc| t}||_d|_d|_d|_t j |_y)zInitialize resource manager with configuration Args: config: Configuration object, loads from global config if None NF)r_config _credentials_agents _initialized threadingLock_lock)selfrs 8/root/.openclaw/workspace/harvester/tools/coordinator.py__init__zResourceManager.__init__%s< >\F 37)- !^^% returnc|j:|j5|j |||_ddd|jS|jS#1swY|jSxYw)zGet singleton instance of resource manager Args: config: Configuration object for initialization Returns: ResourceManager: Singleton instance N)rr)clsrs r get_instancezResourceManager.get_instance4s[ ==  0==($'KCM 0}}s}} 0}}s AA'cT|j5d|_dddy#1swYyxYw)z-Reset singleton instance for testing purposesN)rr)rs rreset_instancezResourceManager.reset_instanceDs'YY ! CM ! ! !s'c |j5|jr dddy |jjj}t |j |j|jj|_ tjdt|j dt|jd|jjj}t||_tjdt|dd|_ dddy#t"$r}tj%d |d}~wwxYw#1swYyxYw) /Initialize resource managers from configurationN)sessionstokensstrategyz%Initialized credentials manager with z sessions and  tokensz$Initialized User-Agent manager with  agentsTz(Failed to initialize resource managers: )rrr global_configgithub_credentialsr r$r%r&valuerloggerinfolen user_agentsrr Exceptionerror)r github_credsr/es r initializezResourceManager.initializeJs0 ZZ      #||99LL $/)22<;N;NYeYnYnYtYt%! ;C @U@Us rrAzResourceManager.get_tokensF  '')335 5  LL04 5 r? prefer_tokenc@|jj|S)aGet next credential with type preference Args: prefer_token: Whether to prefer API tokens over sessions Returns: tuple[str, str]: (credential_value, credential_type) Raises: RuntimeError: If no credentials available )r9get_credential)rrBs rrDzResourceManager.get_credentials##%44\BBrc |jjS#t$r"}tj d|Yd}~yd}~wwxYw)zaGet random User-Agent string Returns: str: Random User-Agent string zFailed to get User-Agent: NzoMozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/138.0.0.0 Safari/537.36)r;getr0r,r1r>s rget_user_agentzResourceManager.get_user_agentsJ  E??$((* * E LL5aS9 :E Er?)NrNT)__name__ __module__ __qualname____doc__rr__annotations__rrrrr classmethodrr!r4r r9rr;strr=rAbooltuplerDrGrrr r s>-1Ix)*1 INN E &x/ & (6"2 >O  !! 8 ! ! F  Xc]  8C=  C4 C5c? C E Err rcLtj}|jy)r#N)r rr4)r s r init_managersrUs**,G rcFtjjS)zGet credentials manager instance Returns: Credentials: Credentials manager instance Raises: RuntimeError: If not initialized )r rr9rSrrget_credentials_managerrWs  ' ' ) 9 9 ;;rcFtjjS)zGet agents manager instance Returns: Agents: Agents manager instance Raises: RuntimeError: If not initialized )r rr;rSrrget_agents_managerrYs  ' ' ) 4 4 66rcFtjjS)znGet next GitHub session token Returns: Optional[str]: Session token or None if not available )r rr=rSrrr=r=s  ' ' ) 5 5 77rcFtjjS)zfGet next GitHub API token Returns: Optional[str]: API token or None if not available )r rrArSrrrArAs  ' ' ) 3 3 55rrBcHtjj|S)aGet next credential with type preference Args: prefer_token: Whether to prefer API tokens over sessions Returns: tuple[str, str]: (credential_value, credential_type) Raises: RuntimeError: If no credentials available )r rrD)rBs rrDrDs  ' ' ) 8 8 FFrcFtjjS)zUGet random User-Agent string Returns: str: Random User-Agent string )r rrGrSrrrGrGs  ' ' ) 8 8 ::rcji} tj}|jrR|jr|jj |d<|j r|j j |d<|S#t $r1}tjd|t||d<Yd}~|Sd}~wwxYw)zpGet statistics from all resource managers Returns: dict: Combined statistics from all managers credentialsagentszFailed to get managers stats: r1N) r rrr get_statsrr0r,r1rP)statsr r3s rget_managers_statsrcs E  !..0   ##'.';';'E'E'Gm$")//";";"=h L   5aS9:Qg L  sA2A88 B2&B--B2cf tj}|jrb|jr|jj |j r|j j t jdyy#t$r"}t jd|Yd}~yd}~wwxYw)z*Reset statistics for all resource managersz Failed to reset managers stats: N) r rrr reset_statsrr,r-r0r1)r r3s rreset_managers_statsrfs =!..0   ##$$002++- KKD E = 7s;<<=sBB B0B++B0r$r%c tj}|jrr|jre|jj ||jj |t jdt|dt|dyyy#t$r"}t jd|Yd}~yd}~wwxYw)zUpdate credentials in the resource manager Args: sessions: New list of session tokens tokens: New list of API tokens zUpdated credentials: z sessions, r'zFailed to update credentials: N) r rrrupdate_sessions update_tokensr,r-r.r0r1)r$r%r r3s rupdate_credentialsrj!s;!..0   G$8$8  0 0 :  . .v 6 KK/H k#f+V]^ _%9  ; 5aS9::;sBB CB;;Cr/c8 tj}|jrK|jr>|jj |t j dt|dyyy#t$r"}t jd|Yd}~yd}~wwxYw)zuUpdate User-Agent strings in the resource manager Args: user_agents: New list of User-Agent strings zUpdated User-Agents: r(zFailed to update User-Agents: N) r rrr update_agentsr,r-r.r0r1)r/r r3s rupdate_user_agentsrm2s ;!..0   GOO OO ) )+ 6 KK/K0@/AI J%4  ; 5aS9::;sA)A.. B7BBrHrI)!rMrtypingrrrconfig.schemasr core.typesragentr credentialr r,r r rUrWrYrPr=rArQrRrDrGdictrcrflistrjrmrSrrrus !$# I REmREl < < 7F 78Xc]868C=6 G GsCx G;;D, =;c;DI;$;" ;DI ;$ ;r