Improve AutoDecoder

Author: ganlvtech

src/AutoDecoder.php

@@ -30,9 +30,9 @@
 use Ganlv\EnphpDecoder\NodeVisitors\GlobalStringNodeVisitor;
 use Ganlv\EnphpDecoder\NodeVisitors\RemoveDefineGlobalVariableNameNodeVisitor;
 use Ganlv\EnphpDecoder\NodeVisitors\RemoveUnusedConstFetchNodeVisitor;
+use Ganlv\EnphpDecoder\PrettyPrinter\StandardPrettyPrinter;
 use PhpParser\NodeTraverser;
 use PhpParser\ParserFactory;
-use PhpParser\PrettyPrinter\Standard;
 
 class AutoDecoder
 {
@@ -53,13 +53,6 @@ public function __construct($ast)
         $this->ast = $ast;
     }
 
-    public static function parseFile($code)
-    {
-        $parser = (new ParserFactory())->create(ParserFactory::PREFER_PHP5);
-        $ast = $parser->parse($code);
-        return $ast;
-    }
-
     public function findAndRemoveGlobalVariableName()
     {
         $nodeVisitor = new FindAndRemoveGlobalVariableNameNodeVisitor();
@@ -165,27 +158,48 @@ public function beautify()
 
     public function prettyPrintFile()
     {
-        $prettyPrinter = new Standard();
-        return $prettyPrinter->prettyPrintFile($this->ast);
+        return StandardPrettyPrinter::prettyPrinter()->prettyPrintFile($this->ast);
     }
 
-    public static function decode($code)
+    /**
+     * @return bool is ast modified
+     */
+    public function autoDecode()
     {
-        $ast = self::parseFile($code);
-        $decoder = new self($ast);
+        $modified = false;
         for ($i = 0; $i < 10; $i++) { // avoid too many loops
-            $decoder->findAndRemoveGlobalVariableName();
-            if ($decoder->dataType === 0) {
+            $this->findAndRemoveGlobalVariableName();
+            if ($this->dataType === 0) {
                 break;
             }
-            $decoder->decodeStringArray();
-            $decoder->removeDefineGlobalVariableName();
-            $decoder->removeUnusedConstFetchNodeVisitor();
-            $decoder->replaceGlobalString();
-            $decoder->replaceFunctionLikeGlobalString();
-            $decoder->renameFunctionLikeLocalVariable();
-            $decoder->beautify();
+            $modified = true;
+            $this->decodeStringArray();
+            $this->removeDefineGlobalVariableName();
+            $this->removeUnusedConstFetchNodeVisitor();
+            $this->replaceGlobalString();
+            $this->replaceFunctionLikeGlobalString();
+            $this->renameFunctionLikeLocalVariable();
+            $this->beautify();
+        }
+        return $modified;
+    }
+
+    public static function parseFile($code)
+    {
+        $parser = (new ParserFactory())->create(ParserFactory::PREFER_PHP5);
+        $ast = $parser->parse($code);
+        return $ast;
+    }
+
+    public static function decode($code)
+    {
+        $ast = self::parseFile($code);
+        $decoder = new self($ast);
+        $modified = $decoder->autoDecode();
+        if ($modified) {
+            return $decoder->prettyPrintFile();
+        } else {
+            return $code;
         }
-        return $decoder->prettyPrintFile();
     }
 }