Generate a new SSH key pair if one was not alredy configured (#1).

Author: tintoy

driver.go

@@ -263,11 +263,19 @@ func (driver *Driver) PreCreateCheck() error {
 
 // Create a new Docker Machine instance on CloudControl.
 func (driver *Driver) Create() error {
-	log.Infof("Importing SSH key...")
-
-	err := driver.importSSHKey()
-	if err != nil {
-		return err
+	var err error
+	if driver.SSHKey != "" {
+		log.Infof("Importing SSH key '%s'...", driver.SSHKey)
+		err = driver.importSSHKey()
+		if err != nil {
+			return err
+		}
+	} else {
+		log.Infof("Generating new SSH key...")
+		err = driver.generateSSHKey()
+		if err != nil {
+			return err
+		}
 	}
 
 	log.Infof("Creating server '%s'...", driver.MachineName)

ssh_bootstrap.go

@@ -106,6 +106,23 @@ func (driver *Driver) installSSHKey() error {
 	return nil
 }
 
+// Generate an SSH key pair, and save it into the machine store folder.
+func (driver *Driver) generateSSHKey() error {
+	if driver.SSHKeyPath != "" {
+		return errors.New("SSH key path already configured")
+	}
+
+	driver.SSHKeyPath = driver.ResolveStorePath("id_rsa")
+	err := ssh.GenerateSSHKey(driver.SSHKeyPath)
+	if err != nil {
+		log.Errorf("Failed to generate SSH key pair: %s", err.Error())
+
+		return err
+	}
+
+	return nil
+}
+
 // Import the configured SSH key files into the machine store folder.
 func (driver *Driver) importSSHKey() error {
 	if driver.SSHKey == "" {
@@ -117,14 +134,14 @@ func (driver *Driver) importSSHKey() error {
 	)
 	err := copySSHKey(driver.SSHKey, driver.SSHKeyPath)
 	if err != nil {
-		log.Infof("Couldn't copy SSH private key : %s", err.Error())
+		log.Infof("Couldn't copy SSH private key: %s", err.Error())
 
 		return err
 	}
 
 	err = copySSHKey(driver.SSHKey+".pub", driver.SSHKeyPath+".pub")
 	if err != nil {
-		log.Infof("Couldn't copy SSH public key : %s", err.Error())
+		log.Infof("Couldn't copy SSH public key: %s", err.Error())
 
 		return err
 	}