package notifications

import (
	"github.com/statping/statping/database"
)

var (
	db database.Database
)

func SetDB(database database.Database) {
	db = database.Model(&Notification{})
}

func Find(method string) (*Notification, error) {
	var n Notification
	q := db.Where("method = ?", method).Find(&n)
	if q.Error() != nil {
		return nil, q.Error()
	}
	return &n, nil
}

func (n *Notification) Create() error {
	var p Notification
	q := db.Where("method = ?", n.Method).Find(&p)
	if q.RecordNotFound() {
		if err := db.Create(n).Error(); err != nil {
			return err
		}
		return nil
	}
	if p.FailureData.String == "" {
		p.FailureData = n.FailureData
	}
	if p.SuccessData.String == "" {
		p.SuccessData = n.SuccessData
	}
	if err := p.Update(); err != nil {
		return err
	}
	return nil
}

func (n *Notification) UpdateFields(notif *Notification) *Notification {
	n.Id = notif.Id
	n.Limits = notif.Limits
	n.Enabled = notif.Enabled
	n.Host = notif.Host
	n.Port = notif.Port
	n.Username = notif.Username
	n.Password = notif.Password
	n.ApiKey = notif.ApiKey
	n.ApiSecret = notif.ApiSecret
	n.Var1 = notif.Var1
	n.Var2 = notif.Var2
	n.SuccessData = notif.SuccessData
	n.FailureData = notif.FailureData
	return n
}

func (n *Notification) Update() error {
	if err := db.Update(n).Error(); err != nil {
		return err
	}
	return nil
}