resource_compute_health_check.go

package google

import (
	"fmt"
	"log"

	"github.com/hashicorp/terraform/helper/schema"
	"google.golang.org/api/compute/v1"
)

func resourceComputeHealthCheck() *schema.Resource {
	return &schema.Resource{
		Create: resourceComputeHealthCheckCreate,
		Read:   resourceComputeHealthCheckRead,
		Delete: resourceComputeHealthCheckDelete,
		Update: resourceComputeHealthCheckUpdate,
		Importer: &schema.ResourceImporter{
			State: schema.ImportStatePassthrough,
		},

		Schema: map[string]*schema.Schema{
			"name": &schema.Schema{
				Type:     schema.TypeString,
				Required: true,
				ForceNew: true,
			},

			"check_interval_sec": &schema.Schema{
				Type:     schema.TypeInt,
				Optional: true,
				Default:  5,
			},

			"description": &schema.Schema{
				Type:     schema.TypeString,
				Optional: true,
			},

			"healthy_threshold": &schema.Schema{
				Type:     schema.TypeInt,
				Optional: true,
				Default:  2,
			},

			"tcp_health_check": &schema.Schema{
				Type:          schema.TypeList,
				Optional:      true,
				MaxItems:      1,
				ConflictsWith: []string{"ssl_health_check", "http_health_check", "https_health_check"},
				Elem: &schema.Resource{
					Schema: map[string]*schema.Schema{
						"port": &schema.Schema{
							Type:     schema.TypeInt,
							Optional: true,
							Default:  80,
						},
						"proxy_header": &schema.Schema{
							Type:     schema.TypeString,
							Optional: true,
							Default:  "NONE",
						},
						"request": &schema.Schema{
							Type:     schema.TypeString,
							Optional: true,
						},
						"response": &schema.Schema{
							Type:     schema.TypeString,
							Optional: true,
						},
					},
				},
			},

			"ssl_health_check": &schema.Schema{
				Type:          schema.TypeList,
				Optional:      true,
				MaxItems:      1,
				ConflictsWith: []string{"tcp_health_check", "http_health_check", "https_health_check"},
				Elem: &schema.Resource{
					Schema: map[string]*schema.Schema{
						"port": &schema.Schema{
							Type:     schema.TypeInt,
							Optional: true,
							Default:  443,
						},
						"proxy_header": &schema.Schema{
							Type:     schema.TypeString,
							Optional: true,
							Default:  "NONE",
						},
						"request": &schema.Schema{
							Type:     schema.TypeString,
							Optional: true,
						},
						"response": &schema.Schema{
							Type:     schema.TypeString,
							Optional: true,
						},
					},
				},
			},

			"http_health_check": &schema.Schema{
				Type:          schema.TypeList,
				Optional:      true,
				MaxItems:      1,
				ConflictsWith: []string{"tcp_health_check", "ssl_health_check", "https_health_check"},
				Elem: &schema.Resource{
					Schema: map[string]*schema.Schema{
						"host": &schema.Schema{
							Type:     schema.TypeString,
							Optional: true,
						},
						"port": &schema.Schema{
							Type:     schema.TypeInt,
							Optional: true,
							Default:  80,
						},
						"proxy_header": &schema.Schema{
							Type:     schema.TypeString,
							Optional: true,
							Default:  "NONE",
						},
						"request_path": &schema.Schema{
							Type:     schema.TypeString,
							Optional: true,
							Default:  "/",
						},
					},
				},
			},

			"https_health_check": &schema.Schema{
				Type:          schema.TypeList,
				Optional:      true,
				MaxItems:      1,
				ConflictsWith: []string{"tcp_health_check", "ssl_health_check", "http_health_check"},
				Elem: &schema.Resource{
					Schema: map[string]*schema.Schema{
						"host": &schema.Schema{
							Type:     schema.TypeString,
							Optional: true,
						},
						"port": &schema.Schema{
							Type:     schema.TypeInt,
							Optional: true,
							Default:  443,
						},
						"proxy_header": &schema.Schema{
							Type:     schema.TypeString,
							Optional: true,
							Default:  "NONE",
						},
						"request_path": &schema.Schema{
							Type:     schema.TypeString,
							Optional: true,
							Default:  "/",
						},
					},
				},
			},

			"project": &schema.Schema{
				Type:     schema.TypeString,
				Optional: true,
				ForceNew: true,
				Computed: true,
			},

			"self_link": &schema.Schema{
				Type:     schema.TypeString,
				Computed: true,
			},

			"timeout_sec": &schema.Schema{
				Type:     schema.TypeInt,
				Optional: true,
				Default:  5,
			},

			"unhealthy_threshold": &schema.Schema{
				Type:     schema.TypeInt,
				Optional: true,
				Default:  2,
			},
		},
	}
}

