cmd-textmessage.go

package main

import (
	"encoding/json"
	"io"
	"os"
	"os/exec"

	"layeh.com/murmur-cli/MurmurRPC"
)

func init() {
	cmd := root.Add("textmessage")

	cmd.Add("send", func(args Args) {
		server := args.MustServer(0)
		if len(args) <= 1 {
			panic("not enough arguments")
		}
		text := args.MustString(len(args) - 1)

		tm := &MurmurRPC.TextMessage{
			Server: server,
			Text:   &text,
		}

		for i := 1; i < len(args)-1; i++ {
			if session, ok := args.PrefixedUint32("sender:", i); ok && tm.Actor == nil {
				tm.Actor = &MurmurRPC.User{
					Server:  server,
					Session: &session,
				}
			} else if session, ok := args.PrefixedUint32("user:", i); ok {
				tm.Users = append(tm.Users, &MurmurRPC.User{
					Server:  server,
					Session: &session,
				})
			} else if id, ok := args.PrefixedUint32("channel:", i); ok {
				tm.Channels = append(tm.Channels, &MurmurRPC.Channel{
					Server: server,
					Id:     &id,
				})
			} else if id, ok := args.PrefixedUint32("tree:", i); ok {
				tm.Trees = append(tm.Trees, &MurmurRPC.Channel{
					Server: server,
					Id:     &id,
				})
			} else {
				panic("unknown argument")
			}
		}

		Output(client.TextMessageSend(ctx, tm))
	})

	cmd.Add("filter", func(args Args) {
		server := args.MustServer(0)
		program := args.MustString(1)
		arguments := args[2:]

		stream, err := client.TextMessageFilter(ctx)
		if err != nil {
			panic(err)
		}
		msg := MurmurRPC.TextMessage_Filter{
			Server: server,
		}
		if err := stream.Send(&msg); err != nil {
			panic(err)
		}

		for {
			req, err := stream.Recv()
			if err != nil {
				if err != io.EOF {
					panic(err)
				}
				return
			}
			var resp MurmurRPC.TextMessage_Filter
			cmd := exec.Command(program, arguments...)
			cmd.Stdout = os.Stdout
			cmd.Stderr = os.Stderr
			pipe, err := cmd.StdinPipe()
			if err == nil {
				encoder := json.NewEncoder(pipe)
				go encoder.Encode(req.Message)
				cmd.Run()
				if cmd.ProcessState != nil {
					if !cmd.ProcessState.Success() {
						resp.Action = MurmurRPC.TextMessage_Filter_Reject.Enum()
					}
				}
			}
			stream.Send(&resp)
		}
	})
}