mirror of
https://github.com/turbot/steampipe.git
synced 2026-02-24 11:00:53 -05:00
dfc35cf637
Update config precedence code to allow baseline defaults for keys with no corresponding flags. Fix default values od database-port and database-listen. Closes #35
114 lines
3.4 KiB
Go
114 lines
3.4 KiB
Go
package cmd
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
|
|
"github.com/hashicorp/go-hclog"
|
|
"github.com/spf13/cobra"
|
|
"github.com/spf13/viper"
|
|
"github.com/turbot/go-kit/helpers"
|
|
"github.com/turbot/steampipe-plugin-sdk/logging"
|
|
"github.com/turbot/steampipe/cmdconfig"
|
|
"github.com/turbot/steampipe/constants"
|
|
"github.com/turbot/steampipe/steampipeconfig"
|
|
"github.com/turbot/steampipe/task"
|
|
"github.com/turbot/steampipe/utils"
|
|
"github.com/turbot/steampipe/version"
|
|
)
|
|
|
|
// rootCmd represents the base command when called without any subcommands
|
|
var rootCmd = &cobra.Command{
|
|
Use: "steampipe [--version] [--help] COMMAND [args]",
|
|
Version: version.String(),
|
|
Short: "Query cloud resources using SQL",
|
|
Long: `Query cloud resources using SQL.
|
|
|
|
The available commands for execution are listed below.
|
|
The most common, useful commands are shown first, followed by
|
|
less common or more advanced commands. If you're just getting
|
|
started with Steampipe, stick with the common commands. For the
|
|
other commands, please read the help and docs before usage.
|
|
|
|
Getting started:
|
|
|
|
# Interactive SQL query console
|
|
steampipe query
|
|
|
|
# Execute a defined SQL query
|
|
steampipe query "select * from aws_s3_bucket"
|
|
|
|
# Install a plugin
|
|
steampipe plugin install azure
|
|
|
|
# Get help for a command
|
|
steampipe help query
|
|
|
|
Documentation available at https://steampipe.io/docs
|
|
`,
|
|
}
|
|
|
|
func InitCmd() {
|
|
rootCmd.PersistentFlags().String(constants.ArgInstallDir, constants.DefaultInstallDir, "Path to the Config Directory")
|
|
viper.BindPFlag(constants.ArgInstallDir, rootCmd.PersistentFlags().Lookup(constants.ArgInstallDir))
|
|
|
|
AddCommands()
|
|
|
|
// the `OnInitialize` callbacks are called right before PreRun
|
|
cobra.OnInitialize(initGlobalConfig, createLogger, task.RunTasks)
|
|
}
|
|
|
|
// initConfig reads in config file and ENV variables if set.
|
|
func initGlobalConfig() {
|
|
cmdconfig.InitViper()
|
|
|
|
// set global containing install dir
|
|
setInstallDir()
|
|
|
|
// load config (this sets the global config steampipeconfig.Config)
|
|
config, err := steampipeconfig.Load()
|
|
utils.FailOnError(err)
|
|
steampipeconfig.Config = config
|
|
|
|
// set viper config defaults from config and env vars
|
|
cmdconfig.SetViperDefaults(steampipeconfig.Config)
|
|
}
|
|
|
|
// CreateLogger :: create a hclog logger with the level specified by the SP_LOG env var
|
|
func createLogger() {
|
|
// TODO GET FROM VIPER
|
|
level := logging.LogLevel()
|
|
|
|
options := &hclog.LoggerOptions{Name: "steampipe", Level: hclog.LevelFromString(level)}
|
|
if options.Output == nil {
|
|
options.Output = os.Stderr
|
|
}
|
|
logger := hclog.New(options)
|
|
log.SetOutput(logger.StandardWriter(&hclog.StandardLoggerOptions{InferLevels: true}))
|
|
log.SetPrefix("")
|
|
log.SetFlags(0)
|
|
}
|
|
|
|
// SteampipeDir :: set the top level ~/.steampipe folder (creates if it doesnt exist)
|
|
func setInstallDir() {
|
|
installDir, err := helpers.Tildefy(viper.GetString(constants.ArgInstallDir))
|
|
utils.FailOnErrorWithMessage(err, fmt.Sprintf("failed to sanitize install directory"))
|
|
if _, err := os.Stat(installDir); os.IsNotExist(err) {
|
|
err = os.MkdirAll(installDir, 0755)
|
|
utils.FailOnErrorWithMessage(err, fmt.Sprintf("could not create installation directory: %s", installDir))
|
|
}
|
|
constants.SteampipeDir = installDir
|
|
}
|
|
|
|
func AddCommands() {
|
|
// explicitly initialise commands here rather than in init functions to allow us to handle errors from the config load
|
|
rootCmd.AddCommand(PluginCmd())
|
|
rootCmd.AddCommand(QueryCmd())
|
|
rootCmd.AddCommand(ServiceCmd())
|
|
}
|
|
|
|
func Execute() {
|
|
rootCmd.Execute()
|
|
}
|