Ticket #16808: 16808-tests.php

<?php
/*
Plugin Name: Trac 16808
Plugin URI: http://wordpress.org
Description: Trac 16808
Author: Alex Koti
Version: 1
Author URI: http://alexkoti.com
*/

/**
 * Activation
 * Create subscriber test user
 * Assign 'argus_visitors' cap to user
 * 
 */
register_activation_hook( __FILE__, 'trac16808_activation' );
function trac16808_activation(){
        if ( ! current_user_can( 'activate_plugins' ) )
                return;
        $plugin = isset( $_REQUEST['plugin'] ) ? $_REQUEST['plugin'] : '';
        check_admin_referer( "activate-plugin_{$plugin}" );
        
        // new subscriber user
        $new_user_id = wp_create_user( 'trac16808', 'trac16808', 'trac16808@trac16808.com' );
        
        // add 'argus_visitors' cap
        $role_subscriber = get_role( 'subscriber' );
        $role_subscriber->add_cap( 'argus_visitors' );
}

/**
 * Deactivation
 * Delete test user and remove 'argus_visitors' cap
 * 
 */
register_deactivation_hook( __FILE__, 'trac16808_deactivation' );
function trac16808_deactivation(){
        if ( ! current_user_can( 'activate_plugins' ) )
                return;
        $plugin = isset( $_REQUEST['plugin'] ) ? $_REQUEST['plugin'] : '';
        check_admin_referer( "deactivate-plugin_{$plugin}" );
        
        // remove 'argus_visitors' cap
        $role_subscriber = get_role( 'subscriber' );
        $role_subscriber->remove_cap( 'argus_visitors' );
        
        $user = get_user_by( 'login', 'trac16808' );
        wp_delete_user( $user->ID, 1 );
}

/**
 * Register post types
 * 
 * 1) Argus > Visitor    : cap > admin, subscriber
 * 2) Argus > Comics     : cap > admin, subscriber
 * 3) Argus > Candies    : cap > admin
 * 4) Top Level > Toys   : cap > admin, subscriber
 * 5) Top Level > Books  : cap > admin
 * 
 */
