Clean up remote avatar temporary files if we fail before saving them into avatars directory (OMB core, OStatus, WikiHowProfile, YammerImport)

This commit is contained in:
Brion Vibber 2010-10-13 11:04:41 -07:00
parent 2291d68e70
commit bca215563f
4 changed files with 65 additions and 45 deletions

View File

@ -328,6 +328,7 @@ class StatusNetOAuthDataStore extends OAuthDataStore
function add_avatar($profile, $url) function add_avatar($profile, $url)
{ {
$temp_filename = tempnam(sys_get_temp_dir(), 'listener_avatar'); $temp_filename = tempnam(sys_get_temp_dir(), 'listener_avatar');
try {
copy($url, $temp_filename); copy($url, $temp_filename);
$imagefile = new ImageFile($profile->id, $temp_filename); $imagefile = new ImageFile($profile->id, $temp_filename);
$filename = Avatar::filename($profile->id, $filename = Avatar::filename($profile->id,
@ -335,6 +336,10 @@ class StatusNetOAuthDataStore extends OAuthDataStore
null, null,
common_timestamp()); common_timestamp());
rename($temp_filename, Avatar::path($filename)); rename($temp_filename, Avatar::path($filename));
} catch (Exception $e) {
unlink($temp_filename);
throw $e;
}
return $profile->setOriginal($filename); return $profile->setOriginal($filename);
} }

View File

@ -1053,6 +1053,7 @@ class Ostatus_profile extends Memcached_DataObject
// @fixme this should be better encapsulated // @fixme this should be better encapsulated
// ripped from oauthstore.php (for old OMB client) // ripped from oauthstore.php (for old OMB client)
$temp_filename = tempnam(sys_get_temp_dir(), 'listener_avatar'); $temp_filename = tempnam(sys_get_temp_dir(), 'listener_avatar');
try {
if (!copy($url, $temp_filename)) { if (!copy($url, $temp_filename)) {
throw new ServerException(sprintf(_m("Unable to fetch avatar from %s."), $url)); throw new ServerException(sprintf(_m("Unable to fetch avatar from %s."), $url));
} }
@ -1069,6 +1070,10 @@ class Ostatus_profile extends Memcached_DataObject
null, null,
common_timestamp()); common_timestamp());
rename($temp_filename, Avatar::path($filename)); rename($temp_filename, Avatar::path($filename));
} catch (Exception $e) {
unlink($temp_filename);
throw $e;
}
// @fixme hardcoded chmod is lame, but seems to be necessary to // @fixme hardcoded chmod is lame, but seems to be necessary to
// keep from accidentally saving images from command-line (queues) // keep from accidentally saving images from command-line (queues)
// that can't be read from web server, which causes hard-to-notice // that can't be read from web server, which causes hard-to-notice

View File

@ -174,6 +174,7 @@ class WikiHowProfilePlugin extends Plugin
// @fixme this should be better encapsulated // @fixme this should be better encapsulated
// ripped from OStatus via oauthstore.php (for old OMB client) // ripped from OStatus via oauthstore.php (for old OMB client)
$temp_filename = tempnam(sys_get_temp_dir(), 'listener_avatar'); $temp_filename = tempnam(sys_get_temp_dir(), 'listener_avatar');
try {
if (!copy($url, $temp_filename)) { if (!copy($url, $temp_filename)) {
throw new ServerException(sprintf(_m("Unable to fetch avatar from %s."), $url)); throw new ServerException(sprintf(_m("Unable to fetch avatar from %s."), $url));
} }
@ -188,6 +189,10 @@ class WikiHowProfilePlugin extends Plugin
null, null,
common_timestamp()); common_timestamp());
rename($temp_filename, Avatar::path($filename)); rename($temp_filename, Avatar::path($filename));
} catch (Exception $e) {
unlink($temp_filename);
throw $e;
}
$profile->setOriginal($filename); $profile->setOriginal($filename);
} }
} }

View File

@ -436,6 +436,7 @@ class YammerImporter
// @fixme this should be better encapsulated // @fixme this should be better encapsulated
// ripped from oauthstore.php (for old OMB client) // ripped from oauthstore.php (for old OMB client)
$temp_filename = tempnam(sys_get_temp_dir(), 'listener_avatar'); $temp_filename = tempnam(sys_get_temp_dir(), 'listener_avatar');
try {
if (!copy($url, $temp_filename)) { if (!copy($url, $temp_filename)) {
throw new ServerException(sprintf(_m("Unable to fetch avatar from %s."), $url)); throw new ServerException(sprintf(_m("Unable to fetch avatar from %s."), $url));
} }
@ -448,6 +449,10 @@ class YammerImporter
null, null,
common_timestamp()); common_timestamp());
rename($temp_filename, Avatar::path($filename)); rename($temp_filename, Avatar::path($filename));
} catch (Exception $e) {
unlink($temp_filename);
throw $e;
}
// @fixme hardcoded chmod is lame, but seems to be necessary to // @fixme hardcoded chmod is lame, but seems to be necessary to
// keep from accidentally saving images from command-line (queues) // keep from accidentally saving images from command-line (queues)
// that can't be read from web server, which causes hard-to-notice // that can't be read from web server, which causes hard-to-notice