
    kjF                         d dl mZ d dl mZ d dlmZ d dlmZ d dlmZ d dlm	Z	 dede
fd	Z G d
 de      Z e       Zy)    )InteractiveContentview)Metadata)
yaml_dumps)
yaml_loads)
DNSMessage)pack_messagemetadatareturnc                 H    t        | j                  xs | j                        S )N)booltcp_messagehttp_message)r	   s    i/home/ubuntu/.cache/pipx/cdb5059130eadca/lib/python3.12/site-packages/mitmproxy/contentviews/_view_dns.py_is_dns_tcpr   	   s    $$=(=(=>>    c                   L    e Zd ZdZdededefdZdededefdZdedede	fdZ
y	)
DNSContentviewyamldatar	   r
   c                     t        |      r|dd  }t        j                  |      j                         }|d= |j	                  dd        t        |      S )N   status_code	timestamp)r   r   unpackto_jsonpopr   )selfr   r	   messages       r   prettifyzDNSContentview.prettify   sQ    
 x 8D##D)113M"K&'""r   
prettifiedc                 |    t        |      }t        j                  |      }t        |t	        |      rd      S d      S )Ntcpudp)r   r   	from_jsonr   r   )r   r    r	   r   r   s        r   reencodezDNSContentview.reencode   s:    
 *%&&t,Gk(.CUOOOOr   c                 "   t        |j                  dk(  xsv t        |j                  xr_ |j                  j                  xrG |j                  j                  j
                  xr% |j                  j                  j
                  d   dv             S )Nzapplication/dns-message   )5   i  )floatcontent_typer   flowserver_connaddress)r   r   r	   s      r   render_priorityzDNSContentview.render_priority%   s    
 !!%>>  GMM--GMM--55G MM--55a8JF	
 	
r   N)__name__
__module____qualname__syntax_highlightbytesr   strr   r%   r)   r.    r   r   r   r      so    
#
# 
# 
	
#PP P 
	P

 
 
	
r   r   N)mitmproxy.contentviews._apir   r   mitmproxy.contentviews._utilsr   r   mitmproxy.dnsr   mitmproxy.proxy.layers.dnsr   r   r   r   dnsr5   r   r   <module>r;      sB    > 0 4 4 2 3?( ?t ?%
+ %
P r   