2dZddlZddlZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z dd l m Z dd lmZej"d Zd efd ZdZd efdZd efdZdej0d efdZded efdZded efdZdZdZdZdZ dedej0fdZ!dZ"defdZ#defd Z$d!Z%d ehfd"Z&dej0d efd#Z'ded efd$Z(d%Z)y)&z*Common flag parsing for management gcloud.N)encoding)folders) constants)errors) properties) resources)yaml)$securitycentermanagement_v1_messagesz [0-9]{1,20}returnc|jr&t|jjdSt|jdS)zReturns the relative path to the parent from args. Args: args: command line args. Returns: The relative path. e.g. 'projects/foo/locations/global', 'folders/1234/locations/global'. z/locations/global)parent _ParseParent RelativeName_GetParentResourceFromArgsargss 4lib/googlecloudsdk/command_lib/scc/manage/parsing.pyGetParentResourceNameFromArgsrsI [[4;;'446 77H II &t , 9 9 ; < +     # # I ))..66::D:I2 $ rct|}|jj}tjj |}|r"|dtj d|jStj|j)a)Returns the specified service name from args if it exists. Otherwise, an exception is raised detailing the parsing error along with the expectation. Args: args: The argument input as the gcloud command. Raises: InvalidServiceNameError: the specified service name was invalid. /) r service_namelowerrSERVICE_INVENTORYgetSERVICE_RESOURCE_PLURAL_NAMEnamerInvalidServiceNameError)rr maybe_service_name_or_abbrservices rGetServiceNameFromArgsr2?sv ) .&#00668  ' ' + +,F G' XQy==>a ~ NN  ( ():): ;;rc|jstjdtj |j}|r|dStj|j)z Returns the module id from args.Nr)module_id_or_namerInvalidCustomModuleIdError_CUSTOM_MODULE_ID_REGEX fullmatch)rmatchs rGetModuleIdFromArgsr9WsU    + +D 11 ! + +D,B,B C% 8O  + +D,B,B CCr module_typec|jstjd|d|d|g}t|jxst |j dk(}|D]=} t jj|j|jcS|r tj|j|t|}t|}|d|d|S#t j$rYwxYw)aReturns the specified module name from args if it exists. Otherwise, an exception is raised detailing the parsing error along with the expectation. Args: args: the args module_type: the module type (see googlecloudsdk.command_lib.scc.manage.constants) Raises: MissingCustomModuleNameOrIdError: no module name or id was specified. InvalidCustomModuleNameError: the specified module name was invalid. InvalidCustomModuleIdError: the specified module id was invalid. z1securitycentermanagement.organizations.locations.z,securitycentermanagement.projects.locations.z+securitycentermanagement.folders.locations.rr()r4r MissingCustomModuleNameOrIdError_IsPossibleResourceNamelenGetSpecifiedArgNamesrrr rRequiredFieldOmittedExceptionInvalidCustomModuleNameErrorrr9)rr: collectionsis_possible_resource_namerr module_ids rGetModuleNameFromArgsrFds "     1 1 33 :+G4[MB3K=A+d445/ T & & ( )Q .  j    % %  Z&    - -    ) .&!$') 81[M9+ ..  2 2   s(8C&&C<;C<r c`|jdr!tjj|dS|jdr%t j j|dS|jdr!tjj|dSt j|)zUExtracts parent name from a string of the form {organizations|projects|folders}/.organizations/rrzfolders/r projects/r) startswithrrr rr"rInvalidParentErrorr s rrrs '(    # #? $ $  " " $ * *9 + %    # #2 $   # #F ++rr.cp|jdxs$|jdxs|jdS)N organizationsprojectsr)rJ)r.s rr>r>s5 ooo&$  $$  #rc|: tj|}tj|tj Sy#tj $r)}tjdj|d}~wwxYw)z5Process the custom config file for the custom module.Nz%Error parsing custom config file [{}]) r loadr DictToMessagemessages CustomConfigYAMLParseErrorrInvalidCustomConfigFileErrorformat)file config_dictypes rGetCustomConfigFromArgsr[sq IIdOk  # #K1F1F GG     / / 1 8 8 = s8>A:$A55A:c tj|}tj|tj S#tj $r)}tjdj|d}~wwxYw)zJProcess the test resource data file for the custom module to test against.z Error parsing resource file [{}]N) r rQrrRrSSimulatedResourcerUrInvalidResourceFileErrorrW)rX resource_dictrZs rGetTestResourceFromArgsr`sdIIdOM  ! !-1K1K LL    ) )*11#6 s8;A7$A22A7c|D tj|}tj|tj j Sy#tj$r)}tjdj|d}~wwxYw)z5Process the config custom file for the custom module.N$Error parsing config value file [{}]) jsonloadsrrRrS EventThreatDetectionCustomModule ConfigValueJSONDecodeErrorrInvalidConfigValueFileErrorrWrXconfiges rGetConfigValueFromArgsrls} zz$f  # # (;;GG      . . 0 7 7 : sAAB$A??Bc|+ tj|}tj|Sy#tj$r)}t j dj |d}~wwxYw)z3Converts the contents of a JSON file into a string.Nrb)rcrddumpsrgrrhrWris r ParseJSONFilerosh zz$f ZZ       . . 0 7 7 : s)/A+$A&&A+enablement_statec|tjjk(rtjj }nQ|tjj k(rtjj }ntjd|d|tjd|j}|dk(r |jS|dk(r |jS|dk(r |jStjd|d) zParse the enablement state.z Module type "z" is not a valid module type.zAError parsing enablement state. Enablement state cannot be empty.ENABLEDDISABLED INHERITED!Error parsing enablement state. "Y" is not a valid enablement state. Please provide one of ENABLED, DISABLED, or INHERITED.)rCustomModuleTypeSHArS#SecurityHealthAnalyticsCustomModuleEnablementStateValueValuesEnumETDrerInvalidModuleTypeErrorInvalidEnablementStateErrorupperrrrsrt)rpr: state_enumstates rGetEnablementStateFromArgsrs I..22244SSi0044411PP  ' '  }$AB   , ,K   "% i             , , +E73I I rc|j |jy|jy|jytjd)z*Create an update mask with the args given.zenablement_state,custom_configrp custom_configzError parsing Update Mask. Either a custom configuration or an enablement state (or both) must be provided to update the custom module.)rpcustom_config_filerInvalidUpdateMaskInputErrorrs rCreateUpdateMaskFromArgsrsR &4+B+B+N + (  *   , ,  rrXc|D tj|}tj|tj j Sy#tjtf$r}tjd|dd}~wwxYw)z/Process the module config file for the service.Nz!Error parsing config value file []) r rQrrRrSSecurityCenterService ModulesValuerUAttributeErrorrrh)rXrjrZs rGetModuleConfigValueFromArgsr!s yyf  # # (00==     0  . . -cU! 4 sAAA?!A::A?ctjj}|y|j}|dk(r |jS|dk(r |j S|dk(r |j Stjd|d)z#Parse the service enablement state.Nrrrsrtrurv) rSr&IntendedEnablementStateValueValuesEnumr~rrrsrtrr})rprrs r!GetServiceEnablementStateFromArgsr1s$$KK    "% i             , , +E73I I rc|j |jy|jy|jytjd)z@Create an update mask with the args given for the given service.z!intended_enablement_state,modulesintended_enablement_statemoduleszError parsing Update Mask. Either a module configuration or an enablement state (or both) must be provided to update the service.)rpmodule_config_filerrrs r"CreateUpdateMaskFromArgsForServicerHsS &4+B+B+N . ( & *   , , N rc|jsgS|jjd}|jd}|Dchc]}|j}}|Scc}w)z)Returns a list of module names from args.z[],)filter_modulesstripsplit)rr modules_listmodule modules_sets rGetModuleListFromArgsrWs[    I    % %d +'s#,.:;lFl+; . Args: parent: The parent string to parse. Returns: The relative path of the parent. Raises: InvalidParentFlagError: The provided parent string is invalid. rHrrrIr)rJrrr rInvalidParentFlagErrorrLs rrrs~ '(    # #? $  {#    # #2 $  %%f--rc2|jr+tjj|jdStjj|jxs4t j jjjddS)z~Returns the parent resource from the given args. Args: args: command line args. Returns: The parent resource. rrTrr) rrrr r#rr$r%r&rs rrrs     # # &J $     ! ! llGj'',,4488$8G0 " r)*__doc__rcreapitools.base.pyr'googlecloudsdk.api_lib.resource_managerr%googlecloudsdk.command_lib.scc.managerrgooglecloudsdk.corerrr Agooglecloudsdk.generated_clients.apis.securitycentermanagement.v1r rScompiler6strrrr2r9rwrFrboolr>r[r`rlrorrrrrrrrrrrrs91 %;;8*)$~$"**]3 O3 O