mirror of
https://github.com/kennethreitz-archive/wp-s3.git
synced 2026-06-05 23:50:19 +00:00
Kenneth Reitz
464 lines
18 KiB
PHP
464 lines
18 KiB
PHP
<?php
|
|
/*
|
|
$Revision$
|
|
$Date$
|
|
$Author$
|
|
*/
|
|
require_once(dirname(__FILE__).'/class-plugin-public.php');
|
|
class TanTanWordPressS3Plugin extends TanTanWordPressS3PluginPublic {
|
|
|
|
function TanTanWordPressS3Plugin() {
|
|
parent::TanTanWordPressS3PluginPublic();
|
|
if (!file_exists(dirname(__FILE__).'/config.php')) {
|
|
add_action('admin_menu', array(&$this, 'settings'));
|
|
}
|
|
if (!$this->options['hideAmazonS3UploadTab']) {
|
|
add_action('load-upload.php', array(&$this, 'addPhotosTab')); // WP < 2.5
|
|
|
|
// WP >= 2.5
|
|
add_action('media_buttons_context', array(&$this, 'media_buttons'));
|
|
add_action('media_upload_tantan-wordpress-s3', array(&$this, 'media_upload_content'));
|
|
}
|
|
add_action('activate_tantan/wordpress-s3.php', array(&$this, 'activate'));
|
|
if ($_GET['tantanActivate'] == 'wordpress-s3') {
|
|
$this->showConfigNotice();
|
|
}
|
|
$this->photos = array();
|
|
$this->albums = array();
|
|
$this->perPage = 1000;
|
|
|
|
|
|
}
|
|
|
|
// this should install the javascripts onto the user's s3.amazonaws.com account
|
|
|
|
function installAjax() {
|
|
$js = array('S3Ajax.js');
|
|
}
|
|
|
|
function activate() {
|
|
wp_redirect('plugins.php?tantanActivate=wordpress-s3');
|
|
exit;
|
|
}
|
|
function deactivate() {}
|
|
|
|
function showConfigNotice() {
|
|
add_action('admin_notices', create_function('', 'echo \'<div id="message" class="updated fade"><p>Amazon S3 Plugin for WordPress <strong>activated</strong>. <a href="options-general.php?page=tantan/wordpress-s3/class-plugin.php">Configure the plugin ></a></p></div>\';'));
|
|
}
|
|
|
|
function settings() {
|
|
add_options_page('Amazon S3', 'Amazon S3', 10, __FILE__, array(&$this, 'admin'));
|
|
$this->version_check();
|
|
}
|
|
function addhooks() {
|
|
parent::addhooks();
|
|
if (!$_POST['disable_amazonS3']) {
|
|
add_filter('wp_update_attachment_metadata', array(&$this, 'wp_update_attachment_metadata'), 9, 2);
|
|
//can't delete mirrored files just yet
|
|
//add_filter('wp_get_attachment_metadata', array(&$this, 'wp_get_attachment_metadata'));
|
|
//add_filter('wp_delete_file', array(&$this, 'wp_delete_file'));
|
|
}
|
|
}
|
|
function version_check() {
|
|
global $TanTanVersionCheck;
|
|
if (is_object($TanTanVersionCheck)) {
|
|
$data = get_plugin_data(dirname(__FILE__).'/../wordpress-s3.php');
|
|
$TanTanVersionCheck->versionCheck(668, $data['Version']);
|
|
}
|
|
}
|
|
function admin() {
|
|
if ($_POST['action'] == 'save') {
|
|
if (!is_array($_POST['options'])) $_POST['options'] = array();
|
|
$options = get_option('tantan_wordpress_s3');
|
|
|
|
$_POST['options']['key'] = trim($_POST['options']['key']);
|
|
$_POST['options']['secret'] = trim($_POST['options']['secret']);
|
|
|
|
if (!$_POST['options']['secret'] || ereg('not shown', $_POST['options']['secret'])) {
|
|
$_POST['options']['secret'] = $options['secret'];
|
|
}
|
|
|
|
update_option('tantan_wordpress_s3', $_POST['options']);
|
|
|
|
if ($_POST['options']['bucket']) {
|
|
$options = get_option('tantan_wordpress_s3');
|
|
require_once(dirname(__FILE__).'/lib.s3.php');
|
|
$s3 = new TanTanS3($options['key'], $options['secret']);
|
|
|
|
if (!in_array($_POST['options']['bucket'], $s3->listBuckets())) {
|
|
if ($s3->createBucket($_POST['options']['bucket'],'public-read')) {
|
|
$message = "Saved settings and created a new bucket: ".$_POST['options']['bucket'];
|
|
} else {
|
|
$error = "There was an error creating the bucket: ".$_POST['options']['bucket'];
|
|
}
|
|
} else {
|
|
$message = "Saved settings.";
|
|
}
|
|
} else {
|
|
$message = "Saved Amazon S3 authentication information. ";
|
|
}
|
|
if (function_exists('dns_get_record') && $_POST['options']['virtual-host']) {
|
|
$record = dns_get_record($_POST['options']['bucket']);
|
|
if (($record[0]['type'] != 'CNAME') || ($record[0]['target'] != $_POST['options']['bucket'].'s3.amazonaws.com')) {
|
|
$error = "Warning: Your DNS doesn't seem to be setup correctly to virtually host the domain <em>".$_POST['options']['bucket']."</em>. ".
|
|
"Double check and make sure the following entry is added to your DNS. ".
|
|
"<a href='http://docs.amazonwebservices.com/AmazonS3/2006-03-01/VirtualHosting.html'>More info ></a>".
|
|
"<br /><br />".
|
|
"<code>".$_POST['options']['bucket']." CNAME ".$_POST['options']['bucket'].".s3.amazonaws.com.</code>".
|
|
"<br /><br />".
|
|
"<small>You can ignore this message if you're sure everything is setup correctly.</small>";
|
|
}
|
|
}
|
|
}
|
|
$options = get_option('tantan_wordpress_s3');
|
|
if ($options['key'] && $options['secret']) {
|
|
require_once(dirname(__FILE__).'/lib.s3.php');
|
|
$s3 = new TanTanS3($options['key'], $options['secret']);
|
|
if (!($buckets = $s3->listBuckets())) {
|
|
$error = $this->getErrorMessage($s3->parsed_xml, $s3->responseCode);
|
|
}
|
|
|
|
$s3->initCacheTables();
|
|
|
|
} elseif ($options['key']) {
|
|
$error = "Please enter your Secret Access Key.";
|
|
} elseif ($options['secret']) {
|
|
$error = "Please enter your Access Key ID.";
|
|
}
|
|
|
|
|
|
include(dirname(__FILE__).'/admin-options.html');
|
|
}
|
|
|
|
|
|
/*
|
|
Delete corresponding file from Amazon S3
|
|
*/
|
|
function wp_delete_file($file) {
|
|
return $file;
|
|
if (!$this->options) $this->options = get_option('tantan_wordpress_s3');
|
|
|
|
if (!$this->options['wp-uploads'] || !$this->options['bucket'] || !$this->options['secret']) {
|
|
return $file;
|
|
}
|
|
|
|
if (is_array($this->meta)) {
|
|
require_once(dirname(__FILE__).'/lib.s3.php');
|
|
$this->s3 = new TanTanS3($this->options['key'], $this->options['secret']);
|
|
$this->s3->setOptions($this->options);
|
|
if (deleteObject($this->meta['bucket'], $this->meta['key'])) {
|
|
|
|
}
|
|
$accessDomain = $this->options['virtual-host'] ? $this->meta['bucket'] : $this->meta['bucket'].'.s3.amazonaws.com';
|
|
return $file;
|
|
//return 'http://'.$accessDomain.'/'.$amazon['key'];
|
|
|
|
}
|
|
return $file;
|
|
}
|
|
function wp_get_attachment_metadata($data=false, $postID=false) {
|
|
if (is_numeric($postID)) $this->meta = get_post_meta($postID, 'amazonS3_info', true);
|
|
return $data;
|
|
}
|
|
/*
|
|
Handle uploads through default WordPress upload handler
|
|
*/
|
|
function wp_update_attachment_metadata($data, $postID) {
|
|
if (!$this->options) $this->options = get_option('tantan_wordpress_s3');
|
|
|
|
if (!$this->options['wp-uploads'] || !$this->options['bucket'] || !$this->options['secret']) {
|
|
return $data;
|
|
}
|
|
|
|
add_filter('option_siteurl', array(&$this, 'upload_path'));
|
|
$uploadDir = wp_upload_dir();
|
|
remove_filter('option_siteurl', array(&$this, 'upload_path'));
|
|
$parts = parse_url($uploadDir['url']);
|
|
|
|
$prefix = substr($parts['path'], 1) .'/';
|
|
$type = get_post_mime_type($postID);
|
|
|
|
$data['file'] = get_attached_file($postID, true);
|
|
|
|
if (file_exists($data['file'])) {
|
|
$file = array(
|
|
'name' => basename($data['file']),
|
|
'type' => $type,
|
|
'tmp_name' => $data['file'],
|
|
'error' => 0,
|
|
'size' => filesize($data['file']),
|
|
);
|
|
|
|
require_once(dirname(__FILE__).'/lib.s3.php');
|
|
$this->s3 = new TanTanS3($this->options['key'], $this->options['secret']);
|
|
$this->s3->setOptions($this->options);
|
|
|
|
if ($this->s3->putObjectStream($this->options['bucket'], $prefix.$file['name'], $file)) {
|
|
|
|
if ($data['thumb']) {
|
|
$thumbpath = str_replace( basename( $data['file'] ), $data['thumb'], $data['file'] );
|
|
$filethumb = array(
|
|
'name' => $data['thumb'],
|
|
'type' => $type,
|
|
'tmp_name' => $thumbpath,
|
|
'size' => filesize($thumbpath),
|
|
);
|
|
|
|
$this->s3->putObjectStream($this->options['bucket'], $prefix.$filethumb['name'], $filethumb);
|
|
} elseif (count($data['sizes'])) foreach ($data['sizes'] as $altName => $altSize) {
|
|
$altPath = str_replace( basename( $data['file'] ), $altSize['file'], $data['file'] );
|
|
$altMeta = array(
|
|
'name' => $altSize['file'],
|
|
'type' => $type,
|
|
'tmp_name' => $altPath,
|
|
'size' => filesize($altPath),
|
|
);
|
|
$this->s3->putObjectStream($this->options['bucket'], $prefix.$altMeta['name'], $altMeta);
|
|
|
|
}
|
|
|
|
|
|
delete_post_meta($postID, 'amazonS3_info');
|
|
add_post_meta($postID, 'amazonS3_info', array(
|
|
'bucket' => $this->options['bucket'],
|
|
'key' => $prefix.$file['name']
|
|
));
|
|
} else {
|
|
|
|
}
|
|
}
|
|
return $data;
|
|
}
|
|
function wp_handle_upload($info) {
|
|
return $info;
|
|
}
|
|
|
|
// figure out the correct path to upload to, for wordpress mu installs
|
|
function upload_path($path='') {
|
|
global $current_blog;
|
|
if (!$current_blog) return $path;
|
|
if ($current_blog->path == '/' && ($current_blog->blog_id != 1)) {
|
|
$dir = substr($current_blog->domain, 0, strpos($current_blog->domain, '.'));
|
|
} else {
|
|
// prepend a directory onto the path for vhosted blogs
|
|
if (constant("VHOST") != 'yes') {
|
|
$dir = '';
|
|
} else {
|
|
$dir = $current_blog->path;
|
|
}
|
|
}
|
|
//echo trim($path.'/'.$dir, '/');
|
|
if ($path == '') {
|
|
$path = $current_blog->path;
|
|
}
|
|
return trim($path.'/'.$dir, '/');
|
|
}
|
|
function media_buttons($context) {
|
|
global $post_ID, $temp_ID;
|
|
$dir = dirname(__FILE__);
|
|
$pluginRootURL = get_option('siteurl').substr($dir, strpos($dir, '/wp-content'));
|
|
$image_btn = $pluginRootURL.'/database.png';
|
|
$image_title = 'Amazon S3';
|
|
|
|
$uploading_iframe_ID = (int) (0 == $post_ID ? $temp_ID : $post_ID);
|
|
|
|
$media_upload_iframe_src = "media-upload.php?post_id=$uploading_iframe_ID";
|
|
$out = ' <a href="'.$media_upload_iframe_src.'&tab=tantan-wordpress-s3&TB_iframe=true&height=500&width=640" class="thickbox" title="'.$image_title.'"><img src="'.$image_btn.'" alt="'.$image_title.'" /></a>';
|
|
return $context.$out;
|
|
}
|
|
function media_upload_content() {
|
|
$this->upload_files_tantan_amazons3(); // process any uploaded files or new folders
|
|
|
|
if (!$this->options) $this->options = get_option('tantan_wordpress_s3');
|
|
//if (!is_object($this->s3)) {
|
|
require_once(dirname(__FILE__).'/lib.s3.php');
|
|
$this->s3 = new TanTanS3($this->options['key'], $this->options['secret']);
|
|
$this->s3->setOptions($this->options);
|
|
//}
|
|
|
|
add_action('admin_print_scripts', array(&$this, 'upload_tabs_scripts'));
|
|
wp_iframe(array(&$this, 'tab'));
|
|
}
|
|
/*
|
|
Display tabs
|
|
*/
|
|
function addPhotosTab() {
|
|
add_filter('wp_upload_tabs', array(&$this, 'wp_upload_tabs'));
|
|
add_action('upload_files_tantan_amazons3', array(&$this, 'upload_files_tantan_amazons3'));
|
|
add_action('upload_files_upload', array(&$this, 'upload_files_upload'));
|
|
add_action('admin_print_scripts', array(&$this, 'upload_tabs_scripts'));
|
|
}
|
|
function wp_upload_tabs ($array) {
|
|
/*
|
|
0 => tab display name,
|
|
1 => required cap,
|
|
2 => function that produces tab content,
|
|
3 => total number objects OR array(total, objects per page),
|
|
4 => add_query_args
|
|
*/
|
|
if (!$this->options) $this->options = get_option('tantan_wordpress_s3');
|
|
require_once(dirname(__FILE__).'/lib.s3.php');
|
|
$this->s3 = new TanTanS3($this->options['key'], $this->options['secret']);
|
|
|
|
|
|
if ($this->options['key'] && $this->options['secret'] && $this->options['bucket']) {
|
|
$paged = array();
|
|
$args = array('prefix' => ''); // this doesn't do anything in WP 2.1.2
|
|
$tab = array(
|
|
'tantan_amazons3' => array('Amazon S3', 'upload_files', array(&$this, 'tab'), $paged, $args),
|
|
//'tantan_amazons3_upload' => array('Upload S3', 'upload_files', array(&$this, 'upload'), $paged, $args),
|
|
);
|
|
|
|
return array_merge($array, $tab);
|
|
} else {
|
|
return $array;
|
|
}
|
|
}
|
|
|
|
function upload_tabs_scripts() {
|
|
//wp_enqueue_script('prototype');
|
|
if (!$this->options) $this->options = get_option('tantan_wordpress_s3');
|
|
|
|
$accessDomain = $this->options['virtual-host'] ? $this->options['bucket'] : $this->options['bucket'].'.s3.amazonaws.com';
|
|
|
|
include(dirname(__FILE__).'/admin-tab-head.html');
|
|
}
|
|
function upload_files_upload() {
|
|
// javascript here to inject javascript and allow the upload from to post to amazon s3 instead
|
|
}
|
|
function upload_files_tantan_amazons3() {
|
|
global $current_blog;
|
|
$restrictPrefix = ''; // restrict to a selected prefix in current bucket
|
|
if ($current_blog) { // if wordpress mu
|
|
$restrictPrefix = ltrim($this->upload_path().'/files/', '/');
|
|
}
|
|
|
|
if (is_array($_FILES['newfile'])) {
|
|
$file = $_FILES['newfile'];
|
|
if (!$this->options) $this->options = get_option('tantan_wordpress_s3');
|
|
require_once(dirname(__FILE__).'/lib.s3.php');
|
|
$this->s3 = new TanTanS3($this->options['key'], $this->options['secret']);
|
|
$this->s3->setOptions($this->options);
|
|
$this->s3->putObjectStream($this->options['bucket'], $restrictPrefix.$_GET['prefix'].$file['name'], $file);
|
|
}
|
|
if ($_POST['newfolder']) {
|
|
if (!$this->options) $this->options = get_option('tantan_wordpress_s3');
|
|
require_once(dirname(__FILE__).'/lib.s3.php');
|
|
$this->s3 = new TanTanS3($this->options['key'], $this->options['secret']);
|
|
|
|
$this->s3->putPrefix($this->options['bucket'], $restrictPrefix.$_POST['prefix'].$_POST['newfolder']);
|
|
}
|
|
}
|
|
function tab() {
|
|
global $current_blog;
|
|
$restrictPrefix = ''; // restrict to a selected prefix in current bucket
|
|
if ($current_blog) { // if wordpress mu
|
|
$restrictPrefix = ltrim($this->upload_path().'/files/', '/');
|
|
}
|
|
|
|
$offsetpage = (int) $_GET['paged'];
|
|
if (!$offsetpage) $offsetpage = 1;
|
|
|
|
if (!$this->options['key'] || !$this->options['secret']) {
|
|
return;
|
|
}
|
|
$bucket = $this->options['bucket'];
|
|
$accessDomain = $this->options['virtual-host'] ? $this->options['bucket'] : $this->options['bucket'].'.s3.amazonaws.com';
|
|
|
|
$prefix = $_GET['prefix'] ? $_GET['prefix'] : '';
|
|
list($prefixes, $keys, $meta, $privateKeys) = $this->getKeys($restrictPrefix.$prefix);
|
|
if ($restrictPrefix) {
|
|
foreach ($prefixes as $k=>$v) {
|
|
$prefixes[$k] = str_replace($restrictPrefix, '', $v);
|
|
}
|
|
}
|
|
include(dirname(__FILE__).'/admin-tab.html');
|
|
}
|
|
|
|
function getErrorMessage($parsed_xml, $responseCode){
|
|
$message = 'Error '.$responseCode.': ' . $parsed_xml->Message;
|
|
if(isset($parsed_xml->StringToSignBytes)) $message .= "<br>Hex-endcoded string to sign: " . $parsed_xml->StringToSignBytes;
|
|
return $message;
|
|
}
|
|
|
|
// turns array('a', 'b', 'c') into $array['a']['b']['c']
|
|
function mapKey($keys, $path) {
|
|
$k =& $keys;
|
|
$size = count($path) - 1;
|
|
$workingPath = '/';
|
|
foreach ($path as $i => $p) {
|
|
if ($i === $size) {
|
|
$k['_size'] = isset($k['_size']) ? $k['_size'] + 1 : 1;
|
|
$k['_path'] = $workingPath;
|
|
$k['_objects'][$k['_size']] = $p;
|
|
} else {
|
|
$k =& $k[$p]; // traverse the tree
|
|
$workingPath .= $p . '/';
|
|
}
|
|
}
|
|
return $keys;
|
|
}
|
|
|
|
// should probably figgure out a way to cache these results to make things more speedy
|
|
function getKeys($prefix) {
|
|
$ret = $this->s3->listKeys($this->options['bucket'], false, urlencode($prefix), '/');//, false, 's3/', '/');
|
|
|
|
if ($this->s3->responseCode >= 400) {
|
|
return array();
|
|
}
|
|
$keys = array();
|
|
$privateKeys = array();
|
|
$prefixes = array();
|
|
$meta = array();
|
|
if ($this->s3->parsed_xml->CommonPrefixes) foreach ($this->s3->parsed_xml->CommonPrefixes as $content) {
|
|
$prefixes[] = (string) $content->Prefix;
|
|
}
|
|
|
|
if ($this->s3->parsed_xml->Contents) foreach ($this->s3->parsed_xml->Contents as $content) {
|
|
$key = (string) $content->Key;
|
|
if ($this->isPublic($key)) $keys[] = $key;
|
|
else {
|
|
if (!($p1 = ereg('^\.', $key)) &&
|
|
!($p2 = ereg('_\$folder\$$', $key)) &&
|
|
!($p3 = ereg('placeholder.ns3', $key))) {
|
|
$privateKeys[] = $key;
|
|
} elseif ($p2) {
|
|
$prefix = ereg_replace('(_\$folder\$$)', '/', $key);
|
|
if (!in_array($prefix, $prefixes)) $prefixes[] = $prefix;
|
|
} else {
|
|
|
|
}
|
|
}
|
|
}
|
|
if ($this->options['permissions'] == 'public') {
|
|
foreach ($privateKeys as $key) {
|
|
$this->s3->setObjectACL($this->options['bucket'], $key, 'public-read');
|
|
$keys[] = $key;
|
|
}
|
|
}
|
|
|
|
foreach ($keys as $i => $key) {
|
|
$meta[$i] = $this->s3->getMetadata($this->options['bucket'], $key);
|
|
}
|
|
natcasesort($keys);
|
|
natcasesort($prefixes);
|
|
|
|
return array($prefixes, $keys, $meta, $privateKeys);
|
|
}
|
|
|
|
function isPublic($key) {
|
|
$everyone = 'http://acs.amazonaws.com/groups/global/AllUsers';
|
|
$this->s3->getObjectACL($this->options['bucket'], $key);
|
|
$acl = (array) $this->s3->parsed_xml->AccessControlList;
|
|
if (is_array($acl['Grant'])) foreach ($acl['Grant'] as $grant) {
|
|
$grant = (array) $grant;
|
|
if ($grant['Grantee'] && (ereg('AllUsers', (string) $grant['Grantee']->URI))) {
|
|
$perm = (string) $grant['Permission'];
|
|
if ($perm == 'READ' || $perm == 'FULL_CONTROL') return true;
|
|
}
|
|
}
|
|
|
|
|
|
}
|
|
}
|
|
?>
|