mirror of https://github.com/statping/statping
107 lines
2.0 KiB
Go
107 lines
2.0 KiB
Go
package core
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"github.com/hunterlong/statup/types"
|
|
"github.com/hunterlong/statup/utils"
|
|
"net/http"
|
|
"time"
|
|
)
|
|
|
|
type Communication types.Communication
|
|
|
|
func LoadDefaultCommunications() {
|
|
emailer := SelectCommunication(1)
|
|
if emailer.Enabled {
|
|
//LoadMailer(emailer)
|
|
//go EmailerQueue()
|
|
}
|
|
}
|
|
|
|
func LoadComms() {
|
|
for _, c := range CoreApp.Communications {
|
|
if c.Enabled {
|
|
|
|
}
|
|
}
|
|
}
|
|
|
|
func Run(c *Communication) {
|
|
|
|
//sample := &Email{
|
|
// To: "info@socialeck.com",
|
|
// Subject: "Test Email from Statup",
|
|
//}
|
|
|
|
//AddEmail(sample)
|
|
}
|
|
|
|
func SelectAllCommunications() ([]*Communication, error) {
|
|
var c []*Communication
|
|
col := DbSession.Collection("communication").Find()
|
|
err := col.All(&c)
|
|
CoreApp.Communications = c
|
|
return c, err
|
|
}
|
|
|
|
func Create(c *Communication) (int64, error) {
|
|
c.CreatedAt = time.Now()
|
|
uuid, err := DbSession.Collection("communication").Insert(c)
|
|
if err != nil {
|
|
utils.Log(3, err)
|
|
}
|
|
if uuid == nil {
|
|
utils.Log(2, err)
|
|
return 0, err
|
|
}
|
|
c.Id = uuid.(int64)
|
|
if CoreApp != nil {
|
|
CoreApp.Communications = append(CoreApp.Communications, c)
|
|
}
|
|
return uuid.(int64), err
|
|
}
|
|
|
|
func Disable(c *Communication) {
|
|
c.Enabled = false
|
|
Update(c)
|
|
}
|
|
|
|
func Enable(c *Communication) {
|
|
c.Enabled = true
|
|
Update(c)
|
|
}
|
|
|
|
func Update(c *Communication) *Communication {
|
|
col := DbSession.Collection("communication").Find("id", c.Id)
|
|
col.Update(c)
|
|
return c
|
|
}
|
|
|
|
func SelectCommunication(id int64) *Communication {
|
|
for _, c := range CoreApp.Communications {
|
|
if c.Id == id {
|
|
return c
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func SendSlackMessage(msg string) error {
|
|
fullMessage := fmt.Sprintf("{\"text\":\"%v\"}", msg)
|
|
utils.Log(1, fmt.Sprintf("Sending JSON to Slack Webhook: %v", fullMessage))
|
|
slack := SelectCommunication(2)
|
|
if slack == nil {
|
|
utils.Log(3, fmt.Sprintf("Slack communication database entry was not found."))
|
|
return nil
|
|
}
|
|
client := http.Client{
|
|
Timeout: 15 * time.Second,
|
|
}
|
|
_, err := client.Post(slack.Host, "application/json", bytes.NewBuffer([]byte(fullMessage)))
|
|
if err != nil {
|
|
utils.Log(3, err)
|
|
}
|
|
return err
|
|
}
|