Implement ReflectionEnum::getCases()

Author: iluuu1994

ext/reflection/php_reflection.c

@@ -1432,6 +1432,21 @@ static void reflection_class_constant_factory(zend_string *name_str, zend_class_
 }
 /* }}} */
 
+static void reflection_enum_case_factory(zend_string *name_str, zend_class_constant *constant, zval *object)
+{
+	reflection_object *intern;
+
+	reflection_instantiate(reflection_enum_case_ptr, object);
+	intern = Z_REFLECTION_P(object);
+	intern->ptr = constant;
+	intern->ref_type = REF_TYPE_CLASS_CONSTANT;
+	intern->ce = constant->ce;
+	intern->ignore_visibility = 0;
+
+	ZVAL_STR_COPY(reflection_prop_name(object), name_str);
+	ZVAL_STR_COPY(reflection_prop_class(object), constant->ce->name);
+}
+
 static int get_parameter_default(zval *result, parameter_reference *param) {
 	if (param->fptr->type == ZEND_INTERNAL_FUNCTION) {
 		if (param->fptr->common.fn_flags & ZEND_ACC_USER_ARG_INFO) {
@@ -6495,6 +6510,29 @@ ZEND_METHOD(ReflectionEnum, hasCase)
 	RETURN_BOOL(class_const->attr & ZEND_CLASS_CONST_IS_CASE);
 }
 
+ZEND_METHOD(ReflectionEnum, getCases)
+{
+	reflection_object *intern;
+	zend_class_entry *ce;
+	zend_string *name;
+	zend_class_constant *constant;
+
+	if (zend_parse_parameters_none() == FAILURE) {
+		RETURN_THROWS();
+	}
+
+	GET_REFLECTION_OBJECT_PTR(ce);
+
+	array_init(return_value);
+	ZEND_HASH_FOREACH_STR_KEY_PTR(&ce->constants_table, name, constant) {
+		if (constant->attr & ZEND_CLASS_CONST_IS_CASE) {
+			zval class_const;
+			reflection_enum_case_factory(name, constant, &class_const);
+			zend_hash_next_index_insert(Z_ARRVAL_P(return_value), &class_const);
+		}
+	} ZEND_HASH_FOREACH_END();
+}
+
 ZEND_METHOD(ReflectionEnum, hasPrimitiveType)
 {
 	reflection_object *intern;
@@ -6712,7 +6750,7 @@ PHP_MINIT_FUNCTION(reflection) /* {{{ */
 	INIT_CLASS_ENTRY(_reflection_entry, "ReflectionEnumCase", class_ReflectionEnumCase_methods);
 	reflection_init_class_handlers(&_reflection_entry);
 	_reflection_entry.ce_flags |= ZEND_ACC_FINAL;
-	reflection_enum_ptr = zend_register_internal_class_ex(&_reflection_entry, reflection_class_constant_ptr);
+	reflection_enum_case_ptr = zend_register_internal_class_ex(&_reflection_entry, reflection_class_constant_ptr);
 
 	REGISTER_REFLECTION_CLASS_CONST_LONG(attribute, "IS_INSTANCEOF", REFLECTION_ATTRIBUTE_IS_INSTANCEOF);
 

ext/reflection/php_reflection.stub.php

@@ -675,7 +675,10 @@ private function __construct() {}
 final class ReflectionEnum extends ReflectionClass
 {
     /** @return bool */
-    public function hasCase(string $name);
+    public function hasCase(string $name) {}
+
+    /** @return ReflectionEnumCase[] */
+    public function getCases() {}
 
     /** @return bool */
     public function hasPrimitiveType() {}

ext/reflection/php_reflection_arginfo.h

@@ -1,5 +1,5 @@
 /* This is a generated file, edit the .stub.php file instead.
- * Stub hash: 07fee3b12c632e12a377bac90ee662a752e9e65e */
+ * Stub hash: 48c4f2eaad51d8a07694748e281b93e6309fe205 */
 
 ZEND_BEGIN_ARG_INFO_EX(arginfo_class_Reflection_getModifierNames, 0, 0, 1)
 	ZEND_ARG_TYPE_INFO(0, modifiers, IS_LONG, 0)
@@ -495,6 +495,8 @@ ZEND_END_ARG_INFO()
 
 #define arginfo_class_ReflectionEnum_hasCase arginfo_class_ReflectionClass_hasMethod
 
+#define arginfo_class_ReflectionEnum_getCases arginfo_class_ReflectionFunctionAbstract_inNamespace
+
 #define arginfo_class_ReflectionEnum_hasPrimitiveType arginfo_class_ReflectionFunctionAbstract_inNamespace
 
 #define arginfo_class_ReflectionEnum_getPrimitiveType arginfo_class_ReflectionFunctionAbstract_inNamespace
@@ -705,6 +707,7 @@ ZEND_METHOD(ReflectionAttribute, newInstance);
 ZEND_METHOD(ReflectionAttribute, __clone);
 ZEND_METHOD(ReflectionAttribute, __construct);
 ZEND_METHOD(ReflectionEnum, hasCase);
+ZEND_METHOD(ReflectionEnum, getCases);
 ZEND_METHOD(ReflectionEnum, hasPrimitiveType);
 ZEND_METHOD(ReflectionEnum, getPrimitiveType);
 ZEND_METHOD(ReflectionEnumCase, getScalar);
@@ -1018,6 +1021,7 @@ static const zend_function_entry class_ReflectionAttribute_methods[] = {
 
 static const zend_function_entry class_ReflectionEnum_methods[] = {
 	ZEND_ME(ReflectionEnum, hasCase, arginfo_class_ReflectionEnum_hasCase, ZEND_ACC_PUBLIC)
+	ZEND_ME(ReflectionEnum, getCases, arginfo_class_ReflectionEnum_getCases, ZEND_ACC_PUBLIC)
 	ZEND_ME(ReflectionEnum, hasPrimitiveType, arginfo_class_ReflectionEnum_hasPrimitiveType, ZEND_ACC_PUBLIC)
 	ZEND_ME(ReflectionEnum, getPrimitiveType, arginfo_class_ReflectionEnum_getPrimitiveType, ZEND_ACC_PUBLIC)
 	ZEND_FE_END

ext/reflection/tests/ReflectionEnum_getCases.phpt

@@ -0,0 +1,31 @@
+--TEST--
+ReflectionEnum::getCases()
+--FILE--
+<?php
+
+enum Foo {
+    case Bar;
+    case Baz;
+    const Qux = self::Bar;
+}
+
+var_dump((new ReflectionEnum(Foo::class))->getCases());
+
+?>
+--EXPECT--
+array(2) {
+  [0]=>
+  object(ReflectionEnumCase)#2 (2) {
+    ["name"]=>
+    string(3) "Bar"
+    ["class"]=>
+    string(3) "Foo"
+  }
+  [1]=>
+  object(ReflectionEnumCase)#3 (2) {
+    ["name"]=>
+    string(3) "Baz"
+    ["class"]=>
+    string(3) "Foo"
+  }
+}