added exception and subBusinessDays method

andrejsstepanovs · Jan 23, 2018 · 487671b · 487671b
1 parent 0a66655
commit 487671b
Showing 1 changed file with 27 additions and 0 deletions.
diff --git a/src/BusinessDays/Calculator.php b/src/BusinessDays/Calculator.php
@@ -120,6 +120,10 @@ private function getFreeWeekDays()
      */
     public function addBusinessDays($howManyDays)
     {
+        if ($howManyDays < 1) {
+            throw new \InvalidArgumentException('The parameter $howManyDays must be greater than 0');
+        }
+
         $iterator = 0;
         while ($iterator < $howManyDays) {
             $this->getDate()->modify('+1 day');
@@ -131,6 +135,29 @@ public function addBusinessDays($howManyDays)
         return $this;
     }
 
+    /**
+     * @param int $howManyDays
+     *
+     * @return $this
+     */
+    public function subBusinessDays($howManyDays)
+    {
+        if ($howManyDays < 1) {
+            throw new \InvalidArgumentException('The parameter $howManyDays must be greater than 0');
+        }
+
+        $iterator = 0;
+        while ($iterator < $howManyDays) {
+            if ($this->isBusinessDay($this->getDate())) {
+                $iterator++;
+            }
+            if ($iterator < $howManyDays) { //Don`t modify the date if we are on the last iteration
+                $this->getDate()->modify('-1 day');
+            }
+        }
+        return $this;
+    }
+
     /**
      * @return \DateTime
      */