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