nginx-ui/agent.go

package main

import (
	"flag"
	"log"
	"nginx-ui/agent"
	"nginx-ui/server/models"
	"os"
)

// 代理服务
func main() {

	serverUrl := flag.String("server", "127.0.0.1:8080/nginx-ui/api", "agent host and path ,eg:127.0.0.1:8080/nginx-ui/api")
	ssl := flag.String("ssl", "N", "use ssl,https or wss,Y or N")
	token := flag.String("token", "", "token")
	flag.Parse()
	envUrl := os.Getenv("SERVER_URL")
	envSSL := os.Getenv("SSL")
	envToken := os.Getenv("TOKEN")
	if len(envUrl) > 0 {
		serverUrl = &envUrl
	}
	if len(envSSL) > 0 {
		ssl = &envSSL
	}
	if len(envToken) > 0 {
		token = &envToken
	}
	log.Printf("agent server url: %s", *serverUrl)
	c := agent.NewAgent(*serverUrl, *ssl, *token)
	c.SetMessageHandler(models.NginxUpdateType, agent.OnNginxUpdated)
	c.SetMessageHandler(models.AgentCmdType, agent.OnCMD)
	c.SetMessageHandler(models.SendFileType, agent.OnSendFile)
	c.SetMessageHandler(models.ServerConnected, agent.OnServerConnected)
	err := c.Run()
	if err != nil {
		log.Fatal(err)
	}

}