Allow array as argument to repairDetails()

1 files changed, 8 insertions, 5 deletions

diff --git a/gsxlib.php b/gsxlib.php
index f405fed..dfea4e3 100644
--- a/gsxlib.php
+++ b/gsxlib.php
@@ -375,13 +375,16 @@ class GsxLib
     */
     public function repairDetails($dispatchId)
     {
-        $dispatchId = trim($dispatchId);
-
-        if( !self::looksLike( $dispatchId, 'dispatchId' )) {
-            exit( 'Invalid dispatch ID: ' . $dispatchId );
+        if (is_string($dispatchId)) {
+            $dispatchId = trim($dispatchId);
+            if( !self::looksLike( $dispatchId, 'dispatchId' )) {
+                $error = sprintf('Invalid dispatch ID: %s', $dispatchId);
+                throw new InvalidArgumentException($error);
+            }
+            $dispatchId = array('dispatchId' => $dispatchId);
         }
 
-        $req = array('RepairDetails' => array('dispatchId' => $dispatchId));
+        $req = array('RepairDetails' => $dispatchId);
         return $this->request($req)->lookupResponseData;
 
     }