<?php if ( ! defined('ABSPATH')) exit('No direct script access allowed');

/**
 * The admin-specific functionality of the plugin.
 *
 * @link       https://philsbury.uk
 * @since      2.0.0
 *
 * @package    Age_Gate
 * @subpackage Age_Gate/admin
 */

/**
 * The access settings of the plugin.
 *
 * Defines the plugin name, version, and two examples hooks for how to
 * enqueue the admin-specific stylesheet and JavaScript.
 *
 * @package    Age_Gate
 * @subpackage Age_Gate/admin
 * @author     Phil Baker
 */
class Age_Gate_Access extends Age_Gate_Common {

  public function __construct() {

		parent::__construct();

	}

  /**
   * Add the sub menu for global Settings
   * @since 2.0.0
   */
  public function add_settings_page()
  {
    add_submenu_page(
			$this->plugin_name,
			'Age Gate Access Settings',
			'Access',
			AGE_GATE_CAP_ACCESS,
			$this->plugin_name . '-access',
			array($this, 'display_options_page')
		);
  }

  /**
   * Display global settings options
   * @since 2.0.0
   */
  public function display_options_page()
  {

    $roles = wp_roles();
    // $roles = apply_filters('editable_roles', $all_roles);
    $values = $this->_filter_values( $this->settings['access'], 0);


    // echo "</";
    // wp_die();
    include AGE_GATE_PATH . 'admin/partials/age-gate-admin-access.php';
  }

  /**
   * Handle settings post from form
   * @return mixed
   * @since 2.0.0
   */
  public function handle_form_submission()
  {
    // Sanitize the post data
    $post = $this->validation->sanitize($_POST);

    if ( ! isset( $post['nonce'] ) || ! wp_verify_nonce( $post['nonce'], 'age_gate_update_access' ) ) {

      $this->_set_admin_notice(array('message' => __('Sorry, your nonce did not verify.', AGE_GATE_TEXT_DOMAIN), 'status' => 'error'));
      // set_transient( 'age_gate_admin_notice',  );

      wp_redirect($post['_wp_http_referer']);
      exit;
    }

    // ALWAYS add Admin
    foreach ($post['ag_settings']['permissions'] as $key => $type) {
      $post['ag_settings']['permissions'][$key] = $this->_prep_roles($post['ag_settings']['permissions'][$key]);
    }

    $this->_update_caps($post['ag_settings']['permissions']['restrict'], AGE_GATE_CAP_RESTRICTIONS);
    $this->_update_caps($post['ag_settings']['permissions']['messaging'], AGE_GATE_CAP_MESSAGING);
    $this->_update_caps($post['ag_settings']['permissions']['appearance'], AGE_GATE_CAP_APPEARANCE);
    $this->_update_caps($post['ag_settings']['permissions']['advanced'], AGE_GATE_CAP_ADVANCED);
    $this->_update_caps($post['ag_settings']['permissions']['settings'], AGE_GATE_CAP_ACCESS);
    $this->_update_caps($post['ag_settings']['permissions']['restrict_individual'], AGE_GATE_CAP_SET_CONTENT);
    $this->_update_caps($post['ag_settings']['permissions']['bypass_individual'], AGE_GATE_CAP_SET_BYPASS);
    $this->_update_caps($post['ag_settings']['permissions']['custom'], AGE_GATE_CAP_SET_CUSTOM_AGE);


    // set empty values so everything is stored
    // this will fix the issue of some settings getting
    // overwritten on update
    //
    // At this point, only post types are stored
    $values = $this->_filter_values( $post['ag_settings']['post_types'], 0);


    update_option('wp_age_gate_access', $values);
    $this->_set_admin_notice( array('message' => __('Settings saved successfully.', AGE_GATE_TEXT_DOMAIN), 'status' => 'success') );
    // set_transient( 'age_gate_admin_notice',  );
    wp_redirect($post['_wp_http_referer']);

  }

  /**
   * Filter to ensure all fields get sent to the DB
   * @param  [type] $data [description]
   * @param  [type] $fill [description]
   * @return [type]       [description]
   * @since   2.0.0
   */
  private function _filter_values($data, $fill)
  {
    $data = (!$data) ? [] : $data;
    $keys = [];

    foreach (get_post_types() as $key => $post_type) {
      $keys[] = $post_type;
    }

    $empties = array_fill_keys($keys, $fill);
    return array_merge($empties, $data);
  }

  /**
   * Update the selected roles with age gate permissions
   * @param  array $roles   Array of user roles
   * @param  string $cap    The capability to assign or remove
   * @since  2.0.0
   */
  private function _update_caps($roles, $cap)
  {
    global $wp_roles;

    foreach ($wp_roles->roles as $key => $value) {
      if(in_array($key, $roles)){
        $wp_roles->add_cap( $key, $cap );

      } else {
        $wp_roles->remove_cap( $key, $cap );

      }
    }
  }

  /**
   * Always make Admin role have the cap
   * @param  array $roles Selected roles
   * @return array
   */
  private function _prep_roles($roles)
  {
    // forces ADMIN to be selected
    return (!$roles) ? array('administrator') : array_unique(array_merge($roles, array('administrator')));
  }

}