terraform/builtin/providers/openstack/resource_openstack_compute_...

295 lines
8.7 KiB
Go
Raw Normal View History

2015-01-08 18:47:10 +01:00
package openstack
import (
"bytes"
2015-01-08 18:47:10 +01:00
"fmt"
"log"
"github.com/hashicorp/terraform/helper/hashcode"
2015-01-08 18:47:10 +01:00
"github.com/hashicorp/terraform/helper/schema"
"github.com/rackspace/gophercloud"
2015-01-08 18:47:10 +01:00
"github.com/rackspace/gophercloud/openstack/compute/v2/extensions/secgroups"
)
2015-01-26 19:09:27 +01:00
func resourceComputeSecGroupV2() *schema.Resource {
2015-01-08 18:47:10 +01:00
return &schema.Resource{
2015-01-26 19:09:27 +01:00
Create: resourceComputeSecGroupV2Create,
Read: resourceComputeSecGroupV2Read,
Update: resourceComputeSecGroupV2Update,
Delete: resourceComputeSecGroupV2Delete,
2015-01-08 18:47:10 +01:00
Schema: map[string]*schema.Schema{
"region": &schema.Schema{
Type: schema.TypeString,
Required: true,
ForceNew: true,
DefaultFunc: envDefaultFuncAllowMissing("OS_REGION_NAME"),
},
2015-01-08 18:47:10 +01:00
"name": &schema.Schema{
Type: schema.TypeString,
Required: true,
ForceNew: false,
},
"description": &schema.Schema{
Type: schema.TypeString,
Required: true,
ForceNew: false,
},
2015-02-01 01:22:19 +01:00
"rule": &schema.Schema{
Type: schema.TypeList,
Optional: true,
Elem: &schema.Resource{
Schema: map[string]*schema.Schema{
"id": &schema.Schema{
Type: schema.TypeString,
Computed: true,
},
"from_port": &schema.Schema{
Type: schema.TypeInt,
Required: true,
ForceNew: false,
},
"to_port": &schema.Schema{
Type: schema.TypeInt,
Required: true,
ForceNew: false,
},
"ip_protocol": &schema.Schema{
Type: schema.TypeString,
Required: true,
ForceNew: false,
},
"cidr": &schema.Schema{
Type: schema.TypeString,
Optional: true,
ForceNew: false,
},
"from_group_id": &schema.Schema{
Type: schema.TypeString,
Optional: true,
ForceNew: false,
},
2015-02-16 22:04:08 +01:00
"self": &schema.Schema{
Type: schema.TypeBool,
Optional: true,
Default: false,
2015-02-16 22:04:08 +01:00
ForceNew: false,
},
},
},
},
2015-01-08 18:47:10 +01:00
},
}
}
2015-01-26 19:09:27 +01:00
func resourceComputeSecGroupV2Create(d *schema.ResourceData, meta interface{}) error {
2015-01-08 18:47:10 +01:00
config := meta.(*Config)
2015-01-31 22:33:54 +01:00
computeClient, err := config.computeV2Client(d.Get("region").(string))
if err != nil {
return fmt.Errorf("Error creating OpenStack compute client: %s", err)
}
2015-01-08 18:47:10 +01:00
createOpts := secgroups.CreateOpts{
Name: d.Get("name").(string),
Description: d.Get("description").(string),
}
2015-03-24 13:59:55 +01:00
log.Printf("[DEBUG] Create Options: %#v", createOpts)
sg, err := secgroups.Create(computeClient, createOpts).Extract()
2015-01-08 18:47:10 +01:00
if err != nil {
return fmt.Errorf("Error creating OpenStack security group: %s", err)
}
d.SetId(sg.ID)
createRuleOptsList := resourceSecGroupRulesV2(d)
for _, createRuleOpts := range createRuleOptsList {
_, err := secgroups.CreateRule(computeClient, createRuleOpts).Extract()
if err != nil {
return fmt.Errorf("Error creating OpenStack security group rule: %s", err)
}
}
2015-01-26 19:09:27 +01:00
return resourceComputeSecGroupV2Read(d, meta)
2015-01-08 18:47:10 +01:00
}
2015-01-26 19:09:27 +01:00
func resourceComputeSecGroupV2Read(d *schema.ResourceData, meta interface{}) error {
2015-01-08 18:47:10 +01:00
config := meta.(*Config)
2015-01-31 22:33:54 +01:00
computeClient, err := config.computeV2Client(d.Get("region").(string))
if err != nil {
return fmt.Errorf("Error creating OpenStack compute client: %s", err)
}
2015-01-08 18:47:10 +01:00
sg, err := secgroups.Get(computeClient, d.Id()).Extract()
2015-01-08 18:47:10 +01:00
if err != nil {
return CheckDeleted(d, err, "security group")
2015-01-08 18:47:10 +01:00
}
d.Set("name", sg.Name)
d.Set("description", sg.Description)
rtm := rulesToMap(sg.Rules)
for _, v := range rtm {
if v["group"] == d.Get("name") {
v["self"] = "1"
} else {
v["self"] = "0"
}
}
log.Printf("[DEBUG] rulesToMap(sg.Rules): %+v", rtm)
d.Set("rule", rtm)
2015-01-08 18:47:10 +01:00
return nil
}
2015-01-26 19:09:27 +01:00
func resourceComputeSecGroupV2Update(d *schema.ResourceData, meta interface{}) error {
2015-01-08 18:47:10 +01:00
config := meta.(*Config)
2015-01-31 22:33:54 +01:00
computeClient, err := config.computeV2Client(d.Get("region").(string))
if err != nil {
return fmt.Errorf("Error creating OpenStack compute client: %s", err)
}
2015-01-08 18:47:10 +01:00
updateOpts := secgroups.UpdateOpts{
2015-01-26 23:45:05 +01:00
Name: d.Get("name").(string),
Description: d.Get("description").(string),
2015-01-08 18:47:10 +01:00
}
log.Printf("[DEBUG] Updating Security Group (%s) with options: %+v", d.Id(), updateOpts)
2015-01-08 18:47:10 +01:00
_, err = secgroups.Update(computeClient, d.Id(), updateOpts).Extract()
if err != nil {
return fmt.Errorf("Error updating OpenStack security group (%s): %s", d.Id(), err)
2015-01-08 18:47:10 +01:00
}
2015-02-01 01:22:19 +01:00
if d.HasChange("rule") {
oldSGRaw, newSGRaw := d.GetChange("rule")
oldSGRSlice, newSGRSlice := oldSGRaw.([]interface{}), newSGRaw.([]interface{})
oldSGRSet := schema.NewSet(secgroupRuleV2Hash, oldSGRSlice)
newSGRSet := schema.NewSet(secgroupRuleV2Hash, newSGRSlice)
secgrouprulesToAdd := newSGRSet.Difference(oldSGRSet)
secgrouprulesToRemove := oldSGRSet.Difference(newSGRSet)
log.Printf("[DEBUG] Security group rules to add: %v", secgrouprulesToAdd)
log.Printf("[DEBUG] Security groups rules to remove: %v", secgrouprulesToRemove)
for _, rawRule := range secgrouprulesToAdd.List() {
createRuleOpts := resourceSecGroupRuleCreateOptsV2(d, rawRule)
rule, err := secgroups.CreateRule(computeClient, createRuleOpts).Extract()
if err != nil {
return fmt.Errorf("Error adding rule to OpenStack security group (%s): %s", d.Id(), err)
}
log.Printf("[DEBUG] Added rule (%s) to OpenStack security group (%s) ", rule.ID, d.Id())
}
for _, r := range secgrouprulesToRemove.List() {
rule := resourceSecGroupRuleV2(d, r)
err := secgroups.DeleteRule(computeClient, rule.ID).ExtractErr()
if err != nil {
errCode, ok := err.(*gophercloud.UnexpectedResponseCodeError)
if !ok {
return fmt.Errorf("Error removing rule (%s) from OpenStack security group (%s): %s", rule.ID, d.Id(), err)
}
if errCode.Actual == 404 {
continue
} else {
return fmt.Errorf("Error removing rule (%s) from OpenStack security group (%s)", rule.ID, d.Id())
}
} else {
log.Printf("[DEBUG] Removed rule (%s) from OpenStack security group (%s): %s", rule.ID, d.Id(), err)
}
}
}
2015-01-26 19:09:27 +01:00
return resourceComputeSecGroupV2Read(d, meta)
2015-01-08 18:47:10 +01:00
}
2015-01-26 19:09:27 +01:00
func resourceComputeSecGroupV2Delete(d *schema.ResourceData, meta interface{}) error {
2015-01-08 18:47:10 +01:00
config := meta.(*Config)
2015-01-31 22:33:54 +01:00
computeClient, err := config.computeV2Client(d.Get("region").(string))
if err != nil {
return fmt.Errorf("Error creating OpenStack compute client: %s", err)
}
2015-01-08 18:47:10 +01:00
err = secgroups.Delete(computeClient, d.Id()).ExtractErr()
2015-01-08 18:47:10 +01:00
if err != nil {
return fmt.Errorf("Error deleting OpenStack security group: %s", err)
}
d.SetId("")
return nil
}
func resourceSecGroupRulesV2(d *schema.ResourceData) []secgroups.CreateRuleOpts {
2015-10-08 14:48:04 +02:00
rawRules := d.Get("rule").([]interface{})
createRuleOptsList := make([]secgroups.CreateRuleOpts, len(rawRules))
for i, raw := range rawRules {
rawMap := raw.(map[string]interface{})
2015-02-16 22:04:08 +01:00
groupId := rawMap["from_group_id"].(string)
if rawMap["self"].(bool) {
groupId = d.Id()
}
createRuleOptsList[i] = secgroups.CreateRuleOpts{
ParentGroupID: d.Id(),
FromPort: rawMap["from_port"].(int),
ToPort: rawMap["to_port"].(int),
IPProtocol: rawMap["ip_protocol"].(string),
CIDR: rawMap["cidr"].(string),
2015-02-16 22:04:08 +01:00
FromGroupID: groupId,
}
}
return createRuleOptsList
}
func resourceSecGroupRuleCreateOptsV2(d *schema.ResourceData, raw interface{}) secgroups.CreateRuleOpts {
rawMap := raw.(map[string]interface{})
2015-02-16 22:04:08 +01:00
groupId := rawMap["from_group_id"].(string)
if rawMap["self"].(bool) {
groupId = d.Id()
}
2015-02-01 01:22:19 +01:00
return secgroups.CreateRuleOpts{
ParentGroupID: d.Id(),
FromPort: rawMap["from_port"].(int),
ToPort: rawMap["to_port"].(int),
IPProtocol: rawMap["ip_protocol"].(string),
CIDR: rawMap["cidr"].(string),
2015-02-16 22:04:08 +01:00
FromGroupID: groupId,
}
}
func resourceSecGroupRuleV2(d *schema.ResourceData, raw interface{}) secgroups.Rule {
rawMap := raw.(map[string]interface{})
return secgroups.Rule{
ID: rawMap["id"].(string),
ParentGroupID: d.Id(),
FromPort: rawMap["from_port"].(int),
ToPort: rawMap["to_port"].(int),
IPProtocol: rawMap["ip_protocol"].(string),
IPRange: secgroups.IPRange{CIDR: rawMap["cidr"].(string)},
}
}
func rulesToMap(sgrs []secgroups.Rule) []map[string]interface{} {
sgrMap := make([]map[string]interface{}, len(sgrs))
for i, sgr := range sgrs {
sgrMap[i] = map[string]interface{}{
"id": sgr.ID,
"from_port": sgr.FromPort,
"to_port": sgr.ToPort,
"ip_protocol": sgr.IPProtocol,
"cidr": sgr.IPRange.CIDR,
"group": sgr.Group.Name,
}
}
return sgrMap
}
func secgroupRuleV2Hash(v interface{}) int {
var buf bytes.Buffer
m := v.(map[string]interface{})
buf.WriteString(fmt.Sprintf("%d-", m["from_port"].(int)))
buf.WriteString(fmt.Sprintf("%d-", m["to_port"].(int)))
buf.WriteString(fmt.Sprintf("%s-", m["ip_protocol"].(string)))
buf.WriteString(fmt.Sprintf("%s-", m["cidr"].(string)))
return hashcode.String(buf.String())
}