enforce ToState when using Duration

LubosD · Oct 31, 2022 · 8a59770 · 8a59770
1 parent b22b544
commit 8a59770
Show file tree

Hide file tree

Showing 4 changed files with 13 additions and 9 deletions.
diff --git a/app.go b/app.go
@@ -75,7 +75,7 @@ func (a *app) RegisterSchedule(s Schedule) {
 	}
 
 	if s.frequency == 0 {
-		log.Fatalln("A schedule must call either Daily() or Every() when built.")
+		panic("A schedule must use either Daily() or Every() when built.")
 	}
 
 	now := time.Now()
@@ -95,6 +95,10 @@ func (a *app) RegisterSchedule(s Schedule) {
 }
 
 func (a *app) RegisterEntityListener(etl EntityListener) {
+	if etl.delay != 0 && etl.toState == "" {
+		panic("EntityListener error: you have to use ToState() when using Duration()")
+	}
+
 	for _, entity := range etl.entityIds {
 		if elList, ok := a.entityListeners[entity]; ok {
 			a.entityListeners[entity] = append(elList, &etl)
@@ -135,7 +139,7 @@ func getSunriseSunset(a *app, sunrise bool, offset []DurationString) carbon.Carb
 	// get next sunrise/sunset time from HA
 	state, err := a.state.Get("sun.sun")
 	if err != nil {
-		log.Fatalln("Couldn't get sun.sun state from HA to calculate", printString)
+		panic("Couldn't get sun.sun state from HA to calculate", printString)
 	}
 
 	nextSetOrRise := carbon.Parse(state.Attributes[attrKey].(string))

diff --git a/entitylistener.go b/entitylistener.go
@@ -67,7 +67,7 @@ type elBuilder1 struct {
 
 func (b elBuilder1) EntityIds(entityIds ...string) elBuilder2 {
 	if len(entityIds) == 0 {
-		log.Fatalln("must pass at least one entityId to EntityIds()")
+		panic("must pass at least one entityId to EntityIds()")
 	} else {
 		b.entityListener.entityIds = entityIds
 	}

diff --git a/internal/http/http.go b/internal/http/http.go
@@ -71,12 +71,12 @@ func get(url, token string) ([]byte, error) {
 // 	defer resp.Body.Close()
 
 // 	if resp.StatusCode == 401 {
-// 		log.Fatalln("ERROR: Auth token is invalid. Please double check it or create a new token in your Home Assistant profile")
+// 		panic("ERROR: Auth token is invalid. Please double check it or create a new token in your Home Assistant profile")
 // 	}
 
 // 	body, err := io.ReadAll(resp.Body)
 // 	if err != nil {
-// 		log.Fatalln(err)
+// 		panic(err)
 // 	}
 
 // 	return body, nil

diff --git a/internal/websocket/websocket.go b/internal/websocket/websocket.go
@@ -59,21 +59,21 @@ func SetupConnection(connString string, authToken string) (*websocket.Conn, cont
 	_, err = ReadMessage(conn, ctx)
 	if err != nil {
 		ctxCancel()
-		log.Fatalln("Unknown error creating websocket client")
+		panic("Unknown error creating websocket client")
 	}
 
 	// Send auth message
 	err = SendAuthMessage(conn, ctx, authToken)
 	if err != nil {
 		ctxCancel()
-		log.Fatalln("Unknown error creating websocket client")
+		panic("Unknown error creating websocket client")
 	}
 
 	// Verify auth message was successful
 	err = VerifyAuthResponse(conn, ctx)
 	if err != nil {
 		ctxCancel()
-		log.Fatalln("ERROR: Auth token is invalid. Please double check it or create a new token in your Home Assistant profile")
+		panic("ERROR: Auth token is invalid. Please double check it or create a new token in your Home Assistant profile")
 	}
 
 	return conn, ctx, ctxCancel
@@ -132,7 +132,7 @@ func SubscribeToEventType(eventType string, conn *websocket.Conn, ctx context.Co
 	}
 	err := WriteMessage(e, conn, ctx)
 	if err != nil {
-		log.Fatalln("Error writing to websocket: ", err)
+		panic("Error writing to websocket: ", err)
 	}
 	// m, _ := ReadMessage(conn, ctx)
 	// log.Default().Println(string(m))