Elgato_dark/fleet
1
0
Fork 0
mirror of https://github.com/fleetdm/fleet synced 2026-05-24 09:28:54 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 30

Start hosts in fleetctl preview (#247)

Browse source 
- Set config context and log in fleetctl
- Start containerized Linux hosts

Closes #173
This commit is contained in:
Zach Wasserman 2021-01-28 18:18:21 -08:00 committed by GitHub
parent 368632b3fa
commit 15baa54d81
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 61 additions and 25 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

86
cmd/fleetctl/preview.go
View file

@ -87,35 +87,28 @@ This command will create a directory fleet-preview in the current working direct
}
}
configPath, context := c.String("config"), c.String("context")
configPath, context := c.String("config"), "default"
val, err := getConfigValue(configPath, context, "address")
contextConfig := Context{
Address: address,
Email: username,
Token: token,
TLSSkipVerify: true,
}
config, err := readConfig(configPath)
if err != nil {
return errors.Wrap(err, "Error checking config")
}
if v, ok := val.(string); ok && v != "" {
fmt.Println("Skipped fleetctl setup because there is an existing fleetctl config.")
fmt.Println("Fleet is now available at https://localhost:8412.")
fmt.Println("Username:", username)
fmt.Println("Password:", password)
fmt.Println("Note: You can safely ignore the browser warning \"Your connection is not private\". Click through this warning using the \"Advanced\" option. Chrome users may need to configure chrome://flags/#allow-insecure-localhost.")
return nil
// No existing config
config.Contexts["default"] = contextConfig
} else {
fmt.Println("Configured fleetctl in the 'preview' context to avoid overwriting existing config.")
context = "preview"
config.Contexts["preview"] = contextConfig
}
c.Set("context", context)
if err := setConfigValue(configPath, context, "email", username); err != nil {
return errors.Wrap(err, "Error setting username")
}
if err := setConfigValue(configPath, context, "token", token); err != nil {
return errors.Wrap(err, "Error setting token")
}
if err := setConfigValue(configPath, context, "tls-skip-verify", "true"); err != nil {
return errors.Wrap(err, "Error setting tls-skip-verify")
}
if err := setConfigValue(configPath, context, "address", address); err != nil {
return errors.Wrap(err, "error setting address")
if err := writeConfig(configPath, config); err != nil {
return errors.Wrap(err, "Error writing fleetctl configuration")
}
fmt.Println("Fleet is now available at https://localhost:8412.")
@ -123,6 +116,49 @@ This command will create a directory fleet-preview in the current working direct
fmt.Println("Password:", password)
fmt.Println("Note: You can safely ignore the browser warning \"Your connection is not private\". Click through this warning using the \"Advanced\" option. Chrome users may need to configure chrome://flags/#allow-insecure-localhost.")
// Create client and get enroll secret
client, err := unauthenticatedClientFromCLI(c)
if err != nil {
return errors.Wrap(err, "Error making fleetctl client")
}
token, err = client.Login(username, password)
if err != nil {
return errors.Wrap(err, "fleetctl login failed")
}
if err := setConfigValue(configPath, context, "token", token); err != nil {
return errors.Wrap(err, "Error setting token for the current context")
}
client.SetToken(token)
secrets, err := client.GetEnrollSecretSpec()
if err != nil {
return errors.Wrap(err, "Error retrieving enroll secret")
}
if len(secrets.Secrets) != 1 || !secrets.Secrets[0].Active {
return errors.New("Expected 1 active enroll secret")
}
if err := os.Chdir(filepath.Join(previewDir, "osquery")); err != nil {
return errors.Wrap(err, "Error getting preview osquery directory")
}
fmt.Println("Starting simulated hosts...")
cmd := exec.Command("docker-compose", "up", "-d")
cmd.Env = append(cmd.Env,
"ENROLL_SECRET="+secrets.Secrets[0].Secret,
"FLEET_URL="+address,
)
out, err = cmd.CombinedOutput()
if err != nil {
fmt.Println(string(out))
return errors.Errorf("Failed to run docker-compose")
}
fmt.Println("Preview environment complete. Enjoy using Fleet!")
return nil
},
}