<?php
/**
 * Adds custom fields to category edit screens
 *
 * These options are for restricting content within categories
 *
 * @package     Restrict Content Pro
 * @subpackage  Admin/Categories
 * @copyright   Copyright (c) 2017, Pippin Williamson
 * @license     http://opensource.org/licenses/gpl-2.0.php GNU Public License
 * @since       2.0
 */

// Exit if accessed directly
if ( ! defined( 'ABSPATH' ) ) exit;

/**
 * Setup actions for taxonomy restricted fields
 *
 * @since 2.5
 * @return void
 */
function rcp_setup_taxonomy_edit_fields() {
	$taxonomies = rcp_get_restricted_taxonomies();

	foreach( $taxonomies as $taxonomy ) {
		add_action( "{$taxonomy}_edit_form_fields", 'rcp_term_edit_meta_fields' );
		add_action( "{$taxonomy}_add_form_fields", 'rcp_term_add_meta_fields' );
	}
}
add_action( 'admin_init', 'rcp_setup_taxonomy_edit_fields' );

/**
 * Add restriction options to the edit term page
 *
 * @access      public
 * @since       2.0
 * @return      void
 */
function rcp_term_edit_meta_fields( $term ) {

	// retrieve the existing value(s) for this meta field. This returns an array
	$term_meta = rcp_get_term_restrictions( $term->term_id );
	$access_level = isset( $term_meta['access_level'] ) ? absint( $term_meta['access_level'] ) : 0;
	$subscription_levels = isset( $term_meta['subscriptions'] ) ? array_map( 'absint', $term_meta['subscriptions'] ) : array();
	?>
	<tr>
		<th scope="row"><?php _e( 'Paid Only?', 'rcp' ); ?></th>
		<td>
			<label for="rcp_category_meta[paid_only]">
				<input type="checkbox" name="rcp_category_meta[paid_only]" id="rcp_category_meta[paid_only]" value="1"<?php checked( true, isset( $term_meta['paid_only'] ) ); ?>>
				<span class="description"><?php _e( 'Restrict items in this category to paid subscribers only?', 'rcp' ); ?></span>
			</label>
		</td>
	</tr>
	<tr>
		<th scope="row"><?php _e( 'Access Level', 'rcp' ); ?></th>
		<td>
			<label for="rcp_category_meta[access_level]">
				<select name="rcp_category_meta[access_level]" id="rcp_category_meta[access_level]">
					<?php foreach( rcp_get_access_levels() as $level ) : ?>
						<option value="<?php echo esc_attr( $level ); ?>"<?php selected( $level, $access_level ); ?>><?php echo $level; ?></option>
					<?php endforeach; ?>
				</select>
				<span class="description"><?php _e( 'Access level required to view content in this category.', 'rcp' ); ?></span>
			</label>
		</td>
	</tr>
	<tr>
		<th scope="row"><?php _e( 'Membership Levels', 'rcp' ); ?></th>
		<td>
			<?php foreach( rcp_get_membership_levels( array( 'number' => 999 ) ) as $level ) : ?>
				<label for="rcp_category_meta[subscriptions][<?php echo esc_attr( $level->get_id() ); ?>]">
					<input type="checkbox" name="rcp_category_meta[subscriptions][<?php echo esc_attr( $level->get_id() ); ?>]" id="rcp_category_meta[subscriptions][<?php echo esc_attr( $level->get_id() ); ?>]" value="1"<?php checked( true, in_array( $level->get_id(), $subscription_levels ) ); ?>>
					<?php echo esc_html( $level->get_name() ); ?>
				</label><br/>
			<?php endforeach; ?>
			<span class="description"><?php _e( 'Membership levels allowed to view content in this category. Leave unchecked for all.', 'rcp' ); ?></span>
			<?php wp_nonce_field( 'rcp_edit_term', 'rcp_edit_term' ); ?>
		</td>
	</tr>
<?php
}


/**
 * Add restriction options to the add term page
 *
 * @param string $taxonomy
 *
 * @access      public
 * @since       2.7
 * @return      void
 */
