towk/makeshift
1
0
Fork 0
Code Issues 1 Pull requests Projects Releases Packages Wiki Activity Actions

Added access-token CLI flag for fetch command

Browse source
This commit is contained in:
David Allen 2024-07-03 12:47:14 -06:00
parent af5e330731
commit c63e9745bb
No known key found for this signature in database
GPG key ID: 717C593FF60A2ACC
1 changed files with 11 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

15
cmd/fetch.go
View file

@ -13,8 +13,9 @@ import (
) )
var ( var (
remoteHost string accessToken string
remotePort int remoteHost string
remotePort int
) )
var fetchCmd = &cobra.Command{ var fetchCmd = &cobra.Command{
@ -26,11 +27,16 @@ var fetchCmd = &cobra.Command{
logrus.Errorf("no '--host' argument set") logrus.Errorf("no '--host' argument set")
return return
} }
for _, target := range targets {
headers := map[string]string{}
if accessToken != "" {
headers["Authorization"] = "Bearer " + accessToken
}
for _, target := range targets {
// make a request for each target // make a request for each target
url := fmt.Sprintf("%s:%d/generate?target=%s", remoteHost, remotePort, target) url := fmt.Sprintf("%s:%d/generate?target=%s", remoteHost, remotePort, target)
res, body, err := util.MakeRequest(url, http.MethodGet, nil, nil) res, body, err := util.MakeRequest(url, http.MethodGet, nil, headers)
if err != nil { if err != nil {
logrus.Errorf("failed to make request: %v", err) logrus.Errorf("failed to make request: %v", err)
return return
@ -49,6 +55,7 @@ func init() {
fetchCmd.Flags().IntVar(&remotePort, "port", 3334, "set the remote configurator port") 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().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().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) rootCmd.AddCommand(fetchCmd)
} }