<?php

/**
 * Set up content restrictions
 *
 * @return void
 */
function onepass_initialise_content_restrictions() {
	add_action( 'template_redirect', 'onepass_truncate_post', 11 );
}

/**
 *
 * Truncate the post if the user is not logged in.
 *
 * Needs to be hooked after the main query has run, but before the content is
 * rendered. 'template_redirect' is ok.
 *
 * @return void
 */
function onepass_truncate_post() {
	global $post;

    /**
	 * If 1Pass is loaded, it will search the post content for the [1pass] shortcode,
	 * truncate the post at that point, and inject the [_1pass] shortcode which
	 * actually loads the embed code
	 */
	if( is_onepass() && onepass_restrict_post( $post ) && !is_feed() ) {
        $content = explode( '[1pass]', $post->post_content );
		$post->post_content = reset( $content ) . '[_1pass]';
    }
}

/**
 * This function decides whether to show a post or not.
 *
 * If it returns true, the post will be truncated.
 *
 * Filter on onepass_restrictions to add a condition.
 *
 * @return bool
 */
function onepass_restrict_post( $post ) {
    return apply_filters( 'onepass_restrictions', false, $post );
}

/**
 * This function sets a rule for which users see the button.
 * If it returns true, the button will be shown.
 * Feel free to unhook it and override it.
 *
 * @return bool
 */
function onepass_reject_unauthorised_users() {
    return !current_user_can( 'edit_posts' );
}

add_filter( 'onepass_restrictions', 'onepass_reject_unauthorised_users' );