Getters that would have returned ResourceContainer(s) now attempt to

return the contained object(s).
dataground · Aug 10, 2017 · df89fd0 · df89fd0
1 parent 4a0ffee
commit df89fd0
Showing 1 changed file with 24 additions and 1 deletion.
diff --git a/src/ClassGenerator/Generator/MethodGenerator.php b/src/ClassGenerator/Generator/MethodGenerator.php
@@ -64,7 +64,30 @@ public static function implementConstructor(Config $config, ClassTemplate $class
      */
     public static function implementMethodsForProperty(Config $config, ClassTemplate $classTemplate, BasePropertyTemplate $propertyTemplate) {
         if ($propertyTemplate->requiresGetter()) {
-            $classTemplate->addMethod(self::createGetter($config, $propertyTemplate));
+            if ($propertyTemplate->getFHIRElementType() === 'ResourceContainer') {
+                $method = new BaseMethodTemplate($config, sprintf('get%s', NameUtils::getPropertyMethodName($propertyTemplate->getName())));
+                $method->setDocumentation($propertyTemplate->getDocumentation());
+                if ($propertyTemplate->isCollection()) {
+                    $classTemplate->addImport(NSUtils::generateRootNamespace($config, 'FHIRResourceContainer'));
+                    $method->addLineToBody("if (count(\$this->{$propertyTemplate->getName()}) > 0) { ");
+                    $method->addLineToBody('    $resources = [];');
+                    $method->addLineToBody("    foreach(\$this->{$propertyTemplate->getName()} as \$container) {");
+                    $method->addLineToBody('        if ($container instanceof FHIRResourceContainer) {');
+                    $method->addLineToBody('            $resources[] = $container->jsonSerialize();');
+                    $method->addLineToBody('        }');
+                    $method->addLineToBody('    }');
+                    $method->addLineToBody('    return $resources;');
+                    $method->addLineToBody('}');
+                    $method->addLineToBody('return [];');
+                    $method->setReturnValueType('array');
+                } else {
+                    $method->addLineToBody("return isset(\$this->{$propertyTemplate->getName()}) ? \$this->{$propertyTemplate->getName()}->jsonSerialize() : null;");
+                    $method->setReturnValueType('mixed');
+                }
+                $classTemplate->addMethod($method);
+            } else {
+                $classTemplate->addMethod(self::createGetter($config, $propertyTemplate));
+            }
         }
 
         if ($propertyTemplate->requireSetter()) {