dZddlmZddlmZddlmZddlZddlmZmZddl m Z ddl m Z d e d e fd Zd e d ee ee ee ffde dee de d e f dZd ee ee ee ffd e de dee de de d e fdZde d ee e ffdZy)zLibrary for fetching TF Files.)absolute_import)division)unicode_literalsN)DictList) prompt_format)files file_pathreturnc0tj|}|S)zuReads the TF file. Args: file_path: The path of the file to read. Returns: The contents of the file. )r ReadFileContents)r files 8lib/googlecloudsdk/api_lib/scc/iac_remediation/prompt.py read_filers    *$ +tfstate_information iam_bindings resource_nametf_filesmemberctj}d|vr|j}n|j}t ||||||S)aeGenerates the prompt for iam policy. Args: tfstate_information: TFState information for the given IAM bindings. iam_bindings: IAM bindings for the resource. resource_name: Resource name for which the finding was generated. tf_files: List of TF files. member: Member for which the prompt is generated. Returns: Prompt string. google_project_iam_policy)rPromptFormatLookupget_policy_prompt_templateget_binding_prompt_template _fetch_prompt)rrrrrprompt_format_data prompt_strs rfetch_input_promptr)sW&%779 $77#>>@J#??AJ   rrcd|zdz}|jD]&\}}||dztj|zdzz }(|jd|}|jdtj|}|jd|}d} |D]!} | d| zdzd zz } | t | d zz } #|jd | }|S) aGenerates the prompt string. Args: iam_bindings: IAM bindings for the resource. tfstate_information: TFState information for the given IAM bindings. resource_name: Resource name for which the finding was generated. tf_files: List of TF files. prompt_str: Prompt file name. member: Member for which the prompt is generated. Returns: Prompt for iam policy. zmember:  z : z{{iam_bindings}}z{{tfstate_information}}z{{resource_name}}z FilePath= ``` z ``` z{{input_tf_files}})itemsjsondumpsreplacer) rrrrrriam_bindings_stractionroles files_strtf_files rrrKs* &(4/#))+mfe(4::e+<rAs %&' D*   sDd3i0013i     D$sDd3i001$$$3i $  $  $ $NcNr