go/r3-netstatus/main.go

   1 // (c) Bernhard Tittelbach, 2013
   2
   3 package main
   4
   5 import (
   6     "./r3xmppbot"
   7     pubsub "github.com/tuxychandru/pubsub"
   8     "flag"
   9     "time"
  10     "fmt"
  11     //~ "./brain"
  12     r3events "svn.spreadspace.org/realraum/go.svn/r3events"
  13 )
  14
  15 type SpaceState struct {
  16     present           bool
  17     buttonpress_until int64
  18     door_locked bool
  19     door_shut bool
  20 }
  21
  22 var (
  23     xmpp_login_ struct {jid string; pass string}
  24     xmpp_bot_authstring_ string
  25     xmpp_state_save_dir_ string
  26     r3eventssub_port_ string
  27     button_press_timeout_ int64 = 3600
  28     brain_connect_addr_ string
  29     enable_syslog_, enable_debug_ bool
  30 )
  31
  32 //-------
  33
  34 func init() {
  35     flag.StringVar(&xmpp_login_.jid, "xjid", "realrauminfo@realraum.at/Tuer", "XMPP Bot Login JID")
  36     flag.StringVar(&xmpp_login_.pass, "xpass", "", "XMPP Bot Login Password")
  37     flag.StringVar(&xmpp_bot_authstring_, "xbotauth", "", "String that user use to authenticate themselves to the bot")
  38     flag.StringVar(&xmpp_state_save_dir_,"xstatedir","/flash/var/lib/r3netstatus/",  "Directory to save XMPP bot state in")
  39     flag.StringVar(&r3eventssub_port_, "eventsubport", "tcp://wuzzler.realraum.at:4244", "zmq address to subscribe r3events")
  40     flag.StringVar(&brain_connect_addr_, "brainconnect", "tcp://wuzzler.realraum.at:4245", "address to ask about most recent stored events")
  41     flag.BoolVar(&enable_syslog_, "syslog", false, "enable logging to syslog")
  42     flag.BoolVar(&enable_debug_, "debug", false, "enable debug output")
  43     flag.Parse()
  44 }
  45
  46 //-------
  47
  48 func IfThenElseStr(c bool, strue, sfalse string) string {
  49     if c {return strue} else {return sfalse}
  50 }
  51
  52 func composeDoorLockMessage(locked bool, frontshut r3events.DoorAjarUpdate, doorcmd r3events.DoorCommandEvent, ts int64) string {
  53     var ajarstring string = "";
  54     if frontshut.Shut == false && frontshut.Ts < doorcmd.Ts {
  55         ajarstring = " (still ajar)"
  56     }
  57     if ts - doorcmd.Ts < 30 {
  58         if len(doorcmd.Who) == 0 || doorcmd.Who == "-" {
  59             return fmt.Sprintf("The%s frontdoor was %s by %s at %s.",ajarstring, IfThenElseStr(locked, "locked","unlocked"), doorcmd.Using, time.Unix(ts,0).String())
  60         } else {
  61             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())
  62         }
  63     } else {
  64         return fmt.Sprintf("The%s frontdoor was %s manually at %s.", ajarstring, IfThenElseStr(locked, "locked","unlocked"), time.Unix(ts,0).String())
  65     }
  66 }
  67
  68 func EventToXMPP(ps *pubsub.PubSub, xmpp_presence_events_chan chan <- interface{}) {
  69     events := ps.Sub("presence","door","buttons","updateinterval")
  70
  71     defer func() {
  72         ps.Unsub(events, "presence","door","buttons","updateinterval")
  73         if x := recover(); x != nil {
  74             //defer ist called _after_ EventToXMPP function has returned. Thus we recover after returning from this function.
  75             Syslog_.Printf("handleIncomingXMPPStanzas: run time panic: %v", x)
  76         }
  77     }()
  78
  79     var present, frontlock bool = false, true
  80     var last_buttonpress int64 = 0
  81     var last_door_cmd r3events.DoorCommandEvent;
  82     var last_frontdoor_ajar r3events.DoorAjarUpdate = r3events.DoorAjarUpdate{true, 0};
  83     button_msg := "Dooom ! The button has been pressed ! Propably someone is bored and in need of company ! ;-)"
  84     present_status := r3xmppbot.XMPPStatusEvent{r3xmppbot.ShowOnline,"Somebody is present"}
  85     notpresent_status := r3xmppbot.XMPPStatusEvent{r3xmppbot.ShowNotAvailabe,"Nobody is here"}
  86     button_status := r3xmppbot.XMPPStatusEvent{r3xmppbot.ShowFreeForChat, "The button has been pressed :-)"}
  87
  88     xmpp_presence_events_chan <- r3xmppbot.XMPPStatusEvent{r3xmppbot.ShowNotAvailabe, "Nobody is here"}
  89
  90     for eventinterface := range(events) {
  91         Debug_.Printf("event2xmpp: %T %+v", eventinterface, eventinterface)
  92         switch event := eventinterface.(type) {
  93             case r3events.PresenceUpdate:
  94                 present = event.Present
  95                 if present {
  96                     xmpp_presence_events_chan <- present_status
  97                 } else {
  98                     xmpp_presence_events_chan <- notpresent_status
  99                 }
 100             case r3events.DoorCommandEvent:
 101                 last_door_cmd = event
 102                 xmpp_presence_events_chan <- fmt.Sprintln("DoorCommand:",event.Command, "using", event.Using, "by", event.Who, time.Unix(event.Ts,0))
 103             case r3events.DoorLockUpdate:
 104                 if frontlock != event.Locked {
 105                     xmpp_presence_events_chan <- r3xmppbot.XMPPMsgEvent{Msg: composeDoorLockMessage(event.Locked, last_frontdoor_ajar, last_door_cmd, event.Ts), DistributeLevel: r3xmppbot.R3OnlineOnlyInfo, RememberAsStatus: true}
 106                 }
 107                 frontlock = event.Locked
 108            case r3events.DoorAjarUpdate:
 109                 if last_frontdoor_ajar.Shut != event.Shut {
 110                     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}
 111                 }
 112                 last_frontdoor_ajar = event
 113            case r3events.BackdoorAjarUpdate:
 114                 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: r3xmppbot.R3OnlineOnlyInfo, RememberAsStatus: false}
 115             case r3events.BoreDoomButtonPressEvent:
 116                 xmpp_presence_events_chan <- r3xmppbot.XMPPMsgEvent{Msg: button_msg, DistributeLevel: r3xmppbot.R3OnlineOnlyInfo}
 117                 xmpp_presence_events_chan <- button_status
 118                 last_buttonpress = event.Ts
 119             case r3events.TimeTick:
 120                 if present && last_buttonpress > 0 && time.Now().Unix() - last_buttonpress > button_press_timeout_ {
 121                     xmpp_presence_events_chan <- present_status
 122                     last_buttonpress = 0
 123                 }
 124         }
 125         }
 126 }
 127
 128 func RunXMPPBot(ps *pubsub.PubSub) {
 129     var xmpperr error
 130     var bot *r3xmppbot.XmppBot
 131     var xmpp_presence_events_chan chan interface{}
 132     for {
 133         bot, xmpp_presence_events_chan, xmpperr = r3xmppbot.NewStartedBot(xmpp_login_.jid, xmpp_login_.pass, xmpp_bot_authstring_, xmpp_state_save_dir_, true)
 134         if xmpperr == nil {
 135             EventToXMPP(ps, xmpp_presence_events_chan)
 136             bot.StopBot()
 137         } else {
 138             Syslog_.Printf("Error starting XMPP Bot: %s", xmpperr.Error())
 139         }
 140         time.Sleep(5 * time.Second)
 141     }
 142 }
 143
 144 func ParseZMQr3Event(lines [][]byte, ps *pubsub.PubSub) {
 145     evnt, pubsubcat, err := r3events.UnmarshalByteByte2Event(lines)
 146     Debug_.Printf("ParseZMQr3Event: %s %s %s", evnt, pubsubcat, err)
 147     if err != nil { return }
 148     ps.Pub(evnt, pubsubcat)
 149 }
 150
 151 func main() {
 152     if enable_syslog_ { LogEnableSyslog(); r3xmppbot.LogEnableSyslog() }
 153     if enable_debug_ { LogEnableDebuglog(); r3xmppbot.LogEnableDebuglog() }
 154     Syslog_.Print("started")
 155     defer Syslog_.Print("exiting")
 156     zmqctx, zmqsub := ZmqsInit(r3eventssub_port_)
 157     defer zmqctx.Close()
 158     if zmqsub != nil {defer zmqsub.Close()}
 159     if zmqsub == nil {
 160         panic("zmq sockets must not be nil !!")
 161     }
 162
 163     ps := pubsub.New(1)
 164     defer ps.Shutdown()
 165     //~ brn := brain.New()
 166     //~ defer brn.Shutdown()
 167
 168     go EventToWeb(ps)
 169     go RunXMPPBot(ps)
 170
 171     // --- get update on most recent events ---
 172     answ := ZmqsAskQuestionsAndClose(zmqctx, brain_connect_addr_, [][][]byte{[][]byte{[]byte("DoorCommandEvent")}, [][]byte{[]byte("DoorLockUpdate")}, [][]byte{[]byte("DoorAjarUpdate")}, [][]byte{[]byte("PresenceUpdate")}, [][]byte{[]byte("IlluminationSensorUpdate")}, [][]byte{[]byte("TempSensorUpdate")}})
 173     for _, a := range(answ) {
 174         ParseZMQr3Event(a, ps)
 175     }
 176
 177     // --- receive and distribute events ---
 178     ticker := time.NewTicker(time.Duration(7) * time.Minute)
 179     for {
 180     select {
 181         case e := <-zmqsub.In():
 182             ParseZMQr3Event(e, ps) //, brn)
 183         case <-ticker.C:
 184             ps.Pub(r3events.TimeTick{time.Now().Unix()}, "updateinterval")
 185         }
 186     }
 187 }