floraachy
/
console
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add agent install env option to install script

This commit is contained in:
liugq 2023-06-06 12:31:52 +08:00
parent 523a6eeb37
commit d92d1e065f
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
modules/agent/api/setup.go
View File

@ -35,6 +35,11 @@ func (h *APIHandler) generateInstallCommand(w http.ResponseWriter, req *http.Req
h.WriteError(w, "user not found", http.StatusInternalServerError) h.WriteError(w, "user not found", http.StatusInternalServerError)
return return
} }
agCfg := common.GetAgentConfig()
if agCfg == nil || agCfg.Setup == nil {
h.WriteError(w, "agent setup config was not found, please configure in the configuration file first", http.StatusInternalServerError)
return
}
var ( var (
t *Token t *Token
tokenStr string tokenStr string
@ -67,7 +72,6 @@ func (h *APIHandler) generateInstallCommand(w http.ResponseWriter, req *http.Req
} }
} }
tokens.Store(tokenStr, t) tokens.Store(tokenStr, t)
agCfg := common.GetAgentConfig()
consoleEndpoint := agCfg.Setup.ConsoleEndpoint consoleEndpoint := agCfg.Setup.ConsoleEndpoint
if consoleEndpoint == "" { if consoleEndpoint == "" {
scheme := "http" scheme := "http"
@ -77,7 +81,7 @@ func (h *APIHandler) generateInstallCommand(w http.ResponseWriter, req *http.Req
consoleEndpoint = fmt.Sprintf("%s://%s", scheme, req.Host) consoleEndpoint = fmt.Sprintf("%s://%s", scheme, req.Host)
} }
h.WriteJSON(w, util.MapStr{ h.WriteJSON(w, util.MapStr{
"script": fmt.Sprintf(`sudo bash -c "$(curl -L '%s/agent/install.sh?token=%s')"`, consoleEndpoint, tokenStr), "script": fmt.Sprintf(`sudo BASE_URL="%s" AGENT_VER="%s" INSTALL_PATH="/opt" bash -c "$(curl -L '%s/agent/install.sh?token=%s')"`, agCfg.Setup.DownloadURL, agCfg.Setup.Version, consoleEndpoint, tokenStr),
"token": tokenStr, "token": tokenStr,
"expired_at": t.CreatedAt.Add(ExpiredIn), "expired_at": t.CreatedAt.Add(ExpiredIn),
}, http.StatusOK) }, http.StatusOK)