statping/core/setup.go

package core

import (
	"github.com/hunterlong/statup/types"
	"github.com/hunterlong/statup/utils"
	"os"
)

func InsertDefaultComms() {
	emailer := SelectCommunication(1)
	if emailer == nil {
		emailer := &types.Communication{
			Method:    "email",
			Removable: false,
			Enabled:   false,
		}
		Create(emailer)
	}
	slack := SelectCommunication(2)
	if slack == nil {
		slack := &types.Communication{
			Method:    "slack",
			Removable: false,
			Enabled:   false,
		}
		Create(slack)
	}
}

func DeleteConfig() {
	err := os.Remove("./config.yml")
	if err != nil {
		utils.Log(3, err)
	}
}

type ErrorResponse struct {
	Error string
}

func LoadSampleData() error {
	utils.Log(1, "Inserting Sample Data...")
	s1 := &Service{
		Name:           "Google",
		Domain:         "https://google.com",
		ExpectedStatus: 200,
		Interval:       10,
		Port:           0,
		Type:           "http",
		Method:         "GET",
	}
	s2 := &Service{
		Name:           "Statup Github",
		Domain:         "https://github.com/hunterlong/statup",
		ExpectedStatus: 200,
		Interval:       30,
		Port:           0,
		Type:           "http",
		Method:         "GET",
	}
	s3 := &Service{
		Name:           "JSON Users Test",
		Domain:         "https://jsonplaceholder.typicode.com/users",
		ExpectedStatus: 200,
		Interval:       60,
		Port:           443,
		Type:           "http",
		Method:         "GET",
	}
	s4 := &Service{
		Name:           "JSON API Tester",
		Domain:         "https://jsonplaceholder.typicode.com/posts",
		ExpectedStatus: 201,
		Expected:       `(title)": "((\\"|[statup])*)"`,
		Interval:       30,
		Type:           "http",
		Method:         "POST",
		PostData:       `{ "title": "statup", "body": "bar", "userId": 19999 }`,
	}
	s1.Create()
	s2.Create()
	s3.Create()
	s4.Create()

	checkin := &Checkin{
		Service:  s2.Id,
		Interval: 30,
		Api:      utils.NewSHA1Hash(18),
	}
	checkin.Create()

	//for i := 0; i < 3; i++ {
	//	s1.Check()
	//	s2.Check()
	//	s3.Check()
	//	s4.Check()
	//}

	utils.Log(1, "Sample data has finished importing")

	return nil
}