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/upload.php | 25 +++++++++++++++++++++++--
 1 file changed, 23 insertions(+), 2 deletions(-)

(limited to 'actions/upload.php')

diff --git a/actions/upload.php b/actions/upload.php
index 5dacd5fec..e7c78c178 100644
--- a/actions/upload.php
+++ b/actions/upload.php
@@ -16,11 +16,18 @@
 	if (!$container_guid)
 		$container_guid == $_SESSION['user']->getGUID();
 
+	$album = get_entity($container_guid);
+
 	$maxfilesize = get_plugin_setting('maxfilesize','tidypics'); 
 	if (!$maxfilesize)
 		$maxfilesize = 5; // default to 5 MB if not set 
 	$maxfilesize = 1024 * 1024 * $maxfilesize; // convert to bytes from MBs
 
+	$quota = get_plugin_setting('quota','tidypics');
+	$quota = 1024 * 1024 * $quota;
+	$image_repo_size_md = get_metadata_byname($album->container_guid, "image_repo_size");
+	$image_repo_size = (int)$image_repo_size_md->value;
+
 	$image_lib = get_plugin_setting('image_lib', 'tidypics');
 	if (!$image_lib)
 		$image_lib = 'GD';
@@ -87,6 +94,15 @@
 			continue;
 		}
 
+		// check quota
+		if ($quota) {
+			if ($image_repo_size + $sent_file['size'] > $quota) {
+				array_push($not_uploaded, $sent_file['name']);
+				array_push($error_msgs, elgg_echo('tidypics:exceed_quota'));
+				continue;
+			}
+		}
+		
 		// make sure file does not exceed memory limit
 		if ($sent_file['size'] > $maxfilesize) {
 			array_push($not_uploaded, $sent_file['name']);
@@ -178,6 +194,9 @@
 		td_get_exif($file);
 		array_push($uploaded_images, $file->guid);
 
+		// update user/group size for checking quota
+		$image_repo_size += $sent_file['size'];
+
 		if($river_view == "all") {
 			add_to_river('river/object/image/create', 'create', $file->getObjectOwnerGUID(), $file->getGUID());
 		}
@@ -207,7 +226,6 @@
 	}
 
 	// successful upload so check if this is a new album and throw river event if so
-	$album = get_entity($container_guid);
 	if ($album->new_album == TP_NEW_ALBUM) {
 		if (function_exists('add_to_river'))
 			add_to_river('river/object/album/create', 'create', $album->owner_guid, $album->guid);
@@ -219,7 +237,10 @@
 			add_to_river('river/object/image/create', 'create', $file_for_river->getObjectOwnerGUID(), $file_for_river->getGUID());
 		}
 	}
-			
+	
+	// update image repo size
+	create_metadata($album->container_guid, "image_repo_size", $image_repo_size, 'integer', $album->container_guid);
+	
 	// plugins can register to be told when a Tidypics album has had images added
 	trigger_elgg_event('upload', 'tp_album', $album);
 	
-- 
cgit v1.2.3