8e6610dff2bec6c0a11c4c8ea0ba2c3a8490fd4d
[svn42.git] / r3-netstatus / main.go
1 package main
2
3 import (
4     "./r3xmppbot"
5     pubsub "github.com/tuxychandru/pubsub"
6     "flag"
7     "time"
8     "fmt"
9     //~ "./brain"
10 )
11
12 type SpaceState struct {
13     present           bool
14     buttonpress_until int64
15     door_locked bool
16     door_shut bool
17 }
18
19 var (
20     presence_socket_path_ string
21     xmpp_presence_events_chan_     chan interface{}
22     xmpp_login_ struct {jid string; pass string}
23     xmpp_bot_authstring_ string
24     xmpp_state_save_dir_ string
25 )
26
27 //-------
28
29 func init() {
30     flag.StringVar(&xmpp_login_.jid, "xjid", "realrauminfo@realraum.at/Tuer", "XMPP Bot Login JID")
31     flag.StringVar(&xmpp_login_.pass, "xpass", "", "XMPP Bot Login Password")
32     flag.StringVar(&xmpp_bot_authstring_, "xbotauth", "", "String that user use to authenticate themselves to the bot")
33     flag.StringVar(&presence_socket_path_,"presencesocket", "/var/run/tuer/presence.socket",  "Path to presence socket")
34     flag.StringVar(&xmpp_state_save_dir_,"xstatedir","/flash/var/lib/r3netstatus/",  "Directory to save XMPP bot state in")
35     flag.Parse()
36 }
37
38 //-------
39
40 func IfThenElseStr(c bool, strue, sfalse string) string {
41     if c {return strue} else {return sfalse}
42 }
43
44 func composeMessage(present, locked, shut bool, who string, ts int64) string {
45     return fmt.Sprintf("%s (Door is %s and %s and was last used%s at %s)",
46         IfThenElseStr(present,  "Somebody is present!" , "Everybody left."),
47         IfThenElseStr(locked, "locked","unlocked"),
48         IfThenElseStr(shut, "shut","ajar"),
49         IfThenElseStr(who == "","", " by " + who),
50         time.Unix(ts,0).String())
51 }
52
53 func EventToXMPP(ps *pubsub.PubSub, xmpp_presence_events_chan_ chan <- interface{}) {
54     events := ps.Sub("presence","door","buttons")
55     var present, locked, shut bool = false, true, true
56     var who string
57     button_msg := "The button has been pressed ! Propably someone is bored and in need of company ! ;-)"
58
59     for eventinterface := range(events) {
60         switch event := eventinterface.(type) {
61             case PresenceUpdate:
62                 present = event.Present
63                 xmpp_presence_events_chan_ <- r3xmppbot.XMPPMsgEvent{Msg: composeMessage(present, locked, shut, who, event.Ts), DistributeLevel: r3xmppbot.R3OnlineOnlyInfo, RememberAsStatus: true}
64             case DoorCommandEvent:
65                 who = event.Who
66                 xmpp_presence_events_chan_ <- fmt.Sprintln("DoorCommand:",event.Command, "using", event.Using, "by", event.Who, time.Unix(event.Ts,0))
67             case DoorStatusUpdate:
68                 locked = event.Locked
69                 shut = event.Shut
70                 xmpp_presence_events_chan_ <- r3xmppbot.XMPPMsgEvent{Msg: composeMessage(present, locked, shut, who, event.Ts), DistributeLevel: r3xmppbot.R3DebugInfo, RememberAsStatus: true}
71             case ButtonPressUpdate:
72                 xmpp_presence_events_chan_ <- r3xmppbot.XMPPMsgEvent{Msg: button_msg, DistributeLevel: r3xmppbot.R3OnlineOnlyInfo}
73         }
74         }
75 }
76
77 func main() {
78     var err error
79     var bot *r3xmppbot.XmppBot
80     bot, xmpp_presence_events_chan_, err = r3xmppbot.NewStartedBot(xmpp_login_.jid, xmpp_login_.pass, xmpp_bot_authstring_, xmpp_state_save_dir_, true)
81     if err != nil {
82         fmt.Println(err)
83         return
84     }
85     defer bot.StopBot()
86
87     newlinequeue := make(chan string, 1)
88     ps := pubsub.New(1)
89     //~ brn := brain.New()
90     defer close(newlinequeue)
91     defer ps.Shutdown()
92     //~ defer brn.Shutdown()
93
94         ticker := time.NewTicker(time.Duration(7) * time.Minute)
95     go EventToWeb(ps)
96     go EventToXMPP(ps, xmpp_presence_events_chan_)
97         go ReadFromUSocket(presence_socket_path_, newlinequeue)
98         for {
99                 select {
100                 case e := <-newlinequeue:
101                         ParseSocketInputLine(e, ps) //, brn)
102                 case <-ticker.C:
103                         ps.Pub(TimeTick{time.Now().Unix()}, "publishjson")
104                 }
105         }
106 }