UdZddlmZddlmZddlmZddlmZddlZddlZddlm Z ddl m Z dd l m Z dd l mZGd d eZd ZdZdZdZy)z=Helper functions for Split Trust Encryption Tool integration.)absolute_import)print_function)division)unicode_literalsN) storage_url)execution_util)temporary_file_util)configceZdZdZdZdZy)StetSubcommandNamez*Enum class for available STET subcommands.encryptdecryptN)__name__ __module__ __qualname____doc__ENCRYPTDECRYPT(platform/gsutil/gslib/utils/stet_util.pyr r s2 ' 'rr c tjdjtjjD]E}tjj |d}tjj |sC|cSy)zBRetrieves STET binary from path if available. Python 2 compatible.PATHstetN)osgetenvsplitpathpathsepjoinexists)path_directory binary_paths r_get_stet_binary_from_pathr$&sS &)//@n'',,~v6K ww~~k" Arctjddt}|s tdtj j ||g}tjddd}|r1|jdtj j |z|jd|z||gtj|\}} |j| y)a Runs a STET transform on a file. Encrypts for uploads. Decrypts for downloads. Automatically populates flags for the STET binary. Args: subcommand (StetSubcommandName): Subcommand to call on STET binary. blob_id (str): Cloud URL that binary uses for validation. in_file_path (str): File to be transformed source. out_file_path (str): Where to write result of transform. logger (logging.Logger): For logging STET binary output. Raises: KeyError: STET binary or config could not be found. GSUtilstet_binary_pathz2Could not find STET binary in boto config or PATH.stet_config_pathNz--config-file=z --blob-id=) r getr$KeyErrorrr expanduserappendextendrExecuteExternalCommanddebug) subcommandblob_id in_file_path out_file_pathloggerr# command_args config_path_stderrs r_stet_transformr9.s  8%7579+  G HH''$$[1:>, 8%7>+(277+=+=k+JJK|g-|]KL33LA)!V,,vrc|j}tj|}|j}t t j ||||tj|S)a)Encrypts a file with STET binary before upload. Args: source_url (StorageUrl): Copy source. destination_url (StorageUrl): Copy destination. logger (logging.Logger): For logging STET binary output. Returns: stet_temporary_file_url (StorageUrl): Path to STET-encrypted file. ) object_namer GetStetTempFileName url_stringr9r rrStorageUrlFromString) source_urldestination_urlr4in_fileout_filer1s rencrypt_uploadrCMsU  " "' 4 4Z @(  & &'$,,gw  ) )( 33rc|}tj|}|j}ttj ||||t j||y)aSTET-decrypts downloaded file. Args: source_url (StorageUrl): Copy source. destination_url (StorageUrl): Copy destination. temporary_file_name (str): Path to temporary file used for download. logger (logging.Logger): For logging STET binary output. N)r r<r=r9r rshutilmove)r?r@temporary_file_namer4rArBr1s rdecrypt_downloadrHasL ' 4 4_ E(  ! !'$,,gw++h r)r __future__rrrrrrEgslibr gslib.utilsrr botor objectr r$r9rCrHrrrrNsID&%' &+ >4(!r