$dZddlmZddlmZddlmZddlmZddlZddlZddlm Z ddl m Z dd l mZdd lmZdd lmZdd lmZdd lmZddlmZddlmZGddej4ZGddej4ZGddej4ZdZdZdZ dZ!dZ"dZ#dZ$dZ%dZ&dZ'd Z(d!Z)d"Z*d#Z+d$Z,d%Z-y)&z2Declarative Hooks for Cloud SCC surface arguments.)absolute_import)division)print_function)unicode_literalsN)encoding)securitycenter_client)InvalidSCCInputError) yaml_data) resource_args)concept_parsers) exceptions) properties)yamlceZdZdZy)InvalidCustomConfigFileErrorz6Error if a custom config file is improperly formatted.N__name__ __module__ __qualname____doc__+lib/googlecloudsdk/command_lib/scc/hooks.pyrr$s>rrceZdZdZy)InvalidTestDataFileErrorz2Error if a test data file is improperly formatted.Nrrrrrr(s:rrceZdZdZy)InvalidResourceFileErrorz1Error if a resource file is improperly formatted.Nrrrrrr,s9rrc tjjd}tjddddd|j g}t j|ggS)z*Add Organization as a positional resource.zscc.organizationz8to be used for the SCC (Security Command Center) command organizationTF)verbnamerequiredprefixes positional resource_datar ResourceYAMLDataFromPathr GetResourcePresentationSpecGetDatar ConceptParser) org_spec_data arg_specss r AppendOrgArgr.0sa,,556HI-//I%--/ 1)  ' ' 2 6 77rc tjjd}tjdddddd|j g}t j|ggS)z$Add Parent as a positional resource.z scc.parentz'to be used for the `gcloud scc` commandparentz{name} organization, folder, or project in the Google Cloud resource hierarchy {verb}. Specify the argument as either [RESOURCE_TYPE/RESOURCE_ID] or [RESOURCE_ID], as shown in the preceding examples.TF)r r! help_textr"r#r$r%r&)parent_spec_datar-s rAppendParentArgr3?sg//88F//8^(0024 )  ' ' 2 6 77rc~tj}tj||jj S)zAHook to capture "key1=val1,key2=val2" as SourceProperties object.) sc_client GetMessagesr DictToMessageFindingSourcePropertiesValue)source_properties_dictmessagess rSourcePropertiesHookr<Os7  " " $(    6 ( 0 0 F F HHrctj}|j}tj||jj |_|S)z>Hook to capture "key1=val1,key2=val2" as SecurityMarks object.)r5r6 SecurityMarksrr7 MarksValuemarks) parsed_dictr;security_markss rSecurityMarksHookrCVsH  " " $())+.!//8))446. rctjd}tjd}|js3tjj jj }n |j}| td|j|s|j|s td|j|r|Sd|zS)+Prepend organizations/ to org if necessary.organizations/[0-9]+[0-9]+zOCould not find Organization argument. Please provide the organization argument.>Organization must match either organizations/[0-9]+ or [0-9]+.organizations/) recompilerrVALUESsccGetr match)argsresource_pattern id_patternrs rGetOrganizationrS_sZZ 67zz(#*  $$((5599;L$$L  D EE   .   < ( H JJL)  L ((rcDtjd}tjd}tjjj j }|j|s|j|s td|j|r|Sd|zS)rErFrGrHrI) rJrKrrLrMrrNrOr )rQrRrs rGetDefaultOrganizationrUssZZ 67zz(#*""&&33779,   .   < ( H JJL)  L ((rcftjd}tjd}tjjj j }|j|rd|z}|j|s-|jds|jds td|S)z@Converts user input to one of: organization, project, or folder.zorganizations/[0-9]+$rGrIz projects/zfolders/zWParent must match either [0-9]+, organizations/[0-9]+, projects/.* or folders/.*.) rJrKrrLrMr0rNrO startswithr )organization_resource_patternrRr0s rGetDefaultParentrYs"$**-D"Ezz(#*    ' ' + + -&f  &F ' - -f 5   K (F,=,=j,I    -rc&|jddS)z2Removes spaces from a field mask provided by user. )replace)masks rCleanUpUserInputr_s c2 rctjd}|j|s td|j d}|ddz|dzS)Nz#(organizations|projects|folders)/.*zrWhen providing a full resource path, it must also include the pattern the organization, project, or folder prefix./r)rJrKrOr split) resource_namerQlist_organization_componentss rGetParentFromResourceNamerfs`ZZ EF    .  7 88"/!4!4S!9 %a (3 .1Ma1P PPrctjd}|j|s td|j d}t |dz|dzdz|dzS)Nz2(organizations|projects|folders)/.*/sources/[0-9]+zfWhen providing a full resource path, it must also include the organization, project, or folder prefix.ra)rJrKrOr rcrf)rdrQlist_source_componentss rGetSourceParentFromResourceNamerks}ZZ:<    .  7 88)..s3 #M 2S 8  # $&) *,B1,E FGrctj} tj|}t j ||j jS#tj$r}tdj|d}~wwxYw)z;Processes the configuration file for the ETD custom module.%Error parsing custom config file [{}]N) r5r6jsonloadsrr7 EventThreatDetectionCustomModule ConfigValueJSONDecodeErrorrformat) file_contentsr;configes rProcessCustomEtdConfigFilerwsz  " " $( ZZ &F  ! !99EE     &/66q9 s>AB(BBctj} tj|}t j ||j S#tj$r}tdj|d}~wwxYw)rs9&%' %IF:>D=*$??#8#8?;44;:44: 8 8 H)( )   Q G  =  $ r