highperfocused/GoSungrow
1
0
Fork 0
mirror of https://github.com/MickMake/GoSungrow.git synced 2025-03-27 10:01:43 +01:00
Code Issues Packages Projects Releases Wiki Activity
GoSungrow/cmd/cmd_mqtt.go
MickMake f149090e53 v2.3.1 - More endpoints.
2022-10-12 22:01:47 +11:00

725 lines
18 KiB
Go
Raw Blame History

package cmd
import (
"GoSungrow/Only"
"GoSungrow/iSolarCloud/api"
"GoSungrow/mmHa"
"errors"
"fmt"
"github.com/MickMake/GoUnify/cmdHelp"
"github.com/MickMake/GoUnify/cmdLog"
"github.com/go-co-op/gocron"
"github.com/spf13/cobra"
"github.com/spf13/viper"
"strings"
"time"
)
const (
flagMqttUsername = "mqtt-user"
flagMqttPassword = "mqtt-password"
flagMqttHost = "mqtt-host"
flagMqttPort = "mqtt-port"
)
//goland:noinspection GoNameStartsWithPackageName
type CmdMqtt struct {
CmdDefault
// HASSIO MQTT
MqttUsername string
MqttPassword string
MqttHost string
MqttPort string
Mqtt *mmHa.Mqtt
// SunGrow *iSolarCloud.SunGrow
}
func NewCmdMqtt() *CmdMqtt {
var ret *CmdMqtt
for range Only.Once {
ret = &CmdMqtt {
CmdDefault: CmdDefault {
Error: nil,
cmd: nil,
SelfCmd: nil,
},
}
}
return ret
}
func (c *CmdMqtt) AttachCommand(cmd *cobra.Command) *cobra.Command {
for range Only.Once {
if cmd == nil {
break
}
c.cmd = cmd
// ******************************************************************************** //
var cmdMqtt = &cobra.Command{
Use: "mqtt",
Aliases: []string{""},
Short: fmt.Sprintf("Connect to a HASSIO broker."),
Long: fmt.Sprintf("Connect to a HASSIO broker."),
DisableFlagParsing: false,
DisableFlagsInUseLine: false,
PreRunE: nil,
RunE: c.CmdMqtt,
Args: cobra.MinimumNArgs(1),
}
cmd.AddCommand(cmdMqtt)
cmdMqtt.Example = cmdHelp.PrintExamples(cmdMqtt, "run", "sync")
// ******************************************************************************** //
var cmdMqttRun = &cobra.Command{
Use: "run",
Aliases: []string{""},
Short: fmt.Sprintf("One-off sync to a HASSIO broker."),
Long: fmt.Sprintf("One-off sync to a HASSIO broker."),
DisableFlagParsing: false,
DisableFlagsInUseLine: false,
PreRunE: func(cmd *cobra.Command, args []string) error {
cmds.Error = cmds.ProcessArgs(cmd, args)
if cmds.Error != nil {
return cmds.Error
}
cmds.Error = cmds.SunGrowArgs(cmd, args)
if cmds.Error != nil {
return cmds.Error
}
cmds.Error = cmds.MqttArgs(cmd, args)
if cmds.Error != nil {
return cmds.Error
}
return nil
},
RunE: cmds.CmdMqttRun,
Args: cobra.RangeArgs(0, 1),
}
cmdMqtt.AddCommand(cmdMqttRun)
cmdMqttRun.Example = cmdHelp.PrintExamples(cmdMqttRun, "")
// ******************************************************************************** //
var cmdMqttSync = &cobra.Command{
Use: "sync",
Aliases: []string{""},
Short: fmt.Sprintf("Sync to a HASSIO MQTT broker."),
Long: fmt.Sprintf("Sync to a HASSIO MQTT broker."),
DisableFlagParsing: false,
DisableFlagsInUseLine: false,
PreRunE: func(cmd *cobra.Command, args []string) error {
cmds.Error = cmds.ProcessArgs(cmd, args)
if cmds.Error != nil {
return cmds.Error
}
cmds.Error = cmds.SunGrowArgs(cmd, args)
if cmds.Error != nil {
return cmds.Error
}
cmds.Error = cmds.MqttArgs(cmd, args)
if cmds.Error != nil {
return cmds.Error
}
return nil
},
RunE: cmds.CmdMqttSync,
Args: cobra.RangeArgs(0, 1),
}
cmdMqtt.AddCommand(cmdMqttSync)
cmdMqttSync.Example = cmdHelp.PrintExamples(cmdMqttSync, "", "all")
}
return c.SelfCmd
}
func (c *CmdMqtt) AttachFlags(cmd *cobra.Command, viper *viper.Viper) {
for range Only.Once {
cmd.PersistentFlags().StringVarP(&c.MqttUsername, flagMqttUsername, "", "", fmt.Sprintf("HASSIO: mqtt username."))
viper.SetDefault(flagMqttUsername, "")
cmd.PersistentFlags().StringVarP(&c.MqttPassword, flagMqttPassword, "", "", fmt.Sprintf("HASSIO: mqtt password."))
viper.SetDefault(flagMqttPassword, "")
cmd.PersistentFlags().StringVarP(&c.MqttHost, flagMqttHost, "", "", fmt.Sprintf("HASSIO: mqtt host."))
viper.SetDefault(flagMqttHost, "")
cmd.PersistentFlags().StringVarP(&c.MqttPort, flagMqttPort, "", "", fmt.Sprintf("HASSIO: mqtt port."))
viper.SetDefault(flagMqttPort, "")
}
}
func (ca *Cmds) MqttArgs(cmd *cobra.Command, args []string) error {
for range Only.Once {
cmdLog.LogPrintDate("Connecting to MQTT HASSIO Service...\n")
ca.Mqtt.Mqtt = mmHa.New(mmHa.Mqtt {
ClientId: "GoSungrow",
EntityPrefix: "GoSungrow",
Username: ca.Mqtt.MqttUsername,
Password: ca.Mqtt.MqttPassword,
Host: ca.Mqtt.MqttHost,
Port: ca.Mqtt.MqttPort,
})
ca.Error = ca.Mqtt.Mqtt.GetError()
if ca.Error != nil {
break
}
cmdLog.LogPrintDate("Connecting to SunGrow...\n")
ca.Mqtt.Mqtt.SungrowDevices, ca.Error = ca.Api.SunGrow.GetDevices(true)
if ca.Error != nil {
break
}
cmdLog.LogPrintDate("Found SunGrow %d devices\n", len(ca.Mqtt.Mqtt.SungrowDevices))
ca.Mqtt.Mqtt.DeviceName = "GoSungrow"
ca.Error = ca.Mqtt.Mqtt.SetDeviceConfig(
ca.Mqtt.Mqtt.DeviceName,
ca.Mqtt.Mqtt.DeviceName,
"virtual",
"virtual",
"",
"",
"Roof",
)
if ca.Error != nil {
break
}
for _, psId := range ca.Mqtt.Mqtt.SungrowDevices {
// ca.Error = ca.Mqtt.Mqtt.SetDeviceConfig("GoSungrow", strconv.FormatInt(id, 10), "GoSungrow", model[0], "Sungrow", "Roof")
parent := psId.PsId.String()
if parent == psId.PsKey.Value() {
parent = ca.Mqtt.Mqtt.DeviceName
}
ca.Error = ca.Mqtt.Mqtt.SetDeviceConfig(
"GoSungrow",
parent,
psId.PsKey.Value(),
psId.DeviceName.Value(),
psId.DeviceModel.Value(),
psId.Vendor.Value(),
"Roof",
)
if ca.Error != nil {
break
}
ca.Mqtt.Mqtt.SungrowPsIds[psId.PsId] = true
}
ca.Error = ca.Mqtt.Mqtt.Connect()
if ca.Error != nil {
break
}
// if c.Mqtt.PsId == 0 {
// c.Mqtt.PsId, c.Error = c.Api.SunGrow.GetPsId()
// if c.Error != nil {
// break
// }
// cmdLog.LogPrintDate("Found SunGrow device %d\n", c.Mqtt.PsId)
// }
}
return ca.Error
}
func (c *CmdMqtt) CmdMqtt(cmd *cobra.Command, _ []string) error {
return cmd.Help()
}
func (ca *Cmds) CmdMqttRun(_ *cobra.Command, _ []string) error {
for range Only.Once {
// switch1 := mmMqtt.BinarySensor {
// Device: mmMqtt.Device {
// Connections: [][]string{{"sungrow_address", "0"}},
// Identifiers: []string{"sungrow_bin_sensor_0"},
// Manufacturer: "MickMake",
// Model: "SunGrow inverter",
// Name: "SunGrow inverter online",
// SwVersion: "GoSungrow https://github.com/MickMake/GoSungrow",
// ViaDevice: "GoSungrow",
// },
// Name: "SunGrow inverter online",
// StateTopic: "homeassistant/binary_sensor/GoSungrow_0/state",
// UniqueId: "sungrow_bin_sensor_0",
// }
// err = foo.Publish("homeassistant/binary_sensor/GoSungrow_0/config", 0, true, switch1.Json())
// if err != nil {
// break
// }
// err = foo.Publish("homeassistant/binary_sensor/GoSungrow_0/state", 0, true, "OFF")
// if err != nil {
// break
// }
ca.Error = ca.MqttCron()
if ca.Error != nil {
break
}
cmdLog.LogPrintDate("Starting ticker...\n")
updateCounter := 0
timer := time.NewTicker(60 * time.Second)
for t := range timer.C {
if updateCounter < 5 {
updateCounter++
cmdLog.LogPrintDate("Sleeping: %d\n", updateCounter)
continue
}
updateCounter = 0
cmdLog.LogPrintDate("Update: %s\n", t.String())
ca.Error = ca.MqttCron()
if ca.Error != nil {
break
}
}
}
return ca.Error
}
func (ca *Cmds) CmdMqttSync(_ *cobra.Command, args []string) error {
for range Only.Once {
// */1 * * * * /dir/command args args
cronString := "*/5 * * * *"
if len(args) > 0 {
cronString = strings.Join(args[0:5], " ")
cronString = strings.ReplaceAll(cronString, ".", "*")
}
cron := gocron.NewScheduler(time.UTC)
cron = cron.Cron(cronString)
cron = cron.SingletonMode()
ca.Error = ca.MqttCron()
if ca.Error != nil {
break
}
var job *gocron.Job
job, ca.Error = cron.Do(ca.MqttCron)
if ca.Error != nil {
break
}
job.IsRunning()
cmdLog.LogPrintDate("Created job schedule using '%s'\n", cronString)
cron.StartBlocking()
if ca.Error != nil {
break
}
}
return ca.Error
}
func (ca *Cmds) MqttCron() error {
for range Only.Once {
if ca.Mqtt == nil {
ca.Error = errors.New("mqtt not available")
break
}
if ca.Api.SunGrow == nil {
ca.Error = errors.New("sungrow not available")
break
}
if ca.Mqtt.Mqtt.IsFirstRun() {
ca.Mqtt.Mqtt.UnsetFirstRun()
} else {
time.Sleep(time.Second * 40) // Takes up to 40 seconds for data to come in.
}
newDay := false
if ca.Mqtt.Mqtt.IsNewDay() {
newDay = true
}
for psId, ok := range ca.Mqtt.Mqtt.SungrowPsIds {
if !ok {
continue
}
for _, endpoint := range All {
ca.Error = ca.Update(endpoint, psId, newDay)
if ca.Error != nil {
break
}
}
}
ca.Mqtt.Mqtt.LastRefresh = time.Now()
}
if ca.Error != nil {
cmdLog.LogPrintDate("Error: %s\n", ca.Error)
}
return ca.Error
}
// var All = []string{ "queryDeviceList", "getPsList", "getPsDetailWithPsType" }
var All = []string{ "getPsDetail" }
func (ca *Cmds) Update(endpoint string, psId api.Integer, newDay bool) error {
for range Only.Once {
// Also getPowerStatistics, getHouseholdStoragePsReport, getPsList, getUpTimePoint,
var data api.DataMap
switch endpoint {
case "queryDeviceList":
ep := ca.Api.SunGrow.QueryDeviceList(psId)
data = ep.GetData()
case "getPsList":
ep := ca.Api.SunGrow.GetPsList()
data = ep.GetData()
case "getPsDetailWithPsType":
ep := ca.Api.SunGrow.GetPsDetailWithPsType(psId)
data = ep.GetData()
case "getPsDetail":
ep := ca.Api.SunGrow.GetPsDetail(psId)
data = ep.GetData()
}
cmdLog.LogPrintDate("Syncing %d entries with HASSIO from %s.\n", len(data.DataPoints), endpoint)
for _, o := range data.Order {
entries := data.DataPoints[o]
r := entries.GetEntry(api.LastEntry) // Gets the last entry
if !r.Point.Valid {
fmt.Printf("\nInvalid: %v\n", r)
continue
}
fullId := string(r.FullId)
if r.Point.GroupName == "alias" {
fullId = mmHa.JoinStringsForId(r.Parent.Key, r.Point.Parents.Index[0], string(r.Point.Id))
}
re := mmHa.EntityConfig {
Name: mmHa.JoinStringsForName(" - ", fullId), // r.Point.Name, // PointName,
SubName: "",
ParentId: r.EndPoint,
ParentName: r.Parent.Key,
UniqueId: string(r.Point.Id),
// UniqueId: r.Id,
FullId: fullId, // string(r.FullId), // WAS r.Point.FullId
// FullName: r.Point.Name,
Units: r.Point.Unit,
ValueName: r.Point.Name,
// ValueName: r.Id,
DeviceClass: "",
StateClass: r.Point.TimeSpan,
Value: r.Value,
// Icon: "",
// ValueTemplate: "",
// LastReset: "",
// LastResetValueTemplate: "",
}
if newDay {
fmt.Printf("C")
ca.Error = ca.Mqtt.Mqtt.BinarySensorPublishConfig(re)
if ca.Error != nil {
break
}
ca.Error = ca.Mqtt.Mqtt.SensorPublishConfig(re)
if ca.Error != nil {
break
}
}
fmt.Printf("U")
ca.Error = ca.Mqtt.Mqtt.BinarySensorPublishValue(re)
if ca.Error != nil {
break
}
ca.Error = ca.Mqtt.Mqtt.SensorPublishValue(re)
if ca.Error != nil {
break
}
}
fmt.Println()
}
if ca.Error != nil {
cmdLog.LogPrintDate("Error: %s\n", ca.Error)
}
return ca.Error
}
// func (ca *Cmds) Update1(psId api.Integer, newDay bool) error {
// for range Only.Once {
// // Also getPowerStatistics, getHouseholdStoragePsReport, getPsList, getUpTimePoint,
// ep := ca.Api.SunGrow.QueryDevice(psId)
// if ep.IsError() {
// ca.Error = ep.GetError()
// break
// }
// data := ep.GetData()
//
// cmdLog.LogPrintDate("Syncing %d entries with HASSIO.\n", len(data.DataPoints))
// for _, o := range data.Order {
// entries := data.DataPoints[o]
// r := entries.GetEntry(api.LastEntry) // Gets the last entry
// if !r.Point.Valid {
// fmt.Printf("\nInvalid: %v\n", r)
// continue
// }
//
// if strings.Contains(strings.ToLower(string(r.Point.Id)), "count") {
// fmt.Sprintf("")
// }
//
// fullId := string(r.FullId)
// if r.Point.GroupName == "alias" {
// fullId = mmHa.JoinStringsForId(r.Parent.Key, r.Point.Parents.Index[0], string(r.Point.Id))
// }
//
// re := mmHa.EntityConfig {
// Name: mmHa.JoinStringsForName(" - ", fullId), // r.Point.Name, // PointName,
// SubName: "",
// ParentId: r.EndPoint,
// ParentName: r.Parent.Key,
// UniqueId: string(r.Point.Id),
// // UniqueId: r.Id,
// FullId: fullId, // string(r.FullId), // WAS r.Point.FullId
// // FullName: r.Point.Name,
// Units: r.Point.Unit,
// ValueName: r.Point.Name,
// // ValueName: r.Id,
// DeviceClass: "",
// StateClass: r.Point.TimeSpan,
// Value: r.Value,
//
// // Icon: "",
// // ValueTemplate: "",
// // LastReset: "",
// // LastResetValueTemplate: "",
// }
//
// if newDay {
// fmt.Printf("C")
// ca.Error = ca.Mqtt.Mqtt.BinarySensorPublishConfig(re)
// if ca.Error != nil {
// break
// }
//
// ca.Error = ca.Mqtt.Mqtt.SensorPublishConfig(re)
// if ca.Error != nil {
// break
// }
// }
//
// fmt.Printf("U")
// ca.Error = ca.Mqtt.Mqtt.BinarySensorPublishValue(re)
// if ca.Error != nil {
// break
// }
//
// ca.Error = ca.Mqtt.Mqtt.SensorPublishValue(re)
// if ca.Error != nil {
// break
// }
// }
// fmt.Println()
// }
//
// if ca.Error != nil {
// cmdLog.LogPrintDate("Error: %s\n", ca.Error)
// }
// return ca.Error
// }
// func (ca *Cmds) Update2(psId api.Integer, newDay bool) error {
// for range Only.Once {
// ep := ca.Api.SunGrow.QueryPs(psId)
// if ep.IsError() {
// ca.Error = ep.GetError()
// break
// }
// data := ep.GetData()
//
// cmdLog.LogPrintDate("Syncing %d entries with HASSIO.\n", len(data.DataPoints))
// for _, o := range data.Order {
// entries := data.DataPoints[o]
// r := entries.GetEntry(api.LastEntry) // Gets the last entry
// if !r.Point.Valid {
// fmt.Printf("\nInvalid: %v\n", r)
// continue
// }
//
// fullId := string(r.FullId)
// if r.Point.GroupName == "alias" {
// fullId = mmHa.JoinStringsForId(r.Parent.Key, r.Point.Parents.Index[0], string(r.Point.Id))
// }
//
// re := mmHa.EntityConfig {
// Name: mmHa.JoinStringsForName(" - ", fullId), // r.Point.Name, // PointName,
// SubName: "",
// ParentId: r.EndPoint,
// ParentName: r.Parent.Key,
// UniqueId: string(r.Point.Id),
// // UniqueId: r.Id,
// FullId: fullId, // string(r.FullId), // WAS r.Point.FullId
// // FullName: r.Point.Name,
// Units: r.Point.Unit,
// ValueName: r.Point.Name,
// // ValueName: r.Id,
// DeviceClass: "",
// StateClass: r.Point.TimeSpan,
// Value: r.Value,
//
// // Icon: "",
// // ValueTemplate: "",
// // LastReset: "",
// // LastResetValueTemplate: "",
// }
//
// if newDay {
// fmt.Printf("C")
// ca.Error = ca.Mqtt.Mqtt.BinarySensorPublishConfig(re)
// if ca.Error != nil {
// break
// }
//
// ca.Error = ca.Mqtt.Mqtt.SensorPublishConfig(re)
// if ca.Error != nil {
// break
// }
// }
//
// fmt.Printf("U")
// ca.Error = ca.Mqtt.Mqtt.BinarySensorPublishValue(re)
// if ca.Error != nil {
// break
// }
//
// ca.Error = ca.Mqtt.Mqtt.SensorPublishValue(re)
// if ca.Error != nil {
// break
// }
// }
// fmt.Println()
// }
//
// if ca.Error != nil {
// cmdLog.LogPrintDate("Error: %s\n", ca.Error)
// }
// return ca.Error
// }
// func (ca *Cmds) Update3(psId api.Integer, newDay bool) error {
// for range Only.Once {
// // Also getPowerStatistics, getHouseholdStoragePsReport, getPsList, getUpTimePoint,
// ep := ca.Api.SunGrow.QueryPs(psId)
// if ep.IsError() {
// ca.Error = ep.GetError()
// break
// }
// data := ep.GetData()
//
// cmdLog.LogPrintDate("Syncing %d entries with HASSIO.\n", len(data.DataPoints))
// for _, o := range data.Order {
// entries := data.DataPoints[o]
// r := entries.GetEntry(api.LastEntry) // Gets the last entry
// if !r.Point.Valid {
// fmt.Printf("\nInvalid: %v\n", r)
// continue
// }
//
// fullId := string(r.FullId)
// if r.Point.GroupName == "alias" {
// fullId = mmHa.JoinStringsForId(r.Parent.Key, r.Point.Parents.Index[0], string(r.Point.Id))
// }
//
// re := mmHa.EntityConfig {
// Name: mmHa.JoinStringsForName(" - ", fullId), // r.Point.Name, // PointName,
// SubName: "",
// ParentId: r.EndPoint,
// ParentName: r.Parent.Key,
// UniqueId: string(r.Point.Id),
// // UniqueId: r.Id,
// FullId: fullId, // string(r.FullId), // WAS r.Point.FullId
// // FullName: r.Point.Name,
// Units: r.Point.Unit,
// ValueName: r.Point.Name,
// // ValueName: r.Id,
// DeviceClass: "",
// StateClass: r.Point.TimeSpan,
// Value: r.Value,
//
// // Icon: "",
// // ValueTemplate: "",
// // LastReset: "",
// // LastResetValueTemplate: "",
// }
//
// if newDay {
// fmt.Printf("C")
// ca.Error = ca.Mqtt.Mqtt.BinarySensorPublishConfig(re)
// if ca.Error != nil {
// break
// }
//
// ca.Error = ca.Mqtt.Mqtt.SensorPublishConfig(re)
// if ca.Error != nil {
// break
// }
// }
//
// fmt.Printf("U")
// ca.Error = ca.Mqtt.Mqtt.BinarySensorPublishValue(re)
// if ca.Error != nil {
// break
// }
//
// ca.Error = ca.Mqtt.Mqtt.SensorPublishValue(re)
// if ca.Error != nil {
// break
// }
// }
// fmt.Println()
// }
//
// if ca.Error != nil {
// cmdLog.LogPrintDate("Error: %s\n", ca.Error)
// }
// return ca.Error
// }
// func toggle(v string) string {
// switch v {
// case "OFF":
// v = "ON"
// case "ON":
// v = "OFF"
// }
// return v
// }
//
// func randoPercent() string {
// t := time.Now()
// min := 0
// max := t.Second()
// i := (rand.Intn(max - min) + min) * t.Minute() // / float64(max)
// return fmt.Sprintf("%.2f", (float64(i) / 3600) * 100)
// }
//
// func randoKWh() string {
// t := time.Now()
// min := 0
// max := t.Minute()
// i := (rand.Intn(max - min) + min) * t.Second() // / float64(max)
// return fmt.Sprintf("%.2f", (float64(i) / 3600) * 11000)
// }
Reference in New Issue View Git Blame Copy Permalink