From 1cfc5b4bda7014b7f9cb83f3ae825590dbd3bb2e Mon Sep 17 00:00:00 2001
From: hunterlong <info@socialeck.com>
Date: Sun, 2 Aug 2020 22:48:35 -0700
Subject: [PATCH] not finalized, notifications works, additional testing

---
 frontend/src/pages/Settings.vue             |  6 +++--
 handlers/notifications.go                   | 19 ++++++---------
 handlers/notifiers_test.go                  | 27 ++++++++++++++++++---
 notifiers/command.go                        |  4 +++
 notifiers/discord.go                        |  4 +++
 notifiers/email.go                          |  4 +++
 notifiers/gotify.go                         |  4 +++
 notifiers/line_notify.go                    |  4 +++
 notifiers/mobile.go                         |  4 +++
 notifiers/pushover.go                       |  4 +++
 notifiers/slack.go                          |  4 +++
 notifiers/statping_emailer.go               |  4 +++
 notifiers/telegram.go                       |  4 +++
 notifiers/twilio.go                         |  4 +++
 notifiers/webhook.go                        |  4 +++
 types/core/database.go                      |  4 +++
 types/notifications/database.go             |  9 +++----
 types/notifications/hooks.go                |  4 ---
 types/notifications/struct.go               | 11 +++++++++
 types/services/notifier.go                  |  5 +++-
 types/services/services_test.go             |  5 +++-
 types/services/services_test_alerts_test.go |  4 +++
 22 files changed, 114 insertions(+), 28 deletions(-)

diff --git a/frontend/src/pages/Settings.vue b/frontend/src/pages/Settings.vue
index 527cf0c5..4f43c733 100644
--- a/frontend/src/pages/Settings.vue
+++ b/frontend/src/pages/Settings.vue
@@ -20,6 +20,8 @@
 
                     <h6 class="mt-4 text-muted">Notifiers</h6>
 
+                  {{notifiers}}
+
                     <div id="notifiers_tabs">
                         <a v-for="(notifier, index) in notifiers" v-bind:key="`${notifier.method}`" @click.prevent="changeTab" class="nav-link text-capitalize" v-bind:class="{active: liClass(`v-pills-${notifier.method.toLowerCase()}-tab`)}" v-bind:id="`v-pills-${notifier.method.toLowerCase()}-tab`" data-toggle="pill" v-bind:href="`#v-pills-${notifier.method.toLowerCase()}`" role="tab" v-bind:aria-controls="`v-pills-${notifier.method.toLowerCase()}`" aria-selected="false">
                             <font-awesome-icon :icon="iconName(notifier.icon)" class="mr-2"/> {{notifier.title}}
