Added "add to blogroll" checkbox

git-svn-id: http://svn.automattic.com/wordpress-mu/trunk@304 7be80a69-a1ef-0310-a953-fb0f7c49ff36

2 files changed, 20 insertions, 1 deletions

diff --git a/wp-inst/wp-admin/invites.php b/wp-inst/wp-admin/invites.php
index dd2213c..d116509 100644
--- a/wp-inst/wp-admin/invites.php
+++ b/wp-inst/wp-admin/invites.php
@@ -46,6 +46,11 @@ if( $_POST[ 'action' ] == 'send' ) {
 	    $wpdb->query( "INSERT INTO ".$wpdb->usermeta." ( `umeta_id` , `user_id` , `meta_key` , `meta_value` ) VALUES ( NULL, '0', '".md5( strtolower( $email ) )."_invited_by' , '$user_ID')" );
 	    $wpdb->query( "INSERT INTO ".$wpdb->usermeta." ( `umeta_id` , `user_id` , `meta_key` , `meta_value` ) VALUES ( NULL, '0', '".md5( strtolower( $email ) )."_to_email' , '{$_POST[ 'email' ]}')" );
 	    $wpdb->query( "INSERT INTO ".$wpdb->usermeta." ( `umeta_id` , `user_id` , `meta_key` , `meta_value` ) VALUES ( NULL, '0', '".md5( strtolower( $email ) )."_to_name' , '{$_POST[ 'fname' ]}')" );
+	    if( $_POST[ 'add_blog_to_blogroll' ] == '1' ) {
+		    $t = array( "blogid" => $wpdb->blogid, "userid" => get_current_user_id() );
+		    $wpdb->query( "INSERT INTO ".$wpdb->usermeta." ( `umeta_id` , `user_id` , `meta_key` , `meta_value` ) VALUES ( NULL, '0', '".md5( strtolower( $email ) )."_add_to_blogroll' , '" . serialize( $t ) . "')" );
+	    }
+
 	    mail( $_POST[ 'email' ], $subject, $msg, "From: $from" );
 	    if( is_site_admin() == false ) {
 		    $invites_left = $invites_left - 1;
@@ -101,6 +106,10 @@ $wpdb->escape( $_GET[ 'to' ] ) ) ?></strong></p></div><?php
         <th width="33%" scope="row"><?php _e('Personal Message:') ?></th> 
         <td><textarea rows="5" cols="60" name="personalmessage" tabindex="5" id="defaultmessage"><?php echo stripslashes( $_POST[ 'personalmessage' ] ) ?></textarea></td> 
       </tr> 
+      <tr valign="top"> 
+        <th width="33%" scope="row"><?php _e('Add blog to my blogroll when created:') ?></th> 
+        <td><input type='checkbox' name='add_blog_to_blogroll' value='1'></td> 
+      </tr> 
     </table> 
     </fieldset> 
     <p class="submit">
diff --git a/wp-inst/wp-content/mu-plugins/invites.php b/wp-inst/wp-content/mu-plugins/invites.php
index b3a315e..3c9fc49 100644
--- a/wp-inst/wp-content/mu-plugins/invites.php
+++ b/wp-inst/wp-content/mu-plugins/invites.php
@@ -124,7 +124,7 @@ function invites_add_field() {
 add_action('newblogform', 'invites_add_field');
 
 function invites_cleanup_db( $val ) {
-    global $wpdb;
+    global $wpdb, $wpmuBaseTablePrefix, $url, $weblog_title;
     if( isset( $_POST[ 'u' ] ) ) {
 	$query = "DELETE FROM ".$wpdb->usermeta."
                   WHERE       meta_key = 'invite'
@@ -133,6 +133,16 @@ function invites_cleanup_db( $val ) {
 	$wpdb->query( "DELETE FROM ".$wpdb->usermeta." WHERE meta_key = '{$_POST[ 'u' ]}_to_email'" );
 	$wpdb->query( "DELETE FROM ".$wpdb->usermeta." WHERE meta_key = '{$_POST[ 'u' ]}_to_name'" );
 
+	$add_to_blogroll = $wpdb->get_var( "SELECT meta_value FROM {$wpdb->usermeta} WHERE meta_key = '{$_POST[ 'u' ]}_add_to_blogroll'" );
+	if( $add_to_blogroll ) {
+		$userdetails = @unserialize( $add_to_blogroll );
+		if( is_array( $userdetails ) ) {
+			$wpdb->query("INSERT INTO {$wpmuBaseTablePrefix}{$userdetails[ 'blogid' ]}_links (link_url, link_name, link_category, link_owner) VALUES('" . addslashes( $url ) . "','" . addslashes( $weblog_title ) . "', '1', '" . intval( $userdetails[ 'userid' ] ) . "' )" );
+		}
+		$wpdb->query( "DELETE FROM ".$wpdb->usermeta." WHERE meta_key = '{$_POST[ 'u' ]}_add_to_blogroll'" );
+	}
+
+
 	$id = $wpdb->get_var( "SELECT ID FROM ".$wpdb->users." WHERE user_login = '" . $_POST[ 'weblog_id' ] . "'" );
 
 	if( $id ) {