From 8f5aa6a87b15ecee226f46539b1400710c9c9047 Mon Sep 17 00:00:00 2001
From: Cash Costello <cash.costello@gmail.com>
Date: Sat, 13 Jun 2009 23:53:17 +0000
Subject: added disk quota for users and groups

---
 actions/delete.php | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

(limited to 'actions/delete.php')

diff --git a/actions/delete.php b/actions/delete.php
index f6b677c55..3deb69c5b 100644
--- a/actions/delete.php
+++ b/actions/delete.php
@@ -30,13 +30,17 @@
 		register_error(elgg_echo("file:deletefailed"));
 		forward($forward_url);
 	}
-	
+
+	$owner_guid = 0; // group or user
 	if ($subtype == 'image') { //deleting an image
+		$album = get_entity($entity->container_guid);
+		$owner_guid = $album->container_guid;
 		$forward_url = $container->getURL(); //forward back to album after deleting pictures
 		$images = array($entity);
 		// plugins can register to be told when a Tidypics image has been deleted
 		trigger_elgg_event('upload', 'tp_album', $entity);
 	} else { //deleting an album
+		$owner_guid = $entity->container_guid;
 		$forward_url = 'pg/photos/owned/' . $container->username;
 		//get all the images from this album as long as less than 999 images
 		$images = get_entities("object", "image", $guid, '', 999); 
@@ -44,6 +48,10 @@
 		trigger_elgg_event('upload', 'tp_album', $entity);
 	}
 
+	// make sure we decrease the repo size for the size quota
+	$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) {
 		$thumbnail = $im->thumbnail;
@@ -72,6 +80,8 @@
 			$delfile = new ElggFile($im->getGUID());
 			$delfile->owner_guid = $im->getOwner();
 			//$delfile->setFilename($im->originalfilename);
+			$image_repo_size -= $delfile->size();
+			
 			if (!$delfile->delete()) {
 				if ($subtype=='image') register_error(elgg_echo("file:deletefailed")); //unable to delete object
 			} else {
@@ -104,6 +114,7 @@
 		}
 	} //end of delete album
 
+	create_metadata($owner_guid, "image_repo_size", $image_repo_size, 'integer', $owner_guid);
 
 	forward($forward_url);
 
-- 
cgit v1.2.3