begin invites, fb setup changes

2 files changed, 52 insertions, 5 deletions

diff --git a/mod/contacts.php b/mod/contacts.php
index 7a97b53d2..388bf08e6 100644
--- a/mod/contacts.php
+++ b/mod/contacts.php
@@ -3,13 +3,26 @@
 require_once('include/Contact.php');
 
 function contacts_init(&$a) {
+	if(! local_user())
+		return;
+
 	require_once('include/group.php');
 	if(! x($a->page,'aside'))
 		$a->page['aside'] = '';
 	$a->page['aside'] .= group_side();
 
-	if($a->config['register_policy'] != REGISTER_CLOSED)
-		$a->page['aside'] .= '<div class="side-link" id="side-invite-link" ><a href="invite" >' . t("Invite Friends") . '</a></div>';
+	$inv = '<div class="side-link" id="side-invite-link" ><a href="invite" >' . t("Invite Friends") . '</a></div>';
+
+	if(get_config('system','invitations_only')) {
+		$x = get_pconfig(local_user(),'system','invites_remaining');
+		if($x || is_site_admin()) {
+			$a->page['aside'] .= '<div class="side-link" id="side-invite-remain">' 
+			. sprintf( tt('%d invitation available','%d invitations available',$x), $x) 
+			. '</div>' . $inv;
+		}
+	}
+	elseif($a->config['register_policy'] != REGISTER_CLOSED)
+		$a->page['aside'] .= $inv;
 
 
 	$a->page['aside'] .= '<div class="side-link" id="side-match-link"><a href="match" >' 
diff --git a/mod/invite.php b/mod/invite.php
index b52aa19f6..5a6745436 100644
--- a/mod/invite.php
+++ b/mod/invite.php
@@ -20,6 +20,13 @@ function invite_post(&$a) {
 
 	$total = 0;
 
+	if(get_config('system','invitations_only')) {
+		$invonly = true;
+		$x = get_pconfig(local_user(),'system','invites_remaining');
+		if((! $x) && (! is_site_admin()))
+			return;
+	}
+
 	foreach($recips as $recip) {
 
 		$recip = trim($recip);
@@ -28,9 +35,25 @@ function invite_post(&$a) {
 			notice(  sprintf( t('%s : Not a valid email address.'), $recip) . EOL);
 			continue;
 		}
+		
+		if($invonly && ($x || is_site_admin())) {
+			$code = autoname(8) . srand(1000,9999);
+			$nmessage = str_replace('$invite_code',$code,$message);
+			// store in db
+
+			if(! is_site_admin()) {
+				$x --;
+				if($x >= 0)
+					set_pconfig(local_user(),'system','invites_remaining,$x);
+				else
+					return;
+			}
+		}
+		else
+			$nmessage = $message;
 
 		$res = mail($recip, sprintf( t('Please join my network on %s'), $a->config['sitename']), 
-			$message, 
+			$nmessage, 
 			"From: " . $a->user['email'] . "\n"
 			. 'Content-type: text/plain; charset=UTF-8' . "\n"
 			. 'Content-transfer-encoding: 8bit' );
@@ -56,14 +79,25 @@ function invite_content(&$a) {
 	}
 
 	$tpl = get_markup_template('invite.tpl');
-	
+	$invonly = false;
+
+	if(get_config('system','invitations_only')) {
+		$invonly = true;
+		$x = get_pconfig(local_user(),'system','invites_remaining');
+		if((! $x) && (! is_site_admin())) {
+			notice( t('You have no more invitations available') . EOL);
+			return '';
+		}
+	}			
+
+
 	$o = replace_macros($tpl, array(
 		'$invite' => t('Send invitations'),
 		'$addr_text' => t('Enter email addresses, one per line:'),
 		'$msg_text' => t('Your message:'),
 		'$default_message' => sprintf(t('Please join my social network on %s'), $a->config['sitename']) . "\r\n" . "\r\n"
 			. t('To accept this invitation, please visit:') . "\r\n" . "\r\n" . $a->get_baseurl()
-			. "\r\n" . "\r\n" . t('Once you have registered, please connect with me via my profile page at:') 
+			. "\r\n" . "\r\n" . (($invonly) ? t('You will need to supply this invitation code: $invite_code') . "\r\n" . "\r\n" : '') .t('Once you have registered, please connect with me via my profile page at:') 
 			. "\r\n" . "\r\n" . $a->get_baseurl() . '/profile/' . $a->user['nickname'] ,
 		'$submit' => t('Submit')
 	));