Added custom mimetype helper

src/CustomMimetypeHelper.php

@@ -0,0 +1,51 @@
+<?php
+
+namespace Http\Message\MultipartStream;
+
+/**
+ * Let you add your own mimetypes. The mimetype lookup will fallback on the ApacheMimeTypeHelper.
+ *
+ * @author Tobias Nyholm <[email protected]>
+ */
+class CustomMimetypeHelper extends ApacheMimetypeHelper
+{
+    /**
+     * @var array
+     */
+    private $mimetypes = [];
+
+    /**
+     * @param array $mimetypes should be of type extension => mimetype
+     */
+    public function __construct(array $mimetypes = [])
+    {
+        $this->mimetypes = $mimetypes;
+    }
+
+    /**
+     * @param string $extension
+     * @param string $mimetype
+     *
+     * @return $this
+     */
+    public function addMimetype($extension, $mimetype)
+    {
+        $this->mimetypes[$extension] = $mimetype;
+
+        return $this;
+    }
+
+    /**
+     * {@inheritdoc}
+     *
+     * Check if we have any defined mimetypes and of not fallback to ApacheMimetypeHelper
+     */
+    public function getMimetypeFromExtension($extension)
+    {
+        $extension = strtolower($extension);
+
+        return isset($this->mimetypes[$extension])
+            ? $this->mimetypes[$extension]
+            : parent::getMimetypeFromExtension($extension);
+    }
+}

tests/CustomMimetypeHelperTest.php

@@ -0,0 +1,18 @@
+<?php
+
+namespace tests\Http\Message\MultipartStream;
+
+use Http\Message\MultipartStream\CustomMimetypeHelper;
+
+class CustomMimetypeHelperTest extends \PHPUnit_Framework_TestCase
+{
+    public function testGetMimetypeFromExtension()
+    {
+        $helper = new CustomMimetypeHelper(['foo'=>'foo/bar']);
+        $this->assertEquals('foo/bar', $helper->getMimetypeFromExtension('foo'));
+
+        $this->assertEquals('application/x-rar-compressed', $helper->getMimetypeFromExtension('rar'));
+        $helper->addMimetype('rar', 'test/test');
+        $this->assertEquals('test/test', $helper->getMimetypeFromExtension('rar'));
+    }
+}