Don't forget the date\!

[cocanwiki.git] / scripts / invite_user_confirm_form.ml

diff --git a/scripts/invite_user_confirm_form.ml b/scripts/invite_user_confirm_form.ml
index a61726d..8a660bb 100644 (file)
--- a/scripts/invite_user_confirm_form.ml
+++ b/scripts/invite_user_confirm_form.ml
@@ -1,7 +1,7 @@
 (* COCANWIKI - a wiki written in Objective CAML.
  * Written by Richard W.M. Jones <rich@merjis.com>.
  * Copyright (C) 2004 Merjis Ltd.
- * $Id: invite_user_confirm_form.ml,v 1.2 2004/10/23 15:00:15 rich Exp $
+ * $Id: invite_user_confirm_form.ml,v 1.6 2006/12/06 09:46:57 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
@@ -28,25 +28,26 @@ open Cocanwiki
 open Cocanwiki_ok
 open Cocanwiki_template
 
-let run r (q : cgi) (dbh : Dbi.connection) hostid _ _ =
-  let template = get_template dbh hostid "invite_user_confirm_form.html" in
+let run r (q : cgi) dbh hostid _ _ =
+  let template = get_template r dbh hostid "invite_user_confirm_form.html" in
 
   (* Get the invite ID. *)
   let invite = q#param "p" in
 
-  let sth = dbh#prepare_cached "select name from users
-                                 where hostid = ? and invite = ?" in
-  sth#execute [`Int hostid; `String invite];
+  let rows = PGSQL(dbh) "select name from users
+                          where hostid = $hostid and invite = $invite" in
 
   let username =
-    try sth#fetch1string ()
-    with
-       Not_found ->
-         error ~title:"Already signed up"
-           q ("It looks like you have already used your invitation.  If " ^
-              "you cannot get to your account, please contact the " ^
-              "administrator.");
-         return () in
+    match rows with
+    | [username] -> username
+    | [] ->
+       error ~title:"Already signed up"
+         r dbh hostid q
+         ("It looks like you have already used your invitation.  If " ^
+            "you cannot get to your account, please contact the " ^
+            "administrator.");
+       return ()
+    | _ -> assert false in
 
   (* Update the template so that the user can set their preferred password. *)
   template#set "username" username;