chore: pass context from cmd

einride · Apr 9, 2024 · 5fa9be5 · 5fa9be5
1 parent a7c5db0
commit 5fa9be5
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/main.go b/main.go
@@ -23,7 +23,7 @@ func main() {
 		Use:     "gh dependabot",
 		Short:   "Manage Dependabot PRs.",
 		Example: "gh dependabot --org einride",
-		RunE: func(_ *cobra.Command, _ []string) error {
+		RunE: func(cmd *cobra.Command, _ []string) error {
 			log.Println("Resolving current user...")
 			username, err := gh.Run("api", "graphql", "-f", "query={viewer{login}}", "--jq", ".data.viewer.login")
 			if err != nil {
@@ -57,7 +57,7 @@ func main() {
 
 			if securityFilter {
 				log.Printf("Matching pull requests to security alerts...")
-				pullRequests, err = filterSecurityPullRequests(client, &pullRequests)
+				pullRequests, err = filterSecurityPullRequests(cmd.Context(), client, &pullRequests)
 				if err != nil {
 					return err
 				}

diff --git a/securityfilter.go b/securityfilter.go
@@ -9,6 +9,7 @@ import (
 )
 
 func filterSecurityPullRequests(
+	ctx context.Context,
 	client *githubv4.Client,
 	pullRequests *[]pullRequest,
 ) ([]pullRequest, error) {
@@ -35,7 +36,7 @@ func filterSecurityPullRequests(
 		}
 		var vulnQ vulnQuery
 
-		if err := client.Query(context.Background(), &vulnQ, vulnVars); err != nil {
+		if err := client.Query(ctx, &vulnQ, vulnVars); err != nil {
 			return nil, fmt.Errorf("load vulnerability reports: %w", err)
 		}
 		for _, vulnAlert := range vulnQ.Repository.VulnerabilityAlerts.Nodes {