AdminerSqlGemini: Wrap returned text to comment

diff --git a/plugins/sql-gemini.php b/plugins/sql-gemini.php
index 48e8af9352d86a169e60ad2953e62b16ea8a399b..37817a50319871ff68813e7a2f05b4a6b480a973 100644 (file)
--- a/plugins/sql-gemini.php
+++ b/plugins/sql-gemini.php
@@ -38,7 +38,14 @@ class AdminerSqlGemini {
                        )));
                        $response = json_decode(file_get_contents("https://generativelanguage.googleapis.com/v1beta/models/$this->model:generateContent?key=$this->apiKey", false, $context));
                        $text = $response->candidates[0]->content->parts[0]->text;
-                       echo preg_replace('~```sql\n(.*\n)```~sU', '\1', $text) . "\n";
+                       $in_code = false;
+                       foreach (preg_split('~(^|\n)```(sql)?(\n|$)~', $text) as $part) {
+                               $part = trim($part);
+                               if ($part) {
+                                       echo ($in_code ? $part : "/*\n$part\n*/") . "\n\n";
+                               }
+                               $in_code = !$in_code;
+                       }
                        exit;
                }
        }