dashboard_widget.go

package client

import (
	"context"
	"fmt"

	"github.com/suzuki-shunsuke/go-graylog"
)

// CreateDashboardWidget creates a new dashboard widget.
func (client *Client) CreateDashboardWidget(dashboardID string, widget graylog.Widget) (graylog.Widget, *ErrorInfo, error) {
	return client.CreateDashboardWidgetContext(context.Background(), dashboardID, widget)
}

// CreateDashboardWidgetContext creates a new dashboard widget with a context.
func (client *Client) CreateDashboardWidgetContext(
	ctx context.Context, dashboardID string, widget graylog.Widget,
) (graylog.Widget, *ErrorInfo, error) {
	if dashboardID == "" {
		return widget, nil, fmt.Errorf("dashboard id is required")
	}

	ret := map[string]string{}
	u, err := client.Endpoints().DashboardWidgets(dashboardID)
	if err != nil {
		return widget, nil, err
	}
	ei, err := client.callPost(ctx, u.String(), widget, &ret)
	if err != nil {
		return widget, ei, err
	}
	if id, ok := ret["widget_id"]; ok {
		widget.ID = id
		return widget, ei, nil
	}
	return widget, ei, fmt.Errorf(`response doesn't have the field "widget_id"`)
}

// DeleteDashboardWidget deletes a given dashboard widget.
func (client *Client) DeleteDashboardWidget(dashboardID, widgetID string) (*ErrorInfo, error) {
	return client.DeleteDashboardWidgetContext(context.Background(), dashboardID, widgetID)
}

// DeleteDashboardWidgetContext deletes a given dashboard widget with a context.
func (client *Client) DeleteDashboardWidgetContext(
	ctx context.Context, dashboardID, widgetID string,
) (*ErrorInfo, error) {
	if dashboardID == "" {
		return nil, fmt.Errorf("dashboard id is required")
	}
	if widgetID == "" {
		return nil, fmt.Errorf("widget id is required")
	}
	u, err := client.Endpoints().DashboardWidget(dashboardID, widgetID)
	if err != nil {
		return nil, err
	}
	return client.callDelete(ctx, u.String(), nil, nil)
}

// GetDashboardWidget gets a dashboard widget.
func (client *Client) GetDashboardWidget(dashboardID, widgetID string) (graylog.Widget, *ErrorInfo, error) {
	return client.GetDashboardWidgetContext(context.Background(), dashboardID, widgetID)
}

// GetDashboardWidgetContext gets a dashboard widget with a context.
func (client *Client) GetDashboardWidgetContext(
	ctx context.Context, dashboardID, widgetID string,
) (graylog.Widget, *ErrorInfo, error) {
	widget := graylog.Widget{}
	if dashboardID == "" {
		return widget, nil, fmt.Errorf("dashboard id is required")
	}
	if widgetID == "" {
		return widget, nil, fmt.Errorf("widget id is required")
	}
	u, err := client.Endpoints().DashboardWidget(dashboardID, widgetID)
	if err != nil {
		return widget, nil, err
	}
	ei, err := client.callGet(ctx, u.String(), nil, &widget)
	return widget, ei, err
}