Ticket #1806: jetpack-follow-link-for-o2.php

<?php
/*
Plugin Name: Jetpack Follow Link for o2
Description: Easily subscribe to an o2 comment thread without commenting using a "Follow" action link like WordPress.com has
*/


class o2_follow {

        // Use the old p2 meta key, for backwards compatiblity.
        const user_meta_key = 'jpflfp2_posts_following';

        /**
         * Construct the plugin
         */
        function __construct() {
                add_action( 'init', array( $this, 'action_init' ) );
        }

        /**
         *
         */
        function action_init() {

                if ( ! current_theme_supports( 'o2' ) || ! class_exists( 'Jetpack_Subscriptions' ) ) {
                        return;
                }

                // Only logged in users should be able to subscribe for now because we use the registered email address
                if ( !is_user_logged_in() )
                        return;

                add_action( 'o2_post_form_extras', array( $this, 'subscription_o2_post_form' ) );

                add_action( 'init', array( $this, 'subscribe_o2_register_post_action_states' ) );
                add_filter( 'o2_filter_post_actions', array( $this, 'subscription_o2_add_comment_subscriber_link' ), 10, 2 );

                add_filter( 'o2_comment_form_extras', array( $this, 'subscribe_o2_comment_form' ), 10, 2 );
                add_filter( 'o2_options', array( $this, 'subscribe_add_o2_options' ) );
                add_filter( 'o2_post_fragment', array( $this, 'subscribe_add_o2_post_fragment' ), 10, 2 );
        }

        function subscribe_o2_comment_form( $comment_form_extras = '', $post_ID = false ) {
                if ( !$post_ID ) {
                        global $post;
                        $post_ID = $post->post_ID;
                }

                $output  = '<p class="comment-subscription-form">';
                $output .= '<input type="checkbox" name="subscribe" id="subscribe" value="subscribe" style="width: auto;"' . checked( $checked, true, false ) . '/> ';
                $output .= '<label class="subscribe-label" id="subscribe-label" for="subscribe" style="display: inline;">' . __( 'Notify me of new comments via email.' ) . '</label>';
                $output .= '</p>';


                $output .= '<p class="post-subscription-form"><input type="checkbox" name="subscribe_blog" id="subscribe_blog" value="subscribe" style="width: auto;"/> ';
                $output .=      '<label class="subscribe-label" id="subscribe-blog-label" for="subscribe_blog"  style="display: inline;">'. __( 'Notify me of new posts via email.' ) .'</label>';
                $output .= '</p>';

                return $comment_form_extras . $output;
        }

        function subscribe_add_o2_options( $options ) {
                global $wpdb;

                $subscribed_ids = (array)get_user_meta( wp_get_current_user()->ID, self::user_meta_key, true );

                $options['options']['followingBlog'] = false;
                $options['options']['followingAllComments'] = false;
                return $options;
        }

        function subscribe_add_o2_post_fragment( $fragment = array(), $post_ID = false ) {
                global $wpdb;
                if ( !$post_ID ) {
                        global $post;
                        $post_ID = $post->post_ID;
                }
                if ( ! array_key_exists( 'postMeta', $fragment ) )
                        $fragment['postMeta'] = array();

                $subscribed_ids = (array)get_user_meta( wp_get_current_user()->ID, self::user_meta_key, true );
                $fragment['postMeta']['isFollowing'] = in_array( get_the_ID(), $subscribed_ids );
                return $fragment;
        }

        /**
        * Add 'subscribe' checkbox to o2 post form
        *
        * @param string
        * @return void
        **/
        function subscription_o2_post_form( $post_form_extras = '' ) {
                $output = '';
                $label = __( 'Notify me of new comments via email.' );
                $output .= '<p style="margin-top: 1.5em;" class="comment-subscription-form"><input type="checkbox" name="post_subscribe" id="post_subscribe" value="post_subscribe" style="margin-left: .5em;"/>';
                $output .= '<label style="font-size: 1.2em; margin-bottom: .5em;" id="post_subscribe_label" for="post_subscribe"><small>' . $label . '</small></label>';
                $output .= '</p>';

                return $post_form_extras . $output;
        }

        function subscribe_o2_register_post_action_states() {
        echo 1;
                if ( ! function_exists( 'o2_register_post_action_states' ) ) {
                        return;
                }
echo 2;
                o2_register_post_action_states( 'follow',
                        array(
                                'normal' => array(
                                        'shortText' => __( 'Follow', 'o2' ),
                                        'title' => __( 'Follow comments', 'o2' ),
                                        'classes' => array(),
                                        'genericon' => 'genericon-subscribe',
                                        'nextState' => 'subscribed'
                                ),
                                'subscribed' => array(
                                        'shortText' => __( 'Following', 'o2' ),
                                        'title' => __( 'Unfollow comments', 'o2' ),
                                        'classes' => array( 'post-comments-subscribed' ),
                                        'genericon' => 'genericon-unsubscribe',
                                        'nextState' => 'normal'
                                )
                        )
                );
        }

        function subscription_o2_add_comment_subscriber_link( $actions, $id = false ) {
                global $post, $wpdb;
                if ( !$id ) {
                        $id = $post->ID;
                }

                if ( comments_open( $id ) ) {
                        $protocol = is_ssl() ? 'https' : 'http';
                        $query_args = array();
                        $initial_state = 'normal';

                        $subscribed_ids = (array)get_user_meta( wp_get_current_user()->ID, self::user_meta_key, true );

                        if ( in_array( $id, $subscribed_ids ) ) {
                                $query_args = array(
                                        'post-id' => $id,
                                        'action' => 'post-comment-unsubscribe',
                                );
                                $initial_state = 'subscribed';
                        } else {
                                $query_args = array(
                                        'post-id' => $id,
                                        'action' => 'post-comment-subscribe',
                                );
                        }

                        $link = add_query_arg( $query_args, home_url( '', $protocol ) );
                        $link = wp_nonce_url( $link, 'post-comment-subscribe' );

                        $actions[31] = array(
                                'action' => 'follow',
                                'href' => $link,
                                'classes' => array( 'subscription-link', 'o2-follow' ),
                                'rel' => false,
                                'initialState' => $initial_state
                        );
                }

                return $actions;
        }
}

global $o2_follow;
$o2_follow = new o2_follow();