[Yaml] add option to dump objects as maps

xabbuh · xabbuh · commit 80f0b6606c4d · 2016-02-15T17:26:39.000+01:00
diff --git a/src/Symfony/Component/Yaml/Inline.php b/src/Symfony/Component/Yaml/Inline.php
@@ -155,6 +155,10 @@ public static function dump($value, $flags = 0)
                     return '!php/object:'.serialize($value);
                 }
 
+                if (Yaml::DUMP_OBJECT_AS_MAP & $flags && ($value instanceof \stdClass || $value instanceof \ArrayObject)) {
+                    return self::dumpArray((array) $value, $flags);
+                }
+
                 if (Yaml::DUMP_EXCEPTION_ON_INVALID_TYPE & $flags) {
                     throw new DumpException('Object support when dumping a YAML file has been disabled.');
                 }
diff --git a/src/Symfony/Component/Yaml/Tests/DumperTest.php b/src/Symfony/Component/Yaml/Tests/DumperTest.php
@@ -276,6 +276,46 @@ public function getEscapeSequences()
             'paragraph-separator' => array("\t\\P", '"\t\\\\P"'),
         );
     }
+
+    /**
+     * @dataProvider objectAsMapProvider
+     */
+    public function testDumpObjectAsMap($object, $expected)
+    {
+
+        $yaml = $this->dumper->dump($object, 0, 0, Yaml::DUMP_OBJECT_AS_MAP);
+
+        $this->assertEquals($expected, Yaml::parse($yaml, Yaml::PARSE_OBJECT_FOR_MAP));
+    }
+
+    public function objectAsMapProvider()
+    {
+        $tests = array();
+
+        $bar = new \stdClass();
+        $bar->class = 'classBar';
+        $bar->args = array('bar');
+        $zar = new \stdClass();
+        $foo = new \stdClass();
+        $foo->bar = $bar;
+        $foo->zar = $zar;
+        $object = new \stdClass();
+        $object->foo = $foo;
+        $tests['stdClass'] = array($object, $object);
+
+        $arrayObject = new \ArrayObject();
+        $arrayObject['foo'] = 'bar';
+        $arrayObject['baz'] = 'foobar';
+        $parsedArrayObject = new \stdClass();
+        $parsedArrayObject->foo = 'bar';
+        $parsedArrayObject->baz = 'foobar';
+        $tests['ArrayObject'] = array($arrayObject, $parsedArrayObject);
+
+        $a = new A();
+        $tests['arbitrary-object'] = array($a, null);
+
+        return $tests;
+    }
 }
 
 class A
diff --git a/src/Symfony/Component/Yaml/Yaml.php b/src/Symfony/Component/Yaml/Yaml.php
@@ -25,6 +25,7 @@ class Yaml
     const PARSE_OBJECT = 4;
     const PARSE_OBJECT_FOR_MAP = 8;
     const DUMP_EXCEPTION_ON_INVALID_TYPE = 16;
+    const DUMP_OBJECT_AS_MAP = 32;
 
     /**
      * Parses YAML into a PHP value.

Original file line number	Diff line number	Diff line change
`@@ -155,6 +155,10 @@ public static function dump($value, $flags = 0)`
`155`	`155`	`return '!php/object:'.serialize($value);`
`156`	`156`	`}`
`157`	`157`
	`158`	`+ if (Yaml::DUMP_OBJECT_AS_MAP & $flags && ($value instanceof \stdClass \|\| $value instanceof \ArrayObject)) {`
	`159`	`+ return self::dumpArray((array) $value, $flags);`
	`160`	`+ }`
	`161`	`+`
`158`	`162`	`if (Yaml::DUMP_EXCEPTION_ON_INVALID_TYPE & $flags) {`
`159`	`163`	`throw new DumpException('Object support when dumping a YAML file has been disabled.');`
`160`	`164`	`}`