o !dR)@sddlZddlZddlZddlZddlmZddlmZddlm Z ddl m Z ddl m Z ddZdd d Zd d ZGddde ZGddde ZGdddeZGdddeZGdddeZddZGddde ZGdddeZdS)N)parse_to_aware_datetime)CloudFrontSigner)CustomArgument)#validate_mutually_exclusive_handler) BasicCommandcs|dt|dt|dtdgdg|dtddgd g|d d d |d d d i|jdjdd|dtdgd g|dfdd dS)Nz!building-command-table.cloudfrontz6building-argument-table.cloudfront.create-invalidationz4operation-args-parsed.cloudfront.create-invalidationZinvalidation_batchpathsz4operation-args-parsed.cloudfront.create-distributionZdefault_root_objectZorigin_domain_nameZdistribution_configz6building-argument-table.cloudfront.create-distributionc[|dt|S)Norigin-domain-name) __setitem__OriginDomainNameargument_tablekwargsrDusr/lib/python3.10/site-packages/awscli/customizations/cloudfront.py+zregister..c[r)Ndefault-root-object)r CreateDefaultRootObjectr rrrr/rztop-level-args-parsed cloudfront)Z unique_idz4operation-args-parsed.cloudfront.update-distributionz6building-argument-table.cloudfront.update-distributioncs|dt|dS)Nr)contextr )r UpdateDefaultRootObjectr rrrr;s )register _add_sign _add_pathsrupdate)Z event_handlerrrrrsJ   rclicCsd|tttddfS)Nz%s-%s-%si@B)inttimerandomrandintprefixrrr unique_string@sr%cKsd|d_t|d<dS)NFzinvalidation-batchr)required PathsArgumentr rrrrDs rc$eZdZfddZddZZS)r'csd}tt|jdd|ddS)NzkThe space-separated paths to be invalidated. Note: --invalidation-batch and --paths are mutually exclusive.r+)nargs help_text)superr'__init__)selfdoc __class__rrr-KszPathsArgument.__init__cCs*|durtt||dd|d<dSdS)NQuantityZItems)CallerReferencePathsZInvalidationBatch)r%lenr. parametersvaluerrr add_to_paramsRs  zPathsArgument.add_to_params__name__ __module__ __qualname__r-r: __classcell__rrr0rr'Is r'cs,eZdZdZ dfdd ZddZZS) ExclusiveArgumentz1%s This argument and --%s are mutually exclusive.distribution-configcs,d||_tt|j||j||fddS)NFr+)r&r,r@r-DOC)r.namer Z exclusive_tor+r0rrr-]s   zExclusiveArgument.__init__cCs4tdgdddddiddddd dd d d d S)Nrr2 placeholderFForwardnone)Z QueryStringZCookies)Enabledr3z allow-all)TargetOriginIdZForwardedValuesZTrustedSignersZViewerProtocolPolicyZMinTTLTrB)r4OriginsDefaultCacheBehaviorrIComment)r%)r.rrrdistribution_config_templatecs z.ExclusiveArgument.distribution_config_template)rArB)r<r=r>rDr-rNr?rrr0rr@Zs r@cr()r cstt|jd|dddS)Nr z The domain name for your origin.rC)r,r r-)r.r r0rrr-zs  zOriginDomainName.__init__cCs|durdS|d|t|d}||dd}|ddr'ddi|d<nd d d d |d <d|gd|dd<||ddd<dS)NDistributionConfigr#rB)Id DomainNameZ OriginPathrQz.s3.amazonaws.comZOriginAccessIdentityZS3OriginConfigPiz http-only)ZHTTPPortZ HTTPSPortZOriginProtocolPolicyZCustomOriginConfigrr2rKrLrJ) setdefaultrNr%endswith)r.r8r9Z origin_iditemrrrr:s$    zOriginDomainName.add_to_paramsr;rrr0rr ys r cs&eZdZdfdd ZddZZS)rrBcstt|jd||p dddS)NrzvThe object that you want CloudFront to return (for example, index.html) when a viewer request points to your root URL.rC)r,rr-)r.r r+r0rrr-s  z CreateDefaultRootObject.__init__cCs,|dur|d|||dd<dSdS)NrODefaultRootObject)rSrNr7rrrr:s z%CreateDefaultRootObject.add_to_params)rBr;rrr0rrsrcr()rcstt|j|dd||_dS)NzThe object that you want CloudFront to return (for example, index.html) when a viewer request points to your root URL. CLI will automatically make a get-distribution-config call to load and preserve your other settings.rC)r,rr-r)r.rr r0rrr-s  z UpdateDefaultRootObject.__init__cCst|dur8|jdjd|jdj|jdj|jdjd}|j|dd}|d|d<|d |d <||d d <dSdS) NsessionrZ parsed_args)Z region_name endpoint_urlZverifyrP)rPZETagZIfMatchrOrV)rZ create_clientZregionrXZ verify_sslZget_distribution_config)r.r8r9Zclientresponserrrr:s      z%UpdateDefaultRootObject.add_to_paramsr;rrr0rrs  rcKst||d<dS)Nsign) SignCommand)Z command_tablerWrrrrrrc@sfeZdZdZdZdZdddddddd d d dd d d dded ddeddddgZddZdS)r[rZzSign a given url.aOSupported formats include: YYYY-MM-DD (which means 0AM UTC of that day), YYYY-MM-DDThh:mm:ss (with default timezone as UTC), YYYY-MM-DDThh:mm:ss+hh:mm or YYYY-MM-DDThh:mm:ss-hh:mm (with offset), or EpochTime (which always means UTC). Do NOT use YYYYMMDD, because it will be treated as EpochTime.urlTzThe URL to be signed)rEZ no_paramfiler&r+z key-pair-idz_The active CloudFront key pair Id for the key pair that you're using to generate the signature.)rEr&r+z private-keyz#file://path/to/your/private-key.pemzdate-less-thanz*The expiration date and time for the URL. zdate-greater-thanz-An optional start date and time for the URL. )rEr+z ip-addressztAn optional IP address or IP address range to allow client making the GET request from. Format: x.x.x.x/x or x.x.x.xcCst|jt|jj}t|j}|j}|durt|}|dus#|jdur<|j |j |||jd}t j |j|j |ddSt j |j|j |ddS)N)date_greater_than ip_address)policy)date_less_thanr)rZ key_pair_id RSASigner private_keyrZrrar^r_Z build_policyr]sysstdoutwriteZgenerate_presigned_url)r.argsZparsed_globalsZsignerrar^r`rrr _run_mains(   zSignCommand._run_mainN)r<r=r>NAMEZ DESCRIPTIONZ DATE_FORMATZ ARG_TABLErhrrrrr[s8 %r[c@seZdZddZddZdS)rbcCstj|d|_dS)Nutf8)rsaZ PrivateKeyZ load_pkcs1encodepriv_key)r.rcrrrr-szRSASigner.__init__cCst||jdS)NzSHA-1)rkrZrm)r.messagerrrrZr\zRSASigner.signN)r<r=r>r-rZrrrrrbs rb)r)rdr r!rkZbotocore.utilsrZbotocore.signersrZawscli.argumentsrZawscli.customizations.utilsrZawscli.customizations.commandsrrr%rr'r@r rrrr[objectrbrrrrs(       &A