scripts/invite_user_confirm.ml

   1 (* COCANWIKI - a wiki written in Objective CAML.
   2  * Written by Richard W.M. Jones <rich@merjis.com>.
   3  * Copyright (C) 2004 Merjis Ltd.
   4  * $Id: invite_user_confirm.ml,v 1.9 2006/12/06 09:46:57 rich Exp $
   5  *
   6  * This program is free software; you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License as published by
   8  * the Free Software Foundation; either version 2 of the License, or
   9  * (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  * GNU General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * along with this program; see the file COPYING.  If not, write to
  18  * the Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  19  * Boston, MA 02111-1307, USA.
  20  *)
  21
  22 open Apache
  23 open Registry
  24 open Cgi
  25 open Printf
  26
  27 open Cocanwiki
  28 open Cocanwiki_ok
  29 open Cocanwiki_strings
  30 open Cocanwiki_template
  31
  32 let run r (q : cgi) dbh hostid { hostname = hostname } _ =
  33   let template = _get_template "invite_user_confirm.txt" in
  34
  35   let username = q#param "username" in
  36   let invite = q#param "invite" in
  37
  38   (* Verify the username, invite combination. *)
  39   let rows = PGSQL(dbh) "select email, id from users
  40                           where hostid = $hostid and
  41                                 name = $username and invite = $invite" in
  42
  43   let email, userid =
  44     match rows with
  45     | [ Some email, userid ] -> Some email, userid
  46     | [ None, userid ] -> None, userid
  47     | [] ->
  48         error ~title:"Already signed up"
  49           r dbh hostid q
  50           "It looks like you have already used your invitation.";
  51         return ()
  52     | _ -> assert false in
  53
  54   let password1 = q#param "password1" in
  55   let password2 = q#param "password2" in
  56
  57   if password1 = "" || password2 = "" then (
  58     error ~back_button:true ~title:"Bad password"
  59       r dbh hostid q "The password you gave is empty.";
  60     return ()
  61   );
  62
  63   if password1 <> password2 then (
  64     error ~back_button:true ~title:"Passwords don't match"
  65       r dbh hostid q "The two passwords you gave aren't identical.";
  66     return ()
  67   );
  68
  69   let password = password1 in
  70
  71   (* Change the password. *)
  72   PGSQL(dbh)
  73     "update users set password = $password, invite = null,
  74                       force_password_change = false
  75       where hostid = $hostid and id = $userid";
  76
  77   (* Send email to this user. *)
  78   (match email with
  79        None -> ()
  80      | Some email ->
  81          template#set "username" username;
  82          template#set "hostname" hostname;
  83
  84          let body = template#to_string in
  85
  86          let subject = "Your new account details" in
  87
  88          let to_addrs = [ "", email ] in
  89          let content_type =
  90            "text/plain", ["charset", Mimestring.mk_param "UTF-8"] in
  91
  92          let msg = Netsendmail.compose ~to_addrs ~subject ~content_type body in
  93          Netsendmail.sendmail msg);
  94
  95   PGOCaml.commit dbh;
  96
  97   (* Redirect to the login page. *)
  98   let redirect =
  99     "http://" ^ hostname ^ "/_bin/login.cmo?" ^
 100     "username=" ^ Cgi_escape.escape_url username ^ "&" ^
 101     "password=" ^ Cgi_escape.escape_url password ^ "&" ^
 102     "permanent=1" in
 103   q#redirect redirect
 104
 105 let () =
 106   register_script run