func resourceComputeHealthCheckCreate(d *schema.ResourceData, meta interface{}) error {
	config := meta.(*Config)

	project, err := getProject(d, config)
	if err != nil {
		return err
	}

	// Build the parameter
	hchk := &compute.HealthCheck{
		Name: d.Get("name").(string),
	}
	// Optional things
	if v, ok := d.GetOk("description"); ok {
		hchk.Description = v.(string)
	}
	if v, ok := d.GetOk("check_interval_sec"); ok {
		hchk.CheckIntervalSec = int64(v.(int))
	}
	if v, ok := d.GetOk("healthy_threshold"); ok {
		hchk.HealthyThreshold = int64(v.(int))
	}
	if v, ok := d.GetOk("timeout_sec"); ok {
		hchk.TimeoutSec = int64(v.(int))
	}
	if v, ok := d.GetOk("unhealthy_threshold"); ok {
		hchk.UnhealthyThreshold = int64(v.(int))
	}

	if v, ok := d.GetOk("tcp_health_check"); ok {
		hchk.Type = "TCP"
		tcpcheck := v.([]interface{})[0].(map[string]interface{})
		tcpHealthCheck := &compute.TCPHealthCheck{}
		if val, ok := tcpcheck["port"]; ok {
			tcpHealthCheck.Port = int64(val.(int))
		}
		if val, ok := tcpcheck["proxy_header"]; ok {
			tcpHealthCheck.ProxyHeader = val.(string)
		}
		if val, ok := tcpcheck["request"]; ok {
			tcpHealthCheck.Request = val.(string)
		}
		if val, ok := tcpcheck["response"]; ok {
			tcpHealthCheck.Response = val.(string)
		}
		hchk.TcpHealthCheck = tcpHealthCheck
	}

	if v, ok := d.GetOk("ssl_health_check"); ok {
		hchk.Type = "SSL"
		sslcheck := v.([]interface{})[0].(map[string]interface{})
		sslHealthCheck := &compute.SSLHealthCheck{}
		if val, ok := sslcheck["port"]; ok {
			sslHealthCheck.Port = int64(val.(int))
		}
		if val, ok := sslcheck["proxy_header"]; ok {
			sslHealthCheck.ProxyHeader = val.(string)
		}
		if val, ok := sslcheck["request"]; ok {
			sslHealthCheck.Request = val.(string)
		}
		if val, ok := sslcheck["response"]; ok {
			sslHealthCheck.Response = val.(string)
		}
		hchk.SslHealthCheck = sslHealthCheck
	}

	if v, ok := d.GetOk("http_health_check"); ok {
		hchk.Type = "HTTP"
		httpcheck := v.([]interface{})[0].(map[string]interface{})
		httpHealthCheck := &compute.HTTPHealthCheck{}
		if val, ok := httpcheck["host"]; ok {
			httpHealthCheck.Host = val.(string)
		}
		if val, ok := httpcheck["port"]; ok {
			httpHealthCheck.Port = int64(val.(int))
		}
		if val, ok := httpcheck["proxy_header"]; ok {
			httpHealthCheck.ProxyHeader = val.(string)
		}
		if val, ok := httpcheck["request_path"]; ok {
			httpHealthCheck.RequestPath = val.(string)
		}
		hchk.HttpHealthCheck = httpHealthCheck
	}

	if v, ok := d.GetOk("https_health_check"); ok {
		hchk.Type = "HTTPS"
		httpscheck := v.([]interface{})[0].(map[string]interface{})
		httpsHealthCheck := &compute.HTTPSHealthCheck{}
		if val, ok := httpscheck["host"]; ok {
			httpsHealthCheck.Host = val.(string)
		}
		if val, ok := httpscheck["port"]; ok {
			httpsHealthCheck.Port = int64(val.(int))
		}
		if val, ok := httpscheck["proxy_header"]; ok {
			httpsHealthCheck.ProxyHeader = val.(string)
		}
		if val, ok := httpscheck["request_path"]; ok {
			httpsHealthCheck.RequestPath = val.(string)
		}
		hchk.HttpsHealthCheck = httpsHealthCheck
	}

	log.Printf("[DEBUG] HealthCheck insert request: %#v", hchk)
	op, err := config.clientCompute.HealthChecks.Insert(
		project, hchk).Do()
	if err != nil {
		return fmt.Errorf("Error creating HealthCheck: %s", err)
	}

	// It probably maybe worked, so store the ID now
	d.SetId(hchk.Name)

	err = computeOperationWait(config.clientCompute, op, project, "Creating Health Check")
	if err != nil {
		return err
	}

	return resourceComputeHealthCheckRead(d, meta)
}

