login remember username

[sql-web.git] / main.php

diff --git a/main.php b/main.php
index 775fbda243922b8f63c481d6b170855d902bdeb9..a6a0f0e3a9053aa0b6c03bc87fbd519c264f682a 100644 (file)
--- a/main.php
+++ b/main.php
@@ -5,16 +5,37 @@ require_once('mitsql.lib.php');
 
 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 databases.';
+               } elseif ($User->addDB($dbname)==false) {
+                       $err1[] = mysql_error();
+               } 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';
 
 ?>