error code color

[sql-web.git] / main.php

diff --git a/main.php b/main.php
index a6a0f0e3a9053aa0b6c03bc87fbd519c264f682a..8a6ac81f5110d51e67c3cac6809bfc315d1b133b 100644 (file)
--- a/main.php
+++ b/main.php
@@ -3,6 +3,8 @@
 require_once('mitsql.cfg.php');
 require_once('mitsql.lib.php');
 
+define('MAXDBS', 20);
+
 if (!isLoggedIn()) redirect('index');
 
 $err1 = $msg1 = array();
@@ -13,9 +15,15 @@ if (isPost()) {
        if (isset($i_newdb)) {
                $dbname = $User->getUsername().DELIMETER.$i_newdb;
                if ($User->isOverQuota()) {
-                       $err1[] = 'You are over your quota. You may not add databases.';
-               } elseif ($User->addDB($dbname)==false) {
-                       $err1[] = mysql_error();
+                       $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 (!$User->addDB($dbname)) {
+                       if (mysql_error()) {
+                               $err1[] = mysql_error();
+                       } else {
+                               $err1[] = 'Database already exists.';
+                       }
                } else {
                        $msg1[] = 'Database `'.$dbname.'` created.';
                }