soju/service.go

package soju

import (
	"flag"
	"fmt"
	"io/ioutil"
	"strings"

	"github.com/google/shlex"
	"gopkg.in/irc.v3"
)

const serviceNick = "BouncerServ"

type serviceCommand struct {
	usage  string
	desc   string
	handle func(dc *downstreamConn, params []string) error
}

func sendServicePRIVMSG(dc *downstreamConn, text string) {
	dc.SendMessage(&irc.Message{
		Prefix:  &irc.Prefix{Name: serviceNick},
		Command: "PRIVMSG",
		Params:  []string{dc.nick, text},
	})
}

func handleServicePRIVMSG(dc *downstreamConn, text string) {
	words, err := shlex.Split(text)
	if err != nil {
		sendServicePRIVMSG(dc, fmt.Sprintf("error: failed to parse command: %v", err))
		return
	}

	var name string
	var params []string
	if len(words) > 0 {
		name = strings.ToLower(words[0])
		params = words[1:]
	}

	cmd, ok := serviceCommands[name]
	if !ok {
		sendServicePRIVMSG(dc, fmt.Sprintf(`error: unknown command %q (type "help" for a list of commands)`, name))
		return
	}

	if err := cmd.handle(dc, params); err != nil {
		sendServicePRIVMSG(dc, fmt.Sprintf("error: %v", err))
	}
}

var serviceCommands map[string]serviceCommand

func init() {
	serviceCommands = map[string]serviceCommand{
		"help": {
			usage:  "[command]",
			desc:   "print help message",
			handle: handleServiceHelp,
		},
		"create-network": {
			usage:  "-addr <addr> [-name name] [-username username] [-pass pass] [-realname realname] [-nick nick]",
			desc:   "add a new network",
			handle: handleServiceCreateNetwork,
		},
	}
}

func handleServiceHelp(dc *downstreamConn, params []string) error {
	if len(params) > 0 {
		name := strings.ToLower(params[0])
		cmd, ok := serviceCommands[name]
		if !ok {
			return fmt.Errorf("unknown command %q", name)
		}

		text := name
		if cmd.usage != "" {
			text += " " + cmd.usage
		}
		text += ": " + cmd.desc

		sendServicePRIVMSG(dc, text)
	} else {
		var l []string
		for name := range serviceCommands {
			l = append(l, name)
		}
		sendServicePRIVMSG(dc, "available commands: "+strings.Join(l, ", "))
	}
	return nil
}

func handleServiceCreateNetwork(dc *downstreamConn, params []string) error {
	fs := flag.NewFlagSet("", flag.ContinueOnError)
	fs.SetOutput(ioutil.Discard)
	addr := fs.String("addr", "", "")
	name := fs.String("name", "", "")
	username := fs.String("username", "", "")
	pass := fs.String("pass", "", "")
	realname := fs.String("realname", "", "")
	nick := fs.String("nick", "", "")

	if err := fs.Parse(params); err != nil {
		return err
	}
	if *addr == "" {
		return fmt.Errorf("flag addr is required")
	}

	if *nick == "" {
		*nick = dc.nick
	}

	var err error
	network, err := dc.user.createNetwork(&Network{
		Addr:     *addr,
		Name:     *name,
		Username: *username,
		Pass:     *pass,
		Realname: *realname,
		Nick:     *nick,
	})
	if err != nil {
		return fmt.Errorf("could not create network: %v", err)
	}

	sendServicePRIVMSG(dc, fmt.Sprintf("created network %s successfully", network.GetName()))
	return nil
}