mirror of
https://github.com/turbot/steampipe.git
synced 2026-02-19 07:00:17 -05:00
1e3db26e8f
Add support for implicit workspace. Closes #2514 If snapshot-location is a local file path, copy snapshot instead of uploading. Closes #2550 Do not show snapshot upload status if --progress=false. Closes #2517 Replace --workspace with --snapshot-location. Closes #2512
46 lines
1.5 KiB
Go
46 lines
1.5 KiB
Go
package initialisation
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/spf13/viper"
|
|
"github.com/turbot/steampipe/pkg/cloud"
|
|
"github.com/turbot/steampipe/pkg/constants"
|
|
"github.com/turbot/steampipe/pkg/steampipeconfig"
|
|
)
|
|
|
|
func getCloudMetadata() (*steampipeconfig.CloudMetadata, error) {
|
|
workspaceDatabase := viper.GetString(constants.ArgWorkspaceDatabase)
|
|
if workspaceDatabase == "local" {
|
|
// local database - nothing to do here
|
|
return nil, nil
|
|
}
|
|
connectionString := workspaceDatabase
|
|
|
|
var cloudMetadata *steampipeconfig.CloudMetadata
|
|
|
|
// so a backend was set - is it a connection string or a database name
|
|
workspaceDatabaseIsConnectionString := strings.HasPrefix(workspaceDatabase, "postgresql://") || strings.HasPrefix(workspaceDatabase, "postgres://")
|
|
if !workspaceDatabaseIsConnectionString {
|
|
// it must be a database name - verify the cloud token was provided
|
|
cloudToken := viper.GetString(constants.ArgCloudToken)
|
|
if cloudToken == "" {
|
|
return nil, fmt.Errorf("cloud token must be set to connect to workspace %s", workspaceDatabase)
|
|
}
|
|
|
|
// so we have a database and a token - build the connection string and set it in viper
|
|
var err error
|
|
if cloudMetadata, err = cloud.GetCloudMetadata(workspaceDatabase, cloudToken); err != nil {
|
|
return nil, err
|
|
}
|
|
// read connection string out of cloudMetadata
|
|
connectionString = cloudMetadata.ConnectionString
|
|
}
|
|
|
|
// now set the connection string in viper
|
|
viper.Set(constants.ArgConnectionString, connectionString)
|
|
|
|
return cloudMetadata, nil
|
|
}
|