who
[svn42.git] / go / door_client_zmq / main.go
1 // (c) Bernhard Tittelbach, 2013
2
3 package main
4
5 import (
6     "fmt"
7     "os"
8     "flag"
9     "log"
10     "bufio"
11     "bytes"
12 )
13
14
15 // ---------- Main Code -------------
16
17 var (
18     cmd_port_ string
19     sub_port_ string
20     cmd_method_ string
21     cmd_user_ string
22 )
23
24 func init() {
25     flag.StringVar(&cmd_port_, "cmdport", "ipc:///run/tuer/door_cmd.ipc", "zmq command socket path")
26     flag.StringVar(&sub_port_, "pubport", "tcp://torwaechter.realraum.at:4242", "zmq subscribe/listen socket path")
27     flag.StringVar(&cmd_method_, "method", "", "zmq cmd method")
28     flag.StringVar(&cmd_user_, "usernick", "", "zmq cmd user identity")
29     flag.Parse()
30 }
31
32 func LineReader(out chan <- [][]byte, stdin * os.File) {
33     linescanner := bufio.NewScanner(stdin)
34     linescanner.Split(bufio.ScanLines)
35     defer close(out)
36     for linescanner.Scan() {
37         if err := linescanner.Err(); err != nil {
38             log.Print(err)
39             return
40         }
41         //text := bytes.Fields(linescanner.Bytes()) //this returns a slice (aka pointer, no array deep-copy here)
42         text := bytes.Fields([]byte(linescanner.Text())) //this allocates a string and slices it -> no race-condition with overwriting any data
43         if len(text) == 0 {
44             continue
45         }
46         out <- text
47     }
48 }
49
50 func ByteArrayToString(bb [][]byte) string {
51     b := bytes.Join(bb, []byte(" "))
52     return string(b)
53 }
54
55 func main() {
56     zmqctx, cmd_chans, sub_chans := ZmqsInit(cmd_port_, sub_port_)
57     defer cmd_chans.Close()
58     defer sub_chans.Close()
59     defer zmqctx.Close()
60     var listen bool
61     var ignore_next uint = 0
62
63     user_input_chan := make(chan [][]byte, 1)
64     go LineReader(user_input_chan, os.Stdin)
65     defer os.Stdin.Close()
66
67     for {
68         select {
69         case input, input_open := (<- user_input_chan):
70             if input_open {
71                 if len(input) == 0 { continue }
72                  switch string(input[0]) {
73                     case "help", "?":
74                         fmt.Println("Available Commands: help, listen, quit. Everything else is passed through to door daemon")
75                     case "listen":
76                         listen = true
77                         fmt.Println("Now listening, @ are broadcasts")
78                     case "quit":
79                         os.Exit(0)
80                     default:
81                         ignore_next = 2
82                         cmd_chans.Out() <- input
83                         reply := <- cmd_chans.In()
84                         fmt.Println(">",ByteArrayToString(reply))
85                 }
86             } else {
87                 os.Exit(0)
88             }
89         case pubsubstuff := <- sub_chans.In():
90             if len(pubsubstuff) == 0 { continue}
91             if ignore_next > 0 {
92                 ignore_next--
93                 continue
94             }
95             if listen {
96                 fmt.Println("@",ByteArrayToString(pubsubstuff))
97             }
98         }
99     }
100 }