mirror of
https://github.com/davidallendj/configurator.git
synced 2025-12-20 03:27:02 -07:00
79 lines
2.3 KiB
Go
79 lines
2.3 KiB
Go
//go:build client || all
|
|
// +build client all
|
|
|
|
package cmd
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"os"
|
|
|
|
"github.com/OpenCHAMI/configurator/pkg/util"
|
|
"github.com/rs/zerolog/log"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
var (
|
|
accessToken string
|
|
remoteHost string
|
|
remotePort int
|
|
)
|
|
|
|
var fetchCmd = &cobra.Command{
|
|
Use: "fetch",
|
|
Short: "Fetch a config file from a remote instance of configurator",
|
|
Long: "This command is simplified to make a HTTP request to the a configurator service.",
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
// make sure a host is set
|
|
if remoteHost == "" {
|
|
log.Error().Msg("no '--host' argument set")
|
|
return
|
|
}
|
|
|
|
// check to see if an access token is available from env
|
|
if config.AccessToken == "" {
|
|
// check if OCHAMI_ACCESS_TOKEN env var is set if no access token is provided and use that instead
|
|
accessToken := os.Getenv("ACCESS_TOKEN")
|
|
if accessToken != "" {
|
|
config.AccessToken = accessToken
|
|
} else {
|
|
// TODO: try and fetch token first if it is needed
|
|
if verbose {
|
|
fmt.Printf("No token found. Attempting to generate config without one...\n")
|
|
}
|
|
}
|
|
}
|
|
|
|
// add the "Authorization" header if an access token is supplied
|
|
headers := map[string]string{}
|
|
if accessToken != "" {
|
|
headers["Authorization"] = "Bearer " + accessToken
|
|
}
|
|
|
|
for _, target := range targets {
|
|
// make a request for each target
|
|
url := fmt.Sprintf("%s:%d/generate?target=%s", remoteHost, remotePort, target)
|
|
res, body, err := util.MakeRequest(url, http.MethodGet, nil, headers)
|
|
if err != nil {
|
|
log.Error().Err(err).Msg("failed to make request")
|
|
return
|
|
}
|
|
// handle getting other error codes other than a 200
|
|
if res != nil {
|
|
if res.StatusCode == http.StatusOK {
|
|
log.Info().Msgf("%s\n", string(body))
|
|
}
|
|
}
|
|
}
|
|
},
|
|
}
|
|
|
|
func init() {
|
|
fetchCmd.Flags().StringVar(&remoteHost, "host", "", "set the remote configurator host")
|
|
fetchCmd.Flags().IntVar(&remotePort, "port", 3334, "set the remote configurator port")
|
|
fetchCmd.Flags().StringSliceVar(&targets, "target", nil, "set the target configs to make")
|
|
fetchCmd.Flags().StringVarP(&outputPath, "output", "o", "", "set the output path for config targets")
|
|
fetchCmd.Flags().StringVar(&accessToken, "access-token", "o", "set the output path for config targets")
|
|
|
|
rootCmd.AddCommand(fetchCmd)
|
|
}
|