nginx-ui/agent.go

package main

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

func look(serverUrl string, ssl string, token string) {
	for {
		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.Println("agent err:", err)
			time.Sleep(5 * time.Second)
		}
	}
}

// 代理服务
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)
	look(*serverUrl, *ssl, *token)
}