Added ability to rewrite post type URLs

Author: azeemhassni

src/Raskoh/PostType.php

@@ -9,8 +9,6 @@
 
 class PostType
 {
-
-
     /**
      * @var
      */
@@ -99,6 +97,13 @@ class PostType
      * @var Taxonomy
      */
     private $taxonomies = array();
+
+    /**
+     * @var array
+     */
+    private $rewrite = array();
+
+
     private $priority;
 
 
@@ -151,6 +156,11 @@ public function hookInWordPress()
             $args[ 'menu_icon' ] = $this->icon;
         }
 
+        if (isset( $this->rewrite[ 'slug' ] )) {
+            $args[ 'rewrite' ] = $this->rewrite;
+        }
+
+
         register_post_type($this->slug, $args);
         flush_rewrite_rules();
     }
@@ -609,5 +619,17 @@ public function setPriority( $priority )
         return $this;
     }
 
+
+    /**
+     * Rewrite post type URL
+     *
+     * @param $slug
+     * @return $this
+     */
+    public function rewrite( $slug )
+    {
+        $this->rewrite[ 'slug' ] = $slug;
+        return $this;
+    }
+
 }
-