From 315d39ae68d2397aec8e125368f5d8aed4d11bf5 Mon Sep 17 00:00:00 2001
From: Aydin Hassan <aydin@hotmail.co.uk>
Date: Tue, 22 Jun 2021 18:04:03 +0100
Subject: [PATCH] Wrap in try/catch patch

---
 src/Patch/WrapInTryCatch.php      | 58 +++++++++++++++++++++++++++++++
 test/Patch/WrapInTryCatchTest.php | 55 +++++++++++++++++++++++++++++
 2 files changed, 113 insertions(+)
 create mode 100644 src/Patch/WrapInTryCatch.php
 create mode 100644 test/Patch/WrapInTryCatchTest.php

diff --git a/src/Patch/WrapInTryCatch.php b/src/Patch/WrapInTryCatch.php
new file mode 100644
index 00000000..c3d86428
--- /dev/null
+++ b/src/Patch/WrapInTryCatch.php
@@ -0,0 +1,58 @@
+<?php
+
+namespace PhpSchool\PhpWorkshop\Patch;
+
+use PhpParser\Node\Stmt;
+use PhpParser\Node\Expr\MethodCall;
+use PhpParser\Node\Expr\Variable;
+use PhpParser\Node\Name;
+use PhpParser\Node\Stmt\Catch_;
+use PhpParser\Node\Stmt\Echo_;
+use PhpParser\Node\Stmt\TryCatch;
+
+class WrapInTryCatch implements Transformer
+{
+    /**
+     * @var string
+     */
+    private $exceptionClass;
+
+    /**
+     * @var array<Stmt>
+     */
+    private $statements;
+
+    /**
+     * @param string $exceptionClass
+     * @param array<Stmt>|null $statements
+     */
+    public function __construct(string $exceptionClass = \Exception::class, array $statements = null)
+    {
+        $this->exceptionClass = $exceptionClass;
+        $this->statements = $statements ?: [
+            new Echo_([
+                new MethodCall(new Variable('e'), 'getMessage')
+            ])
+        ];
+    }
+
+    /**
+     * @param array<Stmt> $statements
+     * @return array<Stmt>
+     */
+    public function transform(array $statements): array
+    {
+        return [
+            new TryCatch(
+                $statements,
+                [
+                    new Catch_(
+                        [new Name($this->exceptionClass)],
+                        new Variable('e'),
+                        $this->statements
+                    )
+                ]
+            )
+        ];
+    }
+}
diff --git a/test/Patch/WrapInTryCatchTest.php b/test/Patch/WrapInTryCatchTest.php
new file mode 100644
index 00000000..a25773d2
--- /dev/null
+++ b/test/Patch/WrapInTryCatchTest.php
@@ -0,0 +1,55 @@
+<?php
+
+namespace PhpSchool\PhpWorkshopTest\Patch;
+
+use PhpParser\Node\Scalar\String_;
+use PhpParser\Node\Stmt\Echo_;
+use PhpParser\ParserFactory;
+use PhpParser\PrettyPrinter\Standard;
+use PhpSchool\PhpWorkshop\Patch\WrapInTryCatch;
+use PHPUnit\Framework\TestCase;
+
+class WrapInTryCatchTest extends TestCase
+{
+    public function testStatementsAreWrappedInTryCatch(): void
+    {
+        $parser = (new ParserFactory())->create(ParserFactory::PREFER_PHP7);
+        $ast = $parser->parse("<?php echo 'Hello World';");
+
+        $transformer = new WrapInTryCatch();
+        $ast = $transformer->transform($ast);
+
+        self::assertSame(
+            "try {\n    echo 'Hello World';\n} catch (Exception \$e) {\n    echo \$e->getMessage();\n}",
+            (new Standard())->prettyPrint($ast)
+        );
+    }
+
+    public function testStatementsAreWrappedInTryCatchWithCustomExceptionClass(): void
+    {
+        $parser = (new ParserFactory())->create(ParserFactory::PREFER_PHP7);
+        $ast = $parser->parse("<?php echo 'Hello World';");
+
+        $transformer = new WrapInTryCatch(\RuntimeException::class);
+        $ast = $transformer->transform($ast);
+
+        self::assertSame(
+            "try {\n    echo 'Hello World';\n} catch (RuntimeException \$e) {\n    echo \$e->getMessage();\n}",
+            (new Standard())->prettyPrint($ast)
+        );
+    }
+
+    public function testStatementsAreWrappedInTryCatchWithStatements(): void
+    {
+        $parser = (new ParserFactory())->create(ParserFactory::PREFER_PHP7);
+        $ast = $parser->parse("<?php echo 'Hello World';");
+
+        $transformer = new WrapInTryCatch(\RuntimeException::class, [new Echo_([new String_('You caught me!')])]);
+        $ast = $transformer->transform($ast);
+
+        self::assertSame(
+            "try {\n    echo 'Hello World';\n} catch (RuntimeException \$e) {\n    echo 'You caught me!';\n}",
+            (new Standard())->prettyPrint($ast)
+        );
+    }
+}