kubevpn/cmd/kubevpn/cmds/serve.go

package cmds

import (
	"context"
	"net/http"

	log "github.com/sirupsen/logrus"
	"github.com/spf13/cobra"

	"github.com/wencaiwulue/kubevpn/pkg/config"
	"github.com/wencaiwulue/kubevpn/pkg/handler"
	"github.com/wencaiwulue/kubevpn/pkg/util"
)

var route handler.Route

func init() {
	ServerCmd.Flags().StringArrayVarP(&route.ServeNodes, "nodeCommand", "L", []string{}, "command needs to be executed")
	ServerCmd.Flags().StringVarP(&route.ChainNode, "chainCommand", "F", "", "command needs to be executed")
	ServerCmd.Flags().BoolVar(&config.Debug, "debug", false, "true/false")
	RootCmd.AddCommand(ServerCmd)
}

var ServerCmd = &cobra.Command{
	Use:   "serve",
	Short: "serve",
	Long:  `serve`,
	PreRun: func(*cobra.Command, []string) {
		util.InitLogger(config.Debug)
		go func() { log.Info(http.ListenAndServe("localhost:6060", nil)) }()
	},
	Run: func(cmd *cobra.Command, args []string) {
		err := handler.Start(context.TODO(), route)
		if err != nil {
			log.Fatal(err)
		}
		select {}
	},
}