sq/cli/cmd_src.go

package cli

import (
	"github.com/neilotoole/sq/cli/flag"
	"github.com/neilotoole/sq/cli/run"
	"github.com/spf13/cobra"
)

func newSrcCommand() *cobra.Command {
	cmd := &cobra.Command{
		Use:  "src [@HANDLE]",
		RunE: execSrc,
		Example: `  # Get active data source
  $ sq src

  # Set @my1 as active data source
  $ sq src @my1`,
		Args:              cobra.MaximumNArgs(1),
		ValidArgsFunction: completeHandle(1),
		Short:             "Get or set active data source",
		Long: `Get or set active data source. If no argument provided, get the active data
source. Otherwise, set @HANDLE as the active data source.`,
	}

	addTextFlags(cmd)
	cmd.Flags().BoolP(flag.JSON, flag.JSONShort, false, flag.JSONUsage)
	cmd.Flags().BoolP(flag.Compact, flag.CompactShort, false, flag.CompactUsage)
	cmd.Flags().BoolP(flag.YAML, flag.YAMLShort, false, flag.YAMLUsage)

	return cmd
}

func execSrc(cmd *cobra.Command, args []string) error {
	ru := run.FromContext(cmd.Context())
	cfg := ru.Config

	if len(args) == 0 {
		// Get the active data source
		src := cfg.Collection.Active()
		if src == nil {
			return nil
		}

		return ru.Writers.Source.Source(cfg.Collection, src)
	}

	src, err := cfg.Collection.SetActive(args[0], false)
	if err != nil {
		return err
	}

	err = ru.ConfigStore.Save(cmd.Context(), cfg)
	if err != nil {
		return err
	}

	return ru.Writers.Source.Source(cfg.Collection, src)
}