<?php
if (!defined('ABSPATH')) exit;

class AIWEST_Memory {
    public static function set($key, $value, $type = 'general') {
        global $wpdb;
        $tables = AIWEST_Database::tables();
        $key = sanitize_key($key);
        $encoded = is_string($value)
            ? $value
            : wp_json_encode($value, JSON_UNESCAPED_UNICODE | JSON_PRETTY_PRINT);
        if ($encoded === false) $encoded = '';

        // Memory is stored in a private plugin table; persistent object caching is not appropriate for this mutable record.
        // phpcs:ignore WordPress.DB.DirectDatabaseQuery.DirectQuery, WordPress.DB.DirectDatabaseQuery.NoCaching
        $existing_id = $wpdb->get_var($wpdb->prepare('SELECT id FROM %i WHERE memory_key = %s', $tables['memory'], $key));

        if ($existing_id) {
            // phpcs:ignore WordPress.DB.DirectDatabaseQuery.DirectQuery, WordPress.DB.DirectDatabaseQuery.NoCaching
            $wpdb->update(
                $tables['memory'],
                [
                    'memory_value' => $encoded,
                    'memory_type'  => sanitize_text_field($type),
                    'updated_at'   => current_time('mysql'),
                ],
                ['id' => absint($existing_id)],
                ['%s', '%s', '%s'],
                ['%d']
            );
            return;
        }

        // phpcs:ignore WordPress.DB.DirectDatabaseQuery.DirectQuery
        $wpdb->insert(
            $tables['memory'],
            [
                'memory_key'   => $key,
                'memory_value' => $encoded,
                'memory_type'  => sanitize_text_field($type),
                'created_at'   => current_time('mysql'),
                'updated_at'   => current_time('mysql'),
            ],
            ['%s', '%s', '%s', '%s', '%s']
        );
    }

    public static function get($key, $default = null) {
        global $wpdb;
        $tables = AIWEST_Database::tables();

        // Memory is read from a private plugin table and may be updated during the same request.
        // phpcs:ignore WordPress.DB.DirectDatabaseQuery.DirectQuery, WordPress.DB.DirectDatabaseQuery.NoCaching
        $value = $wpdb->get_var($wpdb->prepare('SELECT memory_value FROM %i WHERE memory_key = %s', $tables['memory'], sanitize_key($key)));
        if ($value === null) return $default;

        $decoded = json_decode($value, true);
        return json_last_error() === JSON_ERROR_NONE ? $decoded : $value;
    }

    public static function all() {
        global $wpdb;
        $tables = AIWEST_Database::tables();

        // The admin memory screen requires a current, uncached view of the private plugin table.
        // phpcs:ignore WordPress.DB.DirectDatabaseQuery.DirectQuery, WordPress.DB.DirectDatabaseQuery.NoCaching
        return $wpdb->get_results($wpdb->prepare('SELECT * FROM %i ORDER BY updated_at DESC', $tables['memory']));
    }
}