buddypress/bp-messages/bp-messages-classes.php

570 lines
No EOL
19 KiB
PHP

<?php
Class BP_Messages_Thread {
var $thread_id;
var $message_ids;
var $first_post_date;
var $last_post_date;
var $last_sender_id;
var $last_message_id;
var $last_message_subject;
var $last_message_message;
var $last_message_date_sent;
var $messages = null;
var $has_access = false;
var $unread_count = 0;
var $recipients = null;
var $box;
var $get_all_messages;
function bp_messages_thread( $id = null, $get_all_messages = false, $box = 'inbox' ) {
$this->box = $box;
$this->get_all_messages = $get_all_messages;
if ( $id ) {
$this->populate( $id );
}
if ( $this->get_all_messages ) {
$this->messages = $this->get_messages();
$this->recipients = $this->get_recipients();
}
}
function populate( $id ) {
global $wpdb, $bp;
$thread = $wpdb->get_row( $wpdb->prepare( "SELECT * FROM {$bp->messages->table_name_threads} WHERE id = %d", $id ) );
if ( $thread ) {
// If we're only viewing a thread in a list of threads, don't bother using
// up resources checking if we have access. Only check if we're viewing the full
// thread.
if ( $this->get_all_messages )
$this->has_access = $this->check_access($id);
else
$this->has_access = true;
if ( $this->has_access ) {
$this->thread_id = $thread->id;
$this->message_ids = maybe_unserialize($thread->message_ids);
// If we are viewing only the threads in a users inbox/sentbox we need to
// filter the users messages out
if ( !$this->get_all_messages ) {
// Flip the array to start from the newest message
$this->message_ids = array_reverse( $this->message_ids );
foreach ( $this->message_ids as $key => $message_id ) {
if ( 'sentbox' == $this->box ) {
if ( !messages_is_user_sender( $bp->loggedin_user->id, $message_id ) ) {
unset( $this->message_ids[$key] );
} else {
break;
}
} else {
if ( messages_is_user_sender( $bp->loggedin_user->id, $message_id ) ) {
unset( $this->message_ids[$key] );
} else {
break;
}
}
}
// Flip the array back to start from the oldest message
$this->message_ids = array_reverse( $this->message_ids );
}
$this->last_message_id = $this->message_ids[(count($this->message_ids) - 1)];
$this->last_sender_id = messages_get_message_sender( $this->last_message_id );
$this->first_post_date = $thread->first_post_date;
$this->last_post_date = $thread->last_post_date;
if ( !empty($this->message_ids) )
$this->message_ids = implode( ',', $this->message_ids );
else
$this->message_ids = false;
$this->unread_count = $this->get_unread();
$last_message = $wpdb->get_row( $wpdb->prepare( "SELECT * FROM {$bp->messages->table_name_messages} WHERE id = %d", $this->last_message_id ) );
if ( $last_message ) {
$this->last_message_subject = $last_message->subject;
$this->last_message_message = $last_message->message;
$this->last_message_date_sent = $last_message->date_sent;
}
$this->recipients = $this->get_recipients();
}
}
}
function get_messages() {
global $wpdb, $bp;
if ( $this->message_ids)
return $wpdb->get_results( "SELECT * FROM {$bp->messages->table_name_messages} WHERE id IN (" . $wpdb->escape($this->message_ids) . ")" );
else
return false;
}
function get_unread() {
global $wpdb, $bp;
$sql = $wpdb->prepare( "SELECT unread_count FROM {$bp->messages->table_name_recipients} WHERE thread_id = %d AND user_id = %d", $this->thread_id, $bp->loggedin_user->id );
$unread_count = $wpdb->get_var($sql);
return $unread_count;
}
function mark_read() {
BP_Messages_Thread::mark_as_read($this->thread_id);
}
function mark_unread() {
BP_Messages_Thread::mark_as_unread($this->thread_id);
}
function get_recipients() {
global $wpdb, $bp;
$recipients = $wpdb->get_col( $wpdb->prepare( "SELECT user_id FROM {$bp->messages->table_name_recipients} WHERE thread_id = %d", $this->thread_id ) );
for ( $i = 0; $i < count($recipients); $i++ ) {
$recipient = $recipients[$i];
if ( count($recipients) > 1 && $recipient != $bp->loggedin_user->id )
$recipient_ids[] = $recipient;
}
return $recipient_ids;
}
/** Static Functions **/
function delete( $thread_id ) {
global $wpdb, $bp;
$delete_for_user = $wpdb->query( $wpdb->prepare( "UPDATE {$bp->messages->table_name_recipients} SET is_deleted = 1 WHERE thread_id = %d AND user_id = %d", $thread_id, $bp->loggedin_user->id ) );
// Check to see if any more recipients remain for this message
// if not, then delete the message from the database.
$recipients = $wpdb->get_results( $wpdb->prepare( "SELECT id FROM {$bp->messages->table_name_recipients} WHERE thread_id = %d AND is_deleted = 0", $thread_id ) );
if ( !$recipients ) {
// Get message ids:
$message_ids = $wpdb->get_var( $wpdb->prepare( "SELECT message_ids FROM {$bp->messages->table_name_threads} WHERE id = %d", $thread_id ) );
$message_ids = unserialize($message_ids);
// delete thread:
$wpdb->query( $wpdb->prepare( "DELETE FROM {$bp->messages->table_name_threads} WHERE id = %d", $thread_id ) );
// delete messages:
for ( $i = 0; $i < count($message_ids); $i++ ) {
$wpdb->query( $wpdb->prepare( "DELETE FROM {$bp->messages->table_name_messages} WHERE id = %d", $message_ids[$i] ) );
}
// delete the recipients
$wpdb->query( $wpdb->prepare( "DELETE FROM {$bp->messages->table_name_recipients} WHERE id = %d", $thread_id ) );
}
return true;
}
function get_current_threads_for_user( $user_id, $box = 'inbox', $limit = null, $page = null, $type = 'all' ) {
global $wpdb, $bp;
// If we have pagination values set we want to pass those to the query
if ( $limit && $page )
$pag_sql = $wpdb->prepare( " LIMIT %d, %d", intval( ( $page - 1 ) * $limit), intval( $limit ) );
if ( $type == 'unread' )
$type_sql = $wpdb->prepare( " AND r.unread_count != 0 " );
else if ( $type == 'read' )
$type_sql = $wpdb->prepare( " AND r.unread_count = 0 " );
$sql = $wpdb->prepare( "SELECT r.thread_id FROM {$bp->messages->table_name_recipients} r, {$bp->messages->table_name_threads} t WHERE t.id = r.thread_id AND r.is_deleted = 0 AND r.user_id = %d$exclude_sender $type_sql ORDER BY t.last_post_date DESC$pag_sql", $bp->loggedin_user->id );
if ( !$thread_ids = $wpdb->get_results($sql) )
return false;
$threads = false;
for ( $i = 0; $i < count($thread_ids); $i++ ) {
$threads[$i] = new BP_Messages_Thread( $thread_ids[$i]->thread_id, false, $box );
if ( !$threads[$i]->message_ids )
unset($threads[$i]);
}
// reset keys
return array_reverse( array_reverse( $threads ) );
}
function mark_as_read( $thread_id ) {
global $wpdb, $bp;
$sql = $wpdb->prepare( "UPDATE {$bp->messages->table_name_recipients} SET unread_count = 0 WHERE user_id = %d AND thread_id = %d", $bp->loggedin_user->id, $thread_id );
$wpdb->query($sql);
}
function mark_as_unread( $thread_id ) {
global $wpdb, $bp;
$sql = $wpdb->prepare( "UPDATE {$bp->messages->table_name_recipients} SET unread_count = 1 WHERE user_id = %d AND thread_id = %d", $bp->loggedin_user->id, $thread_id );
$wpdb->query($sql);
}
function get_total_threads_for_user( $user_id, $box = 'inbox', $type = 'all' ) {
global $wpdb, $bp;
$exclude_sender = '';
if ( $box != 'sentbox' )
$exclude_sender = ' AND sender_only != 1';
if ( $type == 'unread' )
$type_sql = $wpdb->prepare( " AND unread_count != 0 " );
else if ( $type == 'read' )
$type_sql = $wpdb->prepare( " AND unread_count = 0 " );
return (int) $wpdb->get_var( $wpdb->prepare( "SELECT count(thread_id) FROM {$bp->messages->table_name_recipients} WHERE user_id = %d AND is_deleted = 0$exclude_sender $type_sql", $user_id ) );
}
function user_is_sender($thread_id) {
global $wpdb, $bp;
$sender_ids = $wpdb->get_var( $wpdb->prepare( "SELECT sender_ids FROM {$bp->messages->table_name_threads} WHERE id = %d", $thread_id ) );
if ( !$sender_ids )
return false;
$sender_ids = unserialize($sender_ids);
return in_array( $bp->loggedin_user->id, $sender_ids );
}
function get_last_sender($thread_id) {
global $wpdb, $bp;
$sql = $wpdb->prepare("SELECT last_sender_id FROM {$bp->messages->table_name_threads} WHERE id = %d", $thread_id);
if ( !$sender_id = $wpdb->get_var($sql) )
return false;
return bp_core_get_userlink( $sender_id, true );
}
function get_inbox_count() {
global $wpdb, $bp;
$sql = $wpdb->prepare( "SELECT unread_count FROM {$bp->messages->table_name_recipients} WHERE user_id = %d AND is_deleted = 0", $bp->loggedin_user->id );
if ( !$unread_counts = $wpdb->get_results($sql) )
return false;
$count = 0;
for ( $i = 0; $i < count($unread_counts); $i++ ) {
$count += $unread_counts[$i]->unread_count;
}
return $count;
}
function check_access( $thread_id, $user_id = false ) {
global $wpdb, $bp;
if ( !$user_id )
$user_id = $bp->loggedin_user->id;
return $wpdb->get_var( $wpdb->prepare( "SELECT id FROM {$bp->messages->table_name_recipients} WHERE thread_id = %d AND user_id = %d", $thread_id, $user_id ) );
}
function is_valid( $thread_id ) {
global $wpdb, $bp;
return $wpdb->get_var( $wpdb->prepare( "SELECT id FROM {$bp->messages->table_name_threads} WHERE id = %d", $thread_id ) );
}
function get_recipient_links($recipients) {
if ( count($recipients) >= 5 )
return count($recipients) . __(' Recipients', 'buddypress');
for ( $i = 0; $i < count($recipients); $i++ ) {
$recipient_links[] = bp_core_get_userlink( $recipients[$i] );
}
return implode( ', ', $recipient_links);
}
}
Class BP_Messages_Message {
var $id = null;
var $sender_id;
var $subject;
var $message;
var $date_sent;
var $thread_id;
var $recipients = false;
function bp_messages_message( $id = null ) {
global $bp;
$this->date_sent = time();
$this->sender_id = $bp->loggedin_user->id;
if ( $id ) {
$this->populate($id);
}
}
function populate( $id ) {
global $wpdb, $bp;
$sql = $wpdb->prepare("SELECT * FROM {$bp->messages->table_name_messages} WHERE id = %d", $id);
if ( $message = $wpdb->get_row($sql) ) {
$this->id = $message->id;
$this->sender_id = $message->sender_id;
$this->subject = $message->subject;
$this->message = $message->message;
$this->date_sent = $message->date_sent;
}
}
function send() {
global $wpdb, $bp;
$this->sender_id = apply_filters( 'messages_message_sender_id_before_save', $this->sender_id, $this->id );
$this->subject = apply_filters( 'messages_message_subject_before_save', $this->subject, $this->id );
$this->message = apply_filters( 'messages_message_content_before_save', $this->message, $this->id );
$this->date_sent = apply_filters( 'messages_message_date_sent_before_save', $this->date_sent, $this->id );
do_action( 'messages_message_before_save', $this );
// First insert the message into the messages table
if ( !$wpdb->query( $wpdb->prepare( "INSERT INTO {$bp->messages->table_name_messages} ( sender_id, subject, message, date_sent ) VALUES ( %d, %s, %s, FROM_UNIXTIME(%d) )", $this->sender_id, $this->subject, $this->message, $this->date_sent ) ) )
return false;
// Next, if thread_id is set, we are adding to an existing thread, if not, start a new one.
if ( $this->thread_id ) {
// Select and update the current message ids for the thread.
$the_ids = $wpdb->get_row( $wpdb->prepare( "SELECT message_ids, sender_ids FROM {$bp->messages->table_name_threads} WHERE id = %d", $this->thread_id ) );
$message_ids = unserialize($the_ids->message_ids);
$message_ids[] = $wpdb->insert_id;
$message_ids = serialize($message_ids);
// We need this so we can return the new message ID.
$message_id = $wpdb->insert_id;
// Update the sender ids for the thread
$sender_ids = unserialize($the_ids->sender_ids);
if ( !in_array( $this->sender_id, $sender_ids ) || !$sender_ids )
$sender_ids[] = $this->sender_id;
$sender_ids = serialize($sender_ids);
// Update the thread the message belongs to.
$wpdb->query( $wpdb->prepare( "UPDATE {$bp->messages->table_name_threads} SET message_ids = %s, sender_ids = %s, last_message_id = %d, last_sender_id = %d, last_post_date = FROM_UNIXTIME(%d) WHERE id = %d", $message_ids, $sender_ids, $wpdb->insert_id, $this->sender_id, $this->date_sent, $this->thread_id ) );
// Find the recipients and update the unread counts for each
if ( !$this->recipients )
$this->recipients = $this->get_recipients();
for ( $i = 0; $i < count($this->recipients); $i++ ) {
if ( $this->recipients[$i]->user_id != $bp->loggedin_user->id )
$wpdb->query( $wpdb->prepare( "UPDATE {$bp->messages->table_name_recipients} SET unread_count = unread_count + 1, sender_only = 0, is_deleted = 0 WHERE thread_id = %d AND user_id = %d", $this->thread_id, $this->recipients[$i] ) );
}
} else {
// Create a new thread.
$message_id = $wpdb->insert_id;
$serialized_message_id = serialize( array( (int)$message_id ) );
$serialized_sender_id = serialize( array( (int)$bp->loggedin_user->id ) );
$sql = $wpdb->prepare( "INSERT INTO {$bp->messages->table_name_threads} ( message_ids, sender_ids, first_post_date, last_post_date, last_message_id, last_sender_id ) VALUES ( %s, %s, FROM_UNIXTIME(%d), FROM_UNIXTIME(%d), %d, %d )", $serialized_message_id, $serialized_sender_id, $this->date_sent, $this->date_sent, $message_id, $this->sender_id );
if ( false === $wpdb->query($sql) )
return false;
$this->thread_id = $wpdb->insert_id;
// Add a new entry for each recipient;
for ( $i = 0; $i < count($this->recipients); $i++ ) {
$wpdb->query( $wpdb->prepare( "INSERT INTO {$bp->messages->table_name_recipients} ( user_id, thread_id, unread_count ) VALUES ( %d, %d, 1 )", $this->recipients[$i], $this->thread_id ) );
}
if ( !in_array( $this->sender_id, $this->recipients ) ) {
// Finally, add a recipient entry for the sender, as replies need to go to this person too.
$wpdb->query( $wpdb->prepare( "INSERT INTO {$bp->messages->table_name_recipients} ( user_id, thread_id, unread_count, sender_only ) VALUES ( %d, %d, 0, 0 )", $this->sender_id, $this->thread_id ) );
}
}
$this->id = $message_id;
messages_remove_callback_values();
do_action( 'messages_message_after_save', $this );
return $this->thread_id;
}
function get_recipients() {
global $bp, $wpdb;
return $wpdb->get_results( $wpdb->prepare( "SELECT user_id FROM {$bp->messages->table_name_recipients} WHERE thread_id = %d", $this->thread_id ) );
}
// Static Functions
function get_recipient_ids( $recipient_usernames ) {
if ( !$recipient_usernames )
return false;
if ( is_array($recipient_usernames) ) {
for ( $i = 0; $i < count($recipient_usernames); $i++ ) {
if ( $rid = bp_core_get_userid( trim($recipient_usernames[$i]) ) )
$recipient_ids[] = $rid;
}
}
return $recipient_ids;
}
function get_last_sent_for_user( $thread_id ) {
global $wpdb, $bp;
$message_ids = $wpdb->get_var( $wpdb->prepare( "SELECT message_ids FROM {$bp->messages->table_name_threads} WHERE id = %d", $thread_id ) );
$message_ids = implode( ',', unserialize($message_ids));
$sql = $wpdb->prepare( "SELECT id FROM {$bp->messages->table_name_messages} WHERE sender_id = %d AND id IN (" . $wpdb->escape($message_ids) . ") ORDER BY date_sent DESC LIMIT 1", $bp->loggedin_user->id );
return $wpdb->get_var($sql);
}
function is_user_sender( $user_id, $message_id ) {
global $wpdb, $bp;
return $wpdb->get_var( $wpdb->prepare( "SELECT id FROM {$bp->messages->table_name_messages} WHERE sender_id = %d AND id = %d", $user_id, $message_id ) );
}
function get_message_sender( $message_id ) {
global $wpdb, $bp;
return $wpdb->get_var( $wpdb->prepare( "SELECT sender_id FROM {$bp->messages->table_name_messages} WHERE id = %d", $message_id ) );
}
}
Class BP_Messages_Notice {
var $id = null;
var $subject;
var $message;
var $date_sent;
var $is_active;
function bp_messages_notice($id = null) {
if ( $id ) {
$this->id = $id;
$this->populate($id);
}
}
function populate() {
global $wpdb, $bp;
$notice = $wpdb->get_row( $wpdb->prepare( "SELECT * FROM {$bp->messages->table_name_notices} WHERE id = %d", $this->id ) );
if ( $notice ) {
$this->subject = $notice->subject;
$this->message = $notice->message;
$this->date_sent = $notice->date_sent;
$this->is_active = $notice->is_active;
}
}
function save() {
global $wpdb, $bp;
$this->subject = apply_filters( 'messages_notice_subject_before_save', $this->subject, $this->id );
$this->message = apply_filters( 'messages_notice_message_before_save', $this->message, $this->id );
do_action( 'messages_notice_before_save', $this );
if ( !$this->id ) {
$sql = $wpdb->prepare( "INSERT INTO {$bp->messages->table_name_notices} (subject, message, date_sent, is_active) VALUES (%s, %s, FROM_UNIXTIME(%d), %d)", $this->subject, $this->message, $this->date_sent, $this->is_active );
} else {
$sql = $wpdb->prepare( "UPDATE {$bp->messages->table_name_notices} SET subject = %s, message = %s, is_active = %d WHERE id = %d", $this->subject, $this->message, $this->is_active, $this->id );
}
if ( !$wpdb->query($sql) )
return false;
if ( !$id = $this->id )
$id = $wpdb->insert_id;
// Now deactivate all notices apart from the new one.
$wpdb->query( $wpdb->prepare( "UPDATE {$bp->messages->table_name_notices} SET is_active = 0 WHERE id != %d", $id ) );
update_usermeta( $bp->loggedin_user->id, 'last_activity', date( 'Y-m-d H:i:s' ) );
do_action( 'messages_notice_after_save', $this );
return true;
}
function activate() {
$this->is_active = 1;
if ( !$this->save() )
return false;
return true;
}
function deactivate() {
$this->is_active = 0;
if ( !$this->save() )
return false;
return true;
}
function delete() {
global $wpdb, $bp;
$sql = $wpdb->prepare( "DELETE FROM {$bp->messages->table_name_notices} WHERE id = %d", $this->id );
if ( !$wpdb->query($sql) )
return false;
return true;
}
// Static Functions
function get_notices() {
global $wpdb, $bp;
$notices = $wpdb->get_results( $wpdb->prepare( "SELECT * FROM {$bp->messages->table_name_notices} ORDER BY date_sent DESC" ) );
return $notices;
}
function get_total_notice_count() {
global $wpdb, $bp;
$notice_count = $wpdb->get_var( $wpdb->prepare( "SELECT count(id) FROM " . $bp->messages->table_name_notices ) );
return $notice_count;
}
function get_active() {
global $wpdb, $bp;
$notice_id = $wpdb->get_var( $wpdb->prepare( "SELECT id FROM {$bp->messages->table_name_notices} WHERE is_active = 1") );
return new BP_Messages_Notice($notice_id);
}
}
?>