Gravity Wiz // Gravity Forms // Submit to Access

gw-gravity-forms-submit-to-access.php

<?php
/**
 * Gravity Wiz // Gravity Forms // Submit to Access
 *
 * Require that a form be submitted before a post or page can be accessed.
 *
 * @version   1.1
 * @author    David Smith <david@gravitywiz.com>
 * @license   GPL-2.0+
 * @link      http://gravitywiz.com/submit-a-gravity-form-to-access-content/
 *
 * ### WordPress Plugin Headers ###
 *
 * Plugin Name: Gravity Forms Submit to Access
 * Plugin URI: http://gravitywiz.com/submit-a-gravity-form-to-access-content/
 * Description: Require that a form be submitted before a post or page can be accessed.
 * Author: Gravity Wiz
 * Version: 1.1
 * Author URI: http://gravitywiz.com
 */
class GW_Submit_Access {

	public $version = '1.1';

    public function __construct( $args = array() ) {

        // set our default arguments, parse against the provided arguments, and store for use throughout the class
        $this->_args = wp_parse_args( $args, array(
            'requires_submission_message'  => __( 'Oops! You do not have access to this page.' ),
            'bypass_cache'                 => false,
            'loading_message'              => '', // set later so we can use GFCommon to get URL to GF spinner,
            'enable_user_meta'             => false,
        ) );

        // do version check in the init to make sure if GF is going to be loaded, it is already loaded
        add_action( 'init', array( $this, 'init' ) );

    }

    public function init() {

        // make sure we're running the required minimum version of Gravity Forms
        if( ! property_exists( 'GFCommon', 'version' ) || ! version_compare( GFCommon::$version, '1.8', '>=' ) ) {
            return;
        }

        // setting later so we can use GFCommon::get_base_url() to get GF's spinner URL
        if( empty( $this->_args['loading_message'] ) ) {
            $this->_args['loading_message'] = '<span class="gwsa-loading">Loading content... <img src="' . GFCommon::get_base_url() . '/images/spinner.gif" /></span>';
        }

		add_action( 'wp',         array( $this, 'check_global_requirements' ), 5 );
		add_action( 'admin_init', array( $this, 'check_global_requirements' ), 5 );

        add_action( 'gform_pre_submission', array( $this, 'add_submitted_form' ) );
        add_filter( 'the_content', array( $this, 'maybe_hide_the_content' ) );

        add_action( 'wp_ajax_gwas_get_content', array( $this, 'ajax_get_content' ) );
        add_action( 'wp_ajax_nopriv_gwas_get_content', array( $this, 'ajax_get_content' ) );

    }

	public function check_global_requirements() {

		$global_posts = $this->get_global_posts();
		if( empty( $global_posts ) ) {
			return;
		}

		// if we're already on a global post, don't do anything
		$object = get_queried_object();
		if( is_a( $object, 'WP_Post' ) && in_array( $object->ID, wp_list_pluck( $global_posts, 'ID' ) ) ) {
			return;
		}

		foreach( $global_posts as $global_post ) {
			if( ! $this->has_access( $global_post->ID ) ) {
				wp_redirect( get_permalink( $global_post ) );
				exit;
			}
		}

	}

	public function get_global_posts() {

		$query = array(
			'post_type' => 'any',
			'meta_query' => array(
				'relation' => 'or',
				array(
					'key'   => 'gwsa_require_submission',
					'value' => 'global'
				),
			),
		);

		if( is_user_logged_in() ) {
			$query['meta_query'][] = array(
				'key'   => 'gwsa_require_submission',
				'value' => 'global_logged_in'
			);
		}

		$query        = apply_filters( 'gfsa_get_global_posts_query', $query );
		$global_posts = get_posts( $query );

		return $global_posts;
	}

    public function maybe_hide_the_content( $content ) {
        global $post;

        if( ! $this->requires_access( $post->ID ) ) {
            return $content;
        }

        if( $this->_args['bypass_cache'] ) {
            $content = $this->cache_bypass_content( $content );
        } else if( ! $this->has_access( $post->ID ) ) {
            $content = $this->get_requires_submission_message( $post->ID );
        }

        return $content;
    }

