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