About half way through switching cocanwiki to using the new PG interface.

[cocanwiki.git] / scripts / signup.ml

diff --git a/scripts/signup.ml b/scripts/signup.ml
index ea90b34..b98cede 100644 (file)
--- a/scripts/signup.ml
+++ b/scripts/signup.ml
@@ -1,7 +1,22 @@
-(* COCANWIKI scripts.
+(* COCANWIKI - a wiki written in Objective CAML.
  * Written by Richard W.M. Jones <rich@merjis.com>.
  * Copyright (C) 2004 Merjis Ltd.
- * $Id: signup.ml,v 1.1 2004/09/07 16:58:03 rich Exp $
+ * $Id: signup.ml,v 1.9 2005/11/24 14:54:13 rich Exp $
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; see the file COPYING.  If not, write to
+ * the Free Software Foundation, Inc., 59 Temple Place - Suite 330,
+ * Boston, MA 02111-1307, USA.
  *)
 
 open Apache
@@ -13,7 +28,7 @@ open Cocanwiki
 open Cocanwiki_ok
 open Cocanwiki_strings
 
-let run r (q : cgi) (dbh : Dbi.connection) (hostid, _, _) _ =
+let run r (q : cgi) (dbh : Dbi.connection) hostid _ _ =
   (* Verify that we're allowed to create accounts anonymously
    * on this host.
    *)
@@ -32,32 +47,39 @@ let run r (q : cgi) (dbh : Dbi.connection) (hostid, _, _) _ =
 
   if username = "" || password1 = "" || password2 = "" then (
     error ~back_button:true ~title:"Bad username or password"
-      q "The username or password you gave is empty.";
-    raise CgiExit
+      dbh hostid q "The username or password you gave is empty.";
+    return ()
   );
 
   if password1 <> password2 then (
     error ~back_button:true ~title:"Passwords don't match"
-      q "The two passwords you gave aren't identical.";
-    raise CgiExit
+      dbh hostid q "The two passwords you gave aren't identical.";
+    return ()
   );
 
   let password = password1 in
 
+  (*
+    Uh oh ... Not making UNICODE assumptions ... XXX
+  if String.length username > 32 || String.length password > 32 then
+  *)
+
   let email = trim (q#param "email") in
   let email = if string_is_whitespace email then `Null else `String email in
 
   (* Not a duplicate? *)
-  let sth = dbh#prepare_cached "select id from users where name = ?" in
-  sth#execute [`String username];
+  let sth = dbh#prepare_cached "select id from users
+                                 where hostid = ? and name = ?" in
+  sth#execute [`Int hostid; `String username];
 
   (try
      sth#fetch1 ();
      error ~back_button:true ~title:"Username already taken"
-       q ("Someone, possibly you, has already taken that username. " ^
-         "If you think you have forgotten your password, try going back " ^
-         "and clicking on the 'Forgotten your password?' link.");
-     raise CgiExit
+       dbh hostid q
+       ("Someone, possibly you, has already taken that username. " ^
+       "If you think you have forgotten your password, try going back " ^
+       "and clicking on the 'Forgotten your password?' link.");
+     return ()
    with
        Not_found -> ());
 
@@ -67,7 +89,7 @@ let run r (q : cgi) (dbh : Dbi.connection) (hostid, _, _) _ =
                                 values (?, ?, ?, ?)" in
   sth#execute [`String username; `String password; email; `Int hostid];
 
-  let userid = sth#serial "users_id_seq" in
+  let userid = Int64.to_int (sth#serial "users_id_seq") in
 
   (* Create a cookie. *)
   let cookie = random_sessionid () in
@@ -78,13 +100,14 @@ let run r (q : cgi) (dbh : Dbi.connection) (hostid, _, _) _ =
   dbh#commit ();
 
   let buttons = [ ok_button "/" ] in
-  let cookie = Cookie.cookie ~name:"auth" ~value:cookie ~path:"/" () in
+  let cookie = Cookie.cookie "auth" cookie ~path:"/" in
 
   ok ~title:"Account created"
     ~buttons
     ~cookie
-    q ("An account was created for you, " ^ username ^ ". " ^
-       "We hope you enjoy using this service.")
+    dbh hostid q
+    ("An account was created for you, " ^ username ^ ". " ^
+     "We hope you enjoy using this service.")
 
 let () =
   register_script run