    function cache_bypass_content( $content ) {
        global $post;

        ob_start();
        ?>

        <div id="gwsa-content">
            <?php echo $this->_args['loading_message']; ?>
        </div>

        <script type="text/javascript">

            var ajaxUrl = '<?php echo admin_url( 'admin-ajax.php' ); ?>';

            ( function( $ ) {

                $.post( ajaxUrl, {
                    action: 'gwas_get_content',
                    post:   <?php echo $post->ID; ?>
                }, function( response ) {
                    $( '#gwsa-content' ).html( response );
                } );

            } )( jQuery );

        </script>

        <?php

        return ob_get_clean();
    }

    function ajax_get_content() {

        $post_id = rgpost( 'post' );

        if( $this->has_access( $post_id ) ) {

            $post = get_post( $post_id );
            $GLOBALS['post'] = get_post( $post_id );
            setup_postdata( $post );

            remove_filter( 'the_content', array( $this, 'maybe_hide_the_content' ) );

            // use the_content() so we get the content exactly as WP would have originally displayed it
            ob_start();
            the_content();
            $content = ob_get_clean();

        } else {

            $content = $this->get_requires_submission_message( $post_id );

        }

        die( $content );
    }

    function get_requires_submission_message( $post_id ) {

        $requires_submission_message = get_post_meta( $post_id, 'gwsa_requires_submission_message', true );
        if( ! $requires_submission_message ) {
            $requires_submission_message = $this->_args['requires_submission_message'];
        }

        if( ! $requires_submission_message ) {
            $form_ids = $this->get_form_ids( $post_id );
            if( ! empty( $form_ids ) ) {
                $requires_submission_message = gravity_form( $form_ids[0], false );
            }
        }

        return $requires_submission_message;
    }

    function has_access( $post_id ) {

        if( ! $this->requires_access( $post_id ) ) {
            return true;
        }

        $form_ids        = $this->get_form_ids( $post_id );
        $submitted_forms = $this->get_submitted_forms();

        // if not form-specific and at least one form is submitted, user has access
        if( empty( $form_ids ) && ! empty( $submitted_forms ) ) {
            return true;
        }

        // has specifically required form been submitted?
        $matching_form_ids = array_intersect( $form_ids, $submitted_forms );
        if( ! empty( $matching_form_ids ) ) {
            return true;
        }

        return false;
    }

    function requires_access( $post_id ) {
        return get_post_meta( $post_id, 'gwsa_require_submission', true ) == true;
    }

	function get_submitted_forms() {

		// always check the cookie first; will allow user meta vs cookie to be set per page in the future
		$submitted_forms = (array) json_decode( stripslashes( rgar( $_COOKIE, 'gwsa_submitted_forms' ) ) );

		// if user meta is enabled, merge forms stored there as well
		if( $this->_args['enable_user_meta'] ) {
			$user_meta_forms = (array) wp_get_current_user()->get( 'gwsa_submitted_forms' );
			$submitted_forms = array_merge( $submitted_forms, $user_meta_forms );
		}

		return array_filter( $submitted_forms );
	}

    function add_submitted_form( $form ) {

        $submitted_forms = $this->get_submitted_forms();
        $form_id = $form['id'];

        if( ! in_array( $form_id, $submitted_forms ) && ! headers_sent() ) {

            $submitted_forms[] = $form_id;

	        if( $this->_args['enable_user_meta'] ) {
		        update_user_meta( get_current_user_id(), 'gwsa_submitted_forms', $submitted_forms );
	        } else {
		        setcookie( 'gwsa_submitted_forms', json_encode( $submitted_forms ), null, '/' );
	        }

        }

    }

    function get_form_ids( $post_id ) {
        return array_filter( array_map( 'trim', explode( ',', get_post_meta( $post_id, 'gwsa_form_ids', true ) ) ) );
    }

}

new GW_Submit_Access();