<?php
/**
 * Twitter implementation for Social.
 *
 * @package    Social
 * @subpackage plugins
 */
if (class_exists('Social') and !class_exists('Social_Twitter')) {

final class Social_Twitter {

	/**
	 * Registers Twitter to Social.
	 *
	 * @static
	 * @wp-filter  social_register_service
	 *
	 * @param  array  $services
	 *
	 * @return array
	 */
	public static function register_service(array $services) {
		$services[] = 'twitter';
		return $services;
	}

	/**
	 * Adds to the avatar comment types array.
	 *
	 * @static
	 * @param  array  $types
	 * @return array
	 */
	public static function get_avatar_comment_types(array $types) {
		return array_merge($types, Social_Service_Twitter::comment_types());
	}

	/**
	 * Pre-processor to the comments.
	 *
	 * @wp-filter social_comments_array
	 * @static
	 * @param  array  $comments
	 * @param  int    $post_id
	 * @return array
	 */
	public static function comments_array(array $comments, $post_id) {
		// pre-load the hashes for broadcasted tweets
		$broadcasted_ids = get_post_meta($post_id, '_social_broadcasted_ids', true);
		if (empty($broadcasted_ids)) {
			$broadcasted_ids = array();
		}
		global $wpdb;

		// we need comments to be keyed by ID, check for Tweet comments
		$tweet_comments = $_comments = $comment_ids = array();
		foreach ($comments as $key => $comment) {
			if (is_object($comment)) {
				$_comments['id_'.$comment->comment_ID] = $comment;
				if (in_array($comment->comment_type, Social_Service_Twitter::comment_types())) {
					$comment_ids[] = $comment->comment_ID;
					$tweet_comments['id_'.$comment->comment_ID] = $comment;
				}
			}
			else { // social items
				$_comments[$key] = $comment;
			}
		}

		// if no tweet comments, get out now
		if (!count($tweet_comments)) {
			return $comments;
		}

		// use our keyed array
		$comments = $_comments;
		unset($_comments);

		$social_map = array(); // key = social id, value = comment_ID
		$hash_map = array(); // key = hash, value = comment_ID
		$broadcasted_social_ids = array();
 		$broadcast_retweets = array(); // array of comments

		if (isset($broadcasted_ids['twitter'])) {
			foreach ($broadcasted_ids['twitter'] as $account_id => $broadcasted) {
				foreach ($broadcasted as $id => $data) {
					$broadcasted_social_ids[] = $id;
					// if we don't have a message saved for a tweet, try to get it so that we can use it next time
					if (empty($data['message'])) {
						$url = wp_nonce_url(home_url('index.php?social_controller=aggregation&social_action=retrieve_twitter_content&broadcasted_id='.$id.'&post_id='.$post_id), 'retrieve_twitter_content');
						wp_remote_get(str_replace('&amp;', '&', $url), array(
							'timeout' => 0.01,
							'blocking' => false,
						));
					}
					else {
						// create a hash from the broadcast so we can match retweets to it
						$hash = self::build_hash($data['message']);

						// This is stored as broadcasted and not the ID so we can easily store broadcasted retweets
						// instead of attaching retweets to non-existent comments.
						$hash_map[$hash] = 'broadcasted';
					}
				}
			}
		}

		// Load the comment meta
		$results = $wpdb->get_results("
			SELECT meta_key, meta_value, comment_id
			  FROM $wpdb->commentmeta
			 WHERE comment_id IN (".implode(',', $comment_ids).")
			   AND (
			       meta_key = 'social_in_reply_to_status_id'
			    OR meta_key = 'social_status_id'
			    OR meta_key = 'social_raw_data'
			    OR meta_key = 'social_profile_image_url'
			    OR meta_key = 'social_comment_type'
			)
		");

		// Set up social data for twitter comments
		foreach ($tweet_comments as $key => &$comment) {
			$comment->social_items = array();

			// Attach meta
			foreach ($results as $result) {
				if ($comment->comment_ID == $result->comment_id) {
					switch ($result->meta_key) {
						case 'social_raw_data':
							$comment->social_raw_data = json_decode(base64_decode($result->meta_value));
							break;
						case 'social_status_id':
							$social_map[$result->meta_value] = $result->comment_id;
						default:
							$comment->{$result->meta_key} = $result->meta_value;
					}
				}
			}

			// Attach hash
			if (isset($comment->social_raw_data) and isset($comment->social_raw_data->text)) {
				$text = trim($comment->social_raw_data->text);
			}
			else {
				$text = trim($comment->comment_content);
			}
			$comment->social_hash = self::build_hash($text);

			if (!isset($hash_map[$comment->social_hash])) {
				$hash_map[$comment->social_hash] = $comment->comment_ID;
			}
		}

		// merge data so that $comments has the data we've set up
		$comments = array_merge($comments, $tweet_comments);

		// set-up replies and retweets
		foreach ($tweet_comments as $key => &$comment) {
			if (is_object($comment)) {
				// set reply/comment parent
				if (!empty($comment->social_in_reply_to_status_id) and isset($social_map[$comment->social_in_reply_to_status_id])) {
					$comments[$key]->comment_parent = $social_map[$comment->social_in_reply_to_status_id];
				}

				// set retweets
				$rt_matched = false;
				if (isset($comment->social_raw_data) and isset($comment->social_raw_data->retweeted_status)) {
					// explicit match via API data
					$rt_id = $comment->social_raw_data->retweeted_status->id_str;
					if (in_array($rt_id, $broadcasted_social_ids)) {
						$broadcast_retweets[] = $comment;
						unset($comments[$key]);
						$rt_matched = true;
					}
					else if (isset($social_map[$rt_id]) and isset($comments['id_'.$social_map[$rt_id]])) {
						$comments['id_'.$social_map[$rt_id]]->social_items[$key] = $comment;
						unset($comments[$key]);
						$rt_matched = true;
					}
				}

				if (!$rt_matched) {
					// best guess via hashes
					$hash_match = $hash_map[$comment->social_hash];
					if ($hash_match != $comment->comment_ID) { // hash match to own tweet is expected, at minimum - set above
						if ($hash_match == 'broadcasted') {
							$broadcast_retweets[] = $comment;
						}
						else if (isset($comments['id_'.$hash_match])) {
							$comments['id_'.$hash_match]->social_items[$key] = $comment;
						}
						else {
							// Loop through the broadcasted retweets and see if this is a retweet of one of those.
							foreach ($broadcast_retweets as $retweet) {
								if ($retweet->comment_ID == $hash_match) {
									$broadcast_retweets[] = $comment;
									break;
								}
							}
						}
						unset($comments[$key]);
					}
				}
			}
		}

		if (!isset($comments['social_items'])) {
			$comments['social_items'] = array();
		}

		if (count($broadcast_retweets)) {
			$comments['social_items']['twitter'] = $broadcast_retweets;
		}

		return $comments;
	}


	/**
	 * Sets the raw data for the broadcasted post.
	 *
	 * @wp-filter social_broadcast_response
	 * @static
	 * @param  array                   $data
	 * @param  Social_Service_Account  $account
	 * @param  string                  $service_key
	 * @param  int                     $post_id
	 * @param  Social_Response         $response
	 * @return array
	 */
	public static function social_save_broadcasted_ids_data(array $data, Social_Service_Account $account, $service_key, $post_id, Social_Response $response = null) {
		if ($service_key == 'twitter') {
			if (!empty($response)) {
				$data['message'] = base64_encode(json_encode($response->body()->response));
			}
			$data['account'] = (object) array(
				'user' => $account->as_object()->user
			);
		}

		return $data;
	}

	/**
	 * Strips extra retweet data before comparing.
	 *
	 * @static
	 * @param  string  $text
	 * @return string
	 */
	private static function build_hash($text) {
		$text = explode(' ', $text);
		$content = '';
		foreach ($text as $_content) {
			if (!empty($_content) and strpos($_content, 'http://') === false) {
				if ($_content == 'RT' or preg_match('/@([\w_]+):/i', $_content)) {
					continue;
				}

				$content .= $_content.' ';
			}
		}

		return md5(trim($content));
	}

	/**
	 * Checks for a retweet via twitter API data and user perception.
	 *
	 * @static
	 * @param  stdClass  $comment
	 * @return bool
	 */
	public static function is_retweet($comment = null, $tweet = null) {
		$is_retweet = false;
		if (!is_null($comment)) {
			if (isset($comment->social_raw_data) and !empty($comment->social_raw_data->retweeted_status)) {
				$is_retweet = true;
			}
			if (substr($comment->comment_content, 0, 4) == 'RT @') {
				$is_retweet = true;
			}
		}
		else if (!is_null($tweet)) {
			if (!empty($tweet->retweeted_status)) {
				$is_retweet = true;
			}
			if (substr($tweet->text, 0, 4) == 'RT @') {
				$is_retweet = true;
			}
		}
		return $is_retweet;
	}

	/**
	 * Adds a retweet to the original broadcasted post social items stack.
	 *
	 * @static
	 * @param  int    $comment_id
	 * @param  array  $comments
	 * @param  array  $social_items
	 */
	private static function add_to_social_items($comment_id, &$comments, &$social_items) {
		$object = null;
		$_comments = array();
		foreach ($comments as $id => $comment) {
			if (is_int($id)) {
				if ($comment->comment_ID == $comment_id) {
					$object = $comment;
				}
				else {
					$_comments[] = $comment;
				}
			}
			else {
				if (isset($_comments[$id])) {
					$_comments[$id] = array_merge($_comments[$id], $comment);
				}
				else {
					$_comments[$id] = $comment;
				}
			}
		}
		$comments = $_comments;

		if ($object !== null) {
			if (!isset($social_items['twitter'])) {
				$social_items['twitter'] = array();
			}

			$social_items['twitter'][$comment_id] = $object;
		}
	}

	/**
	 * Adds messaging to the title.
	 *
	 * @static
	 * @param  string  $title
	 * @param  string  $key
	 * @return string
	 */
	public static function social_item_output_title($title, $key) {
		if ($key == 'twitter') {
			$title .= __(' retweeted this', 'social');
		}

		return $title;
	}

	/**
	 * Add a "reply to" field to broadcast form.
	 *
	 * @static
	 * @param  obj  $post
	 * @param  obj  $service
	 * @param  obj  $account
	 * @return void
	 */
	public static function social_broadcast_form_item_edit($post, $service, $account) {
		if ($service->key() != 'twitter') {
			return;
		}
		$field_name = str_replace('_content', '_in_reply_to', $account['field_name_content']);
?>
<a href="#" class="tweet-reply-link"><?php _e('Send as a reply', 'social'); ?></a>
<div class="tweet-reply-fields">
	<label for="<?php echo esc_attr($field_name); ?>"><?php _e('URL of Tweet (to reply to)', 'social'); ?></label>
	<input type="text" class="tweet-reply-field" name="<?php echo esc_attr($field_name); ?>" value="" id="<?php echo esc_attr($field_name); ?>" />
</div>
<?php
	}

} // End Social_Twitter

define('SOCIAL_TWITTER_FILE', __FILE__);

// Filters
add_filter('social_register_service', array('Social_Twitter', 'register_service'));
add_filter('get_avatar_comment_types', array('Social_Twitter', 'get_avatar_comment_types'));
add_filter('social_comments_array', array('Social_Twitter', 'comments_array'), 10, 2);
add_filter('social_save_broadcasted_ids_data', array('Social_Twitter', 'social_save_broadcasted_ids_data'), 10, 5);
add_filter('social_item_output_title', array('Social_Twitter', 'social_item_output_title'), 10, 2);
add_action('social_broadcast_form_item_edit', array('Social_Twitter', 'social_broadcast_form_item_edit'), 10, 3);

}
