<?php
/**
* Handles limitations to post types, uploads and more.
*
* @todo We need to move posts on downgrade.
* @package WP_Ultimo
* @subpackage Limits
* @since 2.0.0
*/
namespace WP_Ultimo\Limits;
// Exit if accessed directly
defined('ABSPATH') || exit;
/**
* Handles limitations to post types, uploads and more.
*
* @since 2.0.0
*/
class Post_Type_Limits {
use \WP_Ultimo\Traits\Singleton;
/**
* Runs on the first and only instantiation.
*
* @since 2.0.0
* @return void
*/
public function init() {
/**
* Emulated post types.
*
* @since 2.0.6
*/
if (is_main_site() && is_network_admin()) {
add_action('init', array($this, 'register_emulated_post_types'), 999);
} // end if;
/**
* Allow plugin developers to short-circuit the limitations.
*
* You can use this filter to run arbitrary code before any of the limits get initiated.
* If you filter returns any truthy value, the process will move on, if it returns any falsy value,
* the code will return and none of the hooks below will run.
*
* @since 1.7.0
* @return bool
*/
if (!apply_filters('wu_apply_plan_limits', wu_get_current_site()->has_limitations())) {
return;
} // end if;
if (!wu_get_current_site()->has_module_limitation('post_types')) {
return;
} // end if;
add_action('load-post-new.php', array($this, 'limit_posts'));
add_filter('wp_handle_upload', array($this, 'limit_media'));
add_filter('media_upload_tabs', array($this, 'limit_tabs'));
add_action('current_screen', array($this, 'limit_restoring'), 10);
add_filter('wp_insert_post_data', array($this, 'limit_draft_publishing'), 10, 2);
} // end init;
/**
* Emulates post types to avoid having to have plugins active on the main site.
*
* @since 2.0.6
* @return void
*/
public function register_emulated_post_types() {
$emulated_post_types = wu_get_setting('emulated_post_types', array());
if (is_array($emulated_post_types) && !empty($emulated_post_types)) {
foreach ($emulated_post_types as $pt) {
$pt = (object) $pt;
$existing_pt = get_post_type_object($pt->post_type);
if ($existing_pt) {
continue;
} // end if;
register_post_type($pt->post_type, array(
'label' => $pt->label,
'exclude_from_search' => true,
'public' => true,
'show_in_menu' => false,
'has_archive' => false,
'can_export' => false,
'delete_with_user' => false,
));
} // end foreach;
} // end if;
} // end register_emulated_post_types;
/**
* Prevents users from trashing posts and restoring them later to bypass the limitation.
*
* @since 2.0.0
* @return void
*/
public function limit_restoring() {
if (isset($_REQUEST['action']) && $_REQUEST['action'] === 'untrash') {
$this->limit_posts();
} // end if;
} // end limit_restoring;
/**
* Limit the posts after the user reach his plan limits
*
* @since 1.0.0
* @since 1.5.4 Checks for blocked post types
*/
public function limit_posts() {
if (is_main_site()) {
return;
} // end if;
$screen = get_current_screen();
if (!wu_get_current_site()->get_limitations()->post_types->{$screen->post_type}->enabled) {
$upgrade_message = __('Your plan does not support this post type.', 'wp-ultimo');
// translators: %s is the URL.
wp_die($upgrade_message, __('Limit Reached', 'wp-ultimo'), array('back_link' => true));
} // end if;
// Check if that is more than our limit
if (wu_get_current_site()->get_limitations()->post_types->is_post_above_limit($screen->post_type)) {
$upgrade_message = __('You reached your plan\'s post limit.', 'wp-ultimo');
// translators: %s is the URL
wp_die($upgrade_message, __('Limit Reached', 'wp-ultimo'), array('back_link' => true));
} // end if;
} // end limit_posts;
/**
* Checks if the user is trying to publish a draft post.
*
* If that's the case, only allow him to do it if the post count is not above the quota.
*
* @since 1.7.0
* @param array $data Info being saved on posts.
* @param array $modified_data Data that is changing. We are interested in publish.
* @return array
*/
public function limit_draft_publishing($data, $modified_data) {
global $current_screen;
if (empty($current_screen)) {
return $data;
} // end if;
if (get_post_status($modified_data['ID']) === 'publish') {
return $data; // If the post is already published, no need to make changes
} // end if;
if (isset($data['post_status']) && $data['post_status'] !== 'publish') {
return $data;
} // end if;
$post_type = isset($data['post_type']) ? $data['post_type'] : 'post';
$post_type_limits = wu_get_current_site()->get_limitations()->post_types;
if (!$post_type_limits->{$current_screen->post_type}->enabled || $post_type_limits->is_post_above_limit($post_type)) {
$data['post_status'] = 'draft';
} // end if;
return $data;
} // end limit_draft_publishing;
/**
* Limits uploads of items to the media library.
*
* @since 2.0.0
*
* @param array $file $_FILE array being passed.
* @return mixed
*/
public function limit_media($file) {
if (!wu_get_current_site()->get_limitations()->post_types->attachment->enabled) {
$file['error'] = __('Your plan does not support media upload.', 'wp-ultimo');
return $file;
} // end if;
$post_count = wp_count_posts('attachment');
$post_count = $post_count->inherit;
$quota = wu_get_current_site()->get_limitations()->post_types->attachment->number;
// This bit is for the flash uploader
if ($file['type'] === 'application/octet-stream' && isset($file['tmp_name'])) {
$file_size = getimagesize($file['tmp_name']);
if (isset($file_size['error']) && $file_size['error'] !== 0) {
$file['error'] = "Unexpected Error: {$file_size['error']}";
return $file;
} else {
$file['type'] = $file_size['mime'];
} // end if;
} // end if;
if ($quota > 0 && $post_count >= $quota) {
// translators: %d is the number of images allowed.
$file['error'] = sprintf(__('You reached your media upload limit of %d images. Upgrade your account to unlock more media uploads.', 'wp-ultimo'), $quota, '#');
} // end if;
return $file;
} // end limit_media;
/**
* Remove the upload tabs if the quota is over.
*
* @since 2.0.0
*
* @param array $tabs Tabs of the media gallery upload modal.
* @return array
*/
public function limit_tabs($tabs) {
$post_count = wp_count_posts('attachment');
$post_count = $post_count->inherit;
$quota = wu_get_current_site()->get_limitations()->post_types->attachment->number;
if ($quota > 0 && $post_count > $quota) {
unset($tabs['type']);
unset($tabs['type_url']);
} // end if;
return $tabs;
} // end limit_tabs;
} // end class Post_Type_Limits;