erclient

var Cmd = &cobra.Command{
	Use:   "er",
	Short: "Issue command to Grip External Resource",
	PersistentPreRun: func(cmd *cobra.Command, args []string) {
		if len(args) > 0 {
			host = args[0]
		}
	},
}

var GetCmd = &cobra.Command{
	Use:   "get <collection> <ids ...>",
	Short: "List rows with field match",
	Args:  cobra.MinimumNArgs(3),
	RunE: func(cmd *cobra.Command, args []string) error {
		client, err := getClient()
		if err != nil {
			return err
		}
		collection := args[1]
		ids := args[2:]

		idChan := make(chan *gripper.RowRequest, 10)

		rows, err := client.GetRowsByID(context.Background(), source, collection, idChan)
		if err != nil {
			return err
		}

		go func() {
			defer close(idChan)
			for _, i := range ids {
				r := gripper.RowRequest{Id: i}
				idChan <- &r
			}
		}()
		jm := protojson.MarshalOptions{}
		for row := range rows {
			if dataOnly {
				fmt.Printf("%s\n", jm.Format(row.Data))
			} else {
				fmt.Printf("%s\t%s\n", row.Id, jm.Format(row.Data))
			}
		}
		return nil
	},
}

var IdsCmd = &cobra.Command{
	Use:   "ids <collection>",
	Short: "List ids from a collection",
	Args:  cobra.ExactArgs(2),
	RunE: func(cmd *cobra.Command, args []string) error {
		client, err := getClient()
		if err != nil {
			return err
		}
		collection := args[1]
		for i := range client.GetIDs(context.Background(), source, collection) {
			fmt.Printf("%s\n", i)
		}
		return nil
	},
}

var InfoCmd = &cobra.Command{
	Use:   "info",
	Short: "Get info about a collection",
	Args:  cobra.ExactArgs(2),
	RunE: func(cmd *cobra.Command, args []string) error {
		client, err := getClient()
		if err != nil {
			return err
		}
		collection := args[1]
		out, err := client.GetCollectionInfo(context.Background(), source, collection)
		if err != nil {
			return err
		}
		jm := protojson.MarshalOptions{}
		fmt.Printf("%s\n", jm.Format(out))
		return nil
	},
}

var ListCmd = &cobra.Command{
	Use:   "list",
	Short: "List collections provided by external resource",
	Args:  cobra.ExactArgs(1),
	RunE: func(cmd *cobra.Command, args []string) error {
		client, err := getClient()
		if err != nil {
			return err
		}
		for name := range client.GetCollections(context.Background(), source) {
			fmt.Printf("%s\n", name)
		}
		return nil
	},
}

var QueryCmd = &cobra.Command{
	Use:   "query <collection> <field> <value>",
	Short: "List rows with field match",
	Args:  cobra.ExactArgs(4),
	RunE: func(cmd *cobra.Command, args []string) error {
		client, err := getClient()
		if err != nil {
			return err
		}
		collection := args[1]
		field := args[2]
		value := args[3]
		jm := protojson.MarshalOptions{}
		rows, err := client.GetRowsByField(context.Background(), source, collection, field, value)
		if err != nil {
			return err
		}
		for row := range rows {
			if dataOnly {
				fmt.Printf("%s\n", jm.Format(row.Data))
			} else {
				fmt.Printf("%s\t%s\n", row.Id, jm.Format(row.Data))
			}
		}
		return nil
	},
}

var RowsCmd = &cobra.Command{
	Use:   "rows <collection>",
	Short: "List rows from a collection",
	Args:  cobra.ExactArgs(2),
	RunE: func(cmd *cobra.Command, args []string) error {
		client, err := getClient()
		if err != nil {
			return err
		}
		collection := args[1]
		jm := protojson.MarshalOptions{}
		for row := range client.GetRows(context.Background(), source, collection) {
			if dataOnly {
				fmt.Printf("%s\n", jm.Format(row.Data))
			} else {
				fmt.Printf("%s\t%s\n", row.Id, jm.Format(row.Data))
			}
		}
		return nil
	},
}