1 // (c) Bernhard Tittelbach, 2013
11 pubsub "github.com/tuxychandru/pubsub"
13 zmq "github.com/vaughan0/go-zmq"
14 r3events "svn.spreadspace.org/realraum/go.svn/r3events"
17 type SpaceState struct {
19 buttonpress_until int64
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
37 type EventToXMPPStartupFinished struct{}
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")
55 func IfThenElseStr(c bool, strue, sfalse string) string {
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)"
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())
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())
75 return fmt.Sprintf("The%s frontdoor was %s manually at %s.", ajarstring, IfThenElseStr(locked, "locked", "unlocked"), time.Unix(ts, 0).String())
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)"
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)}
89 return r3xmppbot.XMPPStatusEvent{r3xmppbot.ShowOnline, fmt.Sprintf(msg, present_msg, temp_cx, light_lothr)}
92 return r3xmppbot.XMPPStatusEvent{r3xmppbot.ShowNotAvailabe, fmt.Sprintf(msg, notpresent_msg, temp_cx, light_lothr)}
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 ! ;-)"
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)
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
112 xmpp_presence_events_chan <- r3xmppbot.XMPPStatusEvent{r3xmppbot.ShowNotAvailabe, "Nobody is here"}
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
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}
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}
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
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 {
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}
166 func RunXMPPBot(ps *pubsub.PubSub, zmqctx *zmq.Context) {
168 var bot *r3xmppbot.XmppBot
169 var xmpp_presence_events_chan chan interface{}
171 bot, xmpp_presence_events_chan, xmpperr = r3xmppbot.NewStartedBot(xmpp_login_.jid, xmpp_login_.pass, xmpp_bot_authstring_, xmpp_state_save_dir_, true)
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")
183 Syslog_.Printf("Error starting XMPP Bot: %s", xmpperr.Error())
185 time.Sleep(20 * time.Second)
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)
195 ps.Pub(evnt, pubsubcat)
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)
215 r3xmppbot.LogEnableSyslog()
219 r3xmppbot.LogEnableDebuglog()
221 Syslog_.Print("started")
222 defer Syslog_.Print("exiting")
223 zmqctx, zmqsub := ZmqsInit(r3eventssub_port_)
229 panic("zmq sockets must not be nil !!")
234 //~ brn := brain.New()
235 //~ defer brn.Shutdown()
238 // --- get update on most recent events ---
239 QueryLatestEventsAndInjectThem(ps, zmqctx)
240 go RunXMPPBot(ps, zmqctx)
242 // --- receive and distribute events ---
243 ticker := time.NewTicker(time.Duration(6) * time.Minute)
246 case e := <-zmqsub.In():
247 ParseZMQr3Event(e, ps) //, brn)
249 ps.Pub(r3events.TimeTick{time.Now().Unix()}, "updateinterval")