Merge pull request #9 from ensi-platform/ef-382

EF-382 экранирование description в enum

Author: MsNatali

config/openapi-client-generator.php

@@ -91,5 +91,10 @@
          * Options for disable patch section "require" composer.json
          */
         'composer_disable_patch_require' => false,
+
+        /**
+         * Options for disable patch enum
+         */
+        'enum_disable_patch' => false,
     ],
 ];

src/Commands/GeneratePhpClient.php

@@ -29,6 +29,7 @@ class GeneratePhpClient extends GenerateClient
     protected string $laravelPackageConfigKey;
 
     private bool $disableComposerPatchRequire;
+    private bool $disableEnumPath;
 
     public function __construct()
     {
@@ -44,11 +45,19 @@ public function __construct()
             "openapi-client-generator.{$this->client}_args.composer_disable_patch_require",
             false
         );
+
+        $this->disableEnumPath = (bool)config(
+            "openapi-client-generator.{$this->client}_args.enum_disable_patch",
+            false
+        );
     }
 
     protected function patchClientPackage(): void
     {
-        //        $this->patchEnums();
+        if (!$this->disableEnumPath) {
+            $this->patchEnums();
+        }
+
         $this->patchComposerPackage();
         $this->patchReadme();
         $this->generateProvider();

src/Core/Patchers/EnumPatcher.php

@@ -0,0 +1,45 @@
+<?php
+
+namespace Ensi\LaravelOpenapiClientGenerator\Core\Patchers;
+
+use Exception;
+
+abstract class PhpClassPatcher
+{
+    public function getMethod(string $name, string $returnType, string $content): string
+    {
+        $pattern = "/([public|protected|private]( static)? function {$name}\(\): {$returnType}.*})/sU";
+        if (!preg_match($pattern, $content, $matches)) {
+            throw new Exception("Не удалось найти метод $name");
+        }
+
+        return $matches[1];
+    }
+
+    public function escapingString(string $string): string
+    {
+        return match (true) {
+            str_starts_with($string, "'") => $this->replacePatternWithinString($string, "/[^\\\\]\K'/", "\\'"),
+            str_starts_with($string, '"') => $this->replacePatternWithinString($string, "/[^\\\\]\K\"/", '\\"'),
+            default => $string
+        };
+    }
+
+    public function replaceValue(string $context, string $oldValue, string $newValue): string
+    {
+        return str_replace($oldValue, $newValue, $context);
+    }
+
+    public function replacePattern(string $context, string $pattern, string $newValue): string
+    {
+        return preg_replace($pattern, $newValue, $context);
+    }
+
+    public function replacePatternWithinString(string $context, string $pattern, string $newValue): string
+    {
+        $subString = substr($context, 1, -1);
+        $replacedSubString = $this->replacePattern($subString, $pattern, $newValue);
+
+        return $this->replaceValue($context, $subString, $replacedSubString);
+    }
+}

src/Core/Patchers/PhpClassPatcher.php

@@ -2,44 +2,50 @@
 
 namespace Ensi\LaravelOpenapiClientGenerator\Core\Patchers;
 
-use Illuminate\Support\Str;
+use Exception;
 
-class PhpEnumPatcher extends EnumPatcher
+class PhpEnumPatcher extends PhpClassPatcher
 {
-    protected function getSpecificationName(): string
+    public function __construct(protected string $enumFile)
     {
-        return $this->toSnakeCase(basename($this->enumFile, '.php'));
     }
 
-    protected function patchEnumFile(array $constants): void
+    /** @throws Exception */
+    public function patch(): void
     {
         if (!file_exists($this->enumFile)) {
-            return;
+            throw new Exception("$this->enumFile not exists");
         }
 
-        $enum = file_get_contents($this->enumFile);
+        $content = file_get_contents($this->enumFile);
 
-        foreach ($constants as $constant) {
-            $enum = $this->patchConstantProperties(
-                $enum,
-                $constant['value'],
-                Str::upper($constant['name']),
-                $constant['title']
-            );
-        }
+        $content = $this->escapingResponseDescription($content);
 
-        file_put_contents($this->enumFile, $enum);
+        file_put_contents($this->enumFile, $content);
     }
 
-    private function patchConstantProperties(string $enum, string $value, string $name, string $title): string
+    /** @throws Exception */
+    protected function escapingResponseDescription(string $content): string
     {
-        // Do some preg replace here to change something
+        $method = $this->getMethod('getDescriptions', 'array', $content);
+
+        $descriptions = $this->parsingResponseDescription($method);
+
+        $escapingMethod = $method;
+        foreach ($descriptions as $description) {
+            $escapedDescription = $this->escapingString($description);
+            $escapingMethod = $this->replaceValue($escapingMethod, $description, $escapedDescription);
+        }
 
-        return $enum;
+        return $this->replaceValue($content, $method, $escapingMethod);
     }
 
-    private function toSnakeCase(string $str): string
+    protected function parsingResponseDescription(string $context): array
     {
-        return strtolower(preg_replace('/(?<!^)[A-Z]/', '_$0', $str));
+        return preg_match_all(
+            '/self::.*=> ([\'|\"].*[\'|\"]),\\n/smU',
+            $context,
+            $constants
+        ) ? $constants[1] : [];
     }
 }