Merge pull request #49 from oba11/add-alerts-plugins-condition

add alerts plugins conditions

api/alert_plugins_conditions.go

@@ -0,0 +1,121 @@
+package api
+
+import (
+	"fmt"
+	"net/url"
+	"strconv"
+)
+
+func (c *Client) queryAlertPluginsConditions(policyID int) ([]AlertPluginsCondition, error) {
+	conditions := []AlertPluginsCondition{}
+
+	reqURL, err := url.Parse("/alerts_plugins_conditions.json")
+	if err != nil {
+		return nil, err
+	}
+
+	qs := reqURL.Query()
+	qs.Set("policy_id", strconv.Itoa(policyID))
+
+	reqURL.RawQuery = qs.Encode()
+
+	nextPath := reqURL.String()
+
+	for nextPath != "" {
+		resp := struct {
+			PluginConditions []AlertPluginsCondition `json:"plugins_conditions,omitempty"`
+		}{}
+
+		nextPath, err = c.Do("GET", nextPath, nil, &resp)
+		if err != nil {
+			return nil, err
+		}
+
+		for _, c := range resp.PluginConditions {
+			c.PolicyID = policyID
+		}
+
+		conditions = append(conditions, resp.PluginConditions...)
+	}
+
+	return conditions, nil
+}
+
+// GetAlertPluginsCondition gets information about a plugin alert condition given an ID and policy ID.
+func (c *Client) GetAlertPluginsCondition(policyID, id int) (*AlertPluginsCondition, error) {
+	conditions, err := c.queryAlertPluginsConditions(policyID)
+	if err != nil {
+		return nil, err
+	}
+
+	for _, condition := range conditions {
+		if condition.ID == id {
+			return &condition, nil
+		}
+	}
+
+	return nil, ErrNotFound
+}
+
+// ListAlertPluginsConditions returns Plugin alert conditions for the specified policy.
+func (c *Client) ListAlertPluginsConditions(policyID int) ([]AlertPluginsCondition, error) {
+	return c.queryAlertPluginsConditions(policyID)
+}
+
+// CreateAlertPluginsCondition creates an Plugin alert condition given the passed configuration.
+func (c *Client) CreateAlertPluginsCondition(condition AlertPluginsCondition) (*AlertPluginsCondition, error) {
+	policyID := condition.PolicyID
+
+	req := struct {
+		Condition AlertPluginsCondition `json:"plugins_condition"`
+	}{
+		Condition: condition,
+	}
+
+	resp := struct {
+		Condition AlertPluginsCondition `json:"plugins_condition,omitempty"`
+	}{}
+
+	u := &url.URL{Path: fmt.Sprintf("/alerts_plugins_conditions/policies/%v.json", policyID)}
+	_, err := c.Do("POST", u.String(), req, &resp)
+	if err != nil {
+		return nil, err
+	}
+
+	resp.Condition.PolicyID = policyID
+
+	return &resp.Condition, nil
+}
+
+// UpdateAlertPluginsCondition updates a Plugin alert condition with the specified changes.
+func (c *Client) UpdateAlertPluginsCondition(condition AlertPluginsCondition) (*AlertPluginsCondition, error) {
+	policyID := condition.PolicyID
+	id := condition.ID
+
+	req := struct {
+		Condition AlertPluginsCondition `json:"plugins_condition"`
+	}{
+		Condition: condition,
+	}
+
+	resp := struct {
+		Condition AlertPluginsCondition `json:"plugins_condition,omitempty"`
+	}{}
+
+	u := &url.URL{Path: fmt.Sprintf("/alerts_plugins_conditions/%v.json", id)}
+	_, err := c.Do("PUT", u.String(), req, &resp)
+	if err != nil {
+		return nil, err
+	}
+
+	resp.Condition.PolicyID = policyID
+
+	return &resp.Condition, nil
+}
+
+// DeleteAlertPluginsCondition removes the Plugin alert condition given the specified ID and policy ID.
+func (c *Client) DeleteAlertPluginsCondition(policyID, id int) error {
+	u := &url.URL{Path: fmt.Sprintf("/alerts_plugins_conditions/%v.json", id)}
+	_, err := c.Do("DELETE", u.String(), nil, nil)
+	return err
+}