add_action( 'init', 'trac16808_register_post_types' );
function trac16808_register_post_types(){
        
        /**
         * CPT Visitors inside Argus admin page 
         * Subscriber have permission to add new posts
         * 
         */
        $args = array(
                'labels' => array (
                        'name' => 'Visitors',
                        'singular_name' => 'Visitor',
                        'add_new' => 'Register New',
                        'add_new_item' => 'Register New Visitor',
                ),
                'public' => true,
                'publicly_queryable' => false,
                'exclude_from_search' => true,
                'show_ui' => true,
                'show_in_menu' => 'argus', // remove this line and the subscriber will can add new post
                'hiearchical' => false,
                'supports' => array(
                        'title',
                        'editor',
                ),
                'capabilities' => array(
                        'edit_post'              => 'argus_visitors',
                        'read_post'              => 'argus_visitors',
                        'delete_post'            => 'argus_visitors',
                        'edit_posts'             => 'argus_visitors',
                        'edit_others_posts'      => 'argus_visitors',
                        'publish_posts'          => 'argus_visitors',
                        'read_private_posts'     => 'argus_visitors',
                        'read'                   => 'argus_visitors',
                        'delete_posts'           => 'argus_visitors',
                        'delete_private_posts'   => 'argus_visitors',
                        'delete_published_posts' => 'argus_visitors',
                        'delete_others_posts'    => 'argus_visitors',
                        'edit_private_posts'     => 'argus_visitors',
                        'edit_published_posts'   => 'argus_visitors',
                ),
        );
        register_post_type( 'visitor', $args );
        
        /**
         * CPT Comics inside Argus admin page 
         * Subscriber have permission to add new posts
         * 
         */
        $args = array(
                'labels' => array (
                        'name' => 'Comics',
                        'singular_name' => 'Comic',
                        'add_new' => 'Add New',
                        'add_new_item' => 'Add New Comic',
                ),
                'public' => true,
                'publicly_queryable' => false,
                'exclude_from_search' => true,
                'show_ui' => true,
                'show_in_menu' => 'argus', // remove this line and the subscriber will can add new post
                'hiearchical' => false,
                'supports' => array(
                        'title',
                        'editor',
                ),
                'capabilities' => array(
                        'edit_post'              => 'argus_visitors',
                        'read_post'              => 'argus_visitors',
                        'delete_post'            => 'argus_visitors',
                        'edit_posts'             => 'argus_visitors',
                        'edit_others_posts'      => 'argus_visitors',
                        'publish_posts'          => 'argus_visitors',
                        'read_private_posts'     => 'argus_visitors',
                        'read'                   => 'argus_visitors',
                        'delete_posts'           => 'argus_visitors',
                        'delete_private_posts'   => 'argus_visitors',
                        'delete_published_posts' => 'argus_visitors',
                        'delete_others_posts'    => 'argus_visitors',
                        'edit_private_posts'     => 'argus_visitors',
                        'edit_published_posts'   => 'argus_visitors',
                ),
        );
        register_post_type( 'comic', $args );
        
        /**
         * CPT Candies inside Argus admin page 
         * Subscriber don't have permission to add new posts
         * 
         */
        $args = array(
                'labels' => array (
                        'name' => 'Candies',
                        'singular_name' => 'Candy',
                        'add_new' => 'Add New',
                        'add_new_item' => 'Add New Candy',
                ),
                'public' => true,
                'publicly_queryable' => false,
                'exclude_from_search' => true,
                'show_ui' => true,
                'show_in_menu' => 'argus',
                'hiearchical' => false,
                'supports' => array(
                        'title',
                        'editor',
                ),
                'capabilities' => array(
                        'edit_post'              => 'manage_options',
                        'read_post'              => 'manage_options',
                        'delete_post'            => 'manage_options',
                        'edit_posts'             => 'manage_options',
                        'edit_others_posts'      => 'manage_options',
                        'publish_posts'          => 'manage_options',
                        'read_private_posts'     => 'manage_options',
                        'read'                   => 'manage_options',
                        'delete_posts'           => 'manage_options',
                        'delete_private_posts'   => 'manage_options',
                        'delete_published_posts' => 'manage_options',
                        'delete_others_posts'    => 'manage_options',
                        'edit_private_posts'     => 'manage_options',
                        'edit_published_posts'   => 'manage_options',
                ),
        );
        register_post_type( 'candy', $args );
        
        /**
         * Regular CPT Books
         * Subscriber have permission to add new posts
         * 
         */
        $args = array(
                'labels' => array (
                        'name' => 'Toys',
                        'singular_name' => 'Toy',
                        'add_new' => 'Add New',
                        'add_new_item' => 'Add New Toy',
                ),
                'public' => true,
                'publicly_queryable' => true,
                'exclude_from_search' => false,
                'show_ui' => true,
                'show_in_menu' => true, 
                'hiearchical' => false,
                'supports' => array(
                        'title',
                        'editor',
                ),
                'capabilities' => array(
                        'edit_post'              => 'argus_visitors',
                        'read_post'              => 'argus_visitors',
                        'delete_post'            => 'argus_visitors',
                        'edit_posts'             => 'argus_visitors',
                        'edit_others_posts'      => 'argus_visitors',
                        'publish_posts'          => 'argus_visitors',
                        'read_private_posts'     => 'argus_visitors',
                        'read'                   => 'argus_visitors',
                        'delete_posts'           => 'argus_visitors',
                        'delete_private_posts'   => 'argus_visitors',
                        'delete_published_posts' => 'argus_visitors',
                        'delete_others_posts'    => 'argus_visitors',
                        'edit_private_posts'     => 'argus_visitors',
                        'edit_published_posts'   => 'argus_visitors',
                ),
        );
        register_post_type( 'toy', $args );
        
        /**
         * Regular CPT Book
         * Subscriber don't have permission to add new posts
         * 
         */
        $args = array(
                'labels' => array (
                        'name' => 'Books',
                        'singular_name' => 'Book',
                        'add_new' => 'Add New',
                        'add_new_item' => 'Add New Book',
                ),
                'public' => true,
                'publicly_queryable' => true,
                'exclude_from_search' => false,
                'show_ui' => true,
                'show_in_menu' => true, 
                'hiearchical' => false,
                'supports' => array(
                        'title',
                        'editor',
                ),
                'capabilities' => array(
                        'edit_post'              => 'manage_options',
                        'read_post'              => 'manage_options',
                        'delete_post'            => 'manage_options',
                        'edit_posts'             => 'manage_options',
                        'edit_others_posts'      => 'manage_options',
                        'publish_posts'          => 'manage_options',
                        'read_private_posts'     => 'manage_options',
                        'read'                   => 'manage_options',
                        'delete_posts'           => 'manage_options',
                        'delete_private_posts'   => 'manage_options',
                        'delete_published_posts' => 'manage_options',
                        'delete_others_posts'    => 'manage_options',
                        'edit_private_posts'     => 'manage_options',
                        'edit_published_posts'   => 'manage_options',
                ),
        );
        register_post_type( 'book', $args );
}

/**
 * Add menus
 * Set priority to 0 just to add submenu on top
 * 
 */
add_action( 'admin_menu', 'trac16808_add_menus', 9 );
function trac16808_add_menus(){
        add_menu_page( 'Argus', 'Argus Admin', 'argus_visitors', 'argus', 'trac16808_admin_page_render', false, 40 );
        add_submenu_page( 'argus', 'Argus Administration', 'Control Panel', 'argus_visitors', 'argus', 'trac16808_admin_page_render' );
}

/**
 * Render admin page
 * 
 */
function trac16808_admin_page_render(){
        echo '<div class="wrap"><h2>Argus Admin Page</h2></div>';
}

/**
 * Workaround
 * Add submenu pages with 'post-new.php?post_type=visitor'
 * Based on wp-includes/post.php function _add_post_type_submenus()
 * 
 */
//add_action( 'admin_menu', 'trac16808_add_post_type_submenus', 99 );
function trac16808_add_post_type_submenus(){
        foreach ( get_post_types( array( 'show_ui' => true ) ) as $ptype ) {
                $ptype_obj = get_post_type_object( $ptype );
                // Sub-menus only.
                if ( ! $ptype_obj->show_in_menu || $ptype_obj->show_in_menu === true )
                        continue;
                add_submenu_page( $ptype_obj->show_in_menu, $ptype_obj->labels->add_new, $ptype_obj->labels->add_new_item, $ptype_obj->cap->edit_posts, "post-new.php?post_type=$ptype" );
        }
}