ZwiiCMS/core/class/jsondb/JsonDb.class.php

<?php

/**
 * Created by PhpStorm.
 * User: Andrey Mistulov
 * Company: Aristos
 * Date: 14.03.2017
 * Time: 15:25
 */

namespace Prowebcraft;

/**
 * Class Data
 * @package Aristos
 */
class JsonDb extends \Prowebcraft\Dot
{
    protected $db = '';
    protected $data = null;
    protected $config = [];
    // Tentative d'écriture après échec
    const MAX_FILE_WRITE_ATTEMPTS = 5;
    // Délais entre deux tentaives
    const RETRY_DELAY_SECONDS = 1;

    public function __construct($config = [])
    {
        $this->config = array_merge([
            'name' => 'data.json',
            'backup' => false,
            'dir' => getcwd()
        ], $config);
        $this->loadData();
        parent::__construct();
    }

    /**
     * Reload data from file
     * @return $this
     */
    public function reload()
    {
        $this->loadData(true);
        return $this;
    }


    /**
     * Set value or array of values to path
     *
     * @param mixed      $key   Path or array of paths and values
     * @param mixed|null $value Value to set if path is not an array
     * @param bool $save Save data to database
     * @return $this
     */
    public function set($key, $value = null, $save = true)
    {
        parent::set($key, $value);
        if ($save)
            $this->save();
        return $this;
    }

    /**
     * Add value or array of values to path
     *
     * @param mixed      $key Path or array of paths and values
     * @param mixed|null $value Value to set if path is not an array
     * @param boolean    $pop Helper to pop out last key if value is an array
     * @param bool       $save    Save data to database
     * @return $this
     */
    public function add($key, $value = null, $pop = false, $save = true)
    {
        parent::add($key, $value, $pop);
        if ($save)
            $this->save();
        return $this;
    }

    /**
     * Delete path or array of paths
     *
     * @param mixed     $key Path or array of paths to delete
     * @param bool      $save Save data to database
     * @return $thisurn $this
     */
    public function delete($key, $save = true)
    {
        parent::delete($key);
        if ($save)
            $this->save();
        return $this;
    }

    /**
     * Delete all data, data from path or array of paths and
     * optionally format path if it doesn't exist
     *
     * @param mixed|null $key Path or array of paths to clean
     * @param boolean    $format Format option
     * @param bool       $save Save data to database
     * @return $this
     */
    public function clear($key = null, $format = false, $save = true)
    {
        parent::clear($key, $format);
        if ($save)
            $this->save();
        return $this;
    }


    /**
     * Charge les données depuis un fichier JSON.
     *
     * @param bool $reload Force le rechargement des données si true
     * 
     * @return array|null Les données chargées ou null si le fichier n'existe pas
     * 
     * @throws \RuntimeException En cas d'erreur lors de la création de la sauvegarde
     * @throws \InvalidArgumentException Si le fichier contient des données JSON invalides
     */
    protected function loadData($reload = false): ?array
    {
        if ($this->data === null || $reload) {
            $this->db = $this->config['dir'] . $this->config['name'];

            if (!file_exists($this->db)) {
                return null; // Rebuild database managed by CMS
            }

            if ($this->config['backup']) {
                $backup_path = $this->config['dir'] . DIRECTORY_SEPARATOR . $this->config['name'] . '.backup';

                try {
                    if (!copy($this->db, $backup_path)) {
                        throw new \RuntimeException('Échec de la création de la sauvegarde');
                    }
                } catch (\Exception $e) {
                    throw new \RuntimeException('Erreur de sauvegarde : ' . $e->getMessage());
                }
            }

            $file_contents = file_get_contents($this->db);

            $this->data = json_decode($file_contents, true);

            if ($this->data === null) {
                throw new \InvalidArgumentException('Le fichier ' . $this->db . ' contient des données invalides.');
            }
        }

        return $this->data;
    }

    /**
     * Charge les données depuis un fichier JSON.
     *
     * @param bool $reload Force le rechargement des données si true
     * 
     * @return array|null Les données chargées ou null si le fichier n'existe pas
     * 
     * @throws \RuntimeException En cas d'erreur lors de la création de la sauvegarde
     * @throws \InvalidArgumentException Si le fichier contient des données JSON invalides
     */

    public function save(): void
    {
        if ($this->data === null) {
            throw new \RuntimeException('Tentative de sauvegarde de données nulles');
        }

        try {
            $encoded_data = json_encode($this->data, JSON_UNESCAPED_UNICODE | JSON_FORCE_OBJECT | JSON_THROW_ON_ERROR);
        } catch (\JsonException $e) {
            throw new \RuntimeException('Erreur d\'encodage JSON : ' . $e->getMessage());
        }

        $encoded_length = strlen($encoded_data);
        $max_attempts = 5;

        for ($attempt = 0; $attempt < $max_attempts; $attempt++) {
            $temp_file = $this->db . '.tmp' . uniqid();

            try {
                $write_result = file_put_contents($temp_file, $encoded_data, LOCK_EX);

                if ($write_result === $encoded_length) {
                    if (rename($temp_file, $this->db)) {
                        return;
                    }
                }

                error_log("Échec sauvegarde : longueur incorrecte ou renommage échoué (tentative " . ($attempt + 1) . ")");
            } catch (\Exception $e) {
                error_log('Erreur de sauvegarde : ' . $e->getMessage());

                if (file_exists($temp_file)) {
                    unlink($temp_file);
                }
            }

            usleep(pow(2, $attempt) * 250000);
        }

        throw new \RuntimeException('Échec de sauvegarde après ' . $max_attempts . ' tentatives');
    }
}