Add refresh method

WyriHaximus · May 23, 2016 · ef515b5 · ef515b5
1 parent 4d702ca
commit ef515b5
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/src/ResourceGenerator.php b/src/ResourceGenerator.php
@@ -181,6 +181,27 @@ protected function createExtendingClass(string $type, array $yaml)
         $class = $factory->class($yaml['class'])
             ->extend('Base' . $yaml['class']);
 
+        $class->addStmt($factory->method('refresh')
+            ->makePublic()
+            ->setReturnType($yaml['class'])
+            ->addStmt(
+                new Node\Stmt\Return_(
+                    new Node\Expr\MethodCall(
+                        new Node\Expr\Variable('this'),
+                        'wait',
+                        [
+                            new Node\Expr\MethodCall(
+                                new Node\Expr\Variable('this'),
+                                'callAsync',
+                                [
+                                    new Node\Scalar\String_('refresh'),
+                                ]
+                            ),
+                        ]
+                    )
+                )
+            ));
+
         $node = $factory->namespace($yaml['namespace'] . '\\' . $type )
             ->addStmt($factory->use($yaml['namespace'] . '\\' . $yaml['class'])->as('Base' . $yaml['class']))
             ->addStmt($class)