mirror of
https://github.com/turbot/steampipe.git
synced 2025-12-19 18:12:43 -05:00
92 lines
2.5 KiB
Go
92 lines
2.5 KiB
Go
package metaquery
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"strings"
|
|
|
|
typeHelpers "github.com/turbot/go-kit/types"
|
|
pconstants "github.com/turbot/pipe-fittings/v2/constants"
|
|
"github.com/turbot/steampipe/v2/pkg/cmdconfig"
|
|
"github.com/turbot/steampipe/v2/pkg/constants"
|
|
"golang.org/x/exp/maps"
|
|
)
|
|
|
|
type handler func(ctx context.Context, input *HandlerInput) error
|
|
|
|
// Handle handles a metaquery execution from the interactive client
|
|
func Handle(ctx context.Context, input *HandlerInput) error {
|
|
cmd, _ := getCmdAndArgs(input.Query)
|
|
metaQueryObj, found := metaQueryDefinitions[cmd]
|
|
if !found {
|
|
return fmt.Errorf("not sure how to handle '%s'", cmd)
|
|
}
|
|
handlerFunction := metaQueryObj.handler
|
|
return handlerFunction(ctx, input)
|
|
}
|
|
|
|
// .header
|
|
// set the ArgHeader viper key with the boolean value evaluated from arg[0]
|
|
func setHeader(_ context.Context, input *HandlerInput) error {
|
|
cmdconfig.Viper().Set(pconstants.ArgHeader, typeHelpers.StringToBool(input.args()[0]))
|
|
return nil
|
|
}
|
|
|
|
// .multi
|
|
// set the ArgMulti viper key with the boolean value evaluated from arg[0]
|
|
func setMultiLine(_ context.Context, input *HandlerInput) error {
|
|
cmdconfig.Viper().Set(pconstants.ArgMultiLine, typeHelpers.StringToBool(input.args()[0]))
|
|
return nil
|
|
}
|
|
|
|
// .timing
|
|
// set the ArgHeader viper key with the boolean value evaluated from arg[0]
|
|
func setTiming(ctx context.Context, input *HandlerInput) error {
|
|
if len(input.args()) == 0 {
|
|
showTimingFlag()
|
|
return nil
|
|
}
|
|
|
|
cmdconfig.Viper().Set(pconstants.ArgTiming, input.args()[0])
|
|
return nil
|
|
}
|
|
|
|
func showTimingFlag() {
|
|
timing := cmdconfig.Viper().GetString(pconstants.ArgTiming)
|
|
|
|
fmt.Printf(`Timing is %s. Available options are: %s`,
|
|
pconstants.Bold(timing),
|
|
pconstants.Bold(strings.Join(maps.Keys(constants.QueryTimingValueLookup), ", ")))
|
|
// add an empty line here so that the rendering buffer can start from the next line
|
|
fmt.Println()
|
|
|
|
return
|
|
}
|
|
|
|
// .separator and .output
|
|
// set the value of `viperKey` in `viper` with the value from `args[0]`
|
|
func setViperConfigFromArg(viperKey string) handler {
|
|
return func(_ context.Context, input *HandlerInput) error {
|
|
cmdconfig.Viper().Set(viperKey, input.args()[0])
|
|
return nil
|
|
}
|
|
}
|
|
|
|
// .exit
|
|
func doExit(_ context.Context, input *HandlerInput) error {
|
|
input.ClosePrompt()
|
|
return nil
|
|
}
|
|
|
|
// .clear
|
|
func clearScreen(_ context.Context, input *HandlerInput) error {
|
|
input.Prompt.ClearScreen()
|
|
return nil
|
|
}
|
|
|
|
// .autocomplete
|
|
func setAutoComplete(_ context.Context, input *HandlerInput) error {
|
|
cmdconfig.Viper().Set(pconstants.ArgAutoComplete, typeHelpers.StringToBool(input.args()[0]))
|
|
return nil
|
|
}
|