add storm_mode

Author: blampe

energysite.go

@@ -255,6 +255,10 @@ func (s *EnergySite) gridImportExportPath() string {
 	return strings.Join([]string{s.basePath(), "grid_import_export"}, "/")
 }
 
+func (s *EnergySite) stormModePath() string {
+	return strings.Join([]string{s.basePath(), "storm_mode"}, "/")
+}
+
 func (s *EnergySite) SetBatteryReserve(percent uint64) error {
 	url := s.basePath() + "/backup"
 	payload := fmt.Sprintf(`{"backup_reserve_percent":%d}`, percent)
@@ -315,6 +319,22 @@ func (s *EnergySite) SetGridExport(mode string) error {
 	return checkCommandResponse(body)
 }
 
+// SetStormMode enables or disables storm watch mode.
+func (s *EnergySite) SetStormMode(enabled bool) error {
+	payload, err := json.Marshal(map[string]bool{"enabled": enabled})
+	if err != nil {
+		return err
+	}
+	body, err := s.c.post(s.stormModePath(), payload)
+	if err != nil {
+		return err
+	}
+	if len(body) == 0 {
+		return nil
+	}
+	return checkCommandResponse(body)
+}
+
 func checkCommandResponse(body []byte) error {
 	var outer struct {
 		Response json.RawMessage `json:"response"`

energysite_test.go

@@ -158,6 +158,47 @@ func TestSetTariff(t *testing.T) {
 	})
 }
 
+var StormModeCommandResponseJSON = `{"response":{"code":201,"message":"Updated"}}`
+
+func TestSetStormMode(t *testing.T) {
+	ts := serveHTTP(t)
+	defer ts.Close()
+
+	client := NewTestClient(ts)
+
+	var lastBody []byte
+	testMux.HandleFunc("/api/1/energy_sites/12345/storm_mode", func(w http.ResponseWriter, req *http.Request) {
+		body, err := io.ReadAll(req.Body)
+		if err != nil {
+			http.Error(w, err.Error(), http.StatusInternalServerError)
+			return
+		}
+		lastBody = body
+		w.WriteHeader(http.StatusOK)
+		_, _ = w.Write([]byte(StormModeCommandResponseJSON))
+	})
+
+	Convey("Should enable storm mode", t, func() {
+		energySite, err := client.EnergySite(12345)
+		So(err, ShouldBeNil)
+		err = energySite.SetStormMode(true)
+		So(err, ShouldBeNil)
+		var m map[string]interface{}
+		So(json.Unmarshal(lastBody, &m), ShouldBeNil)
+		So(m["enabled"], ShouldEqual, true)
+	})
+
+	Convey("Should disable storm mode", t, func() {
+		energySite, err := client.EnergySite(12345)
+		So(err, ShouldBeNil)
+		err = energySite.SetStormMode(false)
+		So(err, ShouldBeNil)
+		var m map[string]interface{}
+		So(json.Unmarshal(lastBody, &m), ShouldBeNil)
+		So(m["enabled"], ShouldEqual, false)
+	})
+}
+
 func TestEnergySitePaths(t *testing.T) {
 	ts := serveHTTP(t)
 	defer ts.Close()

examples/show_energy_site_status/main.go

@@ -14,6 +14,7 @@ var reservePercentage = flag.Int64("backupPercentage", -1, "backup percentage to
 var operationMode = flag.String("operationMode", "", "operating mode: self_consumption, autonomous, backup")
 var gridExport = flag.String("gridExport", "", "grid export rule: battery_ok, pv_only, never")
 var gridCharging = flag.String("gridCharging", "", "grid charging enabled: true, false")
+var stormMode = flag.String("stormMode", "", "storm mode enabled: true, false")
 
 // example that demos fetching of site information and optionally setting the battery reserve percentage for the site
 func main() {
@@ -24,13 +25,13 @@ func main() {
 		os.Exit(1)
 	}
 
-	if err := run(context.Background(), *tokenPath, *reservePercentage, *operationMode, *gridExport, *gridCharging); err != nil {
+	if err := run(context.Background(), *tokenPath, *reservePercentage, *operationMode, *gridExport, *gridCharging, *stormMode); err != nil {
 		fmt.Println(err)
 		os.Exit(1)
 	}
 }
 
-func run(ctx context.Context, tokenPath string, reservePercentage int64, operationMode, gridExport, gridCharging string) error {
+func run(ctx context.Context, tokenPath string, reservePercentage int64, operationMode, gridExport, gridCharging, stormMode string) error {
 	c, err := tesla.NewClient(ctx, tesla.WithTokenFile(tokenPath))
 	if err != nil {
 		return err
@@ -89,6 +90,13 @@ func run(ctx context.Context, tokenPath string, reservePercentage int64, operati
 					os.Exit(1)
 				}
 			}
+			if stormMode != "" {
+				enabled := stormMode == "true"
+				if err := es.SetStormMode(enabled); err != nil {
+					fmt.Printf("error setting storm mode: %+v\n", err)
+					os.Exit(1)
+				}
+			}
 		}
 	}
 	return nil