Add GQL error logging support

superfly · Jun 24, 2020 · 698f1ed · 698f1ed
1 parent ec85266
commit 698f1ed
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/api/client.go b/api/client.go
@@ -7,18 +7,24 @@ import (
 	"errors"
 	"fmt"
 	"net/http"
+	"os"
 	"regexp"
 	"strings"
 
 	"github.com/machinebox/graphql"
 )
 
 var baseURL string
+var errorLog bool
 
 func SetBaseURL(url string) {
 	baseURL = url
 }
 
+func SetErrorLog(log bool) {
+	errorLog = log
+}
+
 type Client struct {
 	httpClient  *http.Client
 	client      *graphql.Client
@@ -58,6 +64,11 @@ func (c *Client) RunWithContext(ctx context.Context, req *graphql.Request) (Quer
 	if err != nil && strings.HasPrefix(err.Error(), "graphql: ") {
 		return resp, errors.New(strings.TrimPrefix(err.Error(), "graphql: "))
 	}
+
+	if resp.Errors != nil && errorLog {
+		fmt.Fprintf(os.Stderr, "Error: %+v\n", resp.Errors)
+	}
+
 	return resp, err
 }
 

diff --git a/flyctl/flyctl.go b/flyctl/flyctl.go
@@ -65,11 +65,13 @@ func initViper() {
 	viper.BindEnv(ConfigAPIToken, "FLY_ACCESS_TOKEN")
 	viper.BindEnv(ConfigAPIToken, "FLY_API_TOKEN")
 	viper.BindEnv(ConfigVerboseOutput, "VERBOSE")
+	viper.BindEnv(ConfigGQLErrorLogging, "GQLErrorLogging")
 
 	viper.SetEnvPrefix("FLY")
 	viper.AutomaticEnv()
 
 	api.SetBaseURL(viper.GetString(ConfigAPIBaseURL))
+	api.SetErrorLog(viper.GetBool(ConfigGQLErrorLogging))
 }
 
 func loadConfig() error {