Ticket #17447: register_post_type_args-hook-v4.diff

trunk/wp-includes/post.php

@@ -1210 +1210 @@
                 '_edit_link'           => 'post.php?post=%d',
         );
         $args = wp_parse_args( $args, $defaults );
+
+        $args['name'] = sanitize_key( $post_type );
+
+        if ( ! apply_filters( 'do_register_post_type', true, $args, $post_type ) ) {
+                return false;
+        }
+
         $args = (object) $args;
 
-        $post_type = sanitize_key( $post_type );
-        $args->name = $post_type;
+        $post_type = $args['name'] = apply_filters( 'register_post_type_name', $args['name'], $args, $post_type );
 
+        $args = (object) $args;
+
+        list(
+                $args->publicly_queryable,
+                $args->show_ui,
+                $args->show_in_menu,
+                $args->show_in_admin_bar,
+                $args->show_in_nav_menus,
+                $args->exclude_from_search,
+        ) = apply_filters( 'register_post_type_flags', array(
+                'publicly_queryable' => $args->publicly_queryable,
+                'show_ui' => $args->show_ui,
+                'show_in_menu' => $args->show_in_menu,
+                'show_in_admin_bar' => $args->show_in_admin_bar,
+                'show_in_nav_menus' => $args->show_in_nav_menus,
+                'exclude_from_search' => $args->exclude_from_search,
+        ), $args, $post_type );
+
         if ( strlen( $post_type ) > 20 )
                 return new WP_Error( 'post_type_too_long', __( 'Post types cannot exceed 20 characters in length' ) );
 
@@ -1242 +1266 @@
         if ( null === $args->exclude_from_search )
                 $args->exclude_from_search = !$args->public;
 
+        list(
+                $args->capabilities,
+                $args->map_meta_cap,
+                $args->capability_type
+        ) = apply_filters( 'register_post_type_capabilities', array(
+                'capabilities' => $args->capabilities,
+                'map_meta_cap' => $args->map_meta_cap,
+                'capability_type' => $args->capability_type
+        ), $args, $post_type );
+
+
         // Back compat with quirky handling in version 3.0. #14122
         if ( empty( $args->capabilities ) && null === $args->map_meta_cap && in_array( $args->capability_type, array( 'post', 'page' ) ) )
                 $args->map_meta_cap = true;
@@ -1272 +1307 @@
                 $wp->add_query_var( $args->query_var );
         }
 
+        $args->rewrite = apply_filters( 'register_post_type_rewrite', $args->rewrite, $args, $post_type );
+
         if ( false !== $args->rewrite && ( is_admin() || '' != get_option( 'permalink_structure' ) ) ) {
                 if ( ! is_array( $args->rewrite ) )
                         $args->rewrite = array();
@@ -1290 +1327 @@
                                 $args->rewrite['ep_mask'] = EP_PERMALINK;
                 }
 
+                $args->hierarchical = apply_filters( 'register_post_type_hierarchical', $args->hierarchical, $args, $post_type );
+
                 if ( $args->hierarchical )
                         add_rewrite_tag( "%$post_type%", '(.+?)', $args->query_var ? "{$args->query_var}=" : "post_type=$post_type&pagename=" );
                 else
                         add_rewrite_tag( "%$post_type%", '([^/]+)', $args->query_var ? "{$args->query_var}=" : "post_type=$post_type&name=" );
 
+                $args->has_archive = apply_filters( 'register_post_type_has_archive', $args->has_archive, $args, $post_type );
+
                 if ( $args->has_archive ) {
                         $archive_slug = $args->has_archive === true ? $args->rewrite['slug'] : $args->has_archive;
                         if ( $args->rewrite['with_front'] )
@@ -1302 +1343 @@
                         else
                                 $archive_slug = $wp_rewrite->root . $archive_slug;
 
+                        $archive_slug = apply_filters( 'register_post_type_archive_slug', $archive_slug, $args, $post_type );
+
                         add_rewrite_rule( "{$archive_slug}/?$", "index.php?post_type=$post_type", 'top' );
                         if ( $args->rewrite['feeds'] && $wp_rewrite->feeds ) {
                                 $feeds = '(' . trim( implode( '|', $wp_rewrite->feeds ) ) . ')';
@@ -1314 +1357 @@
 
                 $permastruct_args = $args->rewrite;
                 $permastruct_args['feed'] = $permastruct_args['feeds'];
+                $permastruct_args = apply_filters( 'register_post_type_permastruct_args', $permastruct_args, $args, $post_type );
                 add_permastruct( $post_type, "{$args->rewrite['slug']}/%$post_type%", $permastruct_args );
         }
 
+        $args->register_meta_box_cb = apply_filters( 'register_post_type_meta_box_cb', $args->register_meta_box_cb, $args, $post_type );
         if ( $args->register_meta_box_cb )
                 add_action( 'add_meta_boxes_' . $post_type, $args->register_meta_box_cb, 10, 1 );
 
-        $args->labels = get_post_type_labels( $args );
+        $args->labels = get_post_type_labels( apply_filters( 'register_post_type_labels', $args, $post_type ) );
         $args->label = $args->labels->name;
 
         $wp_post_types[ $post_type ] = $args;
 
         add_action( 'future_' . $post_type, '_future_post_hook', 5, 2 );
 
+        $args->taxonomies = apply_filters( 'register_post_type_taxonomies', $args->taxonomies, $post_type );
         foreach ( $args->taxonomies as $taxonomy ) {
                 register_taxonomy_for_object_type( $taxonomy, $post_type );
         }
@@ -1432 +1478 @@
         if ( $args->map_meta_cap )
                 _post_type_meta_capabilities( $capabilities );
 
-        return (object) $capabilities;
+        return (object) apply_filters( 'get_post_type_capabilities', $capabilities, $args );
 }
 
 /**