<?php
/**
 * Writes one profiled request as JSON into bloatwright_requests.
 *
 * @package Bloatwright
 */

namespace Bloatwright\Profiler;

use Bloatwright\Db\Schema;

/**
 * Recorder.
 */
class Recorder {

	/**
	 * Cap for the data JSON.
	 */
	const MAX_BYTES = 524288;

	/**
	 * The URL exactly as a profiled request stores it (the runner compares against this form).
	 *
	 * @param string $url URL from the signed profiling context.
	 */
	public static function clean_url( string $url ): string {
		return substr( esc_url_raw( sanitize_text_field( wp_unslash( $url ) ) ), 0, 2048 );
	}

	/**
	 * Store a request record.
	 *
	 * @param int    $session_id Session.
	 * @param string $url URL.
	 * @param int    $pass Pass.
	 * @param array  $summary total_ms, peak_mem_kb, query_count.
	 * @param array  $data Per-owner data plus request meta.
	 * @return int Row id, or 0 when refused (session not running, or this URL/pass was already recorded).
	 */
	public function store( int $session_id, string $url, int $pass, array $summary, array $data ): int {
		global $wpdb;
		$url = substr( $url, 0, 2048 );
		if ( 'running' !== (string) $wpdb->get_var( $wpdb->prepare( 'SELECT status FROM %i WHERE id = %d', Schema::table( 'sessions' ), $session_id ) ) ) {
			return 0;
		}
		// A replayed token must not add a second row for the same request.
		if ( (int) $wpdb->get_var( $wpdb->prepare( 'SELECT id FROM %i WHERE session_id = %d AND url = %s AND pass = %d LIMIT 1', Schema::table( 'requests' ), $session_id, $url, $pass ) ) > 0 ) {
			return 0;
		}
		$json = wp_json_encode( $data );
		if ( ! is_string( $json ) ) {
			$json = '{}';
		}
		if ( strlen( $json ) > self::MAX_BYTES ) {
			$json = wp_json_encode( self::trim( $data ) );
			if ( ! is_string( $json ) ) {
				$json = '{}';
			}
		}
		$wpdb->insert(
			Schema::table( 'requests' ),
			array(
				'session_id'  => $session_id,
				'url'         => $url,
				'pass'        => $pass,
				'total_ms'    => round( (float) ( $summary['total_ms'] ?? 0 ), 2 ),
				'peak_mem_kb' => (int) ( $summary['peak_mem_kb'] ?? 0 ),
				'query_count' => (int) ( $summary['query_count'] ?? 0 ),
				'data'        => $json,
				'created_at'  => current_time( 'mysql', true ),
			),
			array( '%d', '%s', '%d', '%f', '%d', '%d', '%s', '%s' )
		);
		return (int) $wpdb->insert_id;
	}

	/**
	 * Reduce a record to fit the size cap.
	 *
	 * @param array $data Data.
	 */
	public static function trim( array $data ): array {
		foreach ( (array) ( $data['owners'] ?? array() ) as $k => $o ) {
			if ( isset( $o['hooks'] ) ) {
				$data['owners'][ $k ]['hooks'] = array_slice( (array) $o['hooks'], 0, 10 );
			}
			if ( isset( $o['queries']['slowest'] ) ) {
				$data['owners'][ $k ]['queries']['slowest'] = array_slice( (array) $o['queries']['slowest'], 0, 3 );
			}
			if ( isset( $o['assets'] ) ) {
				$data['owners'][ $k ]['assets'] = array_slice( (array) $o['assets'], 0, 20 );
			}
		}
		return $data;
	}
}