func resourceComputeHealthCheckUpdate(d *schema.ResourceData, meta interface{}) error {
	config := meta.(*Config)

	project, err := getProject(d, config)
	if err != nil {
		return err
	}

	// Build the parameter
	hchk := &compute.HealthCheck{
		Name: d.Get("name").(string),
	}

	nullFields := make([]string, 0, 3)

	// Optional things
	if v, ok := d.GetOk("description"); ok {
		hchk.Description = v.(string)
	}
	if v, ok := d.GetOk("check_interval_sec"); ok {
		hchk.CheckIntervalSec = int64(v.(int))
	}
	if v, ok := d.GetOk("healthy_threshold"); ok {
		hchk.HealthyThreshold = int64(v.(int))
	}
	if v, ok := d.GetOk("timeout_sec"); ok {
		hchk.TimeoutSec = int64(v.(int))
	}
	if v, ok := d.GetOk("unhealthy_threshold"); ok {
		hchk.UnhealthyThreshold = int64(v.(int))
	}
	if v, ok := d.GetOk("tcp_health_check"); ok {
		hchk.Type = "TCP"
		tcpcheck := v.([]interface{})[0].(map[string]interface{})
		tcpHealthCheck := &compute.TCPHealthCheck{}
		if val, ok := tcpcheck["port"]; ok {
			tcpHealthCheck.Port = int64(val.(int))
		}
		if val, ok := tcpcheck["proxy_header"]; ok {
			tcpHealthCheck.ProxyHeader = val.(string)
		}
		if val, ok := tcpcheck["request"]; ok {
			tcpHealthCheck.Request = val.(string)
		}
		if val, ok := tcpcheck["response"]; ok {
			tcpHealthCheck.Response = val.(string)
		}
		hchk.TcpHealthCheck = tcpHealthCheck
	} else {
		nullFields = append(nullFields, "TcpHealthCheck")
	}
	if v, ok := d.GetOk("ssl_health_check"); ok {
		hchk.Type = "SSL"
		sslcheck := v.([]interface{})[0].(map[string]interface{})
		sslHealthCheck := &compute.SSLHealthCheck{}
		if val, ok := sslcheck["port"]; ok {
			sslHealthCheck.Port = int64(val.(int))
		}
		if val, ok := sslcheck["proxy_header"]; ok {
			sslHealthCheck.ProxyHeader = val.(string)
		}
		if val, ok := sslcheck["request"]; ok {
			sslHealthCheck.Request = val.(string)
		}
		if val, ok := sslcheck["response"]; ok {
			sslHealthCheck.Response = val.(string)
		}
		hchk.SslHealthCheck = sslHealthCheck
	} else {
		nullFields = append(nullFields, "SslHealthCheck")
	}
	if v, ok := d.GetOk("http_health_check"); ok {
		hchk.Type = "HTTP"
		httpcheck := v.([]interface{})[0].(map[string]interface{})
		httpHealthCheck := &compute.HTTPHealthCheck{}
		if val, ok := httpcheck["host"]; ok {
			httpHealthCheck.Host = val.(string)
		}
		if val, ok := httpcheck["port"]; ok {
			httpHealthCheck.Port = int64(val.(int))
		}
		if val, ok := httpcheck["proxy_header"]; ok {
			httpHealthCheck.ProxyHeader = val.(string)
		}
		if val, ok := httpcheck["request_path"]; ok {
			httpHealthCheck.RequestPath = val.(string)
		}
		hchk.HttpHealthCheck = httpHealthCheck
	} else {
		nullFields = append(nullFields, "HttpHealthCheck")
	}

	if v, ok := d.GetOk("https_health_check"); ok {
		hchk.Type = "HTTPS"
		httpscheck := v.([]interface{})[0].(map[string]interface{})
		httpsHealthCheck := &compute.HTTPSHealthCheck{}
		if val, ok := httpscheck["host"]; ok {
			httpsHealthCheck.Host = val.(string)
		}
		if val, ok := httpscheck["port"]; ok {
			httpsHealthCheck.Port = int64(val.(int))
		}
		if val, ok := httpscheck["proxy_header"]; ok {
			httpsHealthCheck.ProxyHeader = val.(string)
		}
		if val, ok := httpscheck["request_path"]; ok {
			httpsHealthCheck.RequestPath = val.(string)
		}
		hchk.HttpsHealthCheck = httpsHealthCheck
	} else {
		nullFields = append(nullFields, "HttpsHealthCheck")
	}

	hchk.NullFields = nullFields

	log.Printf("[DEBUG] HealthCheck patch request: %#v", hchk)
	op, err := config.clientCompute.HealthChecks.Patch(
		project, hchk.Name, hchk).Do()
	if err != nil {
		return fmt.Errorf("Error patching HealthCheck: %s", err)
	}

	// It probably maybe worked, so store the ID now
	d.SetId(hchk.Name)

	err = computeOperationWait(config.clientCompute, op, project, "Updating Health Check")
	if err != nil {
		return err
	}

	return resourceComputeHealthCheckRead(d, meta)
}

