142 lines
3.1 KiB
Go
142 lines
3.1 KiB
Go
package presence
|
|
|
|
import (
|
|
"automation/device"
|
|
"automation/integration/hue"
|
|
"automation/integration/kasa"
|
|
"automation/integration/ntfy"
|
|
"encoding/json"
|
|
"fmt"
|
|
"log"
|
|
"strings"
|
|
"time"
|
|
|
|
paho "github.com/eclipse/paho.mqtt.golang"
|
|
"github.com/kr/pretty"
|
|
)
|
|
|
|
type Presence struct {
|
|
client paho.Client
|
|
hue *hue.Hue
|
|
ntfy *ntfy.Notify
|
|
provider *device.Provider
|
|
|
|
devices map[string]bool
|
|
presence bool
|
|
}
|
|
|
|
type Message struct {
|
|
State bool `json:"state"`
|
|
Updated int64 `json:"updated"`
|
|
}
|
|
|
|
func (p *Presence) devicePresenceHandler(client paho.Client, msg paho.Message) {
|
|
name := strings.Split(msg.Topic(), "/")[2]
|
|
|
|
if len(msg.Payload()) == 0 {
|
|
delete(p.devices, name)
|
|
} else {
|
|
var message Message
|
|
err := json.Unmarshal(msg.Payload(), &message)
|
|
if err != nil {
|
|
log.Println(err)
|
|
return
|
|
}
|
|
|
|
p.devices[name] = message.State
|
|
}
|
|
|
|
present := false
|
|
pretty.Println(p.devices)
|
|
for _, value := range p.devices {
|
|
if value {
|
|
present = true
|
|
break
|
|
}
|
|
}
|
|
|
|
log.Println(present)
|
|
|
|
if p.presence != present {
|
|
p.presence = present
|
|
|
|
msg, err := json.Marshal(Message{
|
|
State: present,
|
|
Updated: time.Now().UnixMilli(),
|
|
})
|
|
|
|
if err != nil {
|
|
log.Println(err)
|
|
}
|
|
|
|
token := client.Publish("automation/presence", 1, true, msg)
|
|
if token.Wait() && token.Error() != nil {
|
|
log.Println(token.Error())
|
|
}
|
|
}
|
|
}
|
|
|
|
func (p *Presence) overallPresenceHandler(client paho.Client, msg paho.Message) {
|
|
if len(msg.Payload()) == 0 {
|
|
// In this case we clear the persistent message
|
|
return
|
|
}
|
|
var message Message
|
|
err := json.Unmarshal(msg.Payload(), &message)
|
|
if err != nil {
|
|
log.Println(err)
|
|
return
|
|
}
|
|
|
|
fmt.Printf("Presence: %t\n", message.State)
|
|
// Notify users of presence update
|
|
p.ntfy.Presence(p.presence)
|
|
|
|
// Set presence on the hue bridge
|
|
p.hue.SetFlag(41, message.State)
|
|
|
|
if !message.State {
|
|
log.Println("Turn off all the devices")
|
|
|
|
// Turn off all devices
|
|
// @TODO Maybe allow for exceptions, could be a list in the config that we check against?
|
|
for _, dev := range p.provider.Devices.Devices {
|
|
switch d := dev.(type) {
|
|
case *kasa.Kasa:
|
|
d.SetState(false)
|
|
case device.ZigbeeDevice:
|
|
d.SetState(false)
|
|
}
|
|
|
|
}
|
|
|
|
// @TODO Turn off nest thermostat
|
|
} else {
|
|
// @TODO Turn on the nest thermostat again
|
|
}
|
|
}
|
|
|
|
func New(client paho.Client, hue *hue.Hue, ntfy *ntfy.Notify, provider *device.Provider) *Presence {
|
|
p := &Presence{client: client, hue: hue, ntfy: ntfy, provider: provider, devices: make(map[string]bool), presence: false}
|
|
|
|
if token := p.client.Subscribe("automation/presence", 1, p.overallPresenceHandler); token.Wait() && token.Error() != nil {
|
|
log.Println(token.Error())
|
|
}
|
|
|
|
if token := p.client.Subscribe("automation/presence/+", 1, p.devicePresenceHandler); token.Wait() && token.Error() != nil {
|
|
log.Println(token.Error())
|
|
}
|
|
|
|
return p
|
|
}
|
|
|
|
func (p *Presence) Delete() {
|
|
if token := p.client.Unsubscribe("automation/presence"); token.Wait() && token.Error() != nil {
|
|
log.Println(token.Error())
|
|
}
|
|
|
|
if token := p.client.Unsubscribe("automation/presence/+"); token.Wait() && token.Error() != nil {
|
|
log.Println(token.Error())
|
|
}
|
|
}
|