
    i="                         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   target_strippedtarget_lowerpartsplatform_strr   r   r   s
             5/home/ubuntu/.hermes/hermes-agent/gateway/delivery.pyparsezDeliveryTarget.parse-   s    !,,..&,,..8## 	Ds#_"N$."	    sHNdCCCC7""3//// /!!#))#q11E 8>>++L"%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$   1D  D21D26E  E43E4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0 003 00(? 00K[ 00 00 00 [00d
#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)r;   r<   r
   
output_dir)r,   r;   r<   s      r(   __init__zDeliveryRouter.__init__u   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)rH   error)r   r   r   _deliver_local_deliver_to_platformr-   	Exceptionr5   )
r,   rB   rC   rD   rE   rF   resultsr   rI   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joinr5   )
r,   rB   rD   rE   rF   rZ   output_pathlineskeyr+   s
             r(   rK   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.rR   r>   r?   TrT   _z.txt)r   r[   r\   r
   r^   ra   )r,   rB   rD   rZ   out_dirrY   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	 deliveryrD   unknownu4   Cron output truncated (%d chars) — full output: %sNz'

... [truncated, full output saved to ]r   )rF   )r<   getr   r"   r+   r   r!   MAX_PLATFORM_OUTPUTri   loggerinfoTRUNCATED_VISIBLEdictr   send)r,   r   rB   rF   adapterrD   
saved_pathsend_metadatas           r(   rL   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)r0   r1   r2   r3   r   r   r   r	   rA   r5   r   r   r   rP   rK   r   ri   rL   r8   r.   r(   r:   r:   m   s        
@ 
@} 
@Xs]8K 
@ 
@ 
@ 
@  !%"&-1( (( n%( 	(
 3-( 4S>*( 
c3h( ( ( (T,
,
 ,
 3-	,

 4S>*,
 
c3h,
 ,
 ,
 ,
\ c d    [[ [ 4S>*	[
 
c3h[ [ [ [ [ [r.   r:   )r3   loggingpathlibr   r   dataclassesr   typingr   r   r   r	   hermes_cli.configr
   	getLoggerr0   ro   rn   rq   r;   r   r   sessionr   r   r:   r8   r.   r(   <module>r~      sQ                 ! ! ! ! ! ! , , , , , , , , , , , , - - - - - -		8	$	$   + + + + + + + + " " " " " " M# M# M# M# M# M# M# M#`Q[ Q[ Q[ Q[ Q[ Q[ Q[ Q[ Q[ Q[r.   