class-fep-update.php

<?php

/*
Front End PM is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 2 of the License, or
any later version.
 
Front End PM is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
 
You should have received a copy of the GNU General Public License
along with Organized Docs. If not, see <http://www.gnu.org/licenses/>.
*/

class Fep_Update
  {
 	private static $instance;
	
	public static function init()
        {
            if(!self::$instance instanceof self) {
                self::$instance = new self;
            }
            return self::$instance;
        }
		
    function actions_filters()
    {
		add_filter('fep_admin_settings_tabs', array($this, 'sections'));
		add_filter('fep_settings_fields', array($this, 'fields'));
		add_action('fep_admin_settings_field_output_update', array($this, 'field_output'));
		//add_action('wp_ajax_fep_update_ajax', array($this, 'ajax'));
		add_action('fep_update_message_announcement', array($this, 'update_start'));
		
		if( ! empty( $_GET['fep_update'] ) ){
			do_action( 'fep_update_' . $_GET['fep_update'] );
		}
    }
	
	function sections( $tabs)
	{
		$tabs['update'] =  array(
			'priority'			=> 45,
			'tab_title'			=> __('Update', 'front-end-pm'),
			);
		return $tabs;
		
	}
	
	function fields( $fields)
	{
		$fields['update'] =   array(
			'type'	=>	'update',
			'section'	=> 'update',
			'label' => __( 'Update', 'front-end-pm' )
			);
		return $fields;
		
	}
	
	function field_output( $field ){
		
		global $wpdb;

		if( $wpdb->get_var("SHOW TABLES LIKE '". FEP_MESSAGES_TABLE . "'") != FEP_MESSAGES_TABLE ) {
			_e('You are up to date, No need to update.', 'front-end-pm' );
			return;
		}
		if( ! $wpdb->get_var("SELECT COUNT(*) FROM " . FEP_MESSAGES_TABLE . " WHERE id IS NOT NULL LIMIT 1") ) {
			$wpdb->query( "DROP TABLE IF EXISTS ".FEP_MESSAGES_TABLE );
			$wpdb->query( "DROP TABLE IF EXISTS ".FEP_META_TABLE );
			_e('You are up to date, No need to update.', 'front-end-pm' );
			return;
		}
		$args = array(
			'post_type' => 'fep_message',
			'page' => 'fep_settings',
			'tab' => 'update',
			'fep_update' => 'message_announcement',
			'token' => wp_create_nonce('fep_update')
		);
		?><a class="button" href="<?php echo add_query_arg( $args, admin_url( 'edit.php' ) ); ?>"><?php _e('Start Update', 'front-end-pm'); ?></a><?php 
		
	}
	
	function update_start(){
		
		ignore_user_abort( true );

		if ( ! ini_get( 'safe_mode' ) )
		set_time_limit( 0 );
		
		$args = array(
			'post_type' => 'fep_message',
			'page' => 'fep_settings',
			'tab' => 'update'
			);
		
		if( ! $this->update_check() ){
			wp_redirect( add_query_arg( $args, admin_url( 'edit.php' ) ) );
			exit;
		}
		
		if( ! $this->messages_update() ){
			$args['fep_update'] = 'message_announcement';
			$args['token'] = wp_create_nonce('fep_update');
			wp_redirect( add_query_arg( $args, admin_url( 'edit.php' ) ) );
			exit;
		}
		
		if( ! $this->announcement_update() ){
			$args['fep_update'] = 'message_announcement';
			$args['token'] = wp_create_nonce('fep_update');
			wp_redirect( add_query_arg( $args, admin_url( 'edit.php' ) ) );
			exit;
		}
		//$this->finish_update();
		if( ! $this->update_check() ){
			wp_redirect( add_query_arg( $args, admin_url( 'edit.php' ) ) );
			exit;
		} else {
			add_settings_error( 'fep-settings', 'fep_updated', __( 'Something wrong, please try again.', 'front-end-pm' ) );
		}
		
	}
	
	function update_check(){
		
		global $wpdb;
		
		if( ! wp_verify_nonce( $_GET['token'], 'fep_update') )
			wp_die( 'Nonce did not verify' );

		if( $wpdb->get_var("SHOW TABLES LIKE '". FEP_MESSAGES_TABLE . "'") != FEP_MESSAGES_TABLE ) {
			
			add_settings_error( 'fep-settings', 'fep_updated', __( 'You are up to date, No need to update.', 'front-end-pm' ) );
			return false;
		}
		if( ! $wpdb->get_var("SELECT COUNT(*) FROM " . FEP_MESSAGES_TABLE . " WHERE id IS NOT NULL LIMIT 1") ) {
			$wpdb->query( "DROP TABLE IF EXISTS ".FEP_MESSAGES_TABLE );
			$wpdb->query( "DROP TABLE IF EXISTS ".FEP_META_TABLE );
			
			add_settings_error( 'fep-settings', 'fep_updated', __( 'You are up to date, No need to update.', 'front-end-pm' ) );
			return false;
		}
		
		return true;
	}
	
	function messages_update(){
		
		$messages = $this->get_messages();
		
		if( $messages ) {
		foreach ( $messages as $message ) {
			$arr = array(
					'message_title'	=> $message->message_title,
					'message_content'	=> $message->message_contents,
					'fep_parent_id'	=> 0,
					'message_to_id'	=> $message->to_user
				);
			$override = array(
					'post_author'	=> $message->from_user,
					'post_date'	=> $message->send_date
				);
			
			if( $message_id = fep_send_message( $arr, $override ) ) {
				if( 1 == $message->status ) {
					fep_make_read( true, $message_id, $message->to_user );
					fep_make_read( false, $message_id, $message->to_user );
				}
				if( 1 == $message->from_del ) {
					add_post_meta( $message_id, '_fep_delete_by_'. $message->from_user, time(), true ); //No time from previous version
				} elseif( 1 == $message->to_del ) {
					add_post_meta( $message_id, '_fep_delete_by_'. $message->to_user, time(), true );
				}
				
				$this->insert_attachment( $message_id, $message->id, $message->from_user );
				$this->insert_replies( $message_id, $message->id );
				
				$this->delete_message( $message->id );
				
			}
		}
		return false;
	}
	return true;

}
	function announcement_update(){
		
		$announcements = $this->get_announcements();
		
		if( $announcements ) {
		
		foreach( $announcements as $announcement ) {
		
			$arr = array(
					'post_title'	=> $announcement->message_title,
					'post_content'	=> $announcement->message_contents,
					'post_author'	=> $announcement->from_user,
					'post_date'	=> $announcement->send_date,
					'post_type'	=> 'fep_announcement',
					'post_status'    => 'publish'
				);
			if( ! $ann_id = wp_insert_post($arr) )
				continue;
			
			$this->insert_attachment( $ann_id, $announcement->id, $announcement->from_user );

			foreach( array_keys( get_editable_roles() ) as $role ) {
				add_post_meta( $ann_id, '_participant_roles', $role );
			}
			
			$seen = $this->get_announcement_meta( $announcement->id );
			$seen = maybe_unserialize( $seen );
			
			if( $seen && is_array($seen) ) {
				add_post_meta( $ann_id, '_fep_read_by', $seen, true );
			}
			
			$deleted = $this->get_announcement_meta( $announcement->id, 'announcement_deleted_user_id' );
			$deleted = maybe_unserialize( $deleted );
			
			if( $deleted && is_array($deleted) ) {
				foreach( $deleted as $del ) {
					add_post_meta( $ann_id, '_fep_delete_by_'. $del, time(), true );
				}
			}
			
			$this->delete_message( $announcement->id );
			
		}
		delete_metadata( 'user', 0, '_fep_user_announcement_count', '', true );
		return false;
	}
	return true;

}
	
	function insert_attachment( $message_id, $message_prev_id, $author )
	{
		if( ! $attachments = $this->get_attachments( $message_prev_id ) )
		return;
		
		foreach( $attachments as $attachment ) {
			$unserialized_file = maybe_unserialize( $attachment->field_value );
			if ( $unserialized_file['type'] && $unserialized_file['url'] && $unserialized_file['file'] ) {
				// Prepare an array of post data for the attachment.
				$att = array(
					'guid'           => $unserialized_file['url'], 
					'post_mime_type' => $unserialized_file['type'],
					'post_title'     => preg_replace( '/\.[^.]+$/', '', basename( $unserialized_file['url'] ) ),
					'post_content'   => '',
					'post_author'	=> $author,
					'post_status'    => 'inherit'
				);
				
				// Insert the attachment.
				wp_insert_attachment( $att, $unserialized_file['file'], $message_id );
			}
		}
		
	}
	function insert_replies( $message_id, $message_prev_id )
	{
		if( ! $replies = $this->get_replies( $message_prev_id ) )
		return;
		
		foreach( $replies as $reply ) {
			$arr = array(
					'message_title'	=> $reply->message_title,
					'message_content'	=> $reply->message_contents,
					'fep_parent_id'	=> $message_id
				);
			$override = array(
					'post_author'	=> $reply->from_user,
					'post_date'	=> $reply->send_date
				);
			
			if( $reply_id = fep_send_message( $arr, $override ) ) {
				$this->insert_attachment( $reply_id, $reply->id, $reply->from_user );
				
				$this->delete_message( $reply->id );
			}
			
		}
		
	}
	
	function delete_message( $message_id )
    {	global $wpdb;
 
 		$wpdb->query($wpdb->prepare("DELETE FROM ".FEP_MESSAGES_TABLE." WHERE id = %d", $message_id ));
		$wpdb->query($wpdb->prepare("DELETE FROM ".FEP_META_TABLE." WHERE message_id = %d", $message_id ));

    }
	
	function get_messages()
    {	global $wpdb;
 
	 return $wpdb->get_results($wpdb->prepare("SELECT * FROM ".FEP_MESSAGES_TABLE." WHERE parent_id = %d AND (status = 0 OR status = 1) ORDER BY last_date DESC LIMIT 10", 0 ));

    }
	function get_replies( $parent )
    {
 		global $wpdb;
	 return $wpdb->get_results($wpdb->prepare("SELECT * FROM ".FEP_MESSAGES_TABLE." WHERE parent_id = %d AND (status = 0 OR status = 1) ORDER BY last_date DESC", $parent ));

    }
	function get_attachments( $message_id )
    {
 		global $wpdb;
	 return $wpdb->get_results($wpdb->prepare("SELECT * FROM ".FEP_META_TABLE." WHERE message_id = %d AND field_name = %s", $message_id, 'attachment' ));

    }
	function get_announcements()
    {	global $wpdb;
 
	 return $wpdb->get_results($wpdb->prepare("SELECT * FROM ".FEP_MESSAGES_TABLE." WHERE status = %d LIMIT 20", 2 ));

    }
	function get_announcement_meta( $id, $meta = 'announcement_seen_user_id' )
    {	global $wpdb;
 
	 return $wpdb->get_var($wpdb->prepare("SELECT field_value FROM ".FEP_META_TABLE." WHERE message_id = %d AND field_name = %s LIMIT 1", $id, $meta ));

    }
	
	
  } //END CLASS

add_action('admin_init', array(Fep_Update::init(), 'actions_filters'));

function fep_update_script() {

	wp_register_script( 'fep_update_script', FEP_PLUGIN_URL . 'assets/js/fep_update_script.js', array( 'jquery' ), '3.1', true );
}
add_action( 'admin_enqueue_scripts', 'fep_update_script' );

function fep_insert_dummy_message(){
	global $wpdb;
	
	for( $i = 0; $i < 100; $i++ ) {
	$from = mt_rand(1, 5);
	
		$wpdb->insert( FEP_MESSAGES_TABLE, array( 
		'from_user' => $from, 
		'to_user' => mt_rand(1, 5), 
		'message_title' => 'this is title', 
		'message_contents' => 'this is message', 
		'parent_id' => 0, 
		'last_sender' => $from, 
		'send_date' => current_time('mysql'), 
		'last_date' => current_time('mysql'),
		'status' => mt_rand(0, 2) 
		), 
		array( '%d', '%d', '%s', '%s', '%d', '%d', '%s', '%s', '%d' ));
		
	}
}
//add_action('wp_loaded', 'fep_insert_dummy_message' );