7BdZddlmZddlmZddlmZddlmZddlm Z ddl m Z ddl Z d e d efd Zd e d efd Zd ZddZ ddZddZdZddZddZd dZddZd dZdZdZdZddZ ddZddZ ddZ!y)!z"Helper functions for DNS commands.)absolute_import)division)unicode_literals)util)apis)flagsNipreturncN tj|y#t$rYywxYw)zReturns True if ip is an IPv4.TF)ipaddr IPv4Address ValueErrorr s *lib/googlecloudsdk/command_lib/dns/util.pyIsIPv4r* r     $$cN tj|y#t$rYywxYw)zReturns True if ip is an IPv6.TF)r IPv6Addressrrs rIsIPv6r%rrcd}||Bi}||d<||jj||d<|||d<|r|jdi|}|S)aoGenerate a keyspec from the given (unparsed) command line arguments. Args: algorithm: (str) String mnemonic for the DNSSEC algorithm to be specified in the keyspec; must be a value from AlgorithmValueValuesEnum. key_length: (int) The key length value to include in the keyspec. key_type: (KeyTypeValueValuesEnum) Enum value for whether to create a keyspec for a KSK or a ZSK. messages: (module) Module (generally auto-generated by the API build rules) containing the API client's message classes. Returns: A messages.DnsKeySpec instance created from the given arguments. NkeyType algorithm keyLength) DnsKeySpecAlgorithmValueValuesEnum)r key_lengthkey_typemessageskey_spec spec_argss rParseKeyr#.st (j4I#Ii'22KK  i )i $$$1y1h /cd}g}d}|jr/tjd|j|j}d}|jr/tjd|j|j}|dk(r!|j j j}n |j j j}t||j||}||j||dk(r!|j j j}n |j j j}t||j||} | |j| i} |r|| d<t|dd1tj |j|j"| d<|j$2tj&||j|j$| d<| r|j(d i| }|S) aParse all relevant command line arguments and generate a DNSSEC config. Args: args: (dict{str,(str|int)}) Dict of command line arguments; value type dependent on particular command line argument. messages: (module) Module (generally auto-generated by the API build rules) containing the API client's message classes. api_version: api_version that this function should use. Returns: A messages.ManagedZoneDnsSecConfig instance populated from the given command line arguments. Nkskzskv2defaultKeySpecsdenial_of_existence nonExistencestater) ksk_algorithmrGetKeyAlgorithmFlagMapperGetEnumForChoice zsk_algorithmrKeyTypeValueValuesEnum KEY_SIGNING keySigningr#ksk_key_lengthappend ZONE_SIGNING zoneSigningzsk_key_lengthgetattrGetDoeFlagMapperr* dnssec_stateGetDnsSecStateFlagMapperManagedZoneDnsSecConfig) argsr api_version dnssec_config key_specsr-r0key_enumksk_keyzsk_keydnssec_config_argss rParseDnssecConfigArgsrFOs -)- 33 t))*- 33 t))* D""99EEH""99DDH ]D$7$78 L'  WD""99FFH""99EEH ]D$7$78 L'  W,5() T($/; x(99  $ $ &~& ""'"@"@+#t(()w4H44J7IJM r$c g}|jjd}|jjd}||D]}t|r$|j|j|d|2t |r$|j|jd||a|j|jdd||||D]}t|r$|j|j|d|2t |r$|j|jd||a|j|jdd|||j |S)aParses list of forwarding nameservers into ManagedZoneForwardingConfig. Args: messages: (module) Module (generally auto-generated by the API build rules) containing the API client's message classes. server_list: (list) List of IP addresses to use as forwarding targets for the DNS Managed Zone that uses default forwarding logic (based on RFC1918 check). private_server_list: (list) List of IP addresses to use as forwarding targets for the DNS Managed Zone that always use the private VPC path. Returns: A messages.ManagedZoneForwardingConfig instance populated from the given command line arguments. rN ipv4Address ipv6AddressforwardingPath)rJrK domainNamerLtargetNameServers)+ManagedZoneForwardingConfigNameServerTargetForwardingPathValueValuesEnumrr5rManagedZoneForwardingConfig)r server_listprivate_server_listtarget_servers default_enum private_enumnames r2ParseManagedZoneForwardingConfigWithForwardingPathrYs".EEcc ,EEcc ,   @ @ d< A  $<  @ @ d< A    @ @  + A  ,$#   @ @ d+ A - . $<  @ @ d+ A - .   @ @  + A  $*  - - - OOr$ct|}|sgS|Dcgc]"}|j|j$c}Scc}wz.Build PolicyNetwork message from parsed_value.) networkUrl) GetMessages PolicyNetworkSelfLink parsed_valueversionr network_refs rPolicyNetworkProcessorrdsT !(  I& %+ (<(<(>?%  '>ct|dS)z3Build Beta PolicyNetwork message from parsed_value.v1beta2rb)rd)ras rBetaPolicyNetworkProcessorris i @@r$ct|}|sgS|Dcgc]"}|j|j$c}Scc}wr[)r]ResponsePolicyNetworkr_r`s rResponsePolicyNetworkProcessorrlsT !(  I& %+$$ 0D0D0F$G%  rect|}t|r-|j|d|jjdS|jd||jjdS)aBuild a single TargetNameServer based on 'value'. Args: value: (str) A string representation of an IPV4 ip address representing the PrivateTargetNameServer. version: (str) A string indicating the version of the API to be used, should be 'v1' only before removing BetaTargetNameServerType. Returns: A messages.PolicyAlternativeNameServerConfigTargetNameServer instance populated from the given ip address. NrrIr]r1PolicyAlternativeNameServerConfigTargetNameServerrQvaluerbr s rTargetNameServerTyperr !( E]  E EQQoo   F   E EQQoo   F r$ct|}t|r-|j|d|jjdS|jd||jjdS)aBuild a single TargetNameServer based on 'value'. Args: value: (str) A string representation of an IPV4 ip address representing the PrivateTargetNameServer. version: (str) A string indicating the version of the API to be used, should be one of 'v1beta2' or 'v1alpha2'. This function will be removed after promoting v6 address to GA. Returns: A messages.PolicyAlternativeNameServerConfigTargetNameServer instance populated from the given ip address. NrrIrnrps rBetaTargetNameServerTyperu} !( E]  E E : : & &q ) F ++  E E : : & &q ) F ++r$ct|}t|r-|j|d|jjdS|jd||jjdS)aBuild a single PrivateTargetNameServer based on 'value'. Args: value: (str) A string representation of an IPV4 ip address representing the PrivateTargetNameServer. version: (str) A string indicating the version of the API to be used, should be 'v1' only before removing BetaPrivateTargetNameServerType. Returns: A messages.PolicyAlternativeNameServerConfigTargetNameServer instance populated from the given ip address. NrHrIrnrps rPrivateTargetNameServerTyperx6rsr$ct|}t|r-|j|d|jjdS|jd||jjdS)aBuild a single PrivateTargetNameServer based on 'value'. Args: value: (str) A string representation of an IPV4 ip address representing the PrivateTargetNameServer. version: (str) A string indicating the version of the API to be used, should be one of 'v1beta2' or 'v1alpha2'. This function will be removed after promoting v6 address to GA. Returns: A messages.PolicyAlternativeNameServerConfigTargetNameServer instance populated from the given ip address. NrHrIrnrps rBetaPrivateTargetNameServerTyperzVrvr$c4t|||}t||S)z6Build a list of PolicyNetworks from command line args.) ParseNetworksrdrqprojectrbnetworkss rParsePolicyNetworksrus 5'7 3( ' 22r$c |sgStj|}|Dcgc]}|j|dd|i}}|Scc}w)zPBuild a list of PolicyNetworks or ResponsePolicyNetworks from command line args.zcompute.networksr~) collectionparams)api_util GetRegistryParse)rqr~rbregistry network_namers rr|r|{sk  I  ! !' *( Build a list of ResponsePolicyNetworks from command line args.)r|rlr}s rParseResponsePolicyNetworksrs 5'7 3( '' ::r$c|s|syt|}g}|r||Dcgc]}t||c}z }|r||Dcgc]}t||c}z }|j|Scc}wcc}w)aParses list of alternative nameservers into AlternativeNameServerConfig. Args: version: (str) A string indicating the version of the API to be used, should be 'v1' only before removing BetaParseAltNameServers. server_list: (Sequence) List of IP addresses to use as forwarding targets for the DNS managed zone that uses default forwarding logic. private_server_list: (Sequence) List of IP addresses to use as forwarding targets for the DNS Managed Zone that always uses the private VPC path. Returns: A messages.PolicyAlternativeNameServerConfig instance populated from the given command line arguments.Only the not none server list will be parsed and an empty list will be returned if both are none. NrN)r]rrrx!PolicyAlternativeNameServerConfigrbrSrTr result_listr s rParseAltNameServersrs" 0   !(+ L "(W5 LLK;N;NR#B0;NK  3 3# 4 %% M A#A(c|s|syt|}g}|r||Dcgc]}t||c}z }|r||Dcgc]}t||c}z }|j|Scc}wcc}w)a#Parses list of alternative nameservers into AlternativeNameServerConfig. Args: version: (str) A string indicating the version of the API to be used, should be one of 'v1beta2' or 'v1alpha2'. This function will be moved after promoting v6 address to GA. server_list: (Sequence) List of IP addresses to use as forwarding targets for the DNS Managed Zone that uses default forwarding logic. private_server_list: (Sequence) List of IP addresses to use as forwarding targets for the DNS Managed Zone that always uses the private VPC path. Returns: A messages.PolicyAlternativeNameServerConfig instance populated from the given command line arguments.Only the not none server list will be parsed and an empty list will be returned if both are none. NrN)r]rurzrrs rBetaParseAltNameServersrs( 0   !(+KPKb,R9KPPK%%B (G4%K  3 3# 4 %% Qrct|}|jdk(rE|dk(r |jjjS|jjj Sy)z%Parses response policy rule behavior.bypassResponsePolicyr(N)r]behaviorResponsePolicyRuleBehaviorValueValuesEnumBYPASS_RESPONSE_POLICYr)r>rbms r ParseResponsePolicyRulesBehaviorrsj'! ]],,RY]aRa1   7 7 N Nighg{g{hThThihii r$c.tjd|S)Ndns)rGetMessagesModulerhs rr]r]s  w //r$)v1)NN)rg)"__doc__ __future__rrrgooglecloudsdk.api_lib.dnsrrgooglecloudsdk.api_lib.utilrgooglecloudsdk.command_lib.dnsrr strboolrrr#rFrYrdrirlrrrurxrzrr|rrrrr]rr$rrs)&'7,0 ststB@H59CPL A @+>@+>3 ; %@)-04 %F0r$