<?php declare(strict_types = 1);

namespace MailPoet\Migrator;

if (!defined('ABSPATH')) exit;


use MailPoet\Config\Env;
use MailPoetVendor\Doctrine\DBAL\Connection;

class Store {
  /** @var Connection */
  private $connection;

  /** @var string */
  private $table;

  public function __construct(
    Connection $connection
  ) {
    $this->connection = $connection;
    $this->table = Env::$dbPrefix . 'migrations';
  }

  public function getMigrationsTable(): string {
    return $this->table;
  }

  public function startMigration(string $name): void {
    $this->connection->executeStatement("
      INSERT INTO {$this->table} (name, started_at)
      VALUES (?, current_timestamp())
      ON DUPLICATE KEY UPDATE
        started_at = current_timestamp(),
        completed_at = NULL,
        retries = retries + 1,
        error = NULL
    ", [$name]);
  }

  public function completeMigration(string $name): void {
    $this->connection->executeStatement("
      UPDATE {$this->table}
      SET completed_at = current_timestamp()
      WHERE name = ?
    ", [$name]);
  }

  public function failMigration(string $name, string $error): void {
    $this->connection->executeStatement("
      UPDATE {$this->table}
      SET
        completed_at = current_timestamp(),
        error = ?
      WHERE name = ?
    ", [$error ?: 'Unknown error', $name]);
  }

  public function getAll(): array {
    // Some backup plugins may convert NULL values to empty strings,
    // in which case we need to cast the error column value to NULL.
    return $this->connection->fetchAllAssociative("
      SELECT
        id,
        name,
        started_at,
        completed_at,
        retries,
        IF(error = '', NULL, error) AS error
      FROM {$this->table}
      ORDER BY id ASC
    ");
  }

  public function ensureMigrationsTable(): void {
    global $wpdb;
    $collate = $wpdb->get_charset_collate();
    $this->connection->executeStatement("
      CREATE TABLE IF NOT EXISTS {$this->table} (
        id int(11) unsigned NOT NULL AUTO_INCREMENT,
        name varchar(191) NOT NULL,
        started_at timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP,
        completed_at timestamp NULL,
        retries int(11) unsigned NOT NULL DEFAULT 0,
        error text NULL,
        PRIMARY KEY (id),
        UNIQUE KEY (name)
      ) {$collate};
    ");
  }
}
