mirror of
https://github.com/neilotoole/sq.git
synced 2024-12-18 21:52:28 +03:00
111 lines
2.9 KiB
Go
111 lines
2.9 KiB
Go
// Package run holds the run.Run construct, which encapsulates CLI state
|
|
// for a command execution.
|
|
package run
|
|
|
|
import (
|
|
"context"
|
|
"io"
|
|
"os"
|
|
|
|
"github.com/neilotoole/sq/libsq"
|
|
|
|
"github.com/neilotoole/sq/cli/output"
|
|
|
|
"github.com/neilotoole/sq/libsq/core/options"
|
|
|
|
"github.com/neilotoole/sq/cli/config"
|
|
"github.com/neilotoole/sq/libsq/core/cleanup"
|
|
"github.com/neilotoole/sq/libsq/core/errz"
|
|
"github.com/neilotoole/sq/libsq/driver"
|
|
"github.com/neilotoole/sq/libsq/source"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
type runKey struct{}
|
|
|
|
// NewContext returns ctx with ru added as a value.
|
|
func NewContext(ctx context.Context, ru *Run) context.Context {
|
|
if ctx == nil {
|
|
ctx = context.Background()
|
|
}
|
|
|
|
return context.WithValue(ctx, runKey{}, ru)
|
|
}
|
|
|
|
// FromContext extracts the Run added to ctx via NewContext.
|
|
func FromContext(ctx context.Context) *Run {
|
|
return ctx.Value(runKey{}).(*Run)
|
|
}
|
|
|
|
// Run is a container for injectable resources passed
|
|
// to all cobra exec funcs. The Close method should be invoked when
|
|
// the Run is no longer needed.
|
|
type Run struct {
|
|
// Stdin typically is os.Stdin, but can be changed for testing.
|
|
Stdin *os.File
|
|
|
|
// Out is the output destination, typically os.Stdout.
|
|
Out io.Writer
|
|
|
|
// ErrOut is the error output destination, typically os.Stderr.
|
|
ErrOut io.Writer
|
|
|
|
// Cmd is the command instance provided by cobra for
|
|
// the currently executing command. This field will
|
|
// be set before the command's runFunc is invoked.
|
|
Cmd *cobra.Command
|
|
|
|
// Args is the arg slice supplied by cobra for
|
|
// the currently executing command. This field will
|
|
// be set before the command's runFunc is invoked.
|
|
Args []string
|
|
|
|
// Config is the run's config.
|
|
Config *config.Config
|
|
|
|
// ConfigStore manages config persistence.
|
|
ConfigStore config.Store
|
|
|
|
// OptionsRegistry is a registry of CLI options.Opt instances.
|
|
OptionsRegistry *options.Registry
|
|
|
|
// DriverRegistry is a registry of driver implementations.
|
|
DriverRegistry *driver.Registry
|
|
|
|
// Files manages file access.
|
|
Files *source.Files
|
|
|
|
// Databases mediates access to databases.
|
|
Databases *driver.Databases
|
|
|
|
// Writers holds the various writer types that
|
|
// the CLI uses to print output.
|
|
Writers *output.Writers
|
|
|
|
// Cleanup holds cleanup functions.
|
|
Cleanup *cleanup.Cleanup
|
|
}
|
|
|
|
// Close should be invoked to dispose of any open resources
|
|
// held by ru. If an error occurs during Close and ru.Log
|
|
// is not nil, that error is logged at WARN level before
|
|
// being returned.
|
|
func (ru *Run) Close() error {
|
|
if ru == nil {
|
|
return nil
|
|
}
|
|
|
|
return errz.Wrap(ru.Cleanup.Run(), "Close Run")
|
|
}
|
|
|
|
// NewQueryContext returns a *libsq.QueryContext constructed from ru.
|
|
func NewQueryContext(ru *Run, args map[string]string) *libsq.QueryContext {
|
|
return &libsq.QueryContext{
|
|
Collection: ru.Config.Collection,
|
|
DBOpener: ru.Databases,
|
|
JoinDBOpener: ru.Databases,
|
|
ScratchDBOpener: ru.Databases,
|
|
Args: args,
|
|
}
|
|
}
|