func resourceComputeHealthCheckRead(d *schema.ResourceData, meta interface{}) error {
	config := meta.(*Config)

	project, err := getProject(d, config)
	if err != nil {
		return err
	}

	hchk, err := config.clientCompute.HealthChecks.Get(
		project, d.Id()).Do()
	if err != nil {
		return handleNotFoundError(err, d, fmt.Sprintf("Health Check %q", d.Get("name").(string)))
	}

	d.Set("check_interval_sec", hchk.CheckIntervalSec)
	d.Set("healthy_threshold", hchk.HealthyThreshold)
	d.Set("timeout_sec", hchk.TimeoutSec)
	d.Set("unhealthy_threshold", hchk.UnhealthyThreshold)
	d.Set("tcp_health_check", flattenTcpHealthCheck(hchk.TcpHealthCheck))
	d.Set("ssl_health_check", flattenSslHealthCheck(hchk.SslHealthCheck))
	d.Set("http_health_check", flattenHttpHealthCheck(hchk.HttpHealthCheck))
	d.Set("https_health_check", flattenHttpsHealthCheck(hchk.HttpsHealthCheck))
	d.Set("self_link", hchk.SelfLink)
	d.Set("name", hchk.Name)
	d.Set("description", hchk.Description)
	d.Set("project", project)

	return nil
}

func resourceComputeHealthCheckDelete(d *schema.ResourceData, meta interface{}) error {
	config := meta.(*Config)

	project, err := getProject(d, config)
	if err != nil {
		return err
	}

	// Delete the HealthCheck
	op, err := config.clientCompute.HealthChecks.Delete(
		project, d.Id()).Do()
	if err != nil {
		return fmt.Errorf("Error deleting HealthCheck: %s", err)
	}

	err = computeOperationWait(config.clientCompute, op, project, "Deleting Health Check")
	if err != nil {
		return err
	}

	d.SetId("")
	return nil
}

func flattenTcpHealthCheck(hchk *compute.TCPHealthCheck) []map[string]interface{} {
	if hchk == nil {
		return nil
	}

	result := make([]map[string]interface{}, 0, 1)
	data := make(map[string]interface{})
	data["port"] = hchk.Port
	data["proxy_header"] = hchk.ProxyHeader
	data["request"] = hchk.Request
	data["response"] = hchk.Response
	result = append(result, data)
	return result
}

func flattenSslHealthCheck(hchk *compute.SSLHealthCheck) []map[string]interface{} {
	if hchk == nil {
		return nil
	}

	result := make([]map[string]interface{}, 0, 1)
	data := make(map[string]interface{})
	data["port"] = hchk.Port
	data["proxy_header"] = hchk.ProxyHeader
	data["request"] = hchk.Request
	data["response"] = hchk.Response
	result = append(result, data)
	return result
}

func flattenHttpHealthCheck(hchk *compute.HTTPHealthCheck) []map[string]interface{} {
	if hchk == nil {
		return nil
	}

	result := make([]map[string]interface{}, 0, 1)
	data := make(map[string]interface{})
	data["host"] = hchk.Host
	data["port"] = hchk.Port
	data["proxy_header"] = hchk.ProxyHeader
	data["request_path"] = hchk.RequestPath
	result = append(result, data)
	return result
}

func flattenHttpsHealthCheck(hchk *compute.HTTPSHealthCheck) []map[string]interface{} {
	if hchk == nil {
		return nil
	}

	result := make([]map[string]interface{}, 0, 1)
	data := make(map[string]interface{})
	data["host"] = hchk.Host
	data["port"] = hchk.Port
	data["proxy_header"] = hchk.ProxyHeader
	data["request_path"] = hchk.RequestPath
	result = append(result, data)
	return result
}