113 lines
2.9 KiB
Go
113 lines
2.9 KiB
Go
package cmd
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/evcc-io/evcc/api"
|
|
"github.com/evcc-io/evcc/util/config"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
// meterCmd represents the meter command
|
|
var meterCmd = &cobra.Command{
|
|
Use: "meter [name]",
|
|
Short: "Query configured meters",
|
|
Args: cobra.MaximumNArgs(1),
|
|
Run: runMeter,
|
|
}
|
|
|
|
func init() {
|
|
rootCmd.AddCommand(meterCmd)
|
|
withCustomTemplate(meterCmd)
|
|
|
|
meterCmd.Flags().StringP(flagBatteryMode, "b", "", flagBatteryModeDescription)
|
|
meterCmd.Flags().DurationP(flagBatteryModeWait, "w", 0, flagBatteryModeWaitDescription)
|
|
meterCmd.Flags().Bool(flagDiagnose, false, flagDiagnoseDescription)
|
|
meterCmd.Flags().IntP(flagCurtail, "u", -1, flagCurtailDescription)
|
|
meterCmd.Flags().IntP(flagDim, "m", -1, flagDimDescription)
|
|
meterCmd.Flags().BoolP(flagRepeat, "r", false, flagRepeatDescription)
|
|
meterCmd.Flags().Duration(flagRepeatInterval, 0, flagRepeatIntervalDescription)
|
|
meterCmd.Flags().Bool(flagHeartbeat, false, flagHeartbeatDescription)
|
|
meterCmd.Flags().Duration(flagTimeout, time.Second, flagTimeoutDescription)
|
|
}
|
|
|
|
func runMeter(cmd *cobra.Command, args []string) {
|
|
// load config
|
|
if err := loadConfigFile(&conf, !cmd.Flag(flagIgnoreDatabase).Changed); err != nil {
|
|
log.FATAL.Fatal(err)
|
|
}
|
|
|
|
// setup environment
|
|
if err := configureEnvironment(cmd, &conf); err != nil {
|
|
log.FATAL.Fatal(err)
|
|
}
|
|
|
|
if err := configureMeters(conf.Meters, args...); err != nil {
|
|
log.FATAL.Fatal(err)
|
|
}
|
|
|
|
mode := api.BatteryUnknown
|
|
if val := cmd.Flag(flagBatteryMode).Value.String(); val != "" {
|
|
var err error
|
|
mode, err = api.BatteryModeString(val)
|
|
if err != nil {
|
|
log.ERROR.Fatalln(err)
|
|
}
|
|
}
|
|
|
|
meters := config.Meters().Devices()
|
|
|
|
var flagUsed bool
|
|
if mode != api.BatteryUnknown {
|
|
flagUsed = true
|
|
|
|
for _, v := range config.Instances(meters) {
|
|
if b, ok := api.Cap[api.BatteryController](v); ok {
|
|
if err := b.SetBatteryMode(mode); err != nil {
|
|
log.FATAL.Fatalln("set battery mode:", err)
|
|
}
|
|
}
|
|
|
|
if d, err := cmd.Flags().GetDuration(flagBatteryModeWait); d > 0 && err == nil {
|
|
log.INFO.Println("waiting for:", d)
|
|
time.Sleep(d)
|
|
}
|
|
}
|
|
}
|
|
|
|
for _, v := range config.Instances(meters) {
|
|
if handleCurtailFlag(cmd, v) {
|
|
flagUsed = true
|
|
}
|
|
|
|
if handleDimFlag(cmd, v) {
|
|
flagUsed = true
|
|
}
|
|
}
|
|
|
|
if !flagUsed {
|
|
timeout, _ := cmd.Flags().GetDuration(flagTimeout)
|
|
d := dumper{len: len(meters), timeout: timeout}
|
|
flag := cmd.Flag(flagDiagnose).Changed
|
|
|
|
REPEAT:
|
|
for _, dev := range meters {
|
|
v := dev.Instance()
|
|
|
|
d.DumpWithHeader(deviceHeader(dev), v)
|
|
if flag {
|
|
d.DumpDiagnosis(v)
|
|
}
|
|
}
|
|
if ok, _ := cmd.Flags().GetBool(flagRepeat); ok {
|
|
if d, err := cmd.Flags().GetDuration(flagRepeatInterval); d > 0 && err == nil {
|
|
log.INFO.Println("waiting for:", d)
|
|
time.Sleep(d)
|
|
}
|
|
goto REPEAT
|
|
}
|
|
} else if ok, _ := cmd.Flags().GetBool(flagHeartbeat); ok {
|
|
log.INFO.Println("running heartbeat (if any) until interrupted (Ctrl-C to stop)")
|
|
time.Sleep(time.Hour)
|
|
}
|
|
}
|