9.dZddlZddlmZddlmZddlmZddlmZddl m Z ddl m Z ejjd iZejjfd Zejjfd Zd Zd ZdZdZdZdZ ddZ ddZdZy)z API lib for Gemini Cloud Assist.N)apis)waiter)base) cloud_assist)iam_util) resourcesv1alphacTtjdtj|S)zReturns the messages module for the given release track. Args: release_track: The release track to use. Returns: The messages module for the given release track. geminicloudassist)rGetMessagesModule VERSION_MAPget release_tracks 6lib/googlecloudsdk/api_lib/gemini_cloud_assist/util.pyr r %   ;??=9 cTtjdtj|S)zReturns the client instance for the given release track. Args: release_track: The release track to use. Returns: The client instance for the given release track. r )rGetClientInstancer rrs rrr,rrct}t}|jj|j |S)zReturns the investigation for the given investigation resource. Args: investigations_resource_name: The name of the investigation resource. Returns: The investigation for the given investigation resource. name)rr !projects_locations_investigationsGet:GeminicloudassistProjectsLocationsInvestigationsGetRequestinvestigations_resource_nameclientmessagess rGetInvestigationr :sD  &  (  1 1 5 5II+J rc 2t}t}|jjdd}|j i} |dk7r|j| _nd}| j jj|jjjd|jd|jjjjdd|jjj|jjj |r|| _|s|s|s|r|jd d |jjj$|jjj } |r|| _|rK|rI| j(j|j+|j-|j- n|r:| j(j|j+|j- n|r:| j(j|j+|j-nt.j.j1t.j2j4t/j6dz } | j(j|j+| j- |r|| _| j jj|jjj| j:| |j<j?|jA|jj|| S)aCreates the investigation for the given investigation resource. Args: investigation_resource: The investigation resource to create. title: The title of the investigation. issue: The issue of the investigation. start_time: The start time of the investigation. end_time: The end time of the investigation. relevant_resources: The resources of the investigation. Returns: The created investigation. /) observations-N user.projectidtextobservationType observerTypekeyvalueuser.input.logUser Provided Issuer(titler*r+ startTimeendTime)r4)r5)minutes)parentinvestigationId investigation)!rr RelativeNamesplit Investigationrr$additionalPropertiesappendObservationsValueAdditionalProperty ObservationParentObservationTypeValueValuesEnum!OBSERVATION_TYPE_STRUCTURED_INPUTObserverTypeValueValuesEnumOBSERVER_TYPE_USERr2OBSERVATION_TYPE_CLOUD_LOGr) timeIntervalsInterval isoformatdatetimenowtimezoneutc timedeltarelevantResourcesr(rCreate=GeminicloudassistProjectsLocationsInvestigationsCreateRequest) investigation_resourcer2issue start_timeend_timerelevant_resourcesrrinvestigation_idr: observationhalf_an_hour_agos rCreateInvestigationr\Ls'*  &  (+88:@@EbI((b(9-/<<>M1188..AA$$)002vx|~uSz"'22QQss#//KK^^% B   M jH(:&& # ,,KKff))EEXX 'K  kh&&   ",,.8J8J8L   &&   j&:&:&<  = &&   H$6$6$8  9"**..        R ()&&   &6&@&@&B  C&8k#33::00CC D   1 1 8 8LL'..0==?*%M rcd\}}|rJt|jdk\r2|jdj}|jdj}|r/|tjj k7r|j nd}|r/|tjjk7r|j nd}tj||S)a(Calculates the time interval for the given start and end times. Args: start_time: The start time of the investigation. end_time: The end time of the investigation. observation: The observation to get the existing time interval from. Returns: The calculated time interval. )NNrNr3) lenrIr4r5rLminrKmaxr rJ)rVrWrZstart_time_str end_time_strs rCalculateTimeIntervalrds",.,S223q8 ..q1;;N,,Q/77L **.. .     x((,, ,      % %  & rc zt|j}t}t}|j di} g} |j| _| j jj|jjjd|jd|jjjjdd|jjj|jj j"| jd|| jd || _||s|s||jd d |jjj&|jj j" } | jd| jd| jd| jd|| jd|| _|s|rq| jdd } |j jD]} | j*d k(s| j,} n| j.jt1||| || jd|| _| j jj|jjj| j4| |j6j9|j;|jdj=| | S)aIUpdates the investigation for the given investigation resource. Args: investigation_resource: The investigation resource to create. title: The title of the investigation. Pass an empty string to clear. issue: The issue of the investigation. start_time: The start time of the investigation. Pass datetime.datetime.min to clear. end_time: The end time of the investigation. Pass datetime.datetime.max to clear. relevant_resources: The resources of the investigation. Pass an empty array to clear. Returns: The updated investigation. ) dataVersionr$r&r"r#r'r,zobservations.`user.project`Nr2r/r0r1z observations.`user.input.log`.idz#observations.`user.input.log`.titlez-observations.`user.input.log`.observationTypez*observations.`user.input.log`.observerTypez"observations.`user.input.log`.textz+observations.`user.input.log`.timeIntervalsz/observations.`user.input.log`.relevantResources,)r updateMaskr:)r r;rr r=rr$r>r?r@rArBrCr<rDrErFrGr2rHr)r-r.rIrdrQr(rPatch >RR/S rct}t}t|}tj|j ||||j j|j||j|S)a:Adds an IAM policy binding to the given investigation resource. Args: investigations_resource_name: The name of the investigation resource. member: The member to add to the binding. role: The role to add to the binding. Returns: The updated IAM policy for the given investigation resource. policyrsetIamPolicyRequest) rr rrAddBindingToIamPolicyBindingr SetIamPolicyCGeminicloudassistProjectsLocationsInvestigationsSetIamPolicyRequestSetIamPolicyRequestrmemberrolerrrs r AddInvestigationIamPolicyBindingrtsz  &  ( $%A B&   !1!1664H  1 1 > >RR/&::&:IS rct}t}t|}tj||||j j |j||j|S)aIRemoves an IAM policy binding from the given investigation resource. Args: investigations_resource_name: The name of the investigation resource. member: The member to remove from the binding. role: The role to remove from the binding. Returns: The updated IAM policy for the given investigation resource. rr) rr rrRemoveBindingFromIamPolicyrrrrrs r#RemoveInvestigationIamPolicyBindingrsr  &  ( $%A B& %%ffd;  1 1 > >RR/&::&:IS rc t}t}tj||j\}}|j j |j||j||}tj|d|S)a Sets the IAM policy for the given investigation resource. Args: investigations_resource_name: The name of the investigation resource. policy_file: The path to the policy file. Returns: The updated IAM policy for the given investigation resource. )rrirr:) rr rParseYamlOrJsonPolicyFilePolicyrrrrLogSetIamPolicy)r policy_filerrr update_maskresults rSetInvestigationIamPolicyrs  &  ( ::8??&+  3 3 @ @RR/&:: ; S & 7I -r)allUsersz+roles/geminicloudassist.investigationViewer)__doc__rLgooglecloudsdk.api_lib.utilrrgooglecloudsdk.callioper!googlecloudsdk.command_lib.geminirgooglecloudsdk.command_lib.iamrgooglecloudsdk.corer ReleaseTrackALPHAr r rr r\rdrqrrrrrrrrs ',.(:3)  && 2 %)$5$5$;$; %)$5$5$;$; $`F!Hod:(  6:  66r