statping/core/events.go

package core

import (
	"fmt"
	"github.com/fatih/structs"
	"github.com/hunterlong/statup/notifications"
	"github.com/hunterlong/statup/plugin"
	"github.com/hunterlong/statup/types"
	"upper.io/db.v3/lib/sqlbuilder"
)

func OnLoad(db sqlbuilder.Database) {
	for _, p := range AllPlugins {
		p.OnLoad(db)
	}
}

func OnSuccess(s *Service) {
	for _, p := range AllPlugins {
		p.OnSuccess(structs.Map(s))
	}
}

func OnFailure(s *Service, f FailureData) {
	for _, p := range AllPlugins {
		p.OnFailure(structs.Map(s))
	}

	onFailureEmail(s, f)
	onFailureSlack(s, f)
}

func onFailureSlack(s *Service, f FailureData) {
	slack := SelectCommunication(2)
	if slack.Enabled {
		msg := fmt.Sprintf("Service %v is currently offline! Issue: %v", s.Name, f.Issue)
		notifications.SendSlack(msg)
	}
}

type failedEmail struct {
	Service     *Service
	FailureData FailureData
	Domain      string
}

func onFailureEmail(s *Service, f FailureData) {
	email := SelectCommunication(1)
	if email.Enabled {
		data := failedEmail{s, f, CoreApp.Domain}
		admin, _ := SelectUser(1)
		email := &types.Email{
			To:       admin.Email,
			Subject:  fmt.Sprintf("Service %v is Down", s.Name),
			Template: "failure.html",
			Data:     data,
			From:     email.Var1,
		}
		notifications.SendEmail(EmailBox, email)
	}
}

func OnSettingsSaved(c *Core) {
	for _, p := range AllPlugins {
		p.OnSettingsSaved(structs.Map(c))
	}
}

func OnNewUser(u *User) {
	for _, p := range AllPlugins {
		p.OnNewUser(structs.Map(u))
	}
}

func OnNewService(s *Service) {
	for _, p := range AllPlugins {
		p.OnNewService(structs.Map(s))
	}
}

func OnDeletedService(s *Service) {
	for _, p := range AllPlugins {
		p.OnDeletedService(structs.Map(s))
	}
}

func OnUpdateService(s *Service) {
	for _, p := range AllPlugins {
		p.OnUpdatedService(structs.Map(s))
	}
}

func SelectPlugin(name string) plugin.PluginActions {
	for _, p := range AllPlugins {
		if p.GetInfo().Name == name {
			return p
		}
	}
	return plugin.PluginInfo{}
}