add firestore public client

Author: Yevgeny Pats

client/client.go

@@ -1,14 +1,17 @@
 package client
 
 import (
+	"context"
+	"google.golang.org/api/option"
+	"google.golang.org/grpc"
 	"net/http"
 	"time"
 
 	"cloud.google.com/go/firestore"
 )
 
 const FuzzitEndpoint = "https://app.fuzzit.dev"
-const Version = "v2.4.53"
+const Version = "v2.4.54"
 
 type Target struct {
 	Name         string `firestore:"target_name"`
@@ -61,19 +64,27 @@ type FuzzitClient struct {
 	fuzzerFilename  string // this is mainly used by the agent
 }
 
-func NewUnAuthenticatedClient() *FuzzitClient {
-	c := &FuzzitClient{}
-	c.httpClient = &http.Client{Timeout: 120 * time.Second}
-	return c
-}
-
 func NewFuzzitClient(apiKey string) (*FuzzitClient, error) {
+	ctx := context.Background()
+
 	c := &FuzzitClient{}
 	c.httpClient = &http.Client{Timeout: 120 * time.Second}
 	c.ApiKey = apiKey
-	err := c.refreshToken()
+
+	conn, err := grpc.Dial("firestore.googleapis.com", grpc.WithInsecure())
+	if err != nil {
+		return nil, err
+	}
+
+	firestoreClient, err := firestore.NewClient(ctx, "fuzzit-b5fbf", option.WithGRPCConn(conn))
+	c.firestoreClient = firestoreClient
 	if err != nil {
 		return nil, err
 	}
+
+	if err := c.refreshToken(); err != nil {
+		return nil, err
+	}
+
 	return c, nil
 }

client/commands_test.go

@@ -8,7 +8,10 @@ import (
 )
 
 func TestDownloadAndExtractCorpus(t *testing.T) {
-	unAuthenticatedClient := NewUnAuthenticatedClient()
+	unAuthenticatedClient, err := NewFuzzitClient("")
+	if err != nil {
+		t.Fatal(err)
+	}
 	unAuthenticatedClient.Org = "fuzzitdev"
 
 	authenticatedClient, err := NewFuzzitClient(os.Getenv("FUZZIT_API_KEY"))
@@ -54,7 +57,10 @@ func TestDownloadAndExtractCorpus(t *testing.T) {
 }
 
 func TestCreateLocalJob(t *testing.T) {
-	unAuthenticatedClient := NewUnAuthenticatedClient()
+	unAuthenticatedClient, err := NewFuzzitClient("")
+	if err != nil {
+		t.Fatal(err)
+	}
 	unAuthenticatedClient.Org = "fuzzitdev"
 
 	authenticatedClient, err := NewFuzzitClient(os.Getenv("FUZZIT_API_KEY"))

cmd/root.go

@@ -35,13 +35,9 @@ var rootCmd = &cobra.Command{
 	PersistentPreRun: func(cmd *cobra.Command, args []string) {
 		apiKey := viper.GetString("api-key")
 		var err error
-		if apiKey != "" {
-			gFuzzitClient, err = client.NewFuzzitClient(apiKey)
-			if err != nil {
-				log.Fatalln(err)
-			}
-		} else {
-			gFuzzitClient = client.NewUnAuthenticatedClient()
+		gFuzzitClient, err = client.NewFuzzitClient(apiKey)
+		if err != nil {
+			log.Fatalln(err)
 		}
 	},
 }