336 lines
		
	
	
		
			8.6 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
			
		
		
	
	
			336 lines
		
	
	
		
			8.6 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
| <?php
 | |
| namespace W3TC;
 | |
| 
 | |
| /**
 | |
|  * Rackspace Cloud Files CDN engine
 | |
|  */
 | |
| class CdnEngine_RackSpaceCloudFiles extends CdnEngine_Base {
 | |
| 	private $_access_state;
 | |
| 	private $_container;
 | |
| 	private $_api_files;
 | |
| 	private $_api_cdn;
 | |
| 
 | |
| 
 | |
| 
 | |
| 	function __construct( $config = array() ) {
 | |
| 		$config = array_merge( array(
 | |
| 				'user_name' => '',
 | |
| 				'api_key' => '',
 | |
| 				'region' => '',
 | |
| 				'container' => '',
 | |
| 				'cname' => array(),
 | |
| 				'access_state' => ''
 | |
| 			), $config );
 | |
| 
 | |
| 		$this->_container = $config['container'];
 | |
| 		$this->_new_access_state_callback = $config['new_access_state_callback'];
 | |
| 
 | |
| 		// init access state
 | |
| 		$this->_access_state = @json_decode( $config['access_state'], true );
 | |
| 		if ( !is_array( $this->_access_state ) )
 | |
| 			$this->_access_state = array();
 | |
| 		$this->_access_state = array_merge( array(
 | |
| 				'access_token' => '',
 | |
| 				'access_region_descriptor' => array(),
 | |
| 				'host_http' => '',
 | |
| 				'host_https' => ''
 | |
| 			), $this->_access_state );
 | |
| 
 | |
| 		parent::__construct( $config );
 | |
| 		$this->_create_api(
 | |
| 			array( $this, '_on_new_access_requested_api_files' ),
 | |
| 			array( $this, '_on_new_access_requested_api_cdn' ) );
 | |
| 	}
 | |
| 
 | |
| 
 | |
| 
 | |
| 	private function _create_api( $new_access_required_callback_api_files,
 | |
| 		$new_access_required_callback_api_cdn ) {
 | |
| 		$this->_api_files = new Cdn_RackSpace_Api_CloudFiles( array(
 | |
| 				'access_token' => $this->_access_state['access_token'],
 | |
| 				'access_region_descriptor' => $this->_access_state['access_region_descriptor'],
 | |
| 				'new_access_required' => $new_access_required_callback_api_files ) );
 | |
| 
 | |
| 		$this->_api_cdn = new Cdn_RackSpace_Api_CloudFilesCdn( array(
 | |
| 				'access_token' => $this->_access_state['access_token'],
 | |
| 				'access_region_descriptor' => $this->_access_state['access_region_descriptor'],
 | |
| 				'new_access_required' => $new_access_required_callback_api_cdn ) );
 | |
| 	}
 | |
| 
 | |
| 
 | |
| 
 | |
| 	/**
 | |
| 	 * Called when new access token issued by api objects
 | |
| 	 */
 | |
| 	public function _on_new_access_requested_api_files() {
 | |
| 		$this->_on_new_access_requested();
 | |
| 		return $this->_api_files;
 | |
| 	}
 | |
| 
 | |
| 
 | |
| 
 | |
| 	/**
 | |
| 	 * Called when new access token issued by api objects
 | |
| 	 */
 | |
| 	public function _on_new_access_requested_api_cdn() {
 | |
| 		$this->_on_new_access_requested();
 | |
| 		return $this->_api_cdn;
 | |
| 	}
 | |
| 
 | |
| 
 | |
| 
 | |
| 	private function _on_new_access_requested() {
 | |
| 		$r = Cdn_RackSpace_Api_Tokens::authenticate( $this->_config['user_name'],
 | |
| 			$this->_config['api_key'] );
 | |
| 		if ( !isset( $r['access_token'] ) || !isset( $r['services'] ) )
 | |
| 			throw new \Exception( 'Authentication failed' );
 | |
| 		$r['regions'] = Cdn_RackSpace_Api_Tokens::cloudfiles_services_by_region(
 | |
| 			$r['services'] );
 | |
| 
 | |
| 		if ( !isset( $r['regions'][$this->_config['region']] ) )
 | |
| 			throw new \Exception( 'Region ' . $this->_config['region'] . ' not found' );
 | |
| 
 | |
| 		$this->_access_state['access_token'] = $r['access_token'];
 | |
| 		$this->_access_state['access_region_descriptor'] =
 | |
| 			$r['regions'][$this->_config['region']];
 | |
| 
 | |
| 		$this->_create_api(
 | |
| 			array( $this, '_on_new_access_requested_second_time' ),
 | |
| 			array( $this, '_on_new_access_requested_second_time' ) );
 | |
| 
 | |
| 		$c = $this->_api_cdn->container_get( $this->_config['container'] );
 | |
| 
 | |
| 		$this->_access_state['host_http'] = substr( $c['x-cdn-uri'], 7 );
 | |
| 		$this->_access_state['host_https'] = substr( $c['x-cdn-ssl-uri'], 8 );
 | |
| 
 | |
| 		call_user_func( $this->_new_access_state_callback,
 | |
| 			json_encode( $this->_access_state ) );
 | |
| 	}
 | |
| 
 | |
| 
 | |
| 
 | |
| 	private function _on_new_access_requested_second_time() {
 | |
| 		throw new \Exception( 'Authentication failed' );
 | |
| 	}
 | |
| 
 | |
| 
 | |
| 
 | |
| 	/**
 | |
| 	 * Formats URL
 | |
| 	 */
 | |
| 	function _format_url( $path ) {
 | |
| 		$domain = $this->get_domain( $path );
 | |
| 
 | |
| 		if ( $domain ) {
 | |
| 			$scheme = $this->_get_scheme();
 | |
| 
 | |
| 			// it does not support '+', requires '%2B'
 | |
| 			$path = str_replace( '+', '%2B', $path );
 | |
| 			$url = sprintf( '%s://%s/%s', $scheme, $domain, $path );
 | |
| 
 | |
| 			return $url;
 | |
| 		}
 | |
| 
 | |
| 		return false;
 | |
| 	}
 | |
| 
 | |
| 	/**
 | |
| 	 * Uploads files to CDN
 | |
| 	 *
 | |
| 	 * @param array   $files
 | |
| 	 * @param array   $results
 | |
| 	 * @param boolean $force_rewrite
 | |
| 	 * @return boolean
 | |
| 	 */
 | |
| 	function upload( $files, &$results, $force_rewrite = false,
 | |
| 		$timeout_time = NULL ) {
 | |
| 		foreach ( $files as $file ) {
 | |
| 			$local_path = $file['local_path'];
 | |
| 			$remote_path = $file['remote_path'];
 | |
| 
 | |
| 			// process at least one item before timeout so that progress goes on
 | |
| 			if ( !empty( $results ) ) {
 | |
| 				if ( !is_null( $timeout_time ) && time() > $timeout_time ) {
 | |
| 					return 'timeout';
 | |
| 				}
 | |
| 			}
 | |
| 
 | |
| 			if ( !file_exists( $local_path ) ) {
 | |
| 				$results[] = $this->_get_result( $local_path, $remote_path,
 | |
| 					W3TC_CDN_RESULT_ERROR, 'Source file not found.', $file );
 | |
| 
 | |
| 				continue;
 | |
| 			}
 | |
| 
 | |
| 			$file_content = file_get_contents( $local_path );
 | |
| 
 | |
| 			$do_write = true;
 | |
| 
 | |
| 			// rewrite is optional, check md5
 | |
| 			if ( !$force_rewrite ) {
 | |
| 				$object_meta = null;
 | |
| 
 | |
| 				try {
 | |
| 					$object_meta = $this->_api_files->object_get_meta_or_null(
 | |
| 						$this->_container, $remote_path );
 | |
| 				} catch ( \Exception $exception ) {
 | |
| 					$results[] = $this->_get_result( $local_path, $remote_path,
 | |
| 						W3TC_CDN_RESULT_ERROR,
 | |
| 						sprintf( 'Unable to check object (%s).', $exception->getMessage() ),
 | |
| 						$file );
 | |
| 					$do_write = false;
 | |
| 				}
 | |
| 
 | |
| 				if ( is_array( $object_meta ) && isset( $object_meta['etag'] ) ) {
 | |
| 					$md5_actual = md5( $file_content );
 | |
| 
 | |
| 					if ( $md5_actual == $object_meta['etag'] ) {
 | |
| 						$results[] = $this->_get_result( $local_path,
 | |
| 							$remote_path, W3TC_CDN_RESULT_OK,
 | |
| 							'Object up-to-date.', $file );
 | |
| 						$do_write = false;
 | |
| 					}
 | |
| 				}
 | |
| 			}
 | |
| 
 | |
| 			if ( $do_write ) {
 | |
| 				try {
 | |
| 					$this->_api_files->object_create( array(
 | |
| 							'container' => $this->_container,
 | |
| 							'name' => $remote_path,
 | |
| 							'content_type' => Util_Mime::get_mime_type( $local_path ),
 | |
| 							'content' => $file_content ) );
 | |
| 					$results[] = $this->_get_result( $local_path, $remote_path,
 | |
| 						W3TC_CDN_RESULT_OK, 'OK', $file );
 | |
| 				} catch ( \Exception $exception ) {
 | |
| 					$results[] = $this->_get_result( $local_path, $remote_path,
 | |
| 						W3TC_CDN_RESULT_ERROR,
 | |
| 						sprintf( 'Unable to create object (%s).', $exception->getMessage() ),
 | |
| 						$file );
 | |
| 				}
 | |
| 			}
 | |
| 		}
 | |
| 
 | |
| 		return !$this->_is_error( $results );
 | |
| 	}
 | |
| 
 | |
| 	/**
 | |
| 	 * Deletes files from CDN
 | |
| 	 *
 | |
| 	 * @param array   $files
 | |
| 	 * @param array   $results
 | |
| 	 * @return boolean
 | |
| 	 */
 | |
| 	function delete( $files, &$results ) {
 | |
| 		foreach ( $files as $file ) {
 | |
| 			$local_path = $file['local_path'];
 | |
| 			$remote_path = $file['remote_path'];
 | |
| 
 | |
| 			try {
 | |
| 				$this->_api_files->object_delete( $this->_container, $remote_path );
 | |
| 				$results[] = $this->_get_result( $local_path, $remote_path,
 | |
| 					W3TC_CDN_RESULT_OK, 'OK', $file );
 | |
| 			} catch ( \Exception $exception ) {
 | |
| 				$results[] = $this->_get_result( $local_path, $remote_path,
 | |
| 					W3TC_CDN_RESULT_ERROR,
 | |
| 					sprintf( 'Unable to delete object (%s).',
 | |
| 						$exception->getMessage() ),
 | |
| 					$file );
 | |
| 			}
 | |
| 		}
 | |
| 
 | |
| 		return !$this->_is_error( $results );
 | |
| 	}
 | |
| 
 | |
| 	/**
 | |
| 	 * Test CDN connection
 | |
| 	 *
 | |
| 	 * @param string  $error
 | |
| 	 * @return boolean
 | |
| 	 */
 | |
| 	function test( &$error ) {
 | |
| 		$filename = 'test_rscf_' . md5( time() );
 | |
| 
 | |
| 		try {
 | |
| 			$object = $this->_api_files->object_create( array(
 | |
| 					'container' => $this->_container,
 | |
| 					'name' => $filename,
 | |
| 					'content_type' => 'text/plain',
 | |
| 					'content' => $filename ) );
 | |
| 		} catch ( \Exception $exception ) {
 | |
| 			$error = sprintf( 'Unable to write object (%s).', $exception->getMessage() );
 | |
| 			return false;
 | |
| 		}
 | |
| 
 | |
| 		$result = true;
 | |
| 		try {
 | |
| 			$r = wp_remote_get( 'http://' . $this->get_host_http() . '/' . $filename );
 | |
| 
 | |
| 			if ( $r['body'] != $filename ) {
 | |
| 				$error = 'Failed to retrieve object after storing.';
 | |
| 				$result = false;
 | |
| 			}
 | |
| 		} catch ( \Exception $exception ) {
 | |
| 			$error = sprintf( 'Unable to read object (%s).', $exception->getMessage() );
 | |
| 			$result = false;
 | |
| 		}
 | |
| 
 | |
| 		try {
 | |
| 			$this->_api_files->object_delete( $this->_container, $filename );
 | |
| 		} catch ( \Exception $exception ) {
 | |
| 			$error = sprintf( 'Unable to delete object (%s).', $exception->getMessage() );
 | |
| 			$result = false;
 | |
| 		}
 | |
| 
 | |
| 		return $result;
 | |
| 	}
 | |
| 
 | |
| 	/**
 | |
| 	 * Returns CDN domain
 | |
| 	 *
 | |
| 	 * @return array
 | |
| 	 */
 | |
| 	function get_domains() {
 | |
| 		if ( Util_Environment::is_https() ) {
 | |
| 			if ( !empty( $this->_config['cname'] ) ) {
 | |
| 				return (array) $this->_config['cname'];
 | |
| 			}
 | |
| 
 | |
| 			return array( $this->get_host_https() );
 | |
| 		} else {
 | |
| 			if ( !empty( $this->_config['cname'] ) ) {
 | |
| 				return (array) $this->_config['cname'];
 | |
| 			}
 | |
| 
 | |
| 			return array( $this->get_host_http() );
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	/**
 | |
| 	 * Returns VIA string
 | |
| 	 *
 | |
| 	 * @return string
 | |
| 	 */
 | |
| 	function get_via() {
 | |
| 		return sprintf( 'Rackspace Cloud Files: %s', parent::get_via() );
 | |
| 	}
 | |
| 
 | |
| 
 | |
| 
 | |
| 	public function get_host_http() {
 | |
| 		if ( empty( $this->_access_state['host_http'] ) )
 | |
| 			$this->_on_new_access_requested();
 | |
| 
 | |
| 		return $this->_access_state['host_http'];
 | |
| 	}
 | |
| 
 | |
| 
 | |
| 
 | |
| 	public function get_host_https() {
 | |
| 		if ( empty( $this->_access_state['host_https'] ) )
 | |
| 			$this->_on_new_access_requested();
 | |
| 
 | |
| 		return $this->_access_state['host_https'];
 | |
| 	}
 | |
| }
 |