notify.go

package models

import (
	"encoding/json"
	"fmt"
	"os"
	"time"

	"github.com/aws/aws-sdk-go/aws"
	"github.com/aws/aws-sdk-go/service/sns"
	"github.com/convox/rack/client"
	"github.com/ddollar/logger"
)

// NotificationTopic is the SNS topic for notifications
var NotificationTopic = os.Getenv("NOTIFICATION_TOPIC")

// PauseNotifications turns off notifications globally
var PauseNotifications = false

// uniform error handling
func NotifyError(action string, err error, data map[string]string) error {
	data["message"] = err.Error()
	return Notify(action, "error", data)
}

func NotifySuccess(action string, data map[string]string) error {
	return Notify(action, "success", data)
}

func Notify(name, status string, data map[string]string) error {
	if PauseNotifications {
		return nil
	}

	log := logger.New("ns=kernel")
	data["rack"] = os.Getenv("RACK")

	event := &client.NotifyEvent{
		Action:    name,
		Status:    status,
		Data:      data,
		Timestamp: time.Now().UTC(),
	}

	message, err := json.Marshal(event)
	if err != nil {
		return err
	}

	fmt.Printf("models EventSend msg=%q\n", message)

	params := &sns.PublishInput{
		Message:   aws.String(string(message)), // Required
		Subject:   aws.String(name),
		TargetArn: aws.String(NotificationTopic),
	}
	resp, err := SNS().Publish(params)

	if err != nil {
		return err
	}

	log.At("Notify").Log("message-id=%q", *resp.MessageId)

	return nil
}