Enable canar

spinnaker/api/application.go

@@ -31,7 +31,7 @@ func GetApplication(client *gate.GatewayClient, applicationName string, dest int
 	return nil
 }
 
-func CreateApplication(client *gate.GatewayClient, applicationName, email string) error {
+func CreateApplication(client *gate.GatewayClient, applicationName, email string, canary bool) error {
 
 	app := map[string]interface{}{
 		"instancePort": 80,

spinnaker/api/task.go

@@ -0,0 +1,37 @@
+package api
+
+import (
+	"fmt"
+	"net/http"
+
+	gate "github.com/spinnaker/spin/cmd/gateclient"
+)
+
+func EnableApplicationCanary(client *gate.GatewayClient, applicationName string) error {
+	jobSpec := map[string]interface{}{
+		"type": "updateApplication",
+		"application": map[string]interface{}{
+			"name": applicationName,
+			"datasources": map[string]interface{}{
+				"enabled":  "canaryConfigs",
+				"disabled": "",
+			},
+			"user": "[anonymous]",
+		},
+	}
+	enableAppCanaryTask := map[string]interface{}{
+		"job":         []interface{}{jobSpec},
+		"application": applicationName,
+		"description": fmt.Sprintf("Enable canary for: %s", applicationName),
+	}
+
+	_, resp, err := client.TaskControllerApi.TaskUsingPOST1(client.Context, enableAppCanaryTask)
+
+	if err != nil {
+		return err
+	}
+	if resp.StatusCode != http.StatusOK {
+		return fmt.Errorf("Encountered an error enabling canary, status code: %d\n", resp.StatusCode)
+	}
+	return nil
+}

spinnaker/resource_application.go

@@ -18,6 +18,10 @@ func resourceApplication() *schema.Resource {
 				Type:     schema.TypeString,
 				Required: true,
 			},
+			"canary": {
+				Type:     schema.TypeBool,
+				Optional: true,
+			},
 		},
 		Create: resourceApplicationCreate,
 		Read:   resourceApplicationRead,
@@ -39,10 +43,17 @@ func resourceApplicationCreate(data *schema.ResourceData, meta interface{}) erro
 	client := clientConfig.client
 	application := data.Get("application").(string)
 	email := data.Get("email").(string)
+	canary := data.Get("canary").(bool)
 
-	if err := api.CreateApplication(client, application, email); err != nil {
+	if err := api.CreateApplication(client, application, email, canary); err != nil {
 		return err
 	}
+	if canary {
+		err := api.EnableApplicationCanary(client, application)
+		if err != nil {
+			return err
+		}
+	}
 
 	return resourceApplicationRead(data, meta)
 }

spinnaker/resource_application_test.go

@@ -24,6 +24,7 @@ func TestAccSpinnakerApplication_basic(t *testing.T) {
 					testAccCheckApplicationExists(resourceName),
 					resource.TestCheckResourceAttr(resourceName, "application", rName),
 					resource.TestCheckResourceAttr(resourceName, "email", "[email protected]"),
+					resource.TestCheckResourceAttr(resourceName, "canary", "true"),
 				),
 			},
 		},
@@ -67,6 +68,7 @@ func testAccSpinnakerApplication_basic(rName string) string {
 resource "spinnaker_application" "test" {
 	application  = %q
 	email = "[email protected]"
+	canary = true
 }
 `, rName)
 }