<?php
declare(strict_types=1);

/**
 * CakePHP(tm) : Rapid Development Framework (https://cakephp.org)
 * Copyright (c) Cake Software Foundation, Inc. (https://cakefoundation.org)
 *
 * Licensed under The MIT License
 * For full copyright and license information, please see the LICENSE.txt
 * Redistributions of files must retain the above copyright notice.
 *
 * @copyright     Copyright (c) Cake Software Foundation, Inc. (https://cakefoundation.org)
 * @link          https://cakephp.org CakePHP(tm) Project
 * @since         5.1.0
 * @license       https://opensource.org/licenses/mit-license.php MIT License
 */
namespace Cake\Database\Exception;

use Cake\Database\Log\LoggedQuery;
use PDOException;

class QueryException extends PDOException
{
    /**
     * Constructor
     *
     * @param \Cake\Database\Log\LoggedQuery|string $query
     * @param \PDOException $previous
     */
    public function __construct(protected LoggedQuery|string $query, PDOException $previous)
    {
        $message = $previous->getMessage();

        // Prefix with connection name if available
        $connectionName = $this->getConnectionName();
        if ($connectionName !== '') {
            $message = "[{$connectionName}] " . $message;
        }

        $message .= "\nQuery: " . $this->getQueryString();

        parent::__construct($message, (int)$previous->getCode(), $previous);
    }

    /**
     * Get the connection name that caused this exception.
     *
     * @return string
     */
    public function getConnectionName(): string
    {
        if ($this->query instanceof LoggedQuery) {
            return $this->query->getConnectionName();
        }

        return '';
    }

    /**
     * Get the query string that caused this exception.
     *
     * @return string
     */
    public function getQueryString(): string
    {
        if ($this->query instanceof LoggedQuery) {
            return (string)$this->query;
        }

        return $this->query;
    }
}
