welcome back to dyb-tech

2024-05-18 02:28:01 +02:00
parent 9513cdba09
commit 9f30bc98c7
6149 changed files with 668407 additions and 0 deletions
@@ -0,0 +1,197 @@
+<?php
+
+namespace Doctrine\Bundle\DoctrineBundle\Twig;
+
+use Doctrine\SqlFormatter\HtmlHighlighter;
+use Doctrine\SqlFormatter\NullHighlighter;
+use Doctrine\SqlFormatter\SqlFormatter;
+use Symfony\Component\VarDumper\Cloner\Data;
+use Twig\Extension\AbstractExtension;
+use Twig\TwigFilter;
+
+use function addslashes;
+use function array_key_exists;
+use function bin2hex;
+use function implode;
+use function is_array;
+use function is_bool;
+use function is_object;
+use function is_string;
+use function method_exists;
+use function preg_match;
+use function preg_replace_callback;
+use function sprintf;
+use function strtoupper;
+use function substr;
+use function trigger_deprecation;
+
+/**
+ * This class contains the needed functions in order to do the query highlighting
+ *
+ * @internal since 2.11
+ */
+class DoctrineExtension extends AbstractExtension
+{
+    private SqlFormatter $sqlFormatter;
+
+    /**
+     * Define our functions
+     *
+     * @return TwigFilter[]
+     */
+    public function getFilters()
+    {
+        return [
+            new TwigFilter('doctrine_pretty_query', [$this, 'formatQuery'], ['is_safe' => ['html'], 'deprecated' => true]),
+            new TwigFilter('doctrine_prettify_sql', [$this, 'prettifySql'], ['is_safe' => ['html']]),
+            new TwigFilter('doctrine_format_sql', [$this, 'formatSql'], ['is_safe' => ['html']]),
+            new TwigFilter('doctrine_replace_query_parameters', [$this, 'replaceQueryParameters']),
+        ];
+    }
+
+    /**
+     * Escape parameters of a SQL query
+     * DON'T USE THIS FUNCTION OUTSIDE ITS INTENDED SCOPE
+     *
+     * @internal
+     *
+     * @param mixed $parameter
+     *
+     * @return string
+     */
+    public static function escapeFunction($parameter)
+    {
+        $result = $parameter;
+
+        switch (true) {
+            // Check if result is non-unicode string using PCRE_UTF8 modifier
+            case is_string($result) && ! preg_match('//u', $result):
+                $result = '0x' . strtoupper(bin2hex($result));
+                break;
+
+            case is_string($result):
+                $result = "'" . addslashes($result) . "'";
+                break;
+
+            case is_array($result):
+                foreach ($result as &$value) {
+                    $value = static::escapeFunction($value);
+                }
+
+                $result = implode(', ', $result) ?: 'NULL';
+                break;
+
+            case is_object($result) && method_exists($result, '__toString'):
+                $result = addslashes($result->__toString());
+                break;
+
+            case $result === null:
+                $result = 'NULL';
+                break;
+
+            case is_bool($result):
+                $result = $result ? '1' : '0';
+                break;
+        }
+
+        return $result;
+    }
+
+    /**
+     * Return a query with the parameters replaced
+     *
+     * @param string       $query
+     * @param mixed[]|Data $parameters
+     *
+     * @return string
+     */
+    public function replaceQueryParameters($query, $parameters)
+    {
+        if ($parameters instanceof Data) {
+            $parameters = $parameters->getValue(true);
+        }
+
+        $i = 0;
+
+        if (! array_key_exists(0, $parameters) && array_key_exists(1, $parameters)) {
+            $i = 1;
+        }
+
+        return preg_replace_callback(
+            '/\?|((?<!:):[a-z0-9_]+)/i',
+            static function ($matches) use ($parameters, &$i) {
+                $key = substr($matches[0], 1);
+
+                if (! array_key_exists($i, $parameters) && ($key === false || ! array_key_exists($key, $parameters))) {
+                    return $matches[0];
+                }
+
+                $value  = array_key_exists($i, $parameters) ? $parameters[$i] : $parameters[$key];
+                $result = DoctrineExtension::escapeFunction($value);
+                $i++;
+
+                return $result;
+            },
+            $query,
+        );
+    }
+
+    /**
+     * Formats and/or highlights the given SQL statement.
+     *
+     * @param  string $sql
+     * @param  bool   $highlightOnly If true the query is not formatted, just highlighted
+     *
+     * @return string
+     */
+    public function formatQuery($sql, $highlightOnly = false)
+    {
+        trigger_deprecation(
+            'doctrine/doctrine-bundle',
+            '2.1',
+            'The "%s()" method is deprecated and will be removed in doctrine-bundle 3.0.',
+            __METHOD__,
+        );
+
+        $this->setUpSqlFormatter(true, true);
+
+        if ($highlightOnly) {
+            return $this->sqlFormatter->highlight($sql);
+        }
+
+        return sprintf(
+            '<div class="highlight highlight-sql"><pre>%s</pre></div>',
+            $this->sqlFormatter->format($sql),
+        );
+    }
+
+    public function prettifySql(string $sql): string
+    {
+        $this->setUpSqlFormatter();
+
+        return $this->sqlFormatter->highlight($sql);
+    }
+
+    public function formatSql(string $sql, bool $highlight): string
+    {
+        $this->setUpSqlFormatter($highlight);
+
+        return $this->sqlFormatter->format($sql);
+    }
+
+    private function setUpSqlFormatter(bool $highlight = true, bool $legacy = false): void
+    {
+        $this->sqlFormatter = new SqlFormatter($highlight ? new HtmlHighlighter([
+            HtmlHighlighter::HIGHLIGHT_PRE            => 'class="highlight highlight-sql"',
+            HtmlHighlighter::HIGHLIGHT_QUOTE          => 'class="string"',
+            HtmlHighlighter::HIGHLIGHT_BACKTICK_QUOTE => 'class="string"',
+            HtmlHighlighter::HIGHLIGHT_RESERVED       => 'class="keyword"',
+            HtmlHighlighter::HIGHLIGHT_BOUNDARY       => 'class="symbol"',
+            HtmlHighlighter::HIGHLIGHT_NUMBER         => 'class="number"',
+            HtmlHighlighter::HIGHLIGHT_WORD           => 'class="word"',
+            HtmlHighlighter::HIGHLIGHT_ERROR          => 'class="error"',
+            HtmlHighlighter::HIGHLIGHT_COMMENT        => 'class="comment"',
+            HtmlHighlighter::HIGHLIGHT_VARIABLE       => 'class="variable"',
+        ], ! $legacy) : new NullHighlighter());
+    }
+}