Draw based on the id or the instance of the poll

akiyamaSM · May 10, 2019 · 13dd799 · 13dd799
1 parent 6b73bb8
commit 13dd799
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/src/helpers/PollWriter.php b/src/helpers/PollWriter.php
@@ -18,8 +18,16 @@ class PollWriter
      * @param Poll $poll
      * @return string
      */
-    public function draw(Poll $poll)
+    public function draw($poll)
     {
+        if(is_int($poll)){
+            $poll = Poll::findOrFail($poll);
+        }
+
+        if(!$poll instanceof Poll){
+            throw new \InvalidArgumentException("The argument must be an integer or an instance of Poll");
+        }
+
         if ($poll->isComingSoon()) {
             return 'To start soon';
         }