updates for invitations

Add the code to registration to handle invitation codes.

Some edge cases on invitations: is the user already subbed to this
person? Tell them. Is the person already on the system? Sub the user
to them, then, and tell the user.

Add some code to User to auto-sub invitees whenever the email address
changes. Call it from a new registration with an invite code, and also
from confirmaddress.

Some whitespace cleanup in the files touched.

darcs-hash:20080827001927-84dde-b50e5d921ca3f2fb894821730ff93cac09d2ba66.gz

3 files changed, 119 insertions, 54 deletions

diff --git a/actions/confirmaddress.php b/actions/confirmaddress.php
index 3cffda871..44280e08a 100644
--- a/actions/confirmaddress.php
+++ b/actions/confirmaddress.php
@@ -73,6 +73,10 @@ class ConfirmaddressAction extends Action {
             return;
         }
 
+		if ($type == 'email') {
+		    $cur->emailChanged();
+		}
+
         $result = $confirm->delete();
 
         if (!$result) {
diff --git a/actions/invite.php b/actions/invite.php
index e230ec6ee..3a68a60de 100644
--- a/actions/invite.php
+++ b/actions/invite.php
@@ -20,11 +20,11 @@
 if (!defined('LACONICA')) { exit(1); }
 
 class InviteAction extends Action {
-	
-	function is_readonly() {				
+
+	function is_readonly() {
 		return false;
 	}
-	
+
     function handle($args) {
         parent::handle($args);
 		if (!common_logged_in()) {
@@ -32,70 +32,79 @@ class InviteAction extends Action {
 										common_config('site', 'name')));
 			return;
 		} else if ($_SERVER['REQUEST_METHOD'] == 'POST') {
-			$this->send_invitation();
+			$this->send_invitations();
 		} else {
 			$this->show_form();
 		}
 	}
-	
-	function send_invitation() {
-		
+
+	function send_invitations() {
+
 		$user = common_current_user();
 		$profile = $user->getProfile();
-		
+
 		$bestname = $profile->getBestName();
 		$sitename = common_config('site', 'name');
 		$personal = $this->trimmed('personal');
 
 		$addresses = explode("\n", $this->trimmed('addresses'));
-		
+
 		foreach ($addresses as $email) {
-			
 			$email = trim($email);
-			
 			if (!Validate::email($email, true)) {
 				$this->show_form(sprintf(_('Invalid email address: %s'), $email));
 				return;
 			}
 		}
-		
+
+		$already = array();
+		$subbed = array();
+
 		foreach ($addresses as $email) {
-			
-			$email = trim($email);
-			
-			$recipients = array($email);
-			
-			$headers['From'] = mail_notify_from();
-			$headers['To'] = $email;
-			$headers['Subject'] = sprintf(_('%1s has invited you to join them on %2s'), $bestname, $sitename);
-
-			$body = sprintf(_("%1s has invited you to join them on %2s (%3s).\n\n".
-							  "%4s is a micro-blogging service that lets you keep up-to-date with people you know and people who interest you.\n\n".
-							  "You can also share news about yourself, your thoughts, or your life online with people who know about you.\n\n".
-							  "%5s said:\n\n%6s\n\n".
-							  "You can see %7s's profile page on %8s here:\n\n".
-							  "%9s\n\n".
-							  "If you'd like to try the service, click on the link below to accept the invitation.\n\n".
-							  "%10s\n\n".
-							  "If not, you can ignore this message. Thanks for your patience and your time.\n\n".
-							  "Sincerely, %11s\n"),
-							$bestname, $sitename, common_root_url(),
-							$sitename,
-							$bestname, $personal,
-							$bestname, $sitename,
-							common_local_url('showstream', array('nickname' => $user->nickname)),
-							common_local_url('register', array('code' => $invite->code)),
-							$sitename);
-			
-			mail_send($recipients, $headers, $body);
+			$email = common_canonical_email($email);
+			$other = User::staticGet('email', $email);
+			if ($other) {
+				if ($user->isSubscribed($other)) {
+					$already[] = $other;
+				} else {
+					subs_subscribe_to($user, $other);
+					$subbed[] = $other;
+				}
+			} else {
+				$sent[] = $email;
+				$this->send_invitation($email, $user);
+			}
 		}
-		
 
 		common_show_header(_('Invitation(s) sent'));
-		common_element('p', NULL, _('Invitation(s) sent. You will be notified when your invitees accept the invitation and register on the site. Thanks for growing the community!'));
+		if ($already) {
+			common_element('p', NULL, _('You are already subscribed to these users:'));
+			common_element_start('ul');
+			foreach ($already as $other) {
+				common_element('li', NULL, sprintf(_('%s (%s)'), $other->nickname, $other->email));
+			}
+			common_element_end('ul');
+		}
+		if ($subbed) {
+			common_element('p', NULL, _('These people are already users and you were automatically subscribed to them:'));
+			common_element_start('ul');
+			foreach ($subbed as $other) {
+				common_element('li', NULL, sprintf(_('%s (%s)'), $other->nickname, $other->email));
+			}
+			common_element_end('ul');
+		}
+		if ($sent) {
+			common_element('p', NULL, _('Invitation(s) sent to the following people:'));
+			common_element_start('ul');
+			foreach ($sent as $other) {
+				common_element('li', NULL, $sent);
+			}
+			common_element_end('ul');
+			common_element('p', NULL, _('You will be notified when your invitees accept the invitation and register on the site. Thanks for growing the community!'));
+		}
 		common_show_footer();
 	}
-	
+
 	function show_top($error=NULL) {
 		if ($error) {
 			common_element('p', 'error', $error);
@@ -108,7 +117,7 @@ class InviteAction extends Action {
 	}
 
 	function show_form($error=NULL) {
-		
+
 		global $config;
 
 		common_show_header(_('Invite new users'), NULL, $error, array($this, 'show_top'));
@@ -120,15 +129,59 @@ class InviteAction extends Action {
 		common_textarea('addresses', _('Email addresses'),
 						$this->trimmed('addresses'),
 						_('Addresses of friends to invite (one per line)'));
-		
+
 		common_textarea('personal', _('Personal message'),
 						$this->trimmed('personal'),
 						_('Optionally add a personal message to the invitation.'));
-		
+
 		common_submit('preview', _('Preview'));
 
 		common_element_end('form');
-		
+
 		common_show_footer();
 	}
+
+	function send_invitation($email, $user) {
+
+		$email = trim($email);
+
+		$invite = new Invitation();
+
+		$invite->address = $email;
+		$invite->type = 'email';
+		$invite->user_id = $user->id;
+		$invite->created = common_sql_now();
+
+		if (!$invite->insert()) {
+			common_log_db_error($invite, 'INSERT', __FILE__);
+			return false;
+		}
+
+		$recipients = array($email);
+
+		$headers['From'] = mail_notify_from();
+		$headers['To'] = $email;
+		$headers['Subject'] = sprintf(_('%1s has invited you to join them on %2s'), $bestname, $sitename);
+
+		$body = sprintf(_("%1s has invited you to join them on %2s (%3s).\n\n".
+						  "%4s is a micro-blogging service that lets you keep up-to-date with people you know and people who interest you.\n\n".
+						  "You can also share news about yourself, your thoughts, or your life online with people who know about you.\n\n".
+						  "%5s said:\n\n%6s\n\n".
+						  "You can see %7s's profile page on %8s here:\n\n".
+						  "%9s\n\n".
+						  "If you'd like to try the service, click on the link below to accept the invitation.\n\n".
+						  "%10s\n\n".
+						  "If not, you can ignore this message. Thanks for your patience and your time.\n\n".
+						  "Sincerely, %11s\n"),
+						$bestname, $sitename, common_root_url(),
+						$sitename,
+						$bestname, $personal,
+						$bestname, $sitename,
+						common_local_url('showstream', array('nickname' => $user->nickname)),
+						common_local_url('register', array('code' => $invite->code)),
+						$sitename);
+
+		mail_send($recipients, $headers, $body);
+	}
+
 }
diff --git a/actions/register.php b/actions/register.php
index f366bd3e0..9a871dd53 100644
--- a/actions/register.php
+++ b/actions/register.php
@@ -55,6 +55,10 @@ class RegisterAction extends Action {
 		$password = $this->arg('password');
 		$confirm = $this->arg('confirm');
 
+		# invitation code, if any
+
+		$code = $this->trimmed('code');
+
 		# Input scrubbing
 
 		$nickname = common_canonical_nickname($nickname);
@@ -93,8 +97,8 @@ class RegisterAction extends Action {
 		} else if ($password != $confirm) {
 			$this->show_form(_('Passwords don\'t match.'));
 		} else if ($user = User::register(array('nickname' => $nickname, 'password' => $password, 'email' => $email,
-												'fullname' => $fullname, 'homepage' => $homepage, 'bio' => $bio, 
-												'location' => $location))) {
+												'fullname' => $fullname, 'homepage' => $homepage, 'bio' => $bio,
+												'location' => $location, 'code' => $code))) {
 			if (!$user) {
 				$this->show_form(_('Invalid username or password.'));
 				return;
@@ -149,15 +153,19 @@ class RegisterAction extends Action {
 		common_element_start('form', array('method' => 'post',
 										   'id' => 'login',
 										   'action' => common_local_url('register')));
-		common_hidden('token', common_session_token());
 		common_input('nickname', _('Nickname'), $this->trimmed('nickname'),
 					 _('1-64 lowercase letters or numbers, no punctuation or spaces. Required.'));
 		common_password('password', _('Password'),
 						_('6 or more characters. Required.'));
 		common_password('confirm', _('Confirm'),
 						_('Same as password above. Required.'));
-		common_input('email', _('Email'), $this->trimmed('email'),
+		if ($invite && $invite->address_type == 'email') {
+			common_input('email', _('Email'), $invite->address,
 					 _('Used only for updates, announcements, and password recovery'));
+		} else {
+			common_input('email', _('Email'), $this->trimmed('email'),
+						 _('Used only for updates, announcements, and password recovery'));
+		}
 		common_input('fullname', _('Full name'),
 					 $this->trimmed('fullname'),
 					  _('Longer name, preferably your "real" name'));
@@ -170,7 +178,7 @@ class RegisterAction extends Action {
 		common_input('location', _('Location'),
 					 $this->trimmed('location'),
 					 _('Where you are, like "City, State (or Region), Country"'));
-		common_checkbox('rememberme', _('Remember me'), 
+		common_checkbox('rememberme', _('Remember me'),
 						$this->boolean('rememberme'),
 		                _('Automatically login in the future; not for shared computers!'));
 		common_element_start('p');
@@ -191,7 +199,7 @@ class RegisterAction extends Action {
 		common_element_end('form');
 		common_show_footer();
 	}
-						
+
 	function show_success() {
 		$nickname = $this->arg('nickname');
 		common_show_header(_('Registration successful'));
@@ -214,5 +222,5 @@ class RegisterAction extends Action {
 		common_element_end('div');
 		common_show_footer();
 	}
-						
+
 }