Added fallback for $HOME/keys/

AOEpeople · Feb 10, 2017 · b25247c · b25247c
1 parent c49ee00
commit b25247c
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/AwsInspector/Model/Ec2/Instance.php b/src/AwsInspector/Model/Ec2/Instance.php
@@ -5,6 +5,7 @@
 use AwsInspector\Helper\Curl;
 use AwsInspector\Ssh\Connection;
 use AwsInspector\Ssh\PrivateKey;
+use Symfony\Component\Filesystem\Exception\FileNotFoundException;
 
 /**
  * Class Instance
@@ -74,7 +75,11 @@ public function getPrivateKey()
             return null;
             // throw new \Exception('No KeyName found');
         }
-        return PrivateKey::get('keys/' . $keyName . '.pem');
+        try {
+            return PrivateKey::get('keys/' . $keyName . '.pem');
+        } catch(FileNotFoundException $e) {
+            return PrivateKey::get($_SERVER['HOME'] . '/keys/' . $keyName . '.pem');
+        }
     }
 
     /**