sq/cli/cmd_src.go

package cli

import (
	"github.com/spf13/cobra"

	"github.com/neilotoole/sq/libsq/errz"
)

func newSrcCommand() (*cobra.Command, runFunc) {
	cmd := &cobra.Command{
		Use: "src [@HANDLE]",
		Example: `  # get active data source
   sq src
   # set @my1 as active data source
   sq src @my1`,
		// RunE:    execSrc,
		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.`,
	}

	//cmd.Flags().BoolP(flagJSON, flagJSONShort, false, flagJSONUsage)
	//cmd.Flags().BoolP(flagTable, flagTableShort, false, flagTableUsage)
	//cmd.Flags().BoolP(flagHeader, flagHeaderShort, false, flagHeaderUsage)

	return cmd, execSrc
}

func execSrc(rc *RunContext, cmd *cobra.Command, args []string) error {
	if len(args) > 1 {
		return errz.Errorf(msgInvalidArgs)
	}

	cfg := rc.Config

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

		return rc.writers.srcw.Source(src)
	}

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

	err = rc.ConfigStore.Save(cfg)
	if err != nil {
		return err
	}

	return rc.writers.srcw.Source(src)
}