getrecord_bycode and getitme_bycode methods added to db class

[brisk.git] / web / Obj / dbase_pgsql.phh

diff --git a/web/Obj/dbase_pgsql.phh b/web/Obj/dbase_pgsql.phh
index f7f67cd..d5b2770 100644 (file)
--- a/web/Obj/dbase_pgsql.phh
+++ b/web/Obj/dbase_pgsql.phh
@@ -157,15 +157,27 @@ class BriskDB
         return ($user_obj);
     }
 
+    function getrecord_bycode($code) {
+        GLOBAL $G_dbpfx;
+
+        $user_sql = sprintf("SELECT * FROM %susers WHERE code = %d;",  $G_dbpfx, $code);
+        if (($user_pg  = $this->query($user_sql)) == FALSE) {
+            return FALSE;
+        }
+        if (pg_numrows($user_pg) != 1)
+            return FALSE;
+
+        $user_obj = pg_fetch_object($user_pg, 0);
+
+        return ($user_obj);
+    }
+
     function user_update_login_time($code, $lintm)
     {
         GLOBAL $G_dbpfx;
 
         $user_sql = sprintf("UPDATE %susers SET (lintm) = (date 'epoch' + %d * INTERVAL '1 second') WHERE code = %d;", $G_dbpfx, $lintm, $code);
 
-        // $user_pg = $this->query($user_sql);
-        // $row_n = pg_affected_rows($user_pg);
-        // fprintf(STDERR, "query: %s   NUM: %d\n", ($user_pg == FALSE ? "FALSE" : "TRUE"), $row_n);
         if ( ! (($user_pg  = $this->query($user_sql)) != FALSE && pg_affected_rows($user_pg) == 1) ) {
              return FALSE;
         }
@@ -178,7 +190,37 @@ class BriskDB
         GLOBAL $G_dbpfx;
 
         $user_sql = sprintf("UPDATE %susers SET (type, supp_comp) = (%d, '%s') WHERE code = %d;",
-                            $G_dbpfx, $flags, $supp_comp, $code);
+                            $G_dbpfx, $flags, escsql($supp_comp), $code);
+        fprintf(STDERR, "REQUEST [%s]\n", $user_sql);
+        if ( ! (($user_pg  = $this->query($user_sql)) != FALSE && pg_affected_rows($user_pg) == 1) ) {
+             return FALSE;
+        }
+        fprintf(STDERR, "REQUEST GOOD [%s]\n", $user_sql);
+
+        return TRUE;
+    }
+
+    function user_state_update($code, $flags, $disa_reas)
+    {
+        GLOBAL $G_dbpfx;
+
+        $user_sql = sprintf("UPDATE %susers SET (type, disa_reas) = (%d, %d) WHERE code = %d;",
+                            $G_dbpfx, $flags, $disa_reas, $code);
+        fprintf(STDERR, "REQUEST [%s]\n", $user_sql);
+        if ( ! (($user_pg  = $this->query($user_sql)) != FALSE && pg_affected_rows($user_pg) == 1) ) {
+             return FALSE;
+        }
+        fprintf(STDERR, "REQUEST GOOD [%s]\n", $user_sql);
+
+        return TRUE;
+    }
+
+    function user_tos_update($code, $tos_vers)
+    {
+        GLOBAL $G_dbpfx;
+
+        $user_sql = sprintf("UPDATE %susers SET (tos_vers) = ('%s') WHERE code = %d;",
+                            $G_dbpfx, escsql($tos_vers), $code);
         fprintf(STDERR, "REQUEST [%s]\n", $user_sql);
         if ( ! (($user_pg  = $this->query($user_sql)) != FALSE && pg_affected_rows($user_pg) == 1) ) {
              return FALSE;
@@ -255,6 +297,17 @@ class BriskDB
         return (LoginDBItem::LoginDBItemFromRecord($user_obj));
     }
 
+    function getitem_bycode($code) {
+        $ret = FALSE;
+
+        log_main("getitem_bycode: ".$code);
+
+        if (($user_obj = $this->getrecord_bycode($code)) == FALSE)
+            return $ret;
+
+        return (LoginDBItem::LoginDBItemFromRecord($user_obj));
+    }
+
     // TODO FOR DB
     function getmail($login)
     {