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.6 2006/03/27 18:09:46 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 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, email = match user with
  48       User (userid, username, _, prefs) -> userid, username, prefs.email
  49     | _ -> assert false in
  50
  51   let subject = username ^ " has invited you to join " ^ hostname in
  52
  53   (* Get a suitable return address for the email. *)
  54   let from =
  55     match email with
  56       | Some email -> email
  57       | None -> "service@merjis.com" in
  58
  59   (* Add user accounts for these new users.  For users who are already
  60    * registered, we'll send reminder emails.
  61    *)
  62   List.iter
  63     (fun email ->
  64        let sth = dbh#prepare_cached "select id, invite from users
  65                                       where hostid = ? and
  66                                             (email = ? or name = ?)" in
  67        sth#execute [Some hostid; Some email; Some email];
  68
  69        let body =
  70          try
  71            (match sth#fetch1 () with
  72                 [ Some userid; None ] ->
  73                   (* Existing user account - send reminder. *)
  74                   template_exists#set "username" username;
  75                   template_exists#set "hostname" hostname;
  76                   template_exists#to_string
  77
  78               | [ Some userid; Some invite ] ->
  79                   (* Existing user account - resend the invite. *)
  80                   template#set "username" username;
  81                   template#set "hostname" hostname;
  82                   template#set "invite" invite;
  83                   template#to_string
  84
  85               | _ -> assert false)
  86          with
  87              Not_found ->
  88                (* Add user account. *)
  89                let invite = random_sessionid () in
  90                let sth = dbh#prepare_cached "insert into users (hostid, name,
  91                  password, email, invite) values (?, ?, ?, ?, ?)" in
  92                sth#execute [Some hostid; Some email; Some invite;
  93                             Some email; Some invite];
  94
  95                template#set "username" username;
  96                template#set "hostname" hostname;
  97                template#set "invite" invite;
  98                template#to_string in
  99
 100        (* Send the email. *)
 101        Sendmail.send_mail ~subject ~to_addr:[email] ~from body
 102     ) emails;
 103
 104   (* Finish off. *)
 105   PGOCaml.commit dbh;
 106
 107   let buttons = [ ok_button "/_users" ] in
 108   ok ~buttons ~title:"Invitation emails sent"
 109     dbh hostid q "We sent invitations emails to those address(es)."
 110
 111 let () =
 112   register_script ~restrict:[CanManageUsers] ~anonymous:false run