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 }