From 637968f2ddc835de38db6966c4c702b73e14e8ed Mon Sep 17 00:00:00 2001
From: Mitchell Hashimoto <mitchell.hashimoto@gmail.com>
Date: Fri, 13 Dec 2013 19:07:10 -0800
Subject: [PATCH] builder/googlecompute: artifact uses Driver, no more api

---
 builder/googlecompute/api.go         | 23 -----------------
 builder/googlecompute/artifact.go    | 10 ++------
 builder/googlecompute/driver.go      |  3 +++
 builder/googlecompute/driver_gce.go  | 12 +++++++++
 builder/googlecompute/driver_mock.go | 16 ++++++++++++
 builder/googlecompute/wait.go        | 38 ----------------------------
 6 files changed, 33 insertions(+), 69 deletions(-)
 delete mode 100644 builder/googlecompute/api.go
 delete mode 100644 builder/googlecompute/wait.go

diff --git a/builder/googlecompute/api.go b/builder/googlecompute/api.go
deleted file mode 100644
index 46611f02e..000000000
--- a/builder/googlecompute/api.go
+++ /dev/null
@@ -1,23 +0,0 @@
-package googlecompute
-
-import (
-	"code.google.com/p/google-api-go-client/compute/v1beta16"
-)
-
-// GoogleComputeClient represents a GCE client.
-type GoogleComputeClient struct {
-	ProjectId     string
-	Service       *compute.Service
-	Zone          string
-	clientSecrets *clientSecrets
-}
-
-// DeleteImage deletes the named image. Returns a Global Operation.
-func (g *GoogleComputeClient) DeleteImage(name string) (*compute.Operation, error) {
-	imagesDeleteCall := g.Service.Images.Delete(g.ProjectId, name)
-	operation, err := imagesDeleteCall.Do()
-	if err != nil {
-		return nil, err
-	}
-	return operation, nil
-}
diff --git a/builder/googlecompute/artifact.go b/builder/googlecompute/artifact.go
index a6ed7e25c..a09946f9f 100644
--- a/builder/googlecompute/artifact.go
+++ b/builder/googlecompute/artifact.go
@@ -19,14 +19,8 @@ func (*Artifact) BuilderId() string {
 // Destroy destroys the GCE image represented by the artifact.
 func (a *Artifact) Destroy() error {
 	log.Printf("Destroying image: %s", a.imageName)
-	/*
-		// Ignore the operation result as we are not waiting until it completes.
-		_, err := a.client.DeleteImage(a.imageName)
-		if err != nil {
-			return err
-		}
-	*/
-	return nil
+	errCh := a.driver.DeleteImage(a.imageName)
+	return <-errCh
 }
 
 // Files returns the files represented by the artifact.
diff --git a/builder/googlecompute/driver.go b/builder/googlecompute/driver.go
index fc8f010c5..743440e24 100644
--- a/builder/googlecompute/driver.go
+++ b/builder/googlecompute/driver.go
@@ -7,6 +7,9 @@ type Driver interface {
 	// CreateImage creates an image with the given URL in Google Storage.
 	CreateImage(name, description, url string) <-chan error
 
+	// DeleteImage deletes the image with the given name.
+	DeleteImage(name string) <-chan error
+
 	// DeleteInstance deletes the given instance.
 	DeleteInstance(zone, name string) (<-chan error, error)
 
diff --git a/builder/googlecompute/driver_gce.go b/builder/googlecompute/driver_gce.go
index c464ee352..6ab3929e3 100644
--- a/builder/googlecompute/driver_gce.go
+++ b/builder/googlecompute/driver_gce.go
@@ -81,6 +81,18 @@ func (d *driverGCE) CreateImage(name, description, url string) <-chan error {
 	return errCh
 }
 
+func (d *driverGCE) DeleteImage(name string) <-chan error {
+	errCh := make(chan error, 1)
+	op, err := d.service.Images.Delete(d.projectId, name).Do()
+	if err != nil {
+		errCh <- err
+	} else {
+		go waitForState(errCh, "DONE", d.refreshGlobalOp(op))
+	}
+
+	return errCh
+}
+
 func (d *driverGCE) DeleteInstance(zone, name string) (<-chan error, error) {
 	op, err := d.service.Instances.Delete(d.projectId, zone, name).Do()
 	if err != nil {
diff --git a/builder/googlecompute/driver_mock.go b/builder/googlecompute/driver_mock.go
index 87d2528b7..e66727b80 100644
--- a/builder/googlecompute/driver_mock.go
+++ b/builder/googlecompute/driver_mock.go
@@ -8,6 +8,9 @@ type DriverMock struct {
 	CreateImageURL   string
 	CreateImageErrCh <-chan error
 
+	DeleteImageName  string
+	DeleteImageErrCh <-chan error
+
 	DeleteInstanceZone  string
 	DeleteInstanceName  string
 	DeleteInstanceErrCh <-chan error
@@ -43,6 +46,19 @@ func (d *DriverMock) CreateImage(name, description, url string) <-chan error {
 	return resultCh
 }
 
+func (d *DriverMock) DeleteImage(name string) <-chan error {
+	d.DeleteImageName = name
+
+	resultCh := d.DeleteImageErrCh
+	if resultCh == nil {
+		ch := make(chan error)
+		close(ch)
+		resultCh = ch
+	}
+
+	return resultCh
+}
+
 func (d *DriverMock) DeleteInstance(zone, name string) (<-chan error, error) {
 	d.DeleteInstanceZone = zone
 	d.DeleteInstanceName = name
diff --git a/builder/googlecompute/wait.go b/builder/googlecompute/wait.go
deleted file mode 100644
index cd0c6fa10..000000000
--- a/builder/googlecompute/wait.go
+++ /dev/null
@@ -1,38 +0,0 @@
-package googlecompute
-
-import (
-	"time"
-)
-
-// waitForInstanceState.
-func waitForInstanceState(desiredState string, zone string, name string, client *GoogleComputeClient, timeout time.Duration) error {
-	return nil
-	/*
-		f := func() (string, error) {
-			return client.InstanceStatus(zone, name)
-		}
-		return waitForState("instance", desiredState, f, timeout)
-	*/
-}
-
-// waitForZoneOperationState.
-func waitForZoneOperationState(desiredState string, zone string, name string, client *GoogleComputeClient, timeout time.Duration) error {
-	return nil
-	/*
-		f := func() (string, error) {
-			return client.ZoneOperationStatus(zone, name)
-		}
-		return waitForState("operation", desiredState, f, timeout)
-	*/
-}
-
-// waitForGlobalOperationState.
-func waitForGlobalOperationState(desiredState string, name string, client *GoogleComputeClient, timeout time.Duration) error {
-	/*
-		f := func() (string, error) {
-			return client.GlobalOperationStatus(name)
-		}
-		return waitForState("operation", desiredState, f, timeout)
-	*/
-	return nil
-}