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'];
 | 
						|
	}
 | 
						|
}
 |