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