gokins/cmd/cmd.go

package cmd

import (
	"github.com/gokins/core"
	"github.com/gokins/gokins/comm"
	"github.com/gokins/gokins/server"
	hbtp "github.com/mgr9525/HyperByte-Transfer-Protocol"
	"gopkg.in/alecthomas/kingpin.v2"
	"os"
	"os/exec"
	"os/signal"
	"syscall"
)

const Version = "0.1.1"

var app = kingpin.New("gokins", "A golang workflow application.")

func Run() {
	regs()
	kingpin.Version(Version)
	kingpin.MustParse(app.Parse(os.Args[1:]))
}
func regs() {
	app.Flag("web", "gokins web host").Default(":8030").StringVar(&comm.WebHost)
	//app.Flag("hbtp", "gokins hbtp host").Default(":8031").StringVar(&comm.HbtpHost)
	app.Flag("workdir", "gokins work path").Short('w').StringVar(&comm.WorkPath)
	app.Flag("nupass", "can't update password").Hidden().BoolVar(&comm.NotUpPass)
	cmd := app.Command("run", "run process").Default().
		Action(run)
	cmd.Flag("debug", "debug log show").BoolVar(&core.Debug)

	cmd = app.Command("daemon", "run process background").
		Action(start)
}
func getArgs() []string {
	args := make([]string, 0)
	args = append(args, "run")
	if comm.WebHost != "" {
		args = append(args, "--web")
		args = append(args, comm.WebHost)
	}
	/*if comm.HbtpHost != "" {
		args = append(args, "--hbtp")
		args = append(args, comm.HbtpHost)
	}*/
	if comm.WorkPath != "" {
		args = append(args, "--workdir")
		args = append(args, comm.WorkPath)
	}
	if comm.NotUpPass {
		args = append(args, "--nupass")
	}
	return args
}
func start(pc *kingpin.ParseContext) error {
	args := getArgs()
	fullpth, err := os.Executable()
	if err != nil {
		return err
	}
	println("start process")
	cmd := exec.Command(fullpth, args...)
	err = cmd.Start()
	if err != nil {
		return err
	}
	return nil
}
func run(pc *kingpin.ParseContext) error {
	csig := make(chan os.Signal, 1)
	signal.Notify(csig, os.Interrupt, syscall.SIGALRM)
	go func() {
		s := <-csig
		hbtp.Debugf("get signal(%d):%s", s, s.String())
		comm.Cancel()
	}()
	if core.Debug {
		hbtp.Debug = true
	}
	return server.Run()
}