
    ie!                         d Z ddlZddlmZ ddlmZ ddlmZ ddlmZm	Z	m
Z
mZ ddlmZ  ej        e          ZdZd	Zd
dlmZmZ d
dlmZ e G d d                      Z G d d          ZdS )u0  
Delivery routing for cron job outputs and agent responses.

Routes messages to the appropriate destination based on:
- Explicit targets (e.g., "telegram:123456789")
- Platform home channels (e.g., "telegram" → home channel)
- Origin (back to where the job was created)
- Local (always saved to files)
    N)Path)datetime)	dataclass)DictListOptionalAny)get_hermes_homei  i     )PlatformGatewayConfig)SessionSourcec                       e Zd ZU dZeed<   dZee         ed<   dZ	ee         ed<   dZ
eed<   dZeed<   edd	ed
ee         dd fd            ZdefdZdS )DeliveryTargetu   
    A single delivery target.
    
    Represents where a message should be sent:
    - "origin" → back to source
    - "local" → save to local files
    - "telegram" → Telegram home channel
    - "telegram:123456" → specific Telegram chat
    platformNchat_id	thread_idF	is_originis_explicittargetoriginreturnc                    |                                                                 }|dk    r7|r | |j        |j        |j        d          S  | t
          j        d          S |dk    r | t
          j                  S d|v r|                    dd          }|d	         }t          |          d
k    r|d
         nd}t          |          dk    r|d         nd}	 t          |          } | |||d          S # t          $ r  | t
          j                  cY S w xY w	 t          |          } | |          S # t          $ r  | t
          j                  cY S w xY w)u   
        Parse a delivery target string.
        
        Formats:
        - "origin" → back to source
        - "local" → local files only
        - "telegram" → Telegram home channel
        - "telegram:123456" → specific Telegram chat
        r   T)r   r   r   r   )r   r   local)r   :   r   r   N)r   r   r   r   )
striplowerr   r   r   r   LOCALsplitlen
ValueError)clsr   r   partsplatform_strr   r   r   s           8/home/agentuser/.hermes/hermes-agent/gateway/delivery.pyparsezDeliveryTarget.parse-   s    %%''X 	Ds#_"N$."	    sHNdCCCCW3//// &==LLa((E 8L"%e**q..eAhhdG$'JJNNaI4#L11sHg`deeee 4 4 4sHN3333334
	0''H3)))) 	0 	0 	03//////	0s$   C; ; DD"D= = E E c                     | j         rdS | j        t          j        k    rdS | j        r%| j        r| j        j         d| j         d| j         S | j        r| j        j         d| j         S | j        j        S )zConvert back to string format.r   r   r   )r   r   r   r   r   r   value)selfs    r&   	to_stringzDeliveryTarget.to_string^   s    > 	8=HN**7< 	LDN 	Lm)KKDLKK4>KKK< 	;m)::DL:::}""    N)__name__
__module____qualname____doc__r   __annotations__r   r   strr   r   boolr   classmethodr   r'   r+    r,   r&   r   r      s           !GXc]!!!#Ix}###ItK.0 .03 .0(? .0K[ .0 .0 .0 [.0`
#3 
# 
# 
# 
# 
# 
#r,   r   c                   d   e Zd ZdZddedeeef         fdZ	 	 	 dde	de
e         dee	         d	ee	         d
eee	ef                  dee	ef         fdZde	dee	         d	ee	         d
eee	ef                  dee	ef         f
dZde	de	defdZdede	d
eee	ef                  dee	ef         fdZdS )DeliveryRouterz
    Routes messages to appropriate destinations.
    
    Handles the logic of resolving delivery targets and dispatching
    messages to the right platform adapters.
    Nconfigadaptersc                 X    || _         |pi | _        t                      dz  dz  | _        dS )z
        Initialize the delivery router.
        
        Args:
            config: Gateway configuration
            adapters: Dict mapping platforms to their adapter instances
        cronoutputN)r9   r:   r
   
