kilbot/WooCommerce-POS

<?php

/**
 * Administrative Tools
 *
 * @package    WCPOS\Admin_Settings_Tools
 * @author   Paul Kilmurray <paul@kilbot.com.au>
 * @link     http://www.wcpos.com
 */

namespace WCPOS\Admin\Settings;

class Access extends Page {

    protected static $instance;

    /**
     * Each settings tab requires an id and label
     */
    public function __construct() {
        if ( ! current_user_can( 'promote_users' ) ) {
            $this->current_user_authorized = false;
        }

        $this->id    = 'access';
        $this->label = __( 'POS Access', 'woocommerce-pos' );

        // capabilities
        $this->caps = apply_filters( 'woocommerce_pos_capabilities', array(
            'pos' => array(
                'manage_woocommerce_pos', // pos admin
                'access_woocommerce_pos'  // pos frontend
            ),
            'woo' => array(
                'read_private_products',
                'read_private_shop_orders',
                'publish_shop_orders',
                'list_users'
            ),
        ) );
    }

    /**
     * @param bool $key
     * @return array
     */
    public function get( $key = false ) {
        return array(
            'id'    => $this->id,
            'roles' => $this->get_role_caps()
        );
    }

    /**
     * Get: Loop through roles and capabilities
     *
     * @return array
     */
    private function get_role_caps() {
        global $wp_roles;
        $role_caps = array();

        $roles = $wp_roles->roles;
        if ( $roles ): foreach ( $roles as $slug => $role ):
            $role_caps[ $slug ] = array(
                'name'         => $role['name'],
                'capabilities' => array(
                    'pos' => array_intersect_key(
                        array_merge( array_fill_keys( $this->caps['pos'], false ), $role['capabilities'] ),
                        array_flip( $this->caps['pos'] )
                    ),
                    'woo' => array_intersect_key(
                        array_merge( array_fill_keys( $this->caps['woo'], false ), $role['capabilities'] ),
                        array_flip( $this->caps['woo'] )
                    )
                )
            );
        endforeach; endif;

        return $role_caps;
    }

    /**
     * @param array $data
     * @return array
     */
    public function set( array $data ) {
        if ( isset( $data['roles'] ) ) {
            $this->update_capabilities( $data['roles'] );
        }

        return $this->get();
    }

    /**
     * Set: Loop through roles and capabilities
     *
     * @param array $roles
     */
    private function update_capabilities( array $roles ) {
        foreach ( $roles as $slug => $array ):

            $role = get_role( $slug );

            if ( $array['capabilities'] ) : foreach ( $array['capabilities'] as $key => $caps ):
                if ( $caps ): foreach ( $caps as $cap => $grant ):
                    if ( in_array( $cap, $this->caps[ $key ] ) ) {
                        $grant ? $role->add_cap( $cap ) : $role->remove_cap( $cap );
                    }
                endforeach; endif;
            endforeach; endif;

        endforeach;
    }

    /**
     * Delete: Loop through roles and capabilities
     * - add for administrator/shop_manager
     * - remove for everyone else
     *
     * @return bool|mixed|void
     */
    public function delete() {
        global $wp_roles;

        $roles = $wp_roles->roles;
        $caps  = array_merge( $this->caps['pos'], $this->caps['woo'] );

        if ( $roles ): foreach ( $roles as $slug => $array ):
            $role   = get_role( $slug );
            $action = ( $slug == 'administrator' || $slug == 'shop_manager' ) ? 'add_cap' : 'remove_cap';
            if ( $caps ): foreach ( $caps as $cap ):
                $role->$action( $cap );
            endforeach; endif;
        endforeach; endif;

        return $this->get();
    }

}