mirror of
https://github.com/turbot/steampipe.git
synced 2025-12-19 18:12:43 -05:00
123 lines
3.7 KiB
Go
123 lines
3.7 KiB
Go
package cmd
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
|
|
"github.com/mattn/go-isatty"
|
|
"github.com/spf13/cobra"
|
|
"github.com/spf13/viper"
|
|
filehelpers "github.com/turbot/go-kit/files"
|
|
"github.com/turbot/pipe-fittings/v2/app_specific"
|
|
"github.com/turbot/pipe-fittings/v2/constants"
|
|
"github.com/turbot/pipe-fittings/v2/utils"
|
|
"github.com/turbot/steampipe/pkg/error_helpers"
|
|
"github.com/turbot/steampipe/pkg/statushooks"
|
|
)
|
|
|
|
var exitCode int
|
|
|
|
// rootCmd represents the base command when called without any subcommands
|
|
var rootCmd = &cobra.Command{
|
|
Use: "steampipe [--version] [--help] COMMAND [args]",
|
|
Short: "Query cloud resources using SQL",
|
|
Long: `Steampipe: select * from cloud;
|
|
|
|
Dynamically query APIs, code and more with SQL.
|
|
Zero-ETL from 140+ data sources.
|
|
|
|
Common commands:
|
|
|
|
# Interactive SQL query console
|
|
steampipe query
|
|
|
|
# Install a plugin from the hub - https://hub.steampipe.io
|
|
steampipe plugin install aws
|
|
|
|
# Execute a defined SQL query
|
|
steampipe query "select * from aws_s3_bucket"
|
|
|
|
# Get help for a command
|
|
steampipe help query
|
|
|
|
Documentation: https://steampipe.io/docs
|
|
`,
|
|
}
|
|
|
|
func InitCmd() {
|
|
utils.LogTime("cmd.root.InitCmd start")
|
|
defer utils.LogTime("cmd.root.InitCmd end")
|
|
|
|
defaultInstallDir, err := filehelpers.Tildefy(app_specific.DefaultInstallDir)
|
|
error_helpers.FailOnError(err)
|
|
|
|
// Set the version after viper has been initialized
|
|
rootCmd.Version = viper.GetString("main.version")
|
|
rootCmd.SetVersionTemplate("Steampipe v{{.Version}}\n")
|
|
|
|
// global flags
|
|
rootCmd.PersistentFlags().String(constants.ArgWorkspaceProfile, "default", "The workspace profile to use") // workspace profile profile is a global flag since install-dir(global) can be set through the workspace profile
|
|
rootCmd.PersistentFlags().String(constants.ArgInstallDir, defaultInstallDir, "Path to the Config Directory")
|
|
rootCmd.PersistentFlags().Bool(constants.ArgSchemaComments, true, "Include schema comments when importing connection schemas")
|
|
|
|
error_helpers.FailOnError(viper.BindPFlag(constants.ArgInstallDir, rootCmd.PersistentFlags().Lookup(constants.ArgInstallDir)))
|
|
error_helpers.FailOnError(viper.BindPFlag(constants.ArgWorkspaceProfile, rootCmd.PersistentFlags().Lookup(constants.ArgWorkspaceProfile)))
|
|
error_helpers.FailOnError(viper.BindPFlag(constants.ArgSchemaComments, rootCmd.PersistentFlags().Lookup(constants.ArgSchemaComments)))
|
|
|
|
AddCommands()
|
|
|
|
// disable auto completion generation, since we don't want to support
|
|
// powershell yet - and there's no way to disable powershell in the default generator
|
|
rootCmd.CompletionOptions.DisableDefaultCmd = true
|
|
rootCmd.Flags().BoolP(constants.ArgHelp, "h", false, "Help for steampipe")
|
|
|
|
hideRootFlags(constants.ArgSchemaComments)
|
|
|
|
// tell OS to reclaim memory immediately
|
|
os.Setenv("GODEBUG", "madvdontneed=1")
|
|
|
|
}
|
|
|
|
func hideRootFlags(flags ...string) {
|
|
for _, flag := range flags {
|
|
rootCmd.Flag(flag).Hidden = true
|
|
}
|
|
}
|
|
|
|
func AddCommands() {
|
|
// explicitly initialise commands here rather than in init functions to allow us to handle errors from the config load
|
|
rootCmd.AddCommand(
|
|
pluginCmd(),
|
|
queryCmd(),
|
|
serviceCmd(),
|
|
generateCompletionScriptsCmd(),
|
|
pluginManagerCmd(),
|
|
loginCmd(),
|
|
)
|
|
}
|
|
|
|
func Execute() int {
|
|
utils.LogTime("cmd.root.Execute start")
|
|
defer utils.LogTime("cmd.root.Execute end")
|
|
|
|
ctx := createRootContext()
|
|
|
|
err := rootCmd.ExecuteContext(ctx)
|
|
if err != nil {
|
|
exitCode = 1
|
|
}
|
|
return exitCode
|
|
}
|
|
|
|
// create the root context - add a status renderer
|
|
func createRootContext() context.Context {
|
|
statusRenderer := statushooks.NullHooks
|
|
// if the client is a TTY, inject a status spinner
|
|
if isatty.IsTerminal(os.Stdout.Fd()) {
|
|
statusRenderer = statushooks.NewStatusSpinnerHook()
|
|
}
|
|
|
|
ctx := statushooks.AddStatusHooksToContext(context.Background(), statusRenderer)
|
|
return ctx
|
|
}
|