Skip to content

Commit

Permalink
🐛 switch providers after CLI parsing (#2168)
Browse files Browse the repository at this point in the history
Early runtime checks can cause providers to switch. For example: A user
may start with the `aws` provider, but really want to user the `os`
provider to scan an instance in AWS:

```bash
cnquery shell aws ec2 instance-connect ec2-user@i-1234abc --region us-east-1 --profile home
```

In the above example, we use the `aws` provider to parse the CLI.
However, after that it's all about handing off the actual asset to the
`os` provider to execute the scan.

Signed-off-by: Dominik Richter <[email protected]>
  • Loading branch information
arlimus authored Oct 10, 2023
1 parent e3de2b5 commit 9da2623
Showing 1 changed file with 12 additions and 1 deletion.
13 changes: 12 additions & 1 deletion cli/providers/providers.go
Original file line number Diff line number Diff line change
Expand Up @@ -163,7 +163,6 @@ func attachProvidersToCmd(existing providers.Providers, cmd *Command) {
}
}
}

}

func setDefaultConnector(provider *plugin.Provider, connector *plugin.Connector, cmd *Command) {
Expand Down Expand Up @@ -474,6 +473,18 @@ func setConnector(provider *plugin.Provider, connector *plugin.Connector, run fu
runtime.Close()
providers.Coordinator.Shutdown()
log.Fatal().Msg("failed to process CLI arguments, nothing was returned")
return // adding this here as a compiler hint to stop warning about nil-dereferences
}

if cliRes.Asset == nil {
log.Warn().Err(err).Msg("failed to discover assets after processing CLI arguments")
} else {
assetRuntime, err := providers.Coordinator.RuntimeFor(cliRes.Asset, runtime)
if err != nil {
log.Warn().Err(err).Msg("failed to get runtime for an asset that was detected after parsing the CLI")
} else {
runtime = assetRuntime
}
}

run(cc, runtime, cliRes)
Expand Down

0 comments on commit 9da2623

Please sign in to comment.