Added categories to iTunes

src/Entities/Itunes/AbstractChannel.php

@@ -2,6 +2,7 @@
 
 namespace Lukaswhite\FeedWriter\Entities\Itunes;
 
+use Lukaswhite\FeedWriter\Traits\Itunes\HasCategories;
 use Lukaswhite\FeedWriter\Traits\Itunes\HasAuthor;
 use Lukaswhite\FeedWriter\Traits\Itunes\HasBlock;
 use Lukaswhite\FeedWriter\Traits\Itunes\HasExplicit;
@@ -19,6 +20,7 @@ abstract class AbstractChannel extends \Lukaswhite\FeedWriter\Entities\Rss\Chann
     use HasSubtitle,
         HasSummary,
         HasAuthor,
+        HasCategories,
         HasImage,
         HasExplicit,
         HasBlock;

src/Entities/Itunes/Category.php

@@ -0,0 +1,47 @@
+<?php
+
+namespace Lukaswhite\FeedWriter\Entities\Itunes;
+
+use Lukaswhite\FeedWriter\Traits\Itunes\HasCategories;
+
+/**
+ * Class Category
+ *
+ * @package Lukaswhite\FeedWriter\Entities\Media
+ */
+class Category extends \Lukaswhite\FeedWriter\Entities\Atom\Category
+{
+    use HasCategories;
+    /**
+     * @var string
+     */
+    protected $text;
+
+    /**
+     * Create a DOM element that represents this entity.
+     *
+     * @return \DOMElement
+     */
+    public function element( ) : \DOMElement
+    {
+        $category = $this->createElement( 'itunes:category' );
+
+        if ( $this->text ) {
+            $category->setAttribute( 'text', $this->text );
+        }
+
+        $this->addCategoryElements( $category );
+
+        return $category;
+    }
+
+    /**
+     * @param string $text
+     * @return Category
+     */
+    public function text( string $text ) : self
+    {
+        $this->text = $text;
+        return $this;
+    }
+}

src/Entities/Itunes/Channel.php

@@ -2,6 +2,7 @@
 
 namespace Lukaswhite\FeedWriter\Entities\Itunes;
 
+
 /**
  * Class Channel
  *
@@ -21,4 +22,16 @@ public function addItem( ) : Item
         $this->items[ ] = $item;
         return $item;
     }
+
+    /**
+     * Add an item
+     *
+     * @return Item
+     */
+    public function addCategory( ) : Category
+    {
+        $category = new Category( $this->feed );
+        $this->categories[ ] = $category;
+        return $category;
+    }
 }

src/Traits/Itunes/HasCategories.php

@@ -0,0 +1,49 @@
+<?php
+
+namespace Lukaswhite\FeedWriter\Traits\Itunes;
+
+use Lukaswhite\FeedWriter\Entities\Itunes\Category;
+
+
+/**
+ * Trait HasCategories
+ *
+ * @package Lukaswhite\FeedWriter\Traits\Itunes
+ */
+trait HasCategories
+{
+    /**
+     * The categories for this entry / channel
+     *
+     * @var array
+     */
+    protected $categories = [ ];
+
+    /**
+     * Add a contributor
+     *
+     * @return Category
+     */
+    public function addCategory( )
+    {
+        $category = ( new Category( $this->feed ) );
+        $this->categories[ ] = $category;
+        return $category;
+    }
+
+    /**
+    * Add the authors to the specified element.
+    *
+    * @param \DOMElement $el
+    * @return void
+    */
+    protected function addCategoryElements( \DOMElement $el ) : void
+    {
+        if ( count( $this->categories ) ) {
+            foreach( $this->categories as $category ) {
+                /** @var Category $category */
+                $el->appendChild( $category->element( ) );
+            }
+        }
+    }
+}