Allow adding an array of Shortcodes

JayBizzle · Jan 12, 2019 · ca986a8 · ca986a8
1 parent 9e8da78
commit ca986a8
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 4 deletions.
diff --git a/src/Shortcodes.php b/src/Shortcodes.php
@@ -14,12 +14,17 @@ class Shortcodes
     /**
      * Add shortcode hooks.
      *
-     * @param string $tag
-     * @param string $class
+     * @param array|string $classes
      */
-    public function add($class)
+    public function add($classes)
     {
-        $this->shortcodeTags[$class::$shortcode] = $class;
+        if (! is_array($classes)) {
+            $classes = [$classes];
+        }
+
+        foreach ($classes as $class) {
+            $this->shortcodeTags[$class::$shortcode] = $class;
+        }
     }
 
     /**

diff --git a/tests/ParseTest.php b/tests/ParseTest.php
@@ -20,6 +20,25 @@ public function no_tags_added()
         );
     }
 
+    /** @test */
+    public function we_can_add_a_single_shortcode()
+    {
+        $this->shortcodes->add(FooShortcode::class);
+
+        $this->assertCount(1, $this->shortcodes->shortcodeTags);
+    }
+
+    /** @test */
+    public function we_can_add_multiple_shorcodes()
+    {
+        $this->shortcodes->add([
+            FooShortcode::class,
+            BarShortcode::class,
+        ]);
+
+        $this->assertCount(2, $this->shortcodes->shortcodeTags);
+    }
+
     /** @test */
     public function tags_with_closing_slash()
     {