@@ -152,10 +154,10 @@
           }
       },
     mounted() {
-        this.update()
+
       },
     created() {
-          this.update()
+      this.update()
       },
       methods: {
         async update() {
diff --git a/handlers/notifications.go b/handlers/notifications.go
index 457b309f..7d1e9b86 100644
--- a/handlers/notifications.go
+++ b/handlers/notifications.go
@@ -12,12 +12,10 @@ import (
 
 func apiNotifiersHandler(w http.ResponseWriter, r *http.Request) {
 	var notifs []notifications.Notification
-	notifiers := services.AllNotifiers()
-	for _, n := range notifiers {
-		notif := n.Select()
-		notifer, _ := notifications.Find(notif.Method)
-		notif.UpdateFields(notifer)
-		notifs = append(notifs, *notif)
+	for _, n := range services.AllNotifiers() {
+		no := n.Select()
+		notif, _ := notifications.Find(no.Method)
+		notifs = append(notifs, *no.UpdateFields(notif))
 	}
 	sort.Sort(notifications.NotificationOrder(notifs))
 	returnJson(notifs, w, r)
@@ -25,12 +23,12 @@ func apiNotifiersHandler(w http.ResponseWriter, r *http.Request) {
 
 func apiNotifierGetHandler(w http.ResponseWriter, r *http.Request) {
 	vars := mux.Vars(r)
-	notif := services.FindNotifier(vars["notifier"])
-	if notif == nil {
+	notifer := services.FindNotifier(vars["notifier"])
+	if notifer == nil {
 		sendErrorJson(errors.New("could not find notifier"), w, r)
 		return
 	}
-	returnJson(notif, w, r)
+	returnJson(notifer, w, r)
 }
 
 func apiNotifierUpdateHandler(w http.ResponseWriter, r *http.Request) {
@@ -48,8 +46,7 @@ func apiNotifierUpdateHandler(w http.ResponseWriter, r *http.Request) {
 
 	log.Infof("Updating %s Notifier", notifer.Title)
 
-	err = notifer.Update()
-	if err != nil {
+	if err := notifer.Update(); err != nil {
 		sendErrorJson(err, w, r)
 		return
 	}
diff --git a/handlers/notifiers_test.go b/handlers/notifiers_test.go
index 000f9794..08d377eb 100644
--- a/handlers/notifiers_test.go
+++ b/handlers/notifiers_test.go
@@ -5,6 +5,7 @@ import (
 	"github.com/statping/statping/types/services"
 	"github.com/statping/statping/utils"
 	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
 	"testing"
 )
 
@@ -12,7 +13,7 @@ func TestAttachment(t *testing.T) {
 	notifiers.InitNotifiers()
 }
 
-func TestUnAuthenticatedNotifierRoutes(t *testing.T) {
+func TestAuthenticatedNotifierRoutes(t *testing.T) {
 	slackWebhookUrl := utils.Params.GetString("SLACK_URL")
 
 	tests := []HTTPTest{
@@ -42,7 +43,12 @@ func TestUnAuthenticatedNotifierRoutes(t *testing.T) {
 			URL:            "/api/notifier/slack",
 			Method:         "GET",
 			ExpectedStatus: 200,
-			BeforeTest:     SetTestENV,
+			BeforeTest: func(t *testing.T) error {
+				notif := services.FindNotifier("slack")
+				require.NotNil(t, notif)
+				assert.Equal(t, "slack", notif.Method)
+				return SetTestENV(t)
+			},
 		},
 		{
 			Name:   "No Authentication - Update Notifier",
@@ -68,7 +74,20 @@ func TestUnAuthenticatedNotifierRoutes(t *testing.T) {
 					"limits": 55
 				}`,
 			ExpectedStatus: 200,
-			BeforeTest:     SetTestENV,
+			BeforeTest: func(t *testing.T) error {
+				notif := services.FindNotifier("slack")
+				require.NotNil(t, notif)
+				assert.Equal(t, "slack", notif.Method)
+				assert.False(t, notif.Enabled.Bool)
+				return SetTestENV(t)
+			},
+			AfterTest: func(t *testing.T) error {
+				notif := services.FindNotifier("slack")
+				require.NotNil(t, notif)
+				assert.Equal(t, "slack", notif.Method)
+				assert.True(t, notif.Enabled.Bool)
+				return UnsetTestENV(t)
+			},
 		},
 		{
 			Name:   "Test Notifier (OnSuccess)",
@@ -154,7 +173,7 @@ func TestApiNotifiersRoutes(t *testing.T) {
 			URL:              "/api/notifier/slack",
 			Method:           "GET",
 			ExpectedStatus:   200,
-			ExpectedContains: []string{`"method":"slack"`},
+			ExpectedContains: []string{`"method":"slack"`, `"host":"https://slack.api/example/12345"`},
 			BeforeTest:       SetTestENV,
 			SecureRoute:      true,
 		},
diff --git a/notifiers/command.go b/notifiers/command.go
index 1c0a59f6..6f482b74 100644
--- a/notifiers/command.go
+++ b/notifiers/command.go
@@ -22,6 +22,10 @@ func (c *commandLine) Select() *notifications.Notification {
 	return c.Notification
 }
 
+func (c *commandLine) Valid(values notifications.Values) error {
+	return nil
+}
+
 var Command = &commandLine{&notifications.Notification{
 	Method:      "command",
 	Title:       "Command",
diff --git a/notifiers/discord.go b/notifiers/discord.go
index 6d9e39c2..acb6d49a 100644
--- a/notifiers/discord.go
+++ b/notifiers/discord.go
@@ -51,6 +51,10 @@ func (d *discord) Select() *notifications.Notification {
 	return d.Notification
 }
 
+func (d *discord) Valid(values notifications.Values) error {
+	return nil
+}
+
 // OnFailure will trigger failing service
 func (d *discord) OnFailure(s services.Service, f failures.Failure) (string, error) {
 	out, err := d.sendRequest(ReplaceVars(d.FailureData.String, s, f))
diff --git a/notifiers/email.go b/notifiers/email.go
index baad95d4..000ccd46 100644
--- a/notifiers/email.go
+++ b/notifiers/email.go
@@ -28,6 +28,10 @@ func (e *emailer) Select() *notifications.Notification {
 	return e.Notification
 }
 
+func (e *emailer) Valid(values notifications.Values) error {
+	return nil
+}
+
 var email = &emailer{&notifications.Notification{
 	Method:      "email",
 	Title:       "SMTP Mail",
diff --git a/notifiers/gotify.go b/notifiers/gotify.go
index 9666e270..19c2fa5f 100644
--- a/notifiers/gotify.go
+++ b/notifiers/gotify.go
@@ -22,6 +22,10 @@ func (g *gotify) Select() *notifications.Notification {
 	return g.Notification
 }
 
+func (g *gotify) Valid(values notifications.Values) error {
+	return nil
+}
+
 var Gotify = &gotify{&notifications.Notification{
 	Method:      "gotify",
 	Title:       "Gotify",
diff --git a/notifiers/line_notify.go b/notifiers/line_notify.go
index b47ae158..b0dbf304 100644
--- a/notifiers/line_notify.go
+++ b/notifiers/line_notify.go
@@ -26,6 +26,10 @@ func (l *lineNotifier) Select() *notifications.Notification {
 	return l.Notification
 }
 
+func (l *lineNotifier) Valid(values notifications.Values) error {
+	return nil
+}
+
 var LineNotify = &lineNotifier{&notifications.Notification{
 	Method:      lineNotifyMethod,
 	Title:       "LINE Notify",
diff --git a/notifiers/mobile.go b/notifiers/mobile.go
index 7ae0b8eb..a47473e8 100644
--- a/notifiers/mobile.go
+++ b/notifiers/mobile.go
@@ -22,6 +22,10 @@ func (m *mobilePush) Select() *notifications.Notification {
 	return m.Notification
 }
 
+func (m *mobilePush) Valid(values notifications.Values) error {
+	return nil
+}
+
 var Mobile = &mobilePush{&notifications.Notification{
 	Method: "mobile",
 	Title:  "Mobile",
diff --git a/notifiers/pushover.go b/notifiers/pushover.go
index b682a8bb..f0eb2222 100644
--- a/notifiers/pushover.go
+++ b/notifiers/pushover.go
@@ -27,6 +27,10 @@ func (t *pushover) Select() *notifications.Notification {
 	return t.Notification
 }
 
+func (t *pushover) Valid(values notifications.Values) error {
+	return nil
+}
+
 var Pushover = &pushover{&notifications.Notification{
 	Method:      "pushover",
 	Title:       "Pushover",
diff --git a/notifiers/slack.go b/notifiers/slack.go
index 704b48dd..f476da66 100644
--- a/notifiers/slack.go
+++ b/notifiers/slack.go
@@ -91,3 +91,7 @@ func (s *slack) OnSuccess(srv services.Service) (string, error) {
 func (s *slack) OnSave() (string, error) {
 	return "", nil
 }
+
+func (s *slack) Valid(values notifications.Values) error {
+	return nil
+}
diff --git a/notifiers/statping_emailer.go b/notifiers/statping_emailer.go
index 52f6d69b..3fc82afc 100644
--- a/notifiers/statping_emailer.go
+++ b/notifiers/statping_emailer.go
@@ -27,6 +27,10 @@ func (s *statpingEmailer) Select() *notifications.Notification {
 	return s.Notification
 }
 
+func (s *statpingEmailer) Valid(values notifications.Values) error {
+	return nil
+}
+
 var statpingMailer = &statpingEmailer{&notifications.Notification{
 	Method:      statpingEmailerName,
 	Title:       "Email",
diff --git a/notifiers/telegram.go b/notifiers/telegram.go
index d0296bf5..72867fbe 100644
--- a/notifiers/telegram.go
+++ b/notifiers/telegram.go
@@ -25,6 +25,10 @@ func (t *telegram) Select() *notifications.Notification {
 	return t.Notification
 }
 
+func (t *telegram) Valid(values notifications.Values) error {
+	return nil
+}
+
 var Telegram = &telegram{&notifications.Notification{
 	Method:      "telegram",
 	Title:       "Telegram",
diff --git a/notifiers/twilio.go b/notifiers/twilio.go
index 47ae2024..0ea1c902 100644
--- a/notifiers/twilio.go
+++ b/notifiers/twilio.go
@@ -26,6 +26,10 @@ func (t *twilio) Select() *notifications.Notification {
 	return t.Notification
 }
 
+func (t *twilio) Valid(values notifications.Values) error {
+	return nil
+}
+
 var Twilio = &twilio{&notifications.Notification{
 	Method:      "twilio",
 	Title:       "Twilio",
diff --git a/notifiers/webhook.go b/notifiers/webhook.go
index 7ef2e60a..36b6c989 100644
--- a/notifiers/webhook.go
+++ b/notifiers/webhook.go
@@ -80,6 +80,10 @@ func (w *webhooker) Select() *notifications.Notification {
 	return w.Notification
 }
 
+func (w *webhooker) Valid(values notifications.Values) error {
+	return nil
+}
+
 func (w *webhooker) sendHttpWebhook(body string) (*http.Response, error) {
 	utils.Log.Infoln(fmt.Sprintf("sending body: '%v' to %v as a %v request", body, w.Host.String, w.Var1.String))
 	client := new(http.Client)
diff --git a/types/core/database.go b/types/core/database.go
index a38f1f88..40197f91 100644
--- a/types/core/database.go
+++ b/types/core/database.go
@@ -48,6 +48,10 @@ func Select() (*Core, error) {
 func (c *Core) Create() error {
 	if c.ApiSecret == "" {
 		c.ApiSecret = utils.RandomString(16)
+		apiEnv := utils.Params.GetString("API_SECRET")
+		if apiEnv != "" {
+			c.ApiSecret = apiEnv
+		}
 	}
 	q := db.Create(c)
 	utils.Log.Infof("API Key created: %s", c.ApiSecret)
diff --git a/types/notifications/database.go b/types/notifications/database.go
index 63d04073..efd32579 100644
--- a/types/notifications/database.go
+++ b/types/notifications/database.go
@@ -1,7 +1,6 @@
 package notifications
 
 import (
-	"errors"
 	"github.com/statping/statping/database"
 )
 
@@ -16,11 +15,10 @@ func SetDB(database database.Database) {
 func Find(method string) (*Notification, error) {
 	var n Notification
 	q := db.Where("method = ?", method).Find(&n)
-	if &n == nil {
-		return nil, errors.New("cannot find notifier")
+	if q.Error() != nil {
+		return nil, q.Error()
 	}
-	n.UpdateFields(&n)
-	return &n, q.Error()
+	return &n, nil
 }
 
 func (n *Notification) Create() error {
@@ -45,6 +43,7 @@ func (n *Notification) Create() error {
 }
 
 func (n *Notification) UpdateFields(notif *Notification) *Notification {
+	n.Id = notif.Id
 	n.Limits = notif.Limits
 	n.Enabled = notif.Enabled
 	n.Host = notif.Host
diff --git a/types/notifications/hooks.go b/types/notifications/hooks.go
index 05bc9890..756a7cf0 100644
--- a/types/notifications/hooks.go
+++ b/types/notifications/hooks.go
@@ -2,13 +2,9 @@ package notifications
 
 import (
 	"github.com/statping/statping/types/metrics"
-	"github.com/statping/statping/utils"
 )
 
-// AfterFind for Notification will set the timezone
 func (n *Notification) AfterFind() (err error) {
-	n.CreatedAt = utils.Now()
-	n.UpdatedAt = utils.Now()
 	metrics.Query("notifier", "find")
 	return
 }
diff --git a/types/notifications/struct.go b/types/notifications/struct.go
index 3310e3fd..b0f2f217 100644
--- a/types/notifications/struct.go
+++ b/types/notifications/struct.go
@@ -61,6 +61,17 @@ func (n *Notification) Logger() *logrus.Logger {
 
 type RunFunc func(interface{}) error
 
+type Values struct {
+	Host      string
+	Port      int64
+	Username  string
+	Password  string
+	Var1      string
+	Var2      string
+	ApiKey    string
+	ApiSecret string
+}
+
 // NotificationForm contains the HTML fields for each variable/input you want the notifier to accept.
 type NotificationForm struct {
 	Type        string   `json:"type"`        // the html input type (text, password, email)
diff --git a/types/services/notifier.go b/types/services/notifier.go
index a51b9a62..884a28ab 100644
--- a/types/services/notifier.go
+++ b/types/services/notifier.go
@@ -20,7 +20,9 @@ func ReturnNotifier(method string) ServiceNotifier {
 func FindNotifier(method string) *notifications.Notification {
 	n := allNotifiers[method]
 	if n != nil {
-		return n.Select()
+		notif := n.Select()
+		no, _ := notifications.Find(notif.Method)
+		return notif.UpdateFields(no)
 	}
 	return nil
 }
@@ -31,4 +33,5 @@ type ServiceNotifier interface {
 	OnTest() (string, error)                             // OnTest is triggered for testing
 	OnSave() (string, error)                             // OnSave is triggered for testing
 	Select() *notifications.Notification                 // OnTest is triggered for testing
+	Valid(notifications.Values) error                    // Valid checks your form values
 }
diff --git a/types/services/services_test.go b/types/services/services_test.go
index 9591fb29..78a6c924 100644
--- a/types/services/services_test.go
+++ b/types/services/services_test.go
@@ -10,6 +10,7 @@ import (
 	"github.com/statping/statping/types/failures"
 	"github.com/statping/statping/types/hits"
 	"github.com/statping/statping/types/incidents"
+	"github.com/statping/statping/types/notifications"
 	"github.com/statping/statping/types/null"
 	"github.com/statping/statping/utils"
 	"github.com/stretchr/testify/assert"
@@ -182,10 +183,11 @@ func startupDb(t *testing.T) {
 	require.Nil(t, err)
 	db, err := database.OpenTester()
 	require.Nil(t, err)
-	db.AutoMigrate(&Service{}, &hits.Hit{}, &checkins.Checkin{}, &checkins.CheckinHit{}, &failures.Failure{}, &incidents.Incident{}, &incidents.IncidentUpdate{})
+	db.AutoMigrate(&Service{}, &notifications.Notification{}, &hits.Hit{}, &checkins.Checkin{}, &checkins.CheckinHit{}, &failures.Failure{}, &incidents.Incident{}, &incidents.IncidentUpdate{})
 	checkins.SetDB(db)
 	failures.SetDB(db)
 	incidents.SetDB(db)
+	notifications.SetDB(db)
 	hits.SetDB(db)
 	SetDB(db)
 
@@ -198,6 +200,7 @@ func startupDb(t *testing.T) {
 	db.Create(&fail2)
 	db.Create(&incident1)
 	db.Create(&incidentUpdate1)
+	db.Create(&incidentUpdate1)
 }
 
 func TestServices(t *testing.T) {
diff --git a/types/services/services_test_alerts_test.go b/types/services/services_test_alerts_test.go
index c9b064c0..2a8e7aa3 100644
--- a/types/services/services_test_alerts_test.go
+++ b/types/services/services_test_alerts_test.go
@@ -322,3 +322,7 @@ func (e *exampleNotifier) OnTest() (string, error) {
 	e.tests++
 	return "", nil
 }
+
+func (e *exampleNotifier) Valid(form notifications.Values) error {
+	return nil
+}