BE-468: Terraform support for Scan Budget APIs

Author: chirag-sumologic

sumologic/provider.go

@@ -121,6 +121,7 @@ func Provider() terraform.ResourceProvider {
 			"sumologic_rum_source":                               resourceSumologicRumSource(),
 			"sumologic_role_v2":                                  resourceSumologicRoleV2(),
 			"sumologic_azure_event_hub_log_source":               resourceSumologicGenericPollingSource(),
+			"sumologic_scan_budget":                              resourceSumologicScanBudget(),
 		},
 		DataSourcesMap: map[string]*schema.Resource{
 			"sumologic_cse_log_mapping_vendor_product": dataSourceCSELogMappingVendorAndProduct(),

sumologic/resource_sumologic_scan_budget.go

@@ -0,0 +1,172 @@
+package sumologic
+
+import (
+	"log"
+
+	"github.com/hashicorp/terraform-plugin-sdk/helper/schema"
+)
+
+func resourceSumologicScanBudget() *schema.Resource {
+	return &schema.Resource{
+		Create: resourceSumologicScanBudgetCreate,
+		Read:   resourceSumologicScanBudgetRead,
+		Update: resourceSumologicScanBudgetUpdate,
+		Delete: resourceSumologicScanBudgetDelete,
+		Importer: &schema.ResourceImporter{
+			State: schema.ImportStatePassthrough,
+		},
+
+		Schema: map[string]*schema.Schema{
+
+			"name": {
+				Type:     schema.TypeString,
+				Required: true,
+				ForceNew: false,
+			},
+
+			"orgId": {
+				Type:     schema.TypeString,
+				Required: true,
+				ForceNew: false,
+			},
+
+			"capacity": {
+				Type:     schema.TypeInt,
+				Required: true,
+				ForceNew: false,
+			},
+
+			"unit": {
+				Type:     schema.TypeString,
+				Required: true,
+				ForceNew: false,
+			},
+
+			"budgetType": {
+				Type:     schema.TypeString,
+				Required: true,
+				ForceNew: false,
+			},
+
+			"window": {
+				Type:     schema.TypeString,
+				Required: true,
+				ForceNew: false,
+			},
+
+			"groupBy": {
+				Type:     schema.TypeString,
+				Required: true,
+				ForceNew: false,
+			},
+
+			"applicableOn": {
+				Type:     schema.TypeString,
+				Required: true,
+				ForceNew: false,
+			},
+
+			"scope": {
+				Type:     schema.TypeMap,
+				Required: true,
+				ForceNew: false,
+				Elem: &schema.Schema{
+					Type: schema.TypeList,
+				},
+			},
+
+			"action": {
+				Type:     schema.TypeString,
+				Required: true,
+				ForceNew: false,
+			},
+
+			"status": {
+				Type:     schema.TypeString,
+				Required: true,
+				ForceNew: false,
+			},
+		},
+	}
+}
+
+func resourceSumologicScanBudgetCreate(d *schema.ResourceData, meta interface{}) error {
+	c := meta.(*Client)
+
+	if d.Id() == "" {
+		scanBudget := resourceToScanBudget(d)
+		id, err := c.CreateScanBudget(scanBudget)
+		if err != nil {
+			return err
+		}
+
+		d.SetId(id)
+	}
+
+	return resourceSumologicScanBudgetRead(d, meta)
+}
+
+func resourceSumologicScanBudgetRead(d *schema.ResourceData, meta interface{}) error {
+	c := meta.(*Client)
+
+	id := d.Id()
+	ScanBudget, err := c.GetScanBudget(id)
+	if err != nil {
+		return err
+	}
+
+	if ScanBudget == nil {
+		log.Printf("[WARN] ScanBudget not found, removing from state: %v - %v", id, err)
+		d.SetId("")
+		return nil
+	}
+
+	d.Set("name", ScanBudget.Name)
+	d.Set("orgId", ScanBudget.OrgID)
+	d.Set("capacity", ScanBudget.Capacity)
+	d.Set("unit", ScanBudget.Unit)
+	d.Set("budgetType", ScanBudget.BudgetType)
+	d.Set("window", ScanBudget.Window)
+	d.Set("applicableOn", ScanBudget.Grouping)
+	d.Set("groupBy", ScanBudget.GroupingEntity)
+	d.Set("action", ScanBudget.Action)
+	d.Set("scope", ScanBudget.Scope)
+	d.Set("status", ScanBudget.Status)
+
+	return nil
+}
+
+func resourceSumologicScanBudgetDelete(d *schema.ResourceData, meta interface{}) error {
+	c := meta.(*Client)
+
+	return c.DeleteScanBudget(d.Id())
+}
+
+func resourceSumologicScanBudgetUpdate(d *schema.ResourceData, meta interface{}) error {
+	c := meta.(*Client)
+
+	ScanBudget := resourceToScanBudget(d)
+	err := c.UpdateScanBudget(ScanBudget)
+	if err != nil {
+		return err
+	}
+
+	return resourceSumologicScanBudgetRead(d, meta)
+}
+
+func resourceToScanBudget(d *schema.ResourceData) ScanBudget {
+	return ScanBudget{
+		ID:             d.Id(),
+		OrgID:          d.Get("orgId").(string),
+		Name:           d.Get("name").(string),
+		Capacity:       d.Get("capacity").(int),
+		Unit:           d.Get("unit").(string),
+		BudgetType:     d.Get("budgetType").(string),
+		Window:         d.Get("window").(string),
+		Grouping:       d.Get("applicableOn").(string),
+		GroupingEntity: d.Get("groupBy").(string),
+		Action:         d.Get("action").(string),
+		Scope:          d.Get("scope").(map[string]interface{}),
+		Status:         d.Get("status").(string),
+	}
+}

sumologic/sumologic_scan_budget.go

@@ -0,0 +1,95 @@
+package sumologic
+
+import (
+	"encoding/json"
+	"fmt"
+)
+
+func (s *Client) CreateScanBudget(scanBudget ScanBudget) (string, error) {
+	urlWithoutParams := "v1/budgets"
+
+	data, err := s.Post(urlWithoutParams, scanBudget)
+	if err != nil {
+		return "", err
+	}
+
+	var createdScanBudget ScanBudget
+
+	err = json.Unmarshal(data, &createdScanBudget)
+	if err != nil {
+		return "", err
+	}
+
+	return createdScanBudget.ID, nil
+}
+
+func (s *Client) GetScanBudget(id string) (*ScanBudget, error) {
+	urlWithoutParams := "v1/budgets/%s"
+	paramString := ""
+	sprintfArgs := []interface{}{}
+	sprintfArgs = append(sprintfArgs, id)
+
+	urlWithParams := fmt.Sprintf(urlWithoutParams+paramString, sprintfArgs...)
+
+	data, _, err := s.Get(urlWithParams)
+	if err != nil {
+		return nil, err
+	}
+	if data == nil {
+		return nil, nil
+	}
+
+	var scanBudget ScanBudget
+
+	err = json.Unmarshal(data, &scanBudget)
+	if err != nil {
+		return nil, err
+	}
+
+	return &scanBudget, nil
+
+}
+
+func (s *Client) DeleteScanBudget(id string) error {
+	urlWithoutParams := "v1/budgets/%s"
+	paramString := ""
+	sprintfArgs := []interface{}{}
+	sprintfArgs = append(sprintfArgs, id)
+
+	urlWithParams := fmt.Sprintf(urlWithoutParams+paramString, sprintfArgs...)
+
+	_, err := s.Delete(urlWithParams)
+
+	return err
+}
+
+func (s *Client) UpdateScanBudget(scanBudget ScanBudget) error {
+	urlWithoutParams := "v1/budgets/%s"
+	paramString := ""
+	sprintfArgs := []interface{}{}
+	sprintfArgs = append(sprintfArgs, scanBudget.ID)
+
+	urlWithParams := fmt.Sprintf(urlWithoutParams+paramString, sprintfArgs...)
+
+	scanBudget.ID = ""
+
+	_, err := s.Put(urlWithParams, scanBudget)
+
+	return err
+
+}
+
+type ScanBudget struct {
+	ID             string                 `json:"id,omitempty"`
+	OrgID          string                 `json:"orgId"`
+	Name           string                 `json:"name"`
+	Capacity       int                    `json:"capacity"`
+	Unit           string                 `json:"unit"`
+	BudgetType     string                 `json:"budgetType,omitempty"`
+	Window         string                 `json:"window"`
+	Grouping       string                 `json:"applicableOn"`
+	GroupingEntity string                 `json:"groupBy"`
+	Action         string                 `json:"action"`
+	Scope          map[string]interface{} `json:"scope"`
+	Status         string                 `json:"status"`
+}