bugfix in target create command

Author: Yevgeny Pats

client/commands.go

@@ -73,22 +73,28 @@ func (c *fuzzitClient) GetResource(resource string) error {
 	}
 }
 
-func (c *fuzzitClient) CreateTarget(targetConfig Target, seedPath string) (*firestore.DocumentRef, error) {
+func (c *fuzzitClient) CreateTarget(targetName string, seedPath string) (*firestore.DocumentRef, error) {
 	ctx := context.Background()
-	collectionRef := c.firestoreClient.Collection("orgs/" + c.Org + "/targets")
-	doc, _, err := collectionRef.Add(ctx, targetConfig)
-	if err != nil {
-		return nil, err
+	docRef := c.firestoreClient.Doc("orgs/" + c.Org + "/targets/" + targetName)
+	_, err := docRef.Get(ctx)
+	if err == nil {
+		return nil, fmt.Errorf("target %s already exist", targetName)
 	}
 
 	if seedPath != "" {
-		storagePath := fmt.Sprintf("orgs/%s/targets/%s/seed", c.Org, doc.ID)
+		storagePath := fmt.Sprintf("orgs/%s/targets/%s/seed", c.Org, targetName)
 		err := c.uploadFile(seedPath, storagePath, "application/gzip", "seed.tar.gz")
 		if err != nil {
 			return nil, err
 		}
 	}
-	return doc, nil
+
+	_, err = docRef.Set(ctx, Target{Name: targetName})
+	if err != nil {
+		return nil, err
+	}
+
+	return docRef, nil
 }
 
 func (c *fuzzitClient) CreateJob(jobConfig Job, files []string) (*firestore.DocumentRef, error) {

cmd/root.go

@@ -26,7 +26,7 @@ var cfgFile string
 var rootCmd = &cobra.Command{
 	Use:     "Fuzzit",
 	Short:   "Continuous fuzzing made simple CLI",
-	Version: "2.2.0",
+	Version: "2.3.0",
 }
 
 func Execute() {

cmd/target.go

@@ -32,14 +32,12 @@ var targetCmd = &cobra.Command{
 		if err != nil {
 			log.Fatal(err)
 		}
-		newTarget := client.Target{
-			Name: args[0],
-		}
+
 		seed, err := cmd.Flags().GetString("seed")
 		if err != nil {
 			log.Fatal(err)
 		}
-		docRef, err := c.CreateTarget(newTarget, seed)
+		docRef, err := c.CreateTarget(args[0], seed)
 		if err != nil {
 			log.Fatal(err)
 		}