<?php

namespace WP_CLI\Doctor\Check;

use WP_CLI;
use WP_CLI\Doctor\Check;

/**
 * Warns when autoloaded options size exceeds threshold of %threshold_kb% kb.
 */
class Autoload_Options_Size extends Check {

    /**
     * Threshold in kilobytes.
     *
     * @var integer
     */
    protected $threshold_kb = 900;

    public function run() {
        ob_start();
        WP_CLI::run_command(
            array( 'option', 'list' ),
            array(
                'autoload' => 'on',
                'format'   => 'total_bytes',
            )
        );
        $total_bytes = (int) ob_get_clean();

        $threshold_bytes = $this->threshold_kb * 1024;
        $human_threshold = self::format_bytes( $threshold_bytes );
        $human_total     = self::format_bytes( $total_bytes );
        if ( $threshold_bytes < $total_bytes ) {
            $this->set_status( 'warning' );
            $this->set_message( "Autoloaded options size ({$human_total}) exceeds threshold ({$human_threshold})" );
        } else {
            $this->set_status( 'success' );
            $this->set_message( "Autoloaded options size ({$human_total}) is less than threshold ({$human_threshold})." );
        }
    }

    private static function format_bytes( $size, $precision = 2 ) {
        $base     = log( $size, 1024 );
        $suffixes = array( '', 'kb', 'mb', 'g', 't' );
        return round( pow( 1024, $base - floor( $base ) ), $precision ) . $suffixes[ floor( $base ) ];
    }
}