<?php
/**
 *
 *
 * @link       amazon.com
 * @since      2.0.3
 *
 * @package    Amazonpolly
 * @subpackage Amazonpolly/admin
 */

class AmazonAI_S3FileHandler extends AmazonAI_FileHandler {

  private $s3_client;

  /**
	 * Return type of storage which is supported by class (S3).
	 *
	 * @since    2.1.0
	 */
    public function get_type() {
      return "s3";
    }

    public function set_s3_client($new_s3_client) {
      $this->s3_client = $new_s3_client;
    }

    /**
  	 * Function responsible for saving file on local storage file system.
  	 *
  	 * @param           $wp_filesystem         Not used here.
  	 * @param           $file                  File name.
  	 * @param           $translate_langs       Supported translated languages.
  	 * @param           $post_id               ID of the post.
  	 * @since           2.0.3
  	 */
    public function delete($wp_filesystem, $file, $translate_langs, $post_id) {

      // Retrieve the name of the bucket where audio files are stored.
      $s3_bucket  = $this->get_bucket_name();
      $prefix     = $this->get_prefix($post_id);

      // Delete main audio file.
      $this->delete_s3_object( $s3_bucket, $prefix . $file );

      // Delete translations if available.
      foreach ( $translate_langs as $lan ) {
        $value = get_post_meta( $post_id, 'amazon_polly_translation_' . $lan, true );
        if ( ! empty( $value ) ) {
          $s3_key = $prefix . 'amazon_polly_' . $post_id . $lan . '.mp3';
          $this->delete_s3_object( $s3_bucket, $s3_key );
        }
      }

    }


    /**
  	 * Function responsible for saving file on local storage file system.
  	 *
  	 * @param           $wp_filesystem         Not used here.
  	 * @param           $file_temp_full_name   Temporary name of file on local filesystem.
  	 * @param           $dir_final_full_name   Final destination where file should be saved.
  	 * @param           $file_final_full_name  Final name of file.
  	 * @param           $post_id               ID of the post.
  	 * @param           $file_name             Name of the file.
  	 * @since           2.0.3
  	 */
    public function save($wp_filesystem, $file_temp_full_name, $dir_final_full_name, $file_final_full_name, $post_id, $file_name) {
        $media_library_att_id = get_post_meta( $post_id, 'amazon_polly_media_library_attachment_id', true );
  			if ( !empty($media_library_att_id) ) {
  				wp_delete_attachment( $media_library_att_id, true );
  			}

        $key = $this->get_prefix($post_id) . $file_name;

  			// We are storing audio file on Amazon S3.
  			$s3BucketName = $this->get_bucket_name();
  			$audio_location = 's3';
  			$result         = $this->s3_client->putObject(
  				array(
  					'ACL'        => 'public-read',
  					'Bucket'     => $s3BucketName,
  					'Key'        => $key,
  					'SourceFile' => $file_temp_full_name,
  				)
  			);
  			$wp_filesystem->delete( $file_temp_full_name );
  			$cloudfront_domain_name = get_option( 'amazon_polly_cloudfront' );

  			if ( get_option('uploads_use_yearmonth_folders') ) {
  			  $key = get_the_date( 'Y', $post_id ) . '/' . get_the_date( 'm', $post_id ) . '/' . $file_name;
  			} else {
  			  $key = $file_name;
  			}

  			if ( empty( $cloudfront_domain_name ) ) {

  				$selected_region = get_option( 'amazon_polly_region' );

  				$audio_location_link = 'https://s3.' . $selected_region . '.amazonaws.com/' . $s3BucketName . '/' . $key;
  			} else {
  				$audio_location_link = 'https://' . $cloudfront_domain_name . '/' . $key;
  			}

        return $audio_location_link;
    }


    /**
  	 * Method creates (if it doesn't already exists) a S3 bucket
  	 *
  	 * @since       1.0.0
  	 */
  	public function prepare_s3_bucket() {

      $s3BucketName    = $this->get_bucket_name();
      $createNewBucket = false;

      // Check if user specified bucket name in using filter.
      $s3BucketName = apply_filters( 'amazon_polly_s3_bucket_name', $s3BucketName );

      //Check if bucket is provided and can be access.
      if ( empty( $s3BucketName ) ) {
        $createNewBucket = true;
      } else {
        try {
          $result = $this->s3_client->headBucket(array('Bucket' => $s3BucketName));
  			} catch ( Aws\S3\Exception\S3Exception $e ) {
  				$createNewBucket = true;
  			}
  		}

      // If bucket was not provided (or was not accessible), we need to create new bucket.
      // We will try to do it 10 times.
      for ( $i = 0; $i <= 10; $i++ ) {
        if ( $createNewBucket ) {
          try {
            $rand1 = wp_rand( 10000000000, 99999999999 );
            $rand2 = md5( microtime() );
            $name  = 'audio-for-wordpress-' . $rand1 . $rand2;
            $name  = substr( $name, 0, 60 );

            $result = $this->s3_client->createBucket( array( 'Bucket' => $name ) );
  					update_option( 'amazon_polly_s3_bucket', $name );
  					$createNewBucket = false;

  				} catch ( Aws\S3\Exception\S3Exception $e ) {
  					update_option( 'amazon_polly_s3_bucket', '' );
  					update_option( 'amazon_polly_s3', '' );
  				}
  			}
  		}
  	}


        /**
         * Delets object from S3.
         *
         * @param string $post_id ID of the post for which audio should be deleted.
         * @since 2.0.0
         */
        private function delete_s3_object( $bucket, $key ) {

          $this->s3_client->deleteObject(
            array(
              'Bucket' => $bucket,
              'Key'    => $key,
            )
          );

        }


        /**
         * Get S3 bucket name. The method uses filter 'amazon_polly_s3_bucket_name,
         * which allows to use customer S3 bucket name instead of default one.
         *
         * @since  1.0.6
         */
        private function get_bucket_name() {

          $s3BucketName = get_option( 'amazon_polly_s3_bucket' );
          $s3BucketName = apply_filters( 'amazon_polly_s3_bucket_name', $s3BucketName );

          return $s3BucketName;
        }

}