Add support to generate class constants for properties - Close #13

Author: sandrokeil

src/ClassGenerator.php

@@ -12,8 +12,10 @@
 
 use OpenCodeModeling\CodeAst\Builder\ClassBuilder;
 use OpenCodeModeling\CodeAst\Builder\ClassBuilderCollection;
+use OpenCodeModeling\CodeAst\Builder\ClassConstBuilder;
 use OpenCodeModeling\CodeAst\Builder\ClassMethodBuilder;
 use OpenCodeModeling\CodeAst\Builder\ClassPropertyBuilder;
+use OpenCodeModeling\CodeAst\Code\ClassConstGenerator;
 use OpenCodeModeling\CodeAst\Package\ClassInfoList;
 use OpenCodeModeling\JsonSchemaToPhp\Type\ArrayType;
 use OpenCodeModeling\JsonSchemaToPhp\Type\ObjectType;
@@ -167,6 +169,8 @@ public function generateClasses(
     }
 
     /**
+     * Generation of getter methods for value object are skipped.
+     *
      * @param ClassBuilderCollection $classBuilderCollection
      * @param bool $typed
      * @param callable $methodNameFilter Filter the property name to your desired method name e.g. with get prefix
@@ -194,6 +198,39 @@ public function addGetterMethods(
         }
     }
 
+    /**
+     * Generation of constants for value object are skipped.
+     *
+     * @param ClassBuilderCollection $classBuilderCollection
+     * @param callable $constantNameFilter Converts the name to a proper class constant name
+     * @param callable $constantValueFilter Converts the name to a proper class constant value e.g. snake_case or camelCase
+     * @param int $visibility Visibility of the class constant
+     */
+    public function addClassConstantsForProperties(
+        ClassBuilderCollection $classBuilderCollection,
+        callable $constantNameFilter,
+        callable $constantValueFilter,
+        int $visibility = ClassConstGenerator::FLAG_PUBLIC
+    ): void {
+        foreach ($classBuilderCollection as $classBuilder) {
+            foreach ($classBuilder->getProperties() as $classPropertyBuilder) {
+                $constantName = ($constantNameFilter)($classPropertyBuilder->getName());
+
+                if ($this->isValueObject($classBuilder)
+                    || $classBuilder->hasConstant($constantName)) {
+                    continue 2;
+                }
+                $classBuilder->addConstant(
+                    ClassConstBuilder::fromScratch(
+                        $constantName,
+                        ($constantValueFilter)($classPropertyBuilder->getName()),
+                        $visibility
+                    )
+                );
+            }
+        }
+    }
+
     public function generateValueObject(string $className, string $classNamespace, TypeDefinition $definition): ClassBuilder
     {
         $classBuilder = $this->valueObjectFactory->classBuilder($definition);

tests/ClassGeneratorTest.php

@@ -117,6 +117,11 @@ public function it_generates_files(): void
         $this->assertCount(7, $classBuilderCollection);
 
         $this->classGenerator->addGetterMethods($classBuilderCollection, true, FilterFactory::methodNameFilter());
+        $this->classGenerator->addClassConstantsForProperties(
+            $classBuilderCollection,
+            FilterFactory::constantNameFilter(),
+            FilterFactory::constantValueFilter()
+        );
 
         $files = $this->classGenerator->generateFiles($classBuilderCollection, $this->parser, $this->printer);
 
@@ -310,6 +315,8 @@ private function assertOrderFile(string $code): void
 use Acme\ShippingAddresses;
 final class Order
 {
+    public const BILLING_ADDRESS = 'billing_address';
+    public const SHIPPING_ADDRESSES = 'shipping_addresses';
     private Address $billingAddress;
     private ShippingAddresses $shippingAddresses;
     public function billingAddress() : Address
@@ -467,6 +474,9 @@ private function assertAddressFile(string $code): void
 use Acme\State;
 final class Address
 {
+    public const STREET_ADDRESS = 'street_address';
+    public const CITY = 'city';
+    public const FEDERAL_STATE = 'federal_state';
     private StreetAddress $streetAddress;
     private City $city;
     private State $federalState;
@@ -500,6 +510,9 @@ private function assertBillingAddressFile(string $code): void
 use Acme\State;
 final class BillingAddress
 {
+    public const STREET_ADDRESS = 'street_address';
+    public const CITY = 'city';
+    public const FEDERAL_STATE = 'federal_state';
     private StreetAddress $streetAddress;
     private City $city;
     private State $federalState;