X-Git-Url: http://andersk.mit.edu/gitweb/sql-web.git/blobdiff_plain/377015e0d78cce3bd62af0357fcb951fe506efb4..5852fbc4ac1301741cfeca45a3fa6959261c57cb:/main.php

diff --git a/main.php b/main.php
index 775fbda..8fa2c3d 100644
--- a/main.php
+++ b/main.php
@@ -3,18 +3,49 @@
 require_once('mitsql.cfg.php');
 require_once('mitsql.lib.php');
 
+define('MAXDBS', 20);
+
 if (!isLoggedIn()) redirect('index');
 
+$err1 = $msg1 = array();
+
 $User = new User($Login->getUserID());
 
 if (isPost()) {
 	if (isset($i_newdb)) {
-		if ($User->addDB($User->getUsername().DELIMETER.$i_newdb)==false) {
-			$err[] = mysql_error();
+		$dbname = $User->getUsername().DELIMETER.$i_newdb;
+		if ($User->isOverQuota()) {
+			$err1[] = 'You are over your quota. You may not add more databases.';
+		} elseif (count($User->getDBList())>MAXDBS) {
+			$err1[] = 'You have too many databases. You may not add more databases.';
+		} elseif (empty($i_newdb)) {
+			$err1[] = 'Your database name may not be empty.';
+		} elseif (!$User->addDB($dbname)) {
+			if (mysql_error()) {
+				$err1[] = mysql_error();
+			} else {
+				$err1[] = 'Database already exists.';
+			}
+		} else {
+			$msg1[] = 'Database `'.$dbname.'` created.';
 		}
 	}
+	if (isset($i_drop)) {
+		$dropdbs = array_keys($i_drop);
+		foreach($dropdbs as $dbname) {
+			if ($User->delDB($dbname)) {
+				$msg1[] = 'Database `'.$dbname.'` dropped.';
+			} else {
+				$err1[] = mysql_error();
+			}
+		}
+	}
+	if (!count($err1))
+		$User->refresh();
 }
 
+$myDBs = $User->getDBList();
+
 include 'tpl/main.php';
 
 ?>