| 
 | 1 | +package sysdig  | 
 | 2 | + | 
 | 3 | +import (  | 
 | 4 | +	"context"  | 
 | 5 | +	"strconv"  | 
 | 6 | +	"time"  | 
 | 7 | + | 
 | 8 | +	"github.com/hashicorp/terraform-plugin-sdk/v2/diag"  | 
 | 9 | +	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"  | 
 | 10 | + | 
 | 11 | +	"github.com/draios/terraform-provider-sysdig/sysdig/internal/client/monitor"  | 
 | 12 | +)  | 
 | 13 | + | 
 | 14 | +func resourceSysdigMonitorAlertPromql() *schema.Resource {  | 
 | 15 | +	timeout := 5 * time.Minute  | 
 | 16 | + | 
 | 17 | +	return &schema.Resource{  | 
 | 18 | +		CreateContext: resourceSysdigAlertPromqlCreate,  | 
 | 19 | +		UpdateContext: resourceSysdigAlertPromqlUpdate,  | 
 | 20 | +		ReadContext:   resourceSysdigAlertPromqlRead,  | 
 | 21 | +		DeleteContext: resourceSysdigAlertPromqlDelete,  | 
 | 22 | +		Importer: &schema.ResourceImporter{  | 
 | 23 | +			StateContext: schema.ImportStatePassthroughContext,  | 
 | 24 | +		},  | 
 | 25 | + | 
 | 26 | +		Timeouts: &schema.ResourceTimeout{  | 
 | 27 | +			Create: schema.DefaultTimeout(timeout),  | 
 | 28 | +			Update: schema.DefaultTimeout(timeout),  | 
 | 29 | +			Read:   schema.DefaultTimeout(timeout),  | 
 | 30 | +			Delete: schema.DefaultTimeout(timeout),  | 
 | 31 | +		},  | 
 | 32 | + | 
 | 33 | +		Schema: createAlertSchema(map[string]*schema.Schema{  | 
 | 34 | +			"promql": {  | 
 | 35 | +				Type:     schema.TypeString,  | 
 | 36 | +				Required: true,  | 
 | 37 | +			},  | 
 | 38 | +		}),  | 
 | 39 | +	}  | 
 | 40 | +}  | 
 | 41 | + | 
 | 42 | +func resourceSysdigAlertPromqlCreate(ctx context.Context, data *schema.ResourceData, i interface{}) diag.Diagnostics {  | 
 | 43 | +	client, err := i.(SysdigClients).sysdigMonitorClient()  | 
 | 44 | +	if err != nil {  | 
 | 45 | +		return diag.FromErr(err)  | 
 | 46 | +	}  | 
 | 47 | + | 
 | 48 | +	alert, err := promqlAlertFromResourceData(data)  | 
 | 49 | +	if err != nil {  | 
 | 50 | +		return diag.FromErr(err)  | 
 | 51 | +	}  | 
 | 52 | + | 
 | 53 | +	alertCreated, err := client.CreateAlert(ctx, *alert)  | 
 | 54 | +	if err != nil {  | 
 | 55 | +		return diag.FromErr(err)  | 
 | 56 | +	}  | 
 | 57 | + | 
 | 58 | +	data.SetId(strconv.Itoa(alertCreated.ID))  | 
 | 59 | +	data.Set("version", alertCreated.Version)  | 
 | 60 | +	return nil  | 
 | 61 | +}  | 
 | 62 | + | 
 | 63 | +func resourceSysdigAlertPromqlUpdate(ctx context.Context, data *schema.ResourceData, i interface{}) diag.Diagnostics {  | 
 | 64 | +	client, err := i.(SysdigClients).sysdigMonitorClient()  | 
 | 65 | +	if err != nil {  | 
 | 66 | +		return diag.FromErr(err)  | 
 | 67 | +	}  | 
 | 68 | + | 
 | 69 | +	alert, err := promqlAlertFromResourceData(data)  | 
 | 70 | +	if err != nil {  | 
 | 71 | +		return diag.FromErr(err)  | 
 | 72 | +	}  | 
 | 73 | + | 
 | 74 | +	alert.ID, _ = strconv.Atoi(data.Id())  | 
 | 75 | + | 
 | 76 | +	_, err = client.UpdateAlert(ctx, *alert)  | 
 | 77 | +	if err != nil {  | 
 | 78 | +		return diag.FromErr(err)  | 
 | 79 | +	}  | 
 | 80 | + | 
 | 81 | +	return nil  | 
 | 82 | +}  | 
 | 83 | + | 
 | 84 | +func resourceSysdigAlertPromqlRead(ctx context.Context, data *schema.ResourceData, i interface{}) diag.Diagnostics {  | 
 | 85 | +	client, err := i.(SysdigClients).sysdigMonitorClient()  | 
 | 86 | +	if err != nil {  | 
 | 87 | +		return diag.FromErr(err)  | 
 | 88 | +	}  | 
 | 89 | + | 
 | 90 | +	id, err := strconv.Atoi(data.Id())  | 
 | 91 | +	if err != nil {  | 
 | 92 | +		return diag.FromErr(err)  | 
 | 93 | +	}  | 
 | 94 | + | 
 | 95 | +	alert, err := client.GetAlertById(ctx, id)  | 
 | 96 | + | 
 | 97 | +	if err != nil {  | 
 | 98 | +		data.SetId("")  | 
 | 99 | +		return nil  | 
 | 100 | +	}  | 
 | 101 | + | 
 | 102 | +	err = promqlAlertToResourceData(&alert, data)  | 
 | 103 | +	if err != nil {  | 
 | 104 | +		return diag.FromErr(err)  | 
 | 105 | +	}  | 
 | 106 | + | 
 | 107 | +	return nil  | 
 | 108 | +}  | 
 | 109 | + | 
 | 110 | +func resourceSysdigAlertPromqlDelete(ctx context.Context, data *schema.ResourceData, i interface{}) diag.Diagnostics {  | 
 | 111 | +	client, err := i.(SysdigClients).sysdigMonitorClient()  | 
 | 112 | +	if err != nil {  | 
 | 113 | +		return diag.FromErr(err)  | 
 | 114 | +	}  | 
 | 115 | + | 
 | 116 | +	id, err := strconv.Atoi(data.Id())  | 
 | 117 | +	if err != nil {  | 
 | 118 | +		return diag.FromErr(err)  | 
 | 119 | +	}  | 
 | 120 | + | 
 | 121 | +	err = client.DeleteAlert(ctx, id)  | 
 | 122 | +	if err != nil {  | 
 | 123 | +		return diag.FromErr(err)  | 
 | 124 | +	}  | 
 | 125 | + | 
 | 126 | +	return nil  | 
 | 127 | +}  | 
 | 128 | + | 
 | 129 | +func promqlAlertFromResourceData(data *schema.ResourceData) (alert *monitor.Alert, err error) {  | 
 | 130 | +	alert, err = alertFromResourceData(data)  | 
 | 131 | +	if err != nil {  | 
 | 132 | +		return  | 
 | 133 | +	}  | 
 | 134 | + | 
 | 135 | +	alert.Type = "PROMETHEUS"  | 
 | 136 | + | 
 | 137 | +	alert.Condition = data.Get("promql").(string)  | 
 | 138 | + | 
 | 139 | +	return  | 
 | 140 | +}  | 
 | 141 | + | 
 | 142 | +func promqlAlertToResourceData(alert *monitor.Alert, data *schema.ResourceData) (err error) {  | 
 | 143 | +	err = alertToResourceData(alert, data)  | 
 | 144 | +	if err != nil {  | 
 | 145 | +		return  | 
 | 146 | +	}  | 
 | 147 | + | 
 | 148 | +	data.Set("promql", alert.Condition)  | 
 | 149 | + | 
 | 150 | +	return  | 
 | 151 | +}  | 
0 commit comments