output_dir)r*   r9   r:   s      r&   __init__zDeliveryRouter.__init__s   s1      B)++f4x?r,   contenttargetsjob_idjob_namemetadatar   c                 `  K   i }|D ]}	 |j         t          j        k    r|                     ||||          }n|                     |||           d{V }d|d||                                <   i# t          $ r1}	dt          |	          d||                                <   Y d}	~	d}	~	ww xY w|S )a  
        Deliver content to all specified targets.
        
        Args:
            content: The message/output to deliver
            targets: List of delivery targets
            job_id: Optional job ID (for cron jobs)
            job_name: Optional job name
            metadata: Additional metadata to include
        
        Returns:
            Dict with delivery results per target
        NT)successresultF)rF   error)r   r   r   _deliver_local_deliver_to_platformr+   	Exceptionr3   )
r*   r@   rA   rB   rC   rD   resultsr   rG   es
             r&   deliverzDeliveryRouter.deliver   s      *  	 	F?hn44!00&(HUUFF#'#<#<VWh#W#WWWWWWWF  $$/ /((**++    $ VV/ /((**++++++ s   A%A00
B+:'B&&B+c                    t          j                                        d          }|r| j        |z  | dz  }n| j        dz  | dz  }|j                            dd           g }|r|                    d|            n|                    d           |                    d           |                    d	t          j                                        d
                      |r|                    d|            |r5|                                D ] \  }}	|                    d| d|	            !|                    d           |                    d           |                    d           |                    |           |                    d	                    |                     t          |          |dS )zSave content to local files.%Y%m%d_%H%M%Sz.mdmiscTparentsexist_okz# z# Delivery Output z**Timestamp:** z%Y-%m-%d %H:%M:%Sz**Job ID:** z**z:** z---
)path	timestamp)r   nowstrftimer>   parentmkdirappenditems
write_textjoinr3   )
r*   r@   rB   rC   rD   rX   output_pathlineskeyr)   s
             r&   rI   zDeliveryRouter._deliver_local   s    LNN++O<<	 	G/F2	5F5F5FFKK/F2	5F5F5FFK   ===  	.LLh))))LL,---RUx|~~'>'>?R'S'SUUVVV 	2LL000111 	4&nn.. 4 4
U2#225223333RURWtyy//000 $$"
 
 	
r,   c                     t          j                                        d          }t                      dz  dz  }|                    dd           || d| dz  }|                    |           |S )z7Save full cron output to disk and return the file path.rP   r<   r=   TrR   _z.txt)r   rY   rZ   r
   r\   r_   )r*   r@   rB   rX   out_dirrW   s         r&   _save_full_outputz DeliveryRouter._save_full_output   sx    LNN++O<<	!##f,x7dT222F44Y4444   r,   r   c                 h  K   | j                             |j                  }|st          d|j        j                   |j        st          d|j        j         d          t          |          t          k    rm|pi                     dd          }|                     ||          }t          
                    dt          |          |           |dt                   d| d	z   }t          |pi           }|j        rd
|vr
|j        |d
<   |                    |j        ||pd           d{V S )z(Deliver content to a messaging platform.zNo adapter configured for zNo chat ID for z	 deliveryrB   unknownu4   Cron output truncated (%d chars) — full output: %sNz'

... [truncated, full output saved to ]r   )rD   )r:   getr   r"   r)   r   r!   MAX_PLATFORM_OUTPUTrg   loggerinfoTRUNCATED_VISIBLEdictr   send)r*   r   r@   rD   adapterrB   
saved_pathsend_metadatas           r&   rJ   z#DeliveryRouter._deliver_to_platform   sa      -##FO44 	SQ&/:OQQRRR~ 	QOv/DOOOPPP w<<---n"))(I>>F//@@JKKNPST[P\P\^hiii***+KjKKKL 
 X^,, 	:= @ @)/)9M+&\\&.'MDYUY\ZZZZZZZZZr,   r-   )NNN)r.   r/   r0   r1   r   r   r   r	   r?   r3   r   r   r   rN   rI   r   rg   rJ   r6   r,   r&   r8   r8   k   s        
@ 
@} 
@Xs]8K 
@ 
@ 
@ 
@  !%"&-1( (( n%( 	(
 3-( 4S>*( 
c3h( ( ( (T,
,
 ,
 3-	,

 4S>*,
 
c3h,
 ,
 ,
 ,
\ c d    [[ [ 4S>*	[
 
c3h[ [ [ [ [ [r,   r8   )r1   loggingpathlibr   r   dataclassesr   typingr   r   r   r	   hermes_cli.configr
   	getLoggerr.   rm   rl   ro   r9   r   r   sessionr   r   r8   r6   r,   r&   <module>r|      sQ                 ! ! ! ! ! ! , , , , , , , , , , , , - - - - - -		8	$	$   + + + + + + + + " " " " " " K# K# K# K# K# K# K# K#\Q[ Q[ Q[ Q[ Q[ Q[ Q[ Q[ Q[ Q[r,   