From cee2276dc0bd37f7981e885f9dd32cd0e113c1cc Mon Sep 17 00:00:00 2001
From: Cash Costello <cash.costello@gmail.com>
Date: Sat, 23 Oct 2010 12:53:41 +0000
Subject: updating album image list when a photo is deleted

---
 actions/admin/upgrade.php |  2 ++
 actions/delete.php        | 21 +++++++++++----------
 lib/album.php             | 19 +++++++++++++++++++
 lib/image.php             | 13 +++++++++++++
 4 files changed, 45 insertions(+), 10 deletions(-)

diff --git a/actions/admin/upgrade.php b/actions/admin/upgrade.php
index 9c4aef1cf..334e9588e 100644
--- a/actions/admin/upgrade.php
+++ b/actions/admin/upgrade.php
@@ -11,6 +11,8 @@ if ($version <= $local_version) {
 	forward($_SERVER['HTTP_REFERER']);
 }
 
+set_time_limit(0);
+
 $base_dir = $CONFIG->pluginspath . 'tidypics/upgrades';
 
 // taken from engine/lib/version.php
diff --git a/actions/delete.php b/actions/delete.php
index c1e0ed59e..9b8c6ac6f 100644
--- a/actions/delete.php
+++ b/actions/delete.php
@@ -53,7 +53,7 @@ $image_repo_size_md = get_metadata_byname($owner_guid, "image_repo_size");
 $image_repo_size = (int)$image_repo_size_md->value;
 
 //loop through all images and delete them
-foreach($images as $im) {
+foreach ($images as $im) {
 	$thumbnail = $im->thumbnail;
 	$smallthumb = $im->smallthumb;
 	$largethumb = $im->largethumb;
@@ -76,16 +76,17 @@ foreach($images as $im) {
 		$delfile->setFilename($largethumb);
 		$delfile->delete();
 	}
-	if ($im) { //delete actual image file
-		$delfile = new ElggFile($im->getGUID());
-		$delfile->owner_guid = $im->getOwner();
-		//$delfile->setFilename($im->originalfilename);
-		$image_repo_size -= $delfile->size();
+	if ($im) {
+		$image_repo_size -= $im->size();
 
-		if (!$delfile->delete()) {
-			if ($subtype=='image') register_error(elgg_echo("tidypics:deletefailed")); //unable to delete object
+		if (!$im->delete()) {
+			if ($subtype=='image') {
+				register_error(elgg_echo("tidypics:deletefailed")); //unable to delete object
+			}
 		} else {
-			if ($subtype=='image') system_message(elgg_echo("tidypics:deleted")); //successfully deleted object
+			if ($subtype=='image') {
+				system_message(elgg_echo("tidypics:deleted")); //successfully deleted object
+			}
 		}
 	} //end delete actual image file
 } //end looping through each image to delete it
@@ -107,7 +108,7 @@ if ($subtype == 'album') {
 		rmdir($albumdir);
 	}
 
-	//delete object from database
+	//delete album object from database
 	if (!$entity->delete()) {
 		register_error(elgg_echo("tidypics:deletefailed")); //unable to delete object
 	} else {
diff --git a/lib/album.php b/lib/album.php
index 87bb715f4..15450049a 100644
--- a/lib/album.php
+++ b/lib/album.php
@@ -138,4 +138,23 @@ class TidypicsAlbum extends ElggObject {
 		}
 		return $imageList[$key];
 	}
+
+	/**
+	 * Remove an image from the album list
+	 *
+	 * @param int $imageGuid
+	 * @return bool
+	 */
+	public function removeImage($imageGuid)  {
+		$imageList = $this->getImageList();
+		$key = array_search($imageGuid, $imageList);
+		if ($key === FALSE) {
+			return FALSE;
+		}
+		
+		unset($imageList[$key]);
+		$this->setImageList($imageList);
+
+		return TRUE;
+	}
 }
diff --git a/lib/image.php b/lib/image.php
index 4100f3f7b..c60eabfc0 100644
--- a/lib/image.php
+++ b/lib/image.php
@@ -17,6 +17,19 @@ class TidypicsImage extends ElggFile {
 		parent::__construct($guid);
 	}
 
+	/**
+	 * delete image
+	 *
+	 * @return bool
+	 */
+	public function delete() {
+
+		$album = get_entity($this->container_guid);
+		$album->removeImage($this->guid);
+
+		return parent::delete();
+	}
+
 	/**
 	 * Has the photo been tagged with "in this photo" tags
 	 *
-- 
cgit v1.2.3