scripts/invite_user.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.ml,v 1.2 2004/10/23 15:00:15 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_template
  30
  31 let split_re = Pcre.regexp "[\\s,;]+"
  32
  33 let run r (q : cgi) (dbh : Dbi.connection) hostid {hostname = hostname} user =
  34   let template = _get_template "invite_user.txt" in
  35   let template_exists = _get_template "invite_user_exists.txt" in
  36
  37   (* Get the email addresses. *)
  38   let emails = q#param "emails" in
  39   let emails = Pcre.split ~rex:split_re emails in
  40
  41   (* This guy's got no friends ... *)
  42   if emails = [] then (
  43     q#redirect ("http://" ^ hostname ^ "/");
  44     return ()
  45   );
  46
  47   let userid, username = match user with
  48       User (userid, username, _) -> userid, username
  49     | _ -> assert false in
  50
  51   let subject = username ^ " has invited you to join " ^ hostname in
  52
  53   (* Get user's own email address, which will be the return address
  54    * for the email.
  55    *)
  56   let sth = dbh#prepare_cached "select email from users
  57                                  where hostid = ? and id = ?" in
  58   sth#execute [`Int hostid; `Int userid];
  59
  60   let from =
  61     match sth#fetch1 () with
  62         [ `String email ] -> email
  63       | [ `Null ] -> "service@merjis.com"
  64       | _ -> assert false in
  65
  66   (* Add user accounts for these new users.  For users who are already
  67    * registered, we'll send reminder emails.
  68    *)
  69   List.iter
  70     (fun email ->
  71        let sth = dbh#prepare_cached "select id, invite from users
  72                                       where hostid = ? and
  73                                             (email = ? or name = ?)" in
  74        sth#execute [`Int hostid; `String email; `String email];
  75
  76        let body =
  77          try
  78            (match sth#fetch1 () with
  79                 [ `Int userid; `Null ] ->
  80                   (* Existing user account - send reminder. *)
  81                   template_exists#set "username" username;
  82                   template_exists#set "hostname" hostname;
  83                   template_exists#to_string
  84
  85               | [ `Int userid; `String invite ] ->
  86                   (* Existing user account - resend the invite. *)
  87                   template#set "username" username;
  88                   template#set "hostname" hostname;
  89                   template#set "invite" invite;
  90                   template#to_string
  91
  92               | _ -> assert false)
  93          with
  94              Not_found ->
  95                (* Add user account. *)
  96                let invite = random_sessionid () in
  97                let sth = dbh#prepare_cached "insert into users (hostid, name,
  98                  password, email, invite) values (?, ?, ?, ?, ?)" in
  99                sth#execute [`Int hostid; `String email; `String invite;
 100                             `String email; `String invite];
 101
 102                template#set "username" username;
 103                template#set "hostname" hostname;
 104                template#set "invite" invite;
 105                template#to_string in
 106
 107        (* Send the email. *)
 108        Sendmail.send_mail ~subject ~to_addr:[email] ~from ~body ()
 109     ) emails;
 110
 111   (* Finish off. *)
 112   dbh#commit ();
 113
 114   let buttons = [ ok_button "/_users" ] in
 115   ok ~buttons ~title:"Invitation emails sent"
 116     q "We sent invitations emails to those address(es)."
 117
 118 let () =
 119   register_script ~restrict:[CanManageUsers] ~anonymous:false run