warren/cmd/warrend/main.go

89 lines
1.5 KiB
Go

package main
import (
"fmt"
"github.com/dballard/warren/lib/warren"
"log"
"net"
"os"
"os/signal"
"strings"
"syscall"
)
type SockCommand struct {
warren.Command
Run func(cmd *SockCommand, c net.Conn, args ...string)
}
func (cmd *SockCommand) GetName() string {
return cmd.Name
}
func (cmd *SockCommand) Exec(c net.Conn, args []string) {
cmd.Flag.Usage = func() {
// helpFunc(c, c.Name)
}
cmd.Flag.Parse(args)
cmd.Run(cmd, c, cmd.Flag.Args()...)
}
func sockCmdHandle(c net.Conn) {
defer c.Close()
fmt.Println("cmdHandle")
data, err := warren.ReadStringz(c)
if err != nil {
fmt.Println(err)
return
}
fmt.Println("cmdHandle got command:", string(data))
// BAD: TODO: handle quotes "
var args = strings.Split(" ", data)
// parse args
var icmd = warren.GetCommand(args[0])
// get and run cmd
if icmd == nil {
c.Write([]byte("Unknown command " + args[0]))
// usage ?
return
}
icmd.(*SockCommand).Exec(c, args[1:])
c.Write([]byte("echo: " + data))
}
func main() {
warren.RegisterCommand(statusCmd)
cmdSock, err := net.Listen(warren.CmdSockType(), warren.CmdSockAddr())
defer cmdSock.Close()
if err != nil {
log.Fatal("Listen on sock error:", err)
}
// Handle signals
c := make(chan os.Signal, 1)
signal.Notify(c,
syscall.SIGINT,
syscall.SIGTERM,
syscall.SIGQUIT,
syscall.SIGHUP,
)
go func() {
_ = <-c
cmdSock.Close()
}()
for {
fd, err := cmdSock.Accept()
fmt.Println("accept")
if err != nil {
log.Fatal("accpet error:", err)
}
go sockCmdHandle(fd)
}
}