Merge pull request #39 from hi-artem/feature/custom-rules-datasource

Add Custom Rules Datasource

Author: wfg

prismacloudcompute/data_source_custom_rule.go

@@ -0,0 +1,88 @@
+package prismacloudcompute
+
+import (
+	"fmt"
+
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
+	"github.com/paloaltonetworks/prisma-cloud-compute-go/pcc"
+	"github.com/paloaltonetworks/prisma-cloud-compute-go/rule"
+)
+
+func dataSourceCustomRule() *schema.Resource {
+	return &schema.Resource{
+		Description: "Use this data source to retrieve ID of a custom rule.",
+		Read:        dataSourceCustomRuleRead,
+
+		Schema: map[string]*schema.Schema{
+			"id": {
+				Description: "ID of the custom rule.",
+				Type:        schema.TypeString,
+				Computed:    true,
+			},
+			"prisma_id": {
+				Description: "Prisma Cloud Compute ID of the custom rule.",
+				Type:        schema.TypeInt,
+				Computed:    true,
+			},
+			"description": {
+				Type:        schema.TypeString,
+				Optional:    true,
+				Description: "Free-form text description of the custom rule.",
+			},
+			"message": {
+				Type:        schema.TypeString,
+				Computed:    true,
+				Description: "Message to display for a custom rule event.",
+			},
+			"name": {
+				Type:        schema.TypeString,
+				Required:    true,
+				Description: "Unique custom rule name.",
+			},
+			"script": {
+				Type:        schema.TypeString,
+				Computed:    true,
+				Description: "Custom rule expression.",
+			},
+			"type": {
+				Type:        schema.TypeString,
+				Computed:    true,
+				Description: "Custom rule type. Can be set to 'processes', 'filesystem', 'network-outgoing', 'kubernetes-audit', 'waas-request', or 'waas-response'.",
+			},
+		},
+	}
+}
+
+func dataSourceCustomRuleRead(d *schema.ResourceData, meta interface{}) error {
+	client := meta.(*pcc.Client)
+
+	if name := d.Get("name").(string); name != "" {
+		retrievedCustomRule, err := rule.GetCustomRuleByName(*client, name)
+		if err != nil {
+			return fmt.Errorf("error reading custom rule: %s", err)
+		}
+		if err := d.Set("description", retrievedCustomRule.Description); err != nil {
+			return fmt.Errorf("error reading custom rule: %s", err)
+		}
+		if err := d.Set("prisma_id", retrievedCustomRule.Id); err != nil {
+			return fmt.Errorf("error reading custom rule: %s", err)
+		}
+		if err := d.Set("message", retrievedCustomRule.Message); err != nil {
+			return fmt.Errorf("error reading custom rule: %s", err)
+		}
+		if err := d.Set("name", retrievedCustomRule.Name); err != nil {
+			return fmt.Errorf("error reading custom rule: %s", err)
+		}
+		if err := d.Set("script", retrievedCustomRule.Script); err != nil {
+			return fmt.Errorf("error reading custom rule: %s", err)
+		}
+		if err := d.Set("type", retrievedCustomRule.Type); err != nil {
+			return fmt.Errorf("error reading custom rule: %s", err)
+		}
+		d.SetId(retrievedCustomRule.Name)
+
+		return nil
+	}
+
+	return fmt.Errorf("Missing name parameter")
+}

prismacloudcompute/data_source_custom_rule_test.go

@@ -0,0 +1,34 @@
+package prismacloudcompute
+
+import (
+	"fmt"
+	"testing"
+
+	"github.com/hashicorp/terraform-plugin-sdk/helper/acctest"
+	"github.com/hashicorp/terraform-plugin-sdk/helper/resource"
+)
+
+func TestAccDsCustomRule(t *testing.T) {
+	name := fmt.Sprintf("tf%s", acctest.RandString(6))
+
+	resource.Test(t, resource.TestCase{
+		PreCheck:  func() { testAccPreCheck(t) },
+		Providers: testAccProviders,
+		Steps: []resource.TestStep{
+			{
+				Config: testAccDsCustomRule(name),
+				Check: resource.ComposeAggregateTestCheckFunc(
+					resource.TestCheckResourceAttrSet("data.prismacloudcompute_custom_rule.test", "name"),
+				),
+			},
+		},
+	})
+}
+
+func testAccDsCustomRule(name string) string {
+	return fmt.Sprintf(`
+	data "prismacloudcompute_custom_rule" "test" {
+		name = %q
+	}
+	`, name)
+}

prismacloudcompute/provider.go

@@ -71,7 +71,9 @@ func Provider() *schema.Provider {
 			"prismacloudcompute_credential":                    resourceCredentials(),
 		},
 
-		DataSourcesMap: map[string]*schema.Resource{},
+		DataSourcesMap: map[string]*schema.Resource{
+			"prismacloudcompute_custom_rule": dataSourceCustomRule(),
+		},
 
 		ConfigureFunc: configure,
 	}