feat(schedule): adding crontab logic

go.mod

@@ -33,6 +33,7 @@ require (
 
 require (
 	github.com/inconshreveable/mousetrap v1.1.0 // indirect
+	github.com/robfig/cron/v3 v3.0.1
 	github.com/simonvetter/modbus v1.6.0
 	github.com/spf13/pflag v1.0.5 // indirect
 	github.com/spf13/viper v1.18.2

go.sum

@@ -30,6 +30,8 @@ github.com/pelletier/go-toml/v2 v2.1.1/go.mod h1:tJU2Z3ZkXwnxa4DPO899bsyIoywizdU
 github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
 github.com/pmezard/go-difflib v1.0.1-0.20181226105442-5d4384ee4fb2 h1:Jamvg5psRIccs7FGNTlIRMkT8wgtp5eCXdBlqhYGL6U=
 github.com/pmezard/go-difflib v1.0.1-0.20181226105442-5d4384ee4fb2/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
+github.com/robfig/cron/v3 v3.0.1 h1:WdRxkvbJztn8LMz/QEvLN5sBU+xKpSqwwUO1Pjr4qDs=
+github.com/robfig/cron/v3 v3.0.1/go.mod h1:eQICP3HwyT7UooqI/z+Ov+PtYAWygg1TEWWzGIFLtro=
 github.com/rogpeppe/go-internal v1.9.0 h1:73kH8U+JUqXU8lRuOHeVHaa/SZPifC7BkcraZVejAe8=
 github.com/rogpeppe/go-internal v1.9.0/go.mod h1:WtVeX8xhTBvf0smdhujwtBcq4Qrzq/fJaraNFVN+nFs=
 github.com/russross/blackfriday/v2 v2.1.0/go.mod h1:+Rmxgy9KzJVeS9/2gXHxylqXiyQDYRxCVz55jmeOWTM=

home-assistant/addons/sbam/config.json

@@ -2,10 +2,35 @@
   "name": "sbam",
   "version": "1.3.0",
   "slug": "sbam",
+  "init": "false",
   "description": "Smart Battery Avanced Manager",
   "arch": [
     "amd64"
   ],
   "url": "https://github.com/atbore-phx/sbam",
-  "image": "ghcr.io/atbore-phx/ha-sbam"
+  "image": "ghcr.io/atbore-phx/ha-sbam",
+  "options": {
+    "url": "https://api.solcast.com.au/rooftop_sites/<YOUR-SITE>/forecasts?format=json",
+    "apikey": "null",
+    "fronius_ip": "null",
+    "start_hr": "00:00",
+    "end_hr": "06:00",
+    "crontab": "00 00-05 * * *",
+    "pw_consumption": "11000",
+    "max_charge": "3500",
+    "pw_batt_reserve": "3000",
+    "defaults": "true"
+  },
+  "schema": {
+    "url": "str",
+    "apikey": "password",
+    "fronius_ip": "match(^((25[0-5]|(2[0-4]|1\\d|[1-9]|)\\d)(\\.(?!$)|$)){4}$)",
+    "start_hr": "match(^([01]?[0-9]|2[0-3]):[0-5][0-9]$)",
+    "end_hr": "match(^([01]?[0-9]|2[0-3]):[0-5][0-9]$)",
+    "crontab": "str",
+    "pw_consumption": "float",
+    "max_charge": "float",
+    "pw_batt_reserve": "float",
+    "defaults": "bool"
+  }
 }

home-assistant/addons/sbam/run.sh

@@ -1,3 +1,14 @@
 #!/usr/bin/with-contenv bashio
 
-sbam -h
+export URL=$(bashio::config 'url')
+export APIKEY=$(bashio::config 'apikey')
+export FRONIUS_IP=$(bashio::config 'fronius_ip')
+export START_HR=$(bashio::config 'start_hr')
+export END_HR=$(bashio::config 'end_hr')
+export CRONTAB=$(bashio::config 'crontab')
+export PW_CONSUMPTION=$(bashio::config 'pw_consumption')
+export MAX_CHARGE=$(bashio::config 'max_charge')
+export PW_BATT_RESERVE=$(bashio::config 'pw_batt_reserve')
+export DEFAULTS=$(bashio::config 'defaults')
+
+sbam schedule

pkg/cmd/configure.go

@@ -1,8 +1,9 @@
 package cmd
 
 import (
-	"fmt"
+	"errors"
 	"sbam/pkg/fronius"
+	u "sbam/src/utils"
 	"strings"
 
 	"github.com/spf13/cobra"
@@ -15,20 +16,15 @@ var cfgCmd = &cobra.Command{
 	Long:  `connect via modbus to the fronius inverter and set charging`,
 	Run: func(cmd *cobra.Command, args []string) {
 		fronius_ip := viper.GetString("fronius_ip")
-		if len(strings.TrimSpace(fronius_ip)) == 0 {
-			fmt.Println("The --fronius_ip flag must be set")
+		power := viper.GetViper().GetInt("power")
+
+		err := checkConfigure(fronius_ip)
+		if err != nil {
+			u.Log.Error(err)
 			return
 		}
-		if cmd.Flags().Changed("defaults") {
-			fronius.Setdefaults(fronius_ip)
-		} else if cmd.Flags().Changed("force-charge") {
-			power := viper.GetViper().GetInt("power")
-			if power == 0 {
-				fmt.Println("The --power flag must be set when using --force-charge")
-				return
-			}
-			fronius.ForceCharge(fronius_ip, int16(power))
-		}
+
+		configure(fronius_ip, power, cmd)
 
 	},
 }
@@ -42,3 +38,32 @@ func init() {
 	viper.BindPFlag("power", cfgCmd.Flags().Lookup("power"))
 	rootCmd.AddCommand(cfgCmd)
 }
+
+func checkConfigure(fronius_ip string) error {
+	if len(strings.TrimSpace(fronius_ip)) == 0 {
+		err := errors.New("the --fronius_ip flag must be set")
+		return err
+	}
+	return nil
+}
+
+func configure(fronius_ip string, power int, cmd *cobra.Command) {
+	if cmd.Flags().Changed("defaults") {
+		err := fronius.Setdefaults(fronius_ip)
+		if err != nil {
+			u.Log.Error(err)
+			panic(err)
+		}
+	} else if cmd.Flags().Changed("force-charge") {
+		if power == 0 {
+			u.Log.Error("The --power flag must be set when using --force-charge")
+			return
+		}
+		err := fronius.ForceCharge(fronius_ip, int16(power))
+		if err != nil {
+			u.Log.Error(err)
+			panic(err)
+		}
+	}
+
+}

pkg/cmd/estimate.go

@@ -1,9 +1,10 @@
 package cmd
 
 import (
-	"fmt"
+	"errors"
 	"sbam/pkg/power"
 	"sbam/pkg/storage"
+	u "sbam/src/utils"
 	"strings"
 
 	"github.com/spf13/cobra"
@@ -18,28 +19,14 @@ var estCmd = &cobra.Command{
 		url := viper.GetString("url")
 		apiKey := viper.GetString("apikey")
 		fronius_ip := viper.GetString("fronius_ip")
-		if len(strings.TrimSpace(fronius_ip)) == 0 {
-			fmt.Println("The --fronius_ip flag must be set")
-			return
-		} else if len(strings.TrimSpace(apiKey)) == 0 {
-			fmt.Println("The --apiKey flag must be set")
-			return
-		} else if len(strings.TrimSpace(url)) == 0 {
-			fmt.Println("The --url flag must be set")
-			return
-		}
 
-		pwr := power.New()
-		_, err := pwr.Handler(apiKey, url)
+		err := CheckEstimate(apiKey, url, fronius_ip)
 		if err != nil {
-			panic(err)
+			u.Log.Error(err)
+			return
 		}
+		estimate(apiKey, url, fronius_ip)
 
-		str := storage.New()
-		_, _, err = str.Handler(fronius_ip)
-		if err != nil {
-			panic(err)
-		}
 	},
 }
 
@@ -52,3 +39,33 @@ func init() {
 	viper.BindPFlag("fronius_ip", estCmd.Flags().Lookup("fronius_ip"))
 	rootCmd.AddCommand(estCmd)
 }
+
+func CheckEstimate(apiKey string, url string, fronius_ip string) error {
+	if len(strings.TrimSpace(fronius_ip)) == 0 {
+		err := errors.New("the --fronius_ip flag must be set")
+		return err
+	} else if len(strings.TrimSpace(apiKey)) == 0 {
+		err := errors.New("the --apiKey flag must be set")
+		return err
+	} else if len(strings.TrimSpace(url)) == 0 {
+		err := errors.New("the --url flag must be set")
+		return err
+	}
+	return nil
+}
+
+func estimate(apiKey string, url string, fronius_ip string) {
+	pwr := power.New()
+	_, err := pwr.Handler(apiKey, url)
+	if err != nil {
+		u.Log.Error(err)
+		panic(err)
+	}
+
+	str := storage.New()
+	_, _, err = str.Handler(fronius_ip)
+	if err != nil {
+		u.Log.Error(err)
+		panic(err)
+	}
+}