function rcp_term_add_meta_fields( $taxonomy ) {
	?>
	<div class="form-field">
		<h2><?php _e( 'Restriction Options', 'rcp' ); ?></h2>
		<label for="rcp_category_meta[paid_only]">
			<p><?php _e( 'Paid Only?', 'rcp' ); ?></p>
			<input type="checkbox" name="rcp_category_meta[paid_only]" id="rcp_category_meta[paid_only]" value="1">
			<span class="description"><?php _e( 'Restrict items in this category to paid subscribers only?', 'rcp' ); ?></span>
		</label>
	</div>
	<div class="form-field">
		<label for="rcp_category_meta[access_level]">
			<p><?php _e( 'Access Level', 'rcp' ); ?></p>
			<select name="rcp_category_meta[access_level]" id="rcp_category_meta[access_level]">
				<?php foreach( rcp_get_access_levels() as $level ) : ?>
					<option value="<?php echo esc_attr( $level ); ?>"><?php echo $level; ?></option>
				<?php endforeach; ?>
			</select>
			<span class="description"><?php _e( 'Access level required to view content in this category.', 'rcp' ); ?></span>
		</label>
	</div>
	<div class="form-field">
		<p><?php _e( 'Membership Levels', 'rcp' ); ?></p>
		<?php foreach( rcp_get_membership_levels( array( 'number' => 999 ) ) as $level ) : ?>
			<label for="rcp_category_meta[subscriptions][<?php echo esc_attr( $level->get_id() ); ?>]">
				<input type="checkbox" name="rcp_category_meta[subscriptions][<?php echo esc_attr( $level->get_id() ); ?>]" id="rcp_category_meta[subscriptions][<?php echo esc_attr( $level->get_id() ); ?>]" value="1"/>
				<?php echo esc_html( $level->get_name() ); ?>
			</label>
		<?php endforeach; ?>
		<span class="description"><?php _e( 'Membership levels allowed to view content in this category. Leave unchecked for all.', 'rcp' ); ?></span>
		<?php wp_nonce_field( 'rcp_edit_term', 'rcp_edit_term' ); ?>
	</div>
<?php
}

/**
 * Save our custom term meta
 *
 * @param int    $term_id  Term ID.
 * @param int    $tt_id    Term taxonomy ID.
 * @param string $taxonomy Taxonomy slug.
 *
 * @access      public
 * @since       2.0
 * @return      void
 */
function rcp_save_term_meta( $term_id, $tt_id, $taxonomy ) {

	if ( empty( $_POST['rcp_edit_term'] ) || ! wp_verify_nonce( $_POST['rcp_edit_term'], 'rcp_edit_term' ) ) {
		return;
	}

	$restricted_taxonomies = rcp_get_restricted_taxonomies();
	if ( ! in_array( $taxonomy, $restricted_taxonomies ) ) {
		return;
	}

	$fields = ! empty( $_POST['rcp_category_meta'] ) ? $_POST['rcp_category_meta'] : array();

	if( ! empty( $_POST['rcp_category_meta']['subscriptions'] ) ) {
		$fields['subscriptions'] = array_map( 'absint', array_keys( $_POST['rcp_category_meta']['subscriptions'] ) );
	}

	if( ! empty( $fields['access_level'] ) ) {

		if( 'None' == $fields['access_level'] ) {
			unset( $fields['access_level'] );
		}

	}

	if ( function_exists( 'update_term_meta' ) ) {
		if ( ! empty( $fields ) ) {
			update_term_meta( $term_id, 'rcp_restricted_meta', $fields );
		} else {
			delete_term_meta( $term_id, 'rcp_restricted_meta' );
		}
		// remove deprecated data
		delete_option( "rcp_category_meta_$term_id" );
	} else {
		// fallback to older method of handling term meta
		update_option( "rcp_category_meta_$term_id", $fields );
	}

}
add_action( 'edited_term', 'rcp_save_term_meta', 10, 3 );
add_action( 'created_term', 'rcp_save_term_meta', 10, 3 );