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/r3-eventbroker_zmq/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_presence_events_chan_     chan interface{}
  24     xmpp_login_ struct {jid string; pass string}
  25     xmpp_bot_authstring_ string
  26     xmpp_state_save_dir_ string
  27     r3eventssub_port_ string
  28     button_press_timeout_ int64 = 3600
  29     brain_connect_addr_ string
  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.Parse()
  42 }
  43
  44 //-------
  45
  46 func IfThenElseStr(c bool, strue, sfalse string) string {
  47     if c {return strue} else {return sfalse}
  48 }
  49
  50 func composeMessage(present, locked, shut bool, who string, ts int64) string {
  51     return fmt.Sprintf("%s (Door is %s and %s and was last used%s at %s)",
  52         IfThenElseStr(present,  "Somebody is present!" , "Everybody left."),
  53         IfThenElseStr(locked, "locked","unlocked"),
  54         IfThenElseStr(shut, "shut","ajar"),
  55         IfThenElseStr(len(who) == 0,"", " by " + who),
  56         time.Unix(ts,0).String())
  57 }
  58
  59 func EventToXMPP(ps *pubsub.PubSub, xmpp_presence_events_chan_ chan <- interface{}) {
  60     events := ps.Sub("presence","door","buttons","updateinterval")
  61
  62     defer func() {
  63         if x := recover(); x != nil {
  64             fmt.Printf("handleIncomingXMPPStanzas: run time panic: %v", x)
  65             ps.Unsub(events, "presence","door","buttons","updateinterval")
  66             close(xmpp_presence_events_chan_)
  67         }
  68     }()
  69
  70     var present, locked, shut bool = false, true, true
  71     var last_buttonpress int64 = 0
  72     var who string
  73     button_msg := "Dooom ! The button has been pressed ! Propably someone is bored and in need of company ! ;-)"
  74     present_status := r3xmppbot.XMPPStatusEvent{r3xmppbot.ShowOnline,"Somebody is present"}
  75     notpresent_status := r3xmppbot.XMPPStatusEvent{r3xmppbot.ShowNotAvailabe,"Nobody is here"}
  76     button_status := r3xmppbot.XMPPStatusEvent{r3xmppbot.ShowFreeForChat, "The button has been pressed :-)"}
  77
  78     xmpp_presence_events_chan_ <- r3xmppbot.XMPPStatusEvent{r3xmppbot.ShowNotAvailabe, "Nobody is here"}
  79
  80     for eventinterface := range(events) {
  81         fmt.Println("event2xmpp", eventinterface)
  82         switch event := eventinterface.(type) {
  83             case r3events.PresenceUpdate:
  84                 present = event.Present
  85                 xmpp_presence_events_chan_ <- r3xmppbot.XMPPMsgEvent{Msg: composeMessage(present, locked, shut, who, event.Ts), DistributeLevel: r3xmppbot.R3OnlineOnlyInfo, RememberAsStatus: true}
  86                 if present {
  87                     xmpp_presence_events_chan_ <- present_status
  88                 } else {
  89                     xmpp_presence_events_chan_ <- notpresent_status
  90                 }
  91             case r3events.DoorCommandEvent:
  92                 if len(event.Who) > 0 && len(event.Using) > 0 {
  93                     who = fmt.Sprintf("%s (%s)",event.Who, event.Using)
  94                 } else {
  95                     who = event.Who
  96                 }
  97                 xmpp_presence_events_chan_ <- fmt.Sprintln("DoorCommand:",event.Command, "using", event.Using, "by", event.Who, time.Unix(event.Ts,0))
  98             case r3events.DoorLockUpdate:
  99                 locked = event.Locked
 100                 xmpp_presence_events_chan_ <- r3xmppbot.XMPPMsgEvent{Msg: composeMessage(present, locked, shut, who, event.Ts), DistributeLevel: r3xmppbot.R3DebugInfo, RememberAsStatus: true}
 101            case r3events.DoorAjarUpdate:
 102                 shut = event.Shut
 103                 xmpp_presence_events_chan_ <- r3xmppbot.XMPPMsgEvent{Msg: composeMessage(present, locked, shut, who, event.Ts), DistributeLevel: r3xmppbot.R3DebugInfo, RememberAsStatus: true}
 104             case r3events.BoreDoomButtonPressEvent:
 105                 xmpp_presence_events_chan_ <- r3xmppbot.XMPPMsgEvent{Msg: button_msg, DistributeLevel: r3xmppbot.R3OnlineOnlyInfo}
 106                 xmpp_presence_events_chan_ <- button_status
 107                 last_buttonpress = event.Ts
 108             case r3events.TimeTick:
 109                 if present && last_buttonpress > 0 && time.Now().Unix() - last_buttonpress > button_press_timeout_ {
 110                     xmpp_presence_events_chan_ <- present_status
 111                     last_buttonpress = 0
 112                 }
 113         }
 114         }
 115 }
 116
 117 func main() {
 118     zmqctx, zmqsub := ZmqsInit(r3eventssub_port_)
 119     defer zmqctx.Close()
 120     if zmqsub != nil {defer zmqsub.Close()}
 121     if zmqsub == nil {
 122         panic("zmq sockets must not be nil !!")
 123     }
 124
 125     var xmpperr error
 126     var bot *r3xmppbot.XmppBot
 127     bot, xmpp_presence_events_chan_, xmpperr = r3xmppbot.NewStartedBot(xmpp_login_.jid, xmpp_login_.pass, xmpp_bot_authstring_, xmpp_state_save_dir_, true)
 128
 129     ps := pubsub.New(1)
 130     defer ps.Shutdown()
 131     //~ brn := brain.New()
 132     //~ defer brn.Shutdown()
 133
 134     go EventToWeb(ps)
 135     if xmpperr == nil {
 136         defer bot.StopBot()
 137         go EventToXMPP(ps, xmpp_presence_events_chan_)
 138     } else {
 139         fmt.Println(xmpperr)
 140         fmt.Println("XMPP Bot disabled")
 141     }
 142
 143     // --- get update on most recent events ---
 144     answ := ZmqsAskQuestionsAndClose(zmqctx, brain_connect_addr_, [][][]byte{[][]byte{[]byte("DoorLockUpdate")}, [][]byte{[]byte("DoorAjarUpdate")}, [][]byte{[]byte("DoorCommandEvent")}, [][]byte{[]byte("PresenceUpdate")}, [][]byte{[]byte("IlluminationSensorUpdate")}, [][]byte{[]byte("TempSensorUpdate")}})
 145     for _, a := range(answ) {
 146         //~ fmt.Println("recent event:", a)
 147         ParseZMQr3Event(a, ps)
 148     }
 149
 150     // --- receive and distribute events ---
 151     ticker := time.NewTicker(time.Duration(7) * time.Minute)
 152     for {
 153     select {
 154         case e := <-zmqsub.In():
 155             ParseZMQr3Event(e, ps) //, brn)
 156         case <-ticker.C:
 157             ps.Pub(r3events.TimeTick{time.Now().Unix()}, "updateinterval")
 158         }
 159     }
 160 }