Merge pull request #50 from am97/fix/issue49

Search for an existing key by fingerprint before attempting upload

Author: JonasProgrammer

driver.go

@@ -241,14 +241,21 @@ func (d *Driver) Create() error {
 			return errors.Wrap(err, "could not read ssh public key")
 		}
 
-		keyopts := hcloud.SSHKeyCreateOpts{
-			Name:      d.GetMachineName(),
-			PublicKey: string(buf),
-		}
+		key, err := d.getRemoteKeyWithSameFingerprint(buf)
+		if key == nil {
+			log.Infof("SSH key not found in Hetzner. Uploading...")
 
-		key, _, err := d.getClient().SSHKey.Create(context.Background(), keyopts)
-		if err != nil {
-			return errors.Wrap(err, "could not create ssh key")
+			keyopts := hcloud.SSHKeyCreateOpts{
+				Name:      d.GetMachineName(),
+				PublicKey: string(buf),
+			}
+
+			key, _, err = d.getClient().SSHKey.Create(context.Background(), keyopts)
+			if err != nil {
+				return errors.Wrap(err, "could not create ssh key")
+			}
+		} else {
+			log.Debugf("SSH key found in Hetzner. ID: %d", key.ID)
 		}
 
 		d.KeyID = key.ID
@@ -606,6 +613,21 @@ func (d *Driver) getKey() (*hcloud.SSHKey, error) {
 	return stype, nil
 }
 
+func (d *Driver) getRemoteKeyWithSameFingerprint(pubkey_byte []byte) (*hcloud.SSHKey, error) {
+	pubkey, _, _, _, err := ssh.ParseAuthorizedKey(pubkey_byte)
+	if err != nil {
+		return nil, errors.Wrap(err, "could not parse ssh public key")
+	}
+
+	fp := ssh.FingerprintLegacyMD5(pubkey)
+
+	remotekey, _, err := d.getClient().SSHKey.GetByFingerprint(context.Background(), fp)
+	if err != nil {
+		return remotekey, errors.Wrap(err, "could not get sshkey by fingerprint")
+	}
+	return remotekey, nil
+}
+
 func (d *Driver) getServerHandle() (*hcloud.Server, error) {
 	if d.cachedServer != nil {
 		return d.cachedServer, nil