webhookd/pkg/notification/http_notifier.go

package notification

import (
	"bytes"
	"encoding/json"
	"net/http"
	"net/url"
	"strconv"
	"strings"

	"github.com/ncarlier/webhookd/pkg/logger"
	"github.com/ncarlier/webhookd/pkg/model"
)

type notifPayload struct {
	ID    string `json:"id"`
	Name  string `json:"name"`
	Text  string `json:"text"`
	Error error  `json:"error,omitempty"`
}

// HTTPNotifier is able to send a notification to a HTTP endpoint.
type HTTPNotifier struct {
	URL          *url.URL
	PrefixFilter string
}

func newHTTPNotifier(uri *url.URL) *HTTPNotifier {
	logger.Info.Println("using HTTP notification system: ", uri.String())
	return &HTTPNotifier{
		URL:          uri,
		PrefixFilter: getValueOrAlt(uri.Query(), "prefix", "notify:"),
	}
}

// Notify send a notification to a HTTP endpoint.
func (n *HTTPNotifier) Notify(work *model.WorkRequest) error {
	payload := work.GetLogContent(n.PrefixFilter)
	if strings.TrimSpace(payload) == "" {
		// Nothing to notify, abort
		return nil
	}

	notif := &notifPayload{
		ID:    strconv.FormatUint(work.ID, 10),
		Name:  work.Name,
		Text:  payload,
		Error: work.Err,
	}
	notifJSON, err := json.Marshal(notif)
	if err != nil {
		return err
	}

	req, err := http.NewRequest("POST", n.URL.String(), bytes.NewBuffer(notifJSON))
	if err != nil {
		return err
	}
	req.Header.Set("Content-Type", "application/json")

	client := &http.Client{}
	resp, err := client.Do(req)
	if err != nil {
		return err
	}
	resp.Body.Close()
	logger.Info.Printf("job %s#%d notification sent to %s\n", work.Name, work.ID, n.URL.String())
	return nil
}