1 files changed, 79 insertions, 1 deletions

diff --git a/htdocs/src/User.inc.php b/htdocs/src/User.inc.php
index 6ad5ebb..0333520 100644
--- a/htdocs/src/User.inc.php
+++ b/htdocs/src/User.inc.php
@@ -12,8 +12,19 @@ require_once "base.inc.php";
 
 class User {
     private $info;
+    private $indatabase;
 
+    public function __construct($info=null) {
+        $this->info = is_null($info) ? array() : $info;
 
+        if ($info !== null and isset($info['id'])) {
+            $this->indatabase = true;
+
+        } else {
+            $this->indatabase = false;
+        }
+    }
+    
     public static function getById($id) {
         $where = "id='$id'";
 
@@ -36,6 +47,7 @@ class User {
         if ($row) {
             $user = new User();
             $user->info = $row;
+            $user->indatabase = true;
 
             return $user;
 
@@ -47,7 +59,45 @@ class User {
     public function save() {
         $db = getDatabase();
 
-        // TODO: Implement save 
+        // Cleanup Info
+        foreach ($this->info as $key=>$value)
+            $info[$key] = addslashes($value);
+
+        // Save or create?
+        if ($this->indatabase) {
+            try {
+                $db->update('user', $info, "WHERE `id`='"
+                    . $this->getId() ."'");
+                return true;
+
+            } catch (Cif_Database_Exception $e) {
+                return false;
+            }
+
+        } else {
+            // Creating... set special fields.
+            try {
+                $ret = $db->insert('user', $info);
+
+                if ($ret) {
+                    $this->info['id'] = $ret;
+                    $this->indatabase = true;
+                }
+
+                return true;
+
+            } catch (Cif_Database_Exception $e) {
+                return false;
+            }
+        }
+    }
+
+    public function delete() {
+        $db = getDatabase();
+
+        $db->delete('user', 'id=' . $this->getId());
+
+        $this->indatabase = false;
     }
 
     public function getId() {
@@ -58,10 +108,34 @@ class User {
         return $this->info['name'];
     }
 
+    public function setName($value) {
+        $this->info['name'] = $value;
+    }
+
     public function getEmail() {
         return $this->info['email'];
     }
 
+    public function setEmail($value) {
+        $this->info['email'] = $value;
+    }
+
+    public function getNotify() {
+        return $this->info['notify'];
+    }
+
+    public function setNotify($value) {
+        $this->info['notify'] = $value ? 1 : 0;
+    }
+
+    public function getSource() {
+        return $this->info['source_id'];
+    }
+
+    public function setSource($value) {
+        $this->info['source_id'] = $value;
+    }
+
     public function setPassword($password) {
         $this->info['password'] = sha1($password);
     }
@@ -73,6 +147,10 @@ class User {
     public function isAdmin() {
         return $this->info['admin'] == 1;
     }
+
+    public function setAdmin($value) {
+        $this->info['admin'] = $value ? 1 : 0;
+    }
 }
 
 ?>