-
Notifications
You must be signed in to change notification settings - Fork 45
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge remote-tracking branch 'origin/3.x' into sup-2320-bk-pipeline-v…
…iew-w
- Loading branch information
Showing
11 changed files
with
577 additions
and
51 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,145 @@ | ||
| package api | ||
|
|
||
| import ( | ||
| "bytes" | ||
| "encoding/json" | ||
| "fmt" | ||
| "io" | ||
| "net/http" | ||
| "strings" | ||
|
|
||
| "github.com/MakeNowJust/heredoc" | ||
| "github.com/buildkite/cli/v3/pkg/cmd/factory" | ||
| "github.com/buildkite/cli/v3/pkg/cmd/validation" | ||
| "github.com/spf13/cobra" | ||
| ) | ||
|
|
||
| var ( | ||
| method string | ||
| headers []string | ||
| data string | ||
| analytics bool | ||
| queryFile string | ||
| ) | ||
|
|
||
| func NewCmdAPI(f *factory.Factory) *cobra.Command { | ||
| cmd := cobra.Command{ | ||
| Use: "api <endpoint>", | ||
| Short: "Interact with the Buildkite API", | ||
| Long: "Interact with either the REST or GraphQL Buildkite APIs", | ||
| Args: cobra.MaximumNArgs(1), | ||
| Example: heredoc.Doc(` | ||
| # To get a build | ||
| $ bk api /pipelines/example-pipeline/builds/420 | ||
| # To create a pipeline | ||
| $ bk api --method POST /pipelines --data ' | ||
| { | ||
| "name": "My Cool Pipeline", | ||
| "repository": "git@github.com:acme-inc/my-pipeline.git", | ||
| "configuration": "steps:\n - command: env" | ||
| } | ||
| ' | ||
| # To update a cluster | ||
| $ bk api --method PUT /clusters/CLUSTER_UUID --data ' | ||
| { | ||
| "name": "My Updated Cluster", | ||
| } | ||
| ' | ||
| # To get all test suites | ||
| $ bk api --analytics /suites | ||
| `), | ||
| PersistentPreRunE: validation.CheckValidConfiguration(f.Config), | ||
| RunE: func(cmd *cobra.Command, args []string) error { | ||
| if data != "" && !cmd.Flags().Changed("method") { | ||
| method = "POST" | ||
| } | ||
| return apiCaller(cmd, args, f) | ||
| }, | ||
| } | ||
|
|
||
| cmd.Flags().StringVarP(&method, "method", "X", "GET", "HTTP method to use") | ||
| cmd.Flags().StringArrayVarP(&headers, "header", "H", []string{}, "Headers to include in the request") | ||
| cmd.Flags().StringVarP(&data, "data", "d", "", "Data to send in the request body") | ||
| cmd.Flags().BoolVar(&analytics, "analytics", false, "Use the Test Analytics endpoint") | ||
| cmd.Flags().StringVarP(&queryFile, "file", "f", "", "File containing GraphQL query") | ||
|
|
||
| return &cmd | ||
| } | ||
|
|
||
| func apiCaller(cmd *cobra.Command, args []string, f *factory.Factory) error { | ||
| var endpoint string | ||
| var endpointPrefix string | ||
|
|
||
| if len(args) > 1 { | ||
| return fmt.Errorf("Incorrect number of arguments. Expected 1, got %d", len(args)) | ||
| } | ||
|
|
||
| if len(args) == 0 { | ||
| endpoint = "/" | ||
| } else { | ||
| endpoint = args[0] | ||
| } | ||
|
|
||
| if analytics { | ||
| endpointPrefix = fmt.Sprintf("v2/analytics/organizations/%s", f.Config.OrganizationSlug()) | ||
| } else { | ||
| endpointPrefix = fmt.Sprintf("v2/organizations/%s", f.Config.OrganizationSlug()) | ||
| } | ||
|
|
||
| url := f.RestAPIClient.BaseURL.String() + endpointPrefix + endpoint | ||
|
|
||
| var req *http.Request | ||
| var err error | ||
|
|
||
| if data != "" { | ||
| req, err = http.NewRequest(method, url, strings.NewReader(data)) | ||
| } else { | ||
| req, err = http.NewRequest(method, url, nil) | ||
| } | ||
|
|
||
| if err != nil { | ||
| return fmt.Errorf("error creating request: %w", err) | ||
| } | ||
|
|
||
| for _, header := range headers { | ||
| parts := strings.SplitN(header, ":", 2) | ||
| if len(parts) == 2 { | ||
| req.Header.Add(strings.TrimSpace(parts[0]), strings.TrimSpace(parts[1])) | ||
| } | ||
| } | ||
|
|
||
| req.Header.Set("Authorization", "Bearer "+f.Config.APIToken()) | ||
|
|
||
| client := &http.Client{} | ||
| resp, err := client.Do(req) | ||
| if err != nil { | ||
| return fmt.Errorf("error making request: %w", err) | ||
| } | ||
| defer resp.Body.Close() | ||
|
|
||
| body, err := io.ReadAll(resp.Body) | ||
| if err != nil { | ||
| return fmt.Errorf("error reading response: %w", err) | ||
| } | ||
|
|
||
| var jsonResponse interface{} | ||
| err = json.Unmarshal(body, &jsonResponse) | ||
| if err != nil { | ||
| fmt.Println("Response is not valid JSON. Raw response:") | ||
| fmt.Println(string(body)) | ||
| return fmt.Errorf("error parsing response as JSON: %w", err) | ||
| } | ||
|
|
||
| var prettyJSON bytes.Buffer | ||
| err = json.Indent(&prettyJSON, body, "", " ") | ||
| if err != nil { | ||
| return fmt.Errorf("error formatting JSON response: %w", err) | ||
| } | ||
|
|
||
| fmt.Println(prettyJSON.String()) | ||
|
|
||
| return nil | ||
| } |
Oops, something went wrong.