1 files changed, 85 insertions, 0 deletions
diff --git a/src/Exporter/DataFormatterCollection.php b/src/Exporter/DataFormatterCollection.php
new file mode 100644
index 0000000..495c5e3
--- /dev/null
+++ b/src/Exporter/DataFormatterCollection.php
@@ -0,0 +1,85 @@
+<?php
+/**
+* This file is part of the League.csv library
+*
+* @license http://opensource.org/licenses/MIT
+* @link https://github.com/thephpleague/csv/
+* @version 7.0.0
+* @package League.csv
+*
+* For the full copyright and license information, please view the LICENSE
+* file that was distributed with this source code.
+*/
+namespace League\Csv\Exporter;
+
+/**
+ *  A trait to format the row before insertion
+ *
+ * @package League.csv
+ * @since  7.0.0
+ *
+ */
+trait DataFormatterCollection
+{
+    /**
+     * Callables to format the row before insertion
+     *
+     * @var callable[]
+     */
+    protected $formatters = [];
+
+    /**
+     * add a formatter to the collection
+     *
+     * @param callable $callable
+     *
+     * @return $this
+     */
+    public function addFormatter(callable $callable)
+    {
+        $this->formatters[] = $callable;
+
+        return $this;
+    }
+
+    /**
+     * Remove a formatter from the collection
+     *
+     * @param callable $callable
+     *
+     * @return $this
+     */
+    public function removeFormatter(callable $callable)
+    {
+        $res = array_search($callable, $this->formatters, true);
+        if (false !== $res) {
+            unset($this->formatters[$res]);
+        }
+
+        return $this;
+    }
+
+    /**
+     * Detect if the formatter is already registered
+     *
+     * @param callable $callable
+     *
+     * @return bool
+     */
+    public function hasFormatter(callable $callable)
+    {
+        return false !== array_search($callable, $this->formatters, true);
+    }
+
+    /**
+     * Remove all registered formatter
+     *
+     * @return $this
+     */
+    public function clearFormatters()
+    {
+        $this->formatters = [];
+
+        return $this;
+    }
+}