set.go

package main

import (
	firestore "cloud.google.com/go/firestore"
	"context"
	"fmt"
	"github.com/urfave/cli"
)

func setData(
	client *firestore.Client,
	collectionPath string,
	documentPath string,
	id string,
	data string,
	merge bool) error {

	object, err := unmarshallData(data)
	if err != nil {
		return err
	}

	transformExtendedJsonMapToFirestoreMap(object, client)

	var options []firestore.SetOption
	if merge {
		options = append(options, firestore.MergeAll)
	}

	if collectionPath != "" {
		_, err = client.
			Collection(collectionPath).
			Doc(id).
			Set(context.Background(), object, options...)
	} else {
		_, err = client.
			Doc(documentPath).
			Set(context.Background(), object, options...)
	}

	if err != nil {
		return err
	}

	return nil
}

func setCommandAction(c *cli.Context) error {
	argsLength := len(c.Args())

	if argsLength < 2 || argsLength > 3 {
		return cli.NewExitError("Wrong number of arguments", 85)
	}

	merge := c.Bool("merge")

	var collectionPath, id, data, documentPath string

	if argsLength == 3 {
		collectionPath = c.Args().First()
		id = c.Args().Get(1)
		data = c.Args().Get(2)
	} else {
		documentPath = c.Args().First()
		data = c.Args().Get(1)
	}

	client, err := createClient(credentials)
	if err != nil {
		return cliClientError(err)
	}

	err = setData(client, collectionPath, documentPath, id, data, merge)
	if err != nil {
		return cli.NewExitError(fmt.Sprintf("Failed to write data. \n%v", err), 85)
	}

	if collectionPath != "" {
		fmt.Fprintf(c.App.Writer, "%v\n", id)
	} else {
		fmt.Fprintf(c.App.Writer, "%v\n", documentPath)
	}
	defer client.Close()
	return nil
}