Add resource lbvserver_rewritepolicy_binding

Signed-off-by: George Nikolopoulos <[email protected]>

citrixadc/provider.go

@@ -176,6 +176,7 @@ func providerResources() map[string]*schema.Resource {
 		"citrixadc_lbvserver_responderpolicy_binding":       resourceCitrixAdcLbvserver_responderpolicy_binding(),
 		"citrixadc_csvserver_responderpolicy_binding":       resourceCitrixAdcCsvserver_responderpolicy_binding(),
 		"citrixadc_csvserver_rewritepolicy_binding":         resourceCitrixAdcCsvserver_rewritepolicy_binding(),
+		"citrixadc_lbvserver_rewritepolicy_binding":         resourceCitrixAdcLbvserver_rewritepolicy_binding(),
 	}
 }
 

citrixadc/resource_citrixadc_lbvserver_rewritepolicy_binding.go

@@ -0,0 +1,197 @@
+package citrixadc
+
+import (
+	"github.com/chiradeep/go-nitro/config/lb"
+
+	"github.com/chiradeep/go-nitro/netscaler"
+	"github.com/hashicorp/terraform/helper/schema"
+
+	"fmt"
+	"log"
+	"net/url"
+	"strings"
+)
+
+func resourceCitrixAdcLbvserver_rewritepolicy_binding() *schema.Resource {
+	return &schema.Resource{
+		SchemaVersion: 1,
+		Create:        createLbvserver_rewritepolicy_bindingFunc,
+		Read:          readLbvserver_rewritepolicy_bindingFunc,
+		Delete:        deleteLbvserver_rewritepolicy_bindingFunc,
+		Schema: map[string]*schema.Schema{
+			"bindpoint": &schema.Schema{
+				Type:     schema.TypeString,
+				Optional: true,
+				Computed: true,
+				ForceNew: true,
+			},
+			"gotopriorityexpression": &schema.Schema{
+				Type:     schema.TypeString,
+				Optional: true,
+				Computed: true,
+				ForceNew: true,
+			},
+			"invoke": &schema.Schema{
+				Type:     schema.TypeBool,
+				Optional: true,
+				Computed: true,
+				ForceNew: true,
+			},
+			"labelname": &schema.Schema{
+				Type:     schema.TypeString,
+				Optional: true,
+				Computed: true,
+				ForceNew: true,
+			},
+			"labeltype": &schema.Schema{
+				Type:     schema.TypeString,
+				Optional: true,
+				Computed: true,
+				ForceNew: true,
+			},
+			"name": &schema.Schema{
+				Type:     schema.TypeString,
+				Required: true,
+				ForceNew: true,
+			},
+			"policyname": &schema.Schema{
+				Type:     schema.TypeString,
+				Required: true,
+				ForceNew: true,
+			},
+			"priority": &schema.Schema{
+				Type:     schema.TypeInt,
+				Optional: true,
+				Computed: true,
+				ForceNew: true,
+			},
+		},
+	}
+}
+
+func createLbvserver_rewritepolicy_bindingFunc(d *schema.ResourceData, meta interface{}) error {
+	log.Printf("[DEBUG]  citrixadc-provider: In createLbvserver_rewritepolicy_bindingFunc")
+	client := meta.(*NetScalerNitroClient).client
+	name := d.Get("name").(string)
+	policyname := d.Get("policyname").(string)
+	bindingId := fmt.Sprintf("%s,%s", name, policyname)
+	lbvserver_rewritepolicy_binding := lb.Lbvserverrewritepolicybinding{
+		Bindpoint:              d.Get("bindpoint").(string),
+		Gotopriorityexpression: d.Get("gotopriorityexpression").(string),
+		Invoke:                 d.Get("invoke").(bool),
+		Labelname:              d.Get("labelname").(string),
+		Labeltype:              d.Get("labeltype").(string),
+		Name:                   d.Get("name").(string),
+		Policyname:             d.Get("policyname").(string),
+		Priority:               d.Get("priority").(int),
+	}
+
+	err := client.UpdateUnnamedResource(netscaler.Lbvserver_rewritepolicy_binding.Type(), &lbvserver_rewritepolicy_binding)
+	if err != nil {
+		return err
+	}
+
+	d.SetId(bindingId)
+
+	err = readLbvserver_rewritepolicy_bindingFunc(d, meta)
+	if err != nil {
+		log.Printf("[ERROR] netscaler-provider: ?? we just created this lbvserver_rewritepolicy_binding but we can't read it ?? %s", bindingId)
+		return nil
+	}
+	return nil
+}
+
+func readLbvserver_rewritepolicy_bindingFunc(d *schema.ResourceData, meta interface{}) error {
+	log.Printf("[DEBUG] citrixadc-provider:  In readLbvserver_rewritepolicy_bindingFunc")
+	client := meta.(*NetScalerNitroClient).client
+	bindingId := d.Id()
+	idSlice := strings.SplitN(bindingId, ",", 2)
+	name := idSlice[0]
+	policyname := idSlice[1]
+	log.Printf("[DEBUG] citrixadc-provider: Reading lbvserver_rewritepolicy_binding state %s", bindingId)
+
+	findParams := netscaler.FindParams{
+		ResourceType:             "lbvserver_rewritepolicy_binding",
+		ResourceName:             name,
+		ResourceMissingErrorCode: 258,
+	}
+	dataArr, err := client.FindResourceArrayWithParams(findParams)
+
+	// Unexpected error
+	if err != nil {
+		log.Printf("[DEBUG] citrixadc-provider: Error during FindResourceArrayWithParams %s", err.Error())
+		return err
+	}
+
+	// Resource is missing
+	if len(dataArr) == 0 {
+		log.Printf("[DEBUG] citrixadc-provider: FindResourceArrayWithParams returned empty array")
+		log.Printf("[WARN] citrixadc-provider: Clearing lbvserver_rewritepolicy_binding state %s", bindingId)
+		d.SetId("")
+		return nil
+	}
+
+	// Iterate through results to find the one with the right policy name
+	foundIndex := -1
+	for i, v := range dataArr {
+		if v["policyname"].(string) == policyname {
+			foundIndex = i
+			break
+		}
+	}
+
+	// Resource is missing
+	if foundIndex == -1 {
+		log.Printf("[DEBUG] citrixadc-provider: FindResourceArrayWithParams policyname not found in array")
+		log.Printf("[WARN] citrixadc-provider: Clearing lbvserver_rewritepolicy_binding state %s", bindingId)
+		d.SetId("")
+		return nil
+	}
+	// Fallthrough
+
+	data := dataArr[foundIndex]
+
+	d.Set("name", data["name"])
+	d.Set("bindpoint", data["bindpoint"])
+	d.Set("gotopriorityexpression", data["gotopriorityexpression"])
+	d.Set("invoke", data["invoke"])
+	d.Set("labelname", data["labelname"])
+	d.Set("labeltype", data["labeltype"])
+	d.Set("name", data["name"])
+	d.Set("policyname", data["policyname"])
+	d.Set("priority", data["priority"])
+
+	return nil
+
+}
+
+func deleteLbvserver_rewritepolicy_bindingFunc(d *schema.ResourceData, meta interface{}) error {
+	log.Printf("[DEBUG]  citrixadc-provider: In deleteLbvserver_rewritepolicy_bindingFunc")
+	client := meta.(*NetScalerNitroClient).client
+
+	bindingId := d.Id()
+
+	idSlice := strings.SplitN(bindingId, ",", 2)
+
+	name := idSlice[0]
+	policyname := idSlice[1]
+
+	argsMap := make(map[string]string)
+	argsMap["policyname"] = url.QueryEscape(policyname)
+
+	if v, ok := d.GetOk("bindpoint"); ok {
+		argsMap["bindpoint"] = url.QueryEscape(v.(string))
+	}
+
+	if v, ok := d.GetOk("priority"); ok {
+		argsMap["priority"] = url.QueryEscape(fmt.Sprintf("%v", v))
+	}
+	err := client.DeleteResourceWithArgsMap(netscaler.Lbvserver_rewritepolicy_binding.Type(), name, argsMap)
+	if err != nil {
+		return err
+	}
+
+	d.SetId("")
+
+	return nil
+}