change graphql client to use header auth

Author: emmanuelm41

connections/database/graphql.go

@@ -1,12 +1,11 @@
 package database
 
 import (
-	"context"
 	"encoding/json"
 	"fmt"
 	"github.com/hasura/go-graphql-client"
 	"go.uber.org/zap"
-	"golang.org/x/oauth2"
+	"net/http"
 	"time"
 )
 
@@ -27,13 +26,15 @@ type GraphqlSubscriptionClient struct {
 }
 
 func NewGraphqlQueryClient(host string, token string) GraphqlClient {
-	src := oauth2.StaticTokenSource(
-		&oauth2.Token{AccessToken: token},
-	)
-	httpClient := oauth2.NewClient(context.Background(), src)
+	transport := http.DefaultTransport
+	if token != "" {
+		transport = AuthHeaderTransport{Transport: http.DefaultTransport, Token: token}
+	}
+
+	customHttpClient := http.Client{Transport: transport}
 	return GraphqlClient{
 		Host:   host,
-		Client: graphql.NewClient(host, httpClient),
+		Client: graphql.NewClient(host, &customHttpClient),
 	}
 }
 

connections/database/interface.go

@@ -3,8 +3,19 @@ package database
 import (
 	"fmt"
 	"gorm.io/gorm"
+	"net/http"
 )
 
+type AuthHeaderTransport struct {
+	Transport http.RoundTripper
+	Token     string
+}
+
+func (adt AuthHeaderTransport) RoundTrip(req *http.Request) (*http.Response, error) {
+	req.Header.Add("x-hasura-admin-secret", adt.Token)
+	return adt.Transport.RoundTrip(req)
+}
+
 type DBConnection interface {
 	GetDB() *gorm.DB
 }

go.mod

@@ -13,7 +13,6 @@ require (
 	github.com/spf13/cobra v1.1.1
 	github.com/spf13/viper v1.7.1
 	go.uber.org/zap v1.16.0
-	golang.org/x/oauth2 v0.0.0-20190604053449-0f29369cfe45
 	gorm.io/driver/postgres v1.0.6
 	gorm.io/gorm v1.21.12
 )

go.sum

@@ -635,7 +635,6 @@ golang.org/x/net v0.0.0-20201110031124-69a78807bb2b h1:uwuIcX0g4Yl1NC5XAz37xsr2l
 golang.org/x/net v0.0.0-20201110031124-69a78807bb2b/go.mod h1:sp8m0HH+o8qH0wwXwYZr8TS3Oi6o0r6Gce1SSxlDquU=
 golang.org/x/oauth2 v0.0.0-20180821212333-d2e6202438be/go.mod h1:N/0e6XlmueqKjAGxoOufVs8QHGRruUQn6yWY3a++T0U=
 golang.org/x/oauth2 v0.0.0-20190226205417-e64efc72b421/go.mod h1:gOpvHmFTYa4IltrdGE7lF6nIHvwfUNPOp7c8zoXwtLw=
-golang.org/x/oauth2 v0.0.0-20190604053449-0f29369cfe45 h1:SVwTIAaPC2U/AvvLNZ2a7OVsmBpC8L5BlwK1whH3hm0=
 golang.org/x/oauth2 v0.0.0-20190604053449-0f29369cfe45/go.mod h1:gOpvHmFTYa4IltrdGE7lF6nIHvwfUNPOp7c8zoXwtLw=
 golang.org/x/sync v0.0.0-20180314180146-1d60e4601c6f/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
 golang.org/x/sync v0.0.0-20181108010431-42b317875d0f/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
@@ -733,7 +732,6 @@ google.golang.org/api v0.13.0/go.mod h1:iLdEw5Ide6rF15KTC1Kkl0iskquN2gFfn9o9XIsb
 google.golang.org/appengine v1.1.0/go.mod h1:EbEs0AVv82hx2wNQdGPgUI5lhzA/G0D9YwlJXL52JkM=
 google.golang.org/appengine v1.4.0/go.mod h1:xpcJRLb0r/rnEns0DIKYYv+WjYCduHsrkT7/EB5XEv4=
 google.golang.org/appengine v1.5.0/go.mod h1:xpcJRLb0r/rnEns0DIKYYv+WjYCduHsrkT7/EB5XEv4=
-google.golang.org/appengine v1.6.1 h1:QzqyMA1tlu6CgqCDUtU9V+ZKhLFT2dkJuANu5QaxI3I=
 google.golang.org/appengine v1.6.1/go.mod h1:i06prIuMbXzDqacNJfV5OdTW448YApPu5ww/cMBSeb0=
 google.golang.org/genproto v0.0.0-20180817151627-c66870c02cf8/go.mod h1:JiN7NxoALGmiZfu7CAH4rXhgtRTLTxftemlI0sWmxmc=
 google.golang.org/genproto v0.0.0-20190307195333-5fe7a883aa19/go.mod h1:VzzqZJRnGkLBvHegQrXjBqPurQTc5/KpmUdxsrq26oE=