haskal
/
starship
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

feat(aws): add support for getting profile from awsu (#2451)

This commit is contained in:
nils måsén 2021-03-13 09:35:50 +01:00 committed by GitHub
parent 69b9bf72c3
commit e5cdd9c1b3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 25 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
docs/config/README.md
View File

@ -249,6 +249,9 @@ The `aws` module shows the current AWS region and profile. This is based on
When using [aws-vault](https://github.com/99designs/aws-vault) the profile
is read from the `AWS_VAULT` env var.
When using [awsu](https://github.com/kreuzwerker/awsu) the profile
is read from the `AWSU_PROFILE` env var.
### Options
| Option | Default | Description |

30
src/modules/aws.rs
View File

@ -47,14 +47,14 @@ fn get_aws_region_from_config(context: &Context, aws_profile: Option<&str>) -> O
}
fn get_aws_profile_and_region(context: &Context) -> (Option<Profile>, Option<Region>) {
match (
context
.get_env("AWS_VAULT")
.or_else(|| context.get_env("AWS_PROFILE")),
context
.get_env("AWS_DEFAULT_REGION")
.or_else(|| context.get_env("AWS_REGION")),
) {
let profile_env_vars = vec!["AWSU_PROFILE", "AWS_VAULT", "AWS_PROFILE"];
let profile = profile_env_vars
.iter()
.find_map(|env_var| context.get_env(env_var));
let region = context
.get_env("AWS_DEFAULT_REGION")
.or_else(|| context.get_env("AWS_REGION"));
match (profile, region) {
(Some(p), Some(r)) => (Some(p), Some(r)),
(None, Some(r)) => (None, Some(r)),
(Some(ref p), None) => (
@ -200,6 +200,20 @@ mod tests {
assert_eq!(expected, actual);
}
#[test]
fn profile_set_from_awsu() {
let actual = ModuleRenderer::new("aws")
.env("AWSU_PROFILE", "astronauts-awsu")
.env("AWS_PROFILE", "astronauts-profile")
.collect();
let expected = Some(format!(
"on {}",
Color::Yellow.bold().paint("☁️ astronauts-awsu ")
));
assert_eq!(expected, actual);
}
#[test]
fn profile_and_region_set() {
let actual = ModuleRenderer::new("aws")