also register custom formats with the FormatNegotiator

Author: lsmith77

EventListener/MimeTypeListener.php

@@ -11,6 +11,8 @@
 
 namespace FOS\RestBundle\EventListener;
 
+use FOS\RestBundle\Util\FormatNegotiator;
+
 use Symfony\Component\HttpKernel\Event\GetResponseEvent;
 use Symfony\Component\HttpKernel\HttpKernelInterface;
 
@@ -26,14 +28,19 @@ class MimeTypeListener
      */
     private $mimeTypes;
 
+    /**
+     * @var FormatNegotiator
+     */
+    private $formatNegotiator;
     /**
      * Constructor.
      *
      * @param array $mimeTypes key format, value mime type
      */
-    public function __construct(array $mimeTypes)
+    public function __construct(array $mimeTypes, FormatNegotiator $formatNegotiator)
     {
         $this->mimeTypes = $mimeTypes;
+        $this->formatNegotiator = $formatNegotiator;
     }
 
     /**
@@ -48,6 +55,7 @@ public function onKernelRequest(GetResponseEvent $event)
         if (HttpKernelInterface::MASTER_REQUEST === $event->getRequestType()) {
             foreach ($this->mimeTypes as $format => $mimeType) {
                 $request->setFormat($format, $mimeType);
+                $this->formatNegotiator->registerFormat($format, (array) $mimeType, true);
             }
         }
     }

Resources/config/mime_type_listener.xml

@@ -9,6 +9,7 @@
         <service id="fos_rest.mime_type_listener" class="FOS\RestBundle\EventListener\MimeTypeListener">
             <tag name="kernel.event_listener" event="kernel.request" method="onKernelRequest" priority="200" />
             <argument>%fos_rest.mime_types%</argument>
+            <argument type="service" id="fos_rest.format_negotiator" />
         </service>
 
     </services>

Util/FormatNegotiator.php

@@ -54,4 +54,29 @@ public function getBestFormat(Request $request, array $priorities, $preferExtens
 
         return $this->formatNegotiator->getBestFormat($acceptHeader, $priorities);
     }
+
+    /**
+     * Register a new format with its mime types.
+     *
+     * @param string  $format
+     * @param array   $mimeTypes
+     * @param boolean $override
+     */
+    public function registerFormat($format, array $mimeTypes, $override = false)
+    {
+        $this->formatNegotiator->registerFormat($format, $mimeTypes, $override);
+    }
+
+    /**
+     * Returns the format for a given mime type, or null
+     * if not found.
+     *
+     * @param string $mimeType
+     *
+     * @return string|null
+     */
+    public function getFormat($mimeType)
+    {
+        $this->formatNegotiator->getFormat($mimeType);
+    }
 }