use proper context

Author: tok-kkk

cmd/update.go

@@ -6,6 +6,7 @@ import (
 	"io/ioutil"
 	"net/http"
 	"strings"
+	"time"
 
 	"github.com/fatih/color"
 	"github.com/google/go-github/v31/github"
@@ -95,8 +96,11 @@ systemctl --user restart darknode`, url, ver)
 }
 
 func validateVersion(version string) error {
+	ctx, cancel := context.WithTimeout(context.Background(), 5 *time.Second)
+	defer cancel()
+
 	client := github.NewClient(nil)
-	_, response, err := client.Repositories.GetReleaseByTag(context.Background(), "renproject", "darknode-release", version)
+	_, response, err := client.Repositories.GetReleaseByTag(ctx, "renproject", "darknode-release", version)
 	if err != nil {
 		return err
 	}

util/node.go

@@ -10,6 +10,7 @@ import (
 	"path/filepath"
 	"regexp"
 	"strings"
+	"time"
 
 	"github.com/google/go-github/v31/github"
 	"github.com/hashicorp/go-version"
@@ -166,8 +167,11 @@ func ValidateTags(have, required string) bool {
 // LatestStableRelease checks the darknode release repo and return the version
 // of the latest release.
 func LatestStableRelease() (string, error) {
+	ctx, cancel := context.WithTimeout(context.Background(), 5 *time.Second)
+	defer cancel()
+
 	client := github.NewClient(nil)
-	releases, response, err := client.Repositories.ListReleases(context.Background(), "renproject", "darknode-release", nil)
+	releases, response, err := client.Repositories.ListReleases(ctx, "renproject", "darknode-release", nil)
 	if err != nil {
 		return "", err
 	}