<?php
/**
 * Applies saved limits to core's revision count.
 *
 * @package Leafcap_Revision_Limits
 */

namespace Leafcap_Revision_Limits;

use WP_Post;

defined( 'ABSPATH' ) || exit;

/**
 * Filters wp_revisions_to_keep. It only returns a number. Core does any
 * trimming, so this class never writes to the database.
 */
class Limiter {

	/**
	 * Registers the filter.
	 *
	 * @return void
	 */
	public function register_hooks() {
		add_filter( 'wp_revisions_to_keep', array( $this, 'filter_revisions_to_keep' ), 10, 2 );
	}

	/**
	 * Returns the saved limit for the post's type, if it has one.
	 *
	 * @param int     $num  Number of revisions core would keep.
	 * @param WP_Post $post Post being checked.
	 * @return int
	 */
	public function filter_revisions_to_keep( $num, $post ) {
		if ( ! $post instanceof WP_Post || ! post_type_supports( $post->post_type, 'revisions' ) ) {
			return $num;
		}

		$limits = Settings::get_limits();

		return isset( $limits[ $post->post_type ] ) ? $limits[ $post->post_type ] : $num;
	}
}