X-Git-Url: http://andersk.mit.edu/gitweb/sql-web.git/blobdiff_plain/2ea5a69f3075db3f0e2ca4d6c815bcffe6b65dcd..9c70b48184fc9dc34b9725daa6b3b050fd49e6d0:/main.php

diff --git a/main.php b/main.php
index 0e919c3..90bf2ea 100644
--- a/main.php
+++ b/main.php
@@ -10,12 +10,26 @@ $User = new User($Login->getUserID());
 if (isPost()) {
 	if (isset($i_newdb)) {
 		$dbname = $User->getUsername().DELIMETER.$i_newdb;
-		if ($User->addDB($dbname)==false) {
+		if ($User->isOverQuota()) {
+			$err[] = 'You are over your quota. You may not add databases.';
+		} elseif ($User->addDB($dbname)==false) {
 			$err[] = mysql_error();
 		} else {
 			$msg[] = 'Database `'.$dbname.'` created.';
 		}
 	}
+	if (isset($i_drop)) {
+		$dropdbs = array_keys($i_drop);
+		foreach($dropdbs as $dbname) {
+			if ($User->delDB($dbname)) {
+				$msg[] = 'Database `'.$dbname.'` dropped.';
+			} else {
+				$err[] = mysql_error();
+			}
+		}
+	}
+	if (!count($err))
+		$User->refresh();
 }
 
 $myDBs = $User->getDBList();