digitalocean droplet power on & off work

Signed-off-by: Jeff Carr <[email protected]>

1 files changed, 13 insertions, 10 deletions

diff --git a/digitalocean/main.go b/digitalocean/main.go
index de96521..5c40498 100644
--- a/digitalocean/main.go
+++ b/digitalocean/main.go
@@ -58,18 +58,21 @@ func (d *DigitalOcean) Hide() {
 
 func (d *DigitalOcean) Update() bool {
 	if ! d.Ready() {return false}
-	if ! d.ListDroplets() {
+	d.ListSSHKeyID()
+	if d.ListDroplets() {
+		for _, droplet := range d.dpolled {
+			// check if the droplet ID already exists
+			if (d.dropMap[droplet.ID] == nil) {
+				d.dropMap[droplet.ID] = d.NewDroplet(&droplet)
+			} else {
+				log.Info("droplet.Update()", droplet.ID, droplet.Name, "already exists")
+				d.dropMap[droplet.ID].Update(&droplet)
+				continue
+			}
+		}
+	} else {
 		log.Error(d.err, "Error listing droplets")
 		return false
 	}
-	for _, droplet := range d.dpolled {
-		// check if the droplet ID already exists
-		if (d.dropMap[droplet.ID] != nil) {
-			log.Info("droplet.Update()", droplet.ID, droplet.Name, "already exists")
-			d.dropMap[droplet.ID].Update(&droplet)
-			continue
-		}
-		d.dropMap[droplet.ID] = d.NewDroplet(&droplet)
-	}
 	return true
 }