old-new-wiaas/backend/app/plugins/wiaas/includes/admin/class-wiaas-admin-organization.php

<?php

class Wiaas_Admin_Organization {

    public static function init() {
        // role switcher for organization
        add_action( 'admin_bar_menu', array( __CLASS__, 'add_role_switcher_menu' ), 10, 2 );

        // change user role if requested
        add_action('admin_init', array(__CLASS__, 'maybe_change_user_role'));

        add_filter('manage_users_columns', array(__CLASS__, 'manage_users_table_columns'));
        add_filter('views_users', array(__CLASS__, 'hide_users_table_list_navigation'));

        add_filter('get_role_list', array(__CLASS__, 'get_role_list_for_user'), 10, 2);


        add_filter('woocommerce_customer_meta_fields', array(__CLASS__, 'hide_woocommerce_customer_fields'));
    }

    public static function hide_woocommerce_customer_fields() {
        return array();
    }

    public static function hide_users_table_list_navigation() {
        return array();
    }

    public static function get_role_list_for_user($role_list, $user) {
        $organization_id = wiaas_get_user_organization_id($user->ID);

        if (! empty($organization_id)) {
            $organization_roles = wiaas_get_organization_roles($organization_id);

            foreach ($organization_roles as $organization_role) {
                $role_list[$organization_role] = translate_user_role( wp_roles()->role_names[ $organization_role ] );
            }

            if (! empty($organization_role)) {
                unset($role_list['none']);
            }
        }

        return $role_list;
    }

    public static function manage_users_table_columns( $defaults = array() ) {

        $defaults['role'] = __('Roles', 'wiaas');

        unset($defaults['posts']);

        return $defaults;
    }

    /**
     * Adds organization roles switcher menu to admin bar
     *
     * @param $admin_bar
     */
    public static function add_role_switcher_menu($admin_bar) {
        if (is_super_admin()) {
            $roles = array( 'administrator' );
        } else {
            $organization_id = wiaas_get_current_user_organization_id();
            $roles = wiaas_get_organization_roles($organization_id);
        }

        $user = wp_get_current_user();
        $current_role  = $user->roles[0];

        // Add menu item.
        $admin_bar->add_menu( array(
            'id'     => 'wiaas-view-as',
            'parent' => 'top-secondary',
            'title'  => '<span>' .
                __( 'View as:', 'wiaas' ) .
                '</span>' .
                '<span style="margin-left: 6px; font-weight: bold;">'.
                __(translate_user_role( wp_roles()->role_names[ $current_role ] ), 'wiaas') .
                '</span>' .
                '<i style="float: right;" class="ab-icon dashicons dashicons-arrow-down"></i>',
            'href'   => false,
            'meta'   => array(
                'title'    => __( 'View As', 'wiaas' ),
                'tabindex' => '0',
            ),
        ) );

        foreach ($roles as $role) {
            $role_name = translate_user_role( wp_roles()->role_names[ $role ] );

            if ($role === $current_role) {
                // highlight current role
                $title = '<span class="ab-label">' . __( $role_name, 'wiaas' ) . '</span>';

                $url = false;

            } else if ($role === 'customer') {
                // set external link for customer role
                $title = '<span>' . __( $role_name, 'wiaas' ) . '</span>';
                $title .= '<span class="ab-icon dashicons dashicons-external"></span>';

                $url = WIAAS_CUSTOMER_INTERFACE;

                $target = '_blank';

            } else {
                $title = '<span>' . __( $role_name, 'wiaas' ) . '</span>';

                $url = wp_nonce_url(admin_url(), 'wiaas-set-role', 'wiaas-set-role-nonce');
                $url = add_query_arg( 'wiaas-role', $role, $url );

                $target = '_parent';
            }

            $admin_bar->add_menu( array(
                'id'     => 'wiaas_'.$role,
                'parent' => 'wiaas-view-as',
                'title'  => $title,
                'href'   => $url,
                'meta'   => array(
                    'title'    => __( $role_name, 'wiaas' ),
                    'target' => $target
                ),
            ) );
        }
    }


    /**
     *
     * Process organization role switch request if needed
     *
     */
    public static function maybe_change_user_role() {
        if (! empty($_GET['wiaas-set-role-nonce']) && ! empty($_GET['wiaas-role'])) {
            if (wp_verify_nonce($_GET['wiaas-set-role-nonce'], 'wiaas-set-role')) {
                $role_name = sanitize_key($_GET['wiaas-role']);

                $role = wp_roles()->get_role($role_name);

                if (! empty($role) && $role_name !== 'customer') {
                    // get current user
                    $current_user = wp_get_current_user();

                    update_user_meta($current_user->ID, '_wiaas_current_user_admin_role', $role->name);

                    // switch user role
                    $current_user->set_role($role->name);
                }
            }

            wp_safe_redirect( admin_url('index.php') );
            exit;
        }
    }
}

Wiaas_Admin_Organization::init();