Add support for #if macro around classes

Author: adoy

build/gen_stub.php

@@ -2245,6 +2245,8 @@ class ClassInfo {
     public $funcInfos;
     /** @var EnumCaseInfo[] */
     public $enumCaseInfos;
+    /** @var string|null */
+    public $cond;
 
     /**
      * @param Name[] $extends
@@ -2268,7 +2270,8 @@ public function __construct(
         array $constInfos,
         array $propertyInfos,
         array $funcInfos,
-        array $enumCaseInfos
+        array $enumCaseInfos,
+        ?string $cond
     ) {
         $this->name = $name;
         $this->flags = $flags;
@@ -2284,6 +2287,7 @@ public function __construct(
         $this->propertyInfos = $propertyInfos;
         $this->funcInfos = $funcInfos;
         $this->enumCaseInfos = $enumCaseInfos;
+        $this->cond = $cond;
     }
 
     /**
@@ -2301,7 +2305,13 @@ public function getRegistration(iterable $allConstInfos): string
 
         $escapedName = implode("_", $this->name->parts);
 
-        $code = "static zend_class_entry *register_class_$escapedName(" . (empty($params) ? "void" : implode(", ", $params)) . ")\n";
+        $code = '';
+
+        if ($this->cond) {
+            $code .= "#if {$this->cond}\n";
+        }
+
+        $code .= "static zend_class_entry *register_class_$escapedName(" . (empty($params) ? "void" : implode(", ", $params)) . ")\n";
 
         $code .= "{\n";
         if ($this->type == "enum") {
@@ -2362,6 +2372,10 @@ function (Name $item) {
 
         $code .= "}\n";
 
+        if ($this->cond) {
+            $code .= "#endif\n";
+        }
+
         return $code;
     }
 
@@ -3183,7 +3197,8 @@ function parseClass(
     array $consts,
     array $properties,
     array $methods,
-    array $enumCases
+    array $enumCases,
+    ?string $cond
 ): ClassInfo {
     $flags = $class instanceof Class_ ? $class->flags : 0;
     $comment = $class->getDocComment();
@@ -3243,7 +3258,8 @@ function parseClass(
         $consts,
         $properties,
         $methods,
-        $enumCases
+        $enumCases,
+        $cond
     );
 }
 
@@ -3393,7 +3409,7 @@ function handleStatements(FileInfo $fileInfo, array $stmts, PrettyPrinterAbstrac
             }
 
             $fileInfo->classInfos[] = parseClass(
-                $className, $stmt, $constInfos, $propertyInfos, $methodInfos, $enumCaseInfos
+                $className, $stmt, $constInfos, $propertyInfos, $methodInfos, $enumCaseInfos, $cond
             );
             continue;
         }
@@ -3631,7 +3647,7 @@ function (FuncInfo $funcInfo) use($fileInfo, &$generatedFunctionDeclarations) {
         }
 
         foreach ($fileInfo->classInfos as $classInfo) {
-            $code .= generateFunctionEntries($classInfo->name, $classInfo->funcInfos);
+            $code .= generateFunctionEntries($classInfo->name, $classInfo->funcInfos, $classInfo->cond);
         }
     }
 
@@ -3667,22 +3683,30 @@ function generateClassEntryCode(FileInfo $fileInfo, iterable $allConstInfos): st
 }
 
 /** @param FuncInfo[] $funcInfos */
-function generateFunctionEntries(?Name $className, array $funcInfos): string {
-    $code = "";
+function generateFunctionEntries(?Name $className, array $funcInfos, ?string $cond = null): string {
+    $code = "\n\n";
+
+    if ($cond) {
+        $code .= "#if {$cond}\n";
+    }
 
     $functionEntryName = "ext_functions";
     if ($className) {
         $underscoreName = implode("_", $className->parts);
         $functionEntryName = "class_{$underscoreName}_methods";
     }
 
-    $code .= "\n\nstatic const zend_function_entry {$functionEntryName}[] = {\n";
+    $code .= "static const zend_function_entry {$functionEntryName}[] = {\n";
     $code .= generateCodeWithConditions($funcInfos, "", function (FuncInfo $funcInfo) {
         return $funcInfo->getFunctionEntry();
     });
     $code .= "\tZEND_FE_END\n";
     $code .= "};\n";
 
+    if ($cond) {
+        $code .= "#endif\n";
+    }
+
     return $code;
 }