feat(cli): add server_url to agent install cmd (#1224)

* feat(cli): add `server_url` to `agent install` cmd

Follow-on to 070ce61.

Fixes RAIN-54457

Signed-off-by: Nick Schmeller <[email protected]>

* test(cli): fix integration help test with new flag info

Signed-off-by: Salim Afiune Maya <[email protected]>

---------

Signed-off-by: Nick Schmeller <[email protected]>
Signed-off-by: Salim Afiune Maya <[email protected]>
Co-authored-by: Salim Afiune Maya <[email protected]>

Author: nschmeller

cli/cmd/agent.go

@@ -161,6 +161,10 @@ pass it directly via the argument.
 
     lacework agent install <user@host:port>
 
+To explicitly specify the server URL that the agent will connect to:
+
+    lacework agent install --server_url https://api.fra.lacework.net
+
 To list all active agents in your environment. 
 
     lacework agent list
@@ -245,6 +249,9 @@ func init() {
 	agentInstallCmd.Flags().BoolVar(&agentCmdState.InstallTrustHostKey,
 		"trust_host_key", false, "automatically add host keys to the ~/.ssh/known_hosts file",
 	)
+	agentInstallCmd.Flags().StringVar(&agentCmdState.InstallServerURL,
+		"server_url", "https://api.lacework.net", "server URL that agents will talk to, prefixed with `https://`",
+	)
 }
 
 func showAgentToken(_ *cobra.Command, args []string) error {

cli/cmd/agent_install.go

@@ -135,7 +135,7 @@ func installRemoteAgent(_ *cobra.Command, args []string) error {
 			return err
 		}
 	}
-	cmd := fmt.Sprintf("sudo sh -c \"curl -sSL %s | sh -s -- %s\"", agentInstallDownloadURL, token)
+	cmd := fmt.Sprintf("sudo sh -c \"curl -sSL %s | sh -s -- %s -U %s\"", agentInstallDownloadURL, token, agentCmdState.InstallServerURL)
 	return runInstallCommandOnRemoteHost(runner, cmd)
 }
 

integration/test_resources/help/agent_install

@@ -21,6 +21,10 @@ pass it directly via the argument.
 
     lacework agent install <user@host:port>
 
+To explicitly specify the server URL that the agent will connect to:
+
+    lacework agent install --server_url https://api.fra.lacework.net
+
 To list all active agents in your environment. 
 
     lacework agent list
@@ -34,6 +38,7 @@ Flags:
       --force                  override any pre-installed agent
   -h, --help                   help for install
   -i, --identity_file string   identity (private key) for public key authentication (default "~/.ssh/id_rsa")
+      --server_url https://    server URL that agents will talk to, prefixed with https:// (default "https://api.lacework.net")
       --ssh_password string    password for authentication
       --ssh_port int           port to connect to on the remote host (default 22)
       --ssh_username string    username to login with