go/r3-netstatus/main.go

   1 // (c) Bernhard Tittelbach, 2013
   2
   3 package main
   4
   5 import (
   6         "flag"
   7         "fmt"
   8         "time"
   9
  10         "./r3xmppbot"
  11         pubsub "github.com/tuxychandru/pubsub"
  12         //~ "./brain"
  13         zmq "github.com/vaughan0/go-zmq"
  14         r3events "svn.spreadspace.org/realraum/go.svn/r3events"
  15 )
  16
  17 type SpaceState struct {
  18         present           bool
  19         buttonpress_until int64
  20         door_locked       bool
  21         door_shut         bool
  22 }
  23
  24 var (
  25         xmpp_login_ struct {
  26                 jid  string
  27                 pass string
  28         }
  29         xmpp_bot_authstring_          string
  30         xmpp_state_save_dir_          string
  31         r3eventssub_port_             string
  32         button_press_timeout_         int64 = 3600
  33         brain_connect_addr_           string
  34         enable_syslog_, enable_debug_ bool
  35 )
  36
  37 type EventToXMPPStartupFinished struct{}
  38
  39 //-------
  40
  41 func init() {
  42         flag.StringVar(&xmpp_login_.jid, "xjid", "realrauminfo@realraum.at/Tuer", "XMPP Bot Login JID")
  43         flag.StringVar(&xmpp_login_.pass, "xpass", "", "XMPP Bot Login Password")
  44         flag.StringVar(&xmpp_bot_authstring_, "xbotauth", "", "String that users use to authenticate themselves to the bot")
  45         flag.StringVar(&xmpp_state_save_dir_, "xstatedir", "/flash/var/lib/r3netstatus/", "Directory to save XMPP bot state in")
  46         flag.StringVar(&r3eventssub_port_, "eventsubport", "tcp://torwaechter.realraum.at:4244", "zmq address to subscribe r3events")
  47         flag.StringVar(&brain_connect_addr_, "brainconnect", "tcp://torwaechter.realraum.at:4245", "address to ask about most recent stored events")
  48         flag.BoolVar(&enable_syslog_, "syslog", false, "enable logging to syslog")
  49         flag.BoolVar(&enable_debug_, "debug", false, "enable debug output")
  50         flag.Parse()
  51 }
  52
  53 //-------
  54
  55 func IfThenElseStr(c bool, strue, sfalse string) string {
  56         if c {
  57                 return strue
  58         } else {
  59                 return sfalse
  60         }
  61 }
  62
  63 func composeDoorLockMessage(locked bool, frontshut r3events.DoorAjarUpdate, doorcmd r3events.DoorCommandEvent, ts int64) string {
  64         var ajarstring string = ""
  65         if frontshut.Shut == false && frontshut.Ts < doorcmd.Ts {
  66                 ajarstring = " (still ajar)"
  67         }
  68         if ts-doorcmd.Ts < 30 {
  69                 if len(doorcmd.Who) == 0 || doorcmd.Who == "-" {
  70                         return fmt.Sprintf("The%s frontdoor was %s by %s at %s.", ajarstring, IfThenElseStr(locked, "locked", "unlocked"), doorcmd.Using, time.Unix(ts, 0).String())
  71                 } else {
  72                         return fmt.Sprintf("%s %s the%s frontdoor by %s at %s.", doorcmd.Who, IfThenElseStr(locked, "locked", "unlocked"), ajarstring, doorcmd.Using, time.Unix(ts, 0).String())
  73                 }
  74         } else {
  75                 return fmt.Sprintf("The%s frontdoor was %s manually at %s.", ajarstring, IfThenElseStr(locked, "locked", "unlocked"), time.Unix(ts, 0).String())
  76         }
  77 }
  78
  79 func composePresence(present bool, temp_cx float64, light_lothr, last_buttonpress int64) r3xmppbot.XMPPStatusEvent {
  80         present_msg := "Somebody is present"
  81         notpresent_msg := "Nobody is here"
  82         button_msg := "The button has been pressed :-)"
  83         msg := "%s (CX: %.2f°C, LoTHR light: %d)"
  84
  85         if present {
  86                 if last_buttonpress > 0 && time.Now().Unix()-last_buttonpress < button_press_timeout_ {
  87                         return r3xmppbot.XMPPStatusEvent{r3xmppbot.ShowFreeForChat, fmt.Sprintf(msg, button_msg, temp_cx, light_lothr)}
  88                 } else {
  89                         return r3xmppbot.XMPPStatusEvent{r3xmppbot.ShowOnline, fmt.Sprintf(msg, present_msg, temp_cx, light_lothr)}
  90                 }
  91         } else {
  92                 return r3xmppbot.XMPPStatusEvent{r3xmppbot.ShowNotAvailabe, fmt.Sprintf(msg, notpresent_msg, temp_cx, light_lothr)}
  93         }
  94 }
  95
  96 func EventToXMPP(bot *r3xmppbot.XmppBot, events <-chan interface{}, xmpp_presence_events_chan chan<- interface{}) {
  97         button_msg := "Dooom ! The button has been pressed ! Propably someone is bored and in need of company ! ;-)"
  98         defer func() {
  99                 if x := recover(); x != nil {
 100                         //defer ist called _after_ EventToXMPP function has returned. Thus we recover after returning from this function.
 101                         Syslog_.Printf("handleIncomingXMPPStanzas: run time panic: %v", x)
 102                 }
 103         }()
 104
 105         var present, frontlock bool = false, true
 106         var last_buttonpress, light_lothr int64 = 0, 0
 107         var temp_cx float64 = 0.0
 108         var last_door_cmd r3events.DoorCommandEvent
 109         var last_frontdoor_ajar r3events.DoorAjarUpdate = r3events.DoorAjarUpdate{true, 0}
 110         var standard_distribute_level r3xmppbot.R3JIDDesire = r3xmppbot.R3DebugInfo // initial state, changed after startup finished event recieved
 111
 112         xmpp_presence_events_chan <- r3xmppbot.XMPPStatusEvent{r3xmppbot.ShowNotAvailabe, "Nobody is here"}
 113
 114         for eventinterface := range events {
 115                 Debug_.Printf("event2xmpp: %T %+v", eventinterface, eventinterface)
 116                 switch event := eventinterface.(type) {
 117                 case EventToXMPPStartupFinished:
 118                         //after we received all events from QueryLatestEventsAndInjectThem, we get this event and start sending new events normally
 119                         standard_distribute_level = r3xmppbot.R3OnlineOnlyInfo
 120                 case r3events.PresenceUpdate:
 121                         present = event.Present
 122                         if !present {
 123                                 last_buttonpress = 0
 124                         }
 125                         xmpp_presence_events_chan <- composePresence(present, temp_cx, light_lothr, last_buttonpress)
 126                 case r3events.DoorCommandEvent:
 127                         last_door_cmd = event
 128                         xmpp_presence_events_chan <- fmt.Sprintln("DoorCommand:", event.Command, "using", event.Using, "by", event.Who, time.Unix(event.Ts, 0))
 129                 case r3events.DoorLockUpdate:
 130                         if frontlock != event.Locked {
 131                                 xmpp_presence_events_chan <- r3xmppbot.XMPPMsgEvent{Msg: composeDoorLockMessage(event.Locked, last_frontdoor_ajar, last_door_cmd, event.Ts), DistributeLevel: standard_distribute_level, RememberAsStatus: true}
 132                         }
 133                         frontlock = event.Locked
 134                 case r3events.DoorAjarUpdate:
 135                         if last_frontdoor_ajar.Shut != event.Shut {
 136                                 xmpp_presence_events_chan <- r3xmppbot.XMPPMsgEvent{Msg: fmt.Sprintf("Frontdoor is %s  (%s)", IfThenElseStr(event.Shut, "now shut.", "ajar."), time.Unix(event.Ts, 0).String()), DistributeLevel: r3xmppbot.R3DebugInfo, RememberAsStatus: false}
 137                         }
 138                         last_frontdoor_ajar = event
 139                 case r3events.BackdoorAjarUpdate:
 140                         xmpp_presence_events_chan <- r3xmppbot.XMPPMsgEvent{Msg: fmt.Sprintf("Backdoor is %s  (%s)", IfThenElseStr(event.Shut, "now shut.", "ajar!"), time.Unix(event.Ts, 0).String()), DistributeLevel: standard_distribute_level, RememberAsStatus: false}
 141                 case r3events.GasLeakAlert:
 142                         xmpp_presence_events_chan <- r3xmppbot.XMPPMsgEvent{Msg: fmt.Sprintf("ALERT !! GasLeak Detected !!! (%s)", time.Unix(event.Ts, 0).String()), DistributeLevel: r3xmppbot.R3NeverInfo, RememberAsStatus: false}
 143                 case r3events.IlluminationSensorUpdate:
 144                         light_lothr = event.Value
 145                 case r3events.TempSensorUpdate:
 146                         if event.Sensorindex == 1 {
 147                                 temp_cx = event.Value
 148                         }
 149                 case r3events.BoreDoomButtonPressEvent:
 150                         last_buttonpress = event.Ts
 151                         xmpp_presence_events_chan <- composePresence(present, temp_cx, light_lothr, last_buttonpress)
 152                         xmpp_presence_events_chan <- r3xmppbot.XMPPMsgEvent{Msg: button_msg, DistributeLevel: standard_distribute_level}
 153                 case r3events.TimeTick:
 154                         // update presence text with sensor and button info
 155                         xmpp_presence_events_chan <- composePresence(present, temp_cx, light_lothr, last_buttonpress)
 156                         // Try to XMPP Ping the server and if that fails, quit XMPPBot
 157                         if bot.PingServer(900) == false && bot.PingServer(900) == false && bot.PingServer(900) == false && bot.PingServer(900) == false && bot.PingServer(900) == false {
 158                                 return
 159                         }
 160                 case r3events.DoorProblemEvent:
 161                         xmpp_presence_events_chan <- r3xmppbot.XMPPMsgEvent{Msg: fmt.Sprintf("Door Problem: %s. SeverityLevel: %d (%s)", event.Problem, event.Severity, time.Unix(event.Ts, 0).String()), DistributeLevel: r3xmppbot.R3OnlineOnlyInfo, RememberAsStatus: false}
 162                 }
 163         }
 164 }
 165
 166 func RunXMPPBot(ps *pubsub.PubSub, zmqctx *zmq.Context) {
 167         var xmpperr error
 168         var bot *r3xmppbot.XmppBot
 169         var xmpp_presence_events_chan chan interface{}
 170         for {
 171                 bot, xmpp_presence_events_chan, xmpperr = r3xmppbot.NewStartedBot(xmpp_login_.jid, xmpp_login_.pass, xmpp_bot_authstring_, xmpp_state_save_dir_, true)
 172                 if xmpperr == nil {
 173                         Syslog_.Printf("Successfully (re)started XMPP Bot")
 174                         // subscribe before QueryLatestEventsAndInjectThem and EventToXMPP
 175                         psevents := ps.Sub("presence", "door", "buttons", "updateinterval", "sensors", "xmppmeta")
 176                         QueryLatestEventsAndInjectThem(ps, zmqctx)
 177                         ps.Pub(EventToXMPPStartupFinished{}, "xmppmeta")
 178                         EventToXMPP(bot, psevents, xmpp_presence_events_chan)
 179                         // unsubscribe right away, since we don't known when reconnect will succeed and we don't want to block PubSub
 180                         ps.Unsub(psevents, "presence", "door", "buttons", "updateinterval", "sensors", "xmppmeta")
 181                         bot.StopBot()
 182                 } else {
 183                         Syslog_.Printf("Error starting XMPP Bot: %s", xmpperr.Error())
 184                 }
 185                 time.Sleep(20 * time.Second)
 186         }
 187 }
 188
 189 func ParseZMQr3Event(lines [][]byte, ps *pubsub.PubSub) {
 190         evnt, pubsubcat, err := r3events.UnmarshalByteByte2Event(lines)
 191         Debug_.Printf("ParseZMQr3Event: %s %s %s", evnt, pubsubcat, err)
 192         if err != nil {
 193                 return
 194         }
 195         ps.Pub(evnt, pubsubcat)
 196 }
 197
 198 func QueryLatestEventsAndInjectThem(ps *pubsub.PubSub, zmqctx *zmq.Context) {
 199         answ := ZmqsAskQuestionsAndClose(zmqctx, brain_connect_addr_, [][][]byte{
 200                 [][]byte{[]byte("BackdoorAjarUpdate")},
 201                 [][]byte{[]byte("DoorCommandEvent")},
 202                 [][]byte{[]byte("DoorLockUpdate")},
 203                 [][]byte{[]byte("DoorAjarUpdate")},
 204                 [][]byte{[]byte("PresenceUpdate")},
 205                 [][]byte{[]byte("IlluminationSensorUpdate")},
 206                 [][]byte{[]byte("TempSensorUpdate")}})
 207         for _, a := range answ {
 208                 ParseZMQr3Event(a, ps)
 209         }
 210 }
 211
 212 func main() {
 213         if enable_syslog_ {
 214                 LogEnableSyslog()
 215                 r3xmppbot.LogEnableSyslog()
 216         }
 217         if enable_debug_ {
 218                 LogEnableDebuglog()
 219                 r3xmppbot.LogEnableDebuglog()
 220         }
 221         Syslog_.Print("started")
 222         defer Syslog_.Print("exiting")
 223         zmqctx, zmqsub := ZmqsInit(r3eventssub_port_)
 224         defer zmqctx.Close()
 225         if zmqsub != nil {
 226                 defer zmqsub.Close()
 227         }
 228         if zmqsub == nil {
 229                 panic("zmq sockets must not be nil !!")
 230         }
 231
 232         ps := pubsub.New(10)
 233         defer ps.Shutdown()
 234         //~ brn := brain.New()
 235         //~ defer brn.Shutdown()
 236
 237         go EventToWeb(ps)
 238         // --- get update on most recent events ---
 239         QueryLatestEventsAndInjectThem(ps, zmqctx)
 240         go RunXMPPBot(ps, zmqctx)
 241
 242         // --- receive and distribute events ---
 243         ticker := time.NewTicker(time.Duration(6) * time.Minute)
 244         for {
 245                 select {
 246                 case e := <-zmqsub.In():
 247                         ParseZMQr3Event(e, ps) //, brn)
 248                 case <-ticker.C:
 249                         ps.Pub(r3events.TimeTick{time.Now().Unix()}, "updateinterval")
 250                 }
 251         }
 252 }