warren/cmd/warrend/main.go

package main

import (
	"fmt"
	"github.com/dballard/warren/lib/warren"
	"log"
	"net"
	"os"
	"os/signal"
	"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))

	// parse args
	// get and run cmd

	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)
	}

}