mudhorn/oh-my-posh
1
0
Fork 0
mirror of https://github.com/JanDeDobbeleer/oh-my-posh.git synced 2024-12-28 20:39:40 -08:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat(az): display info origin

Browse source 
resolves #1562
This commit is contained in:
Jan De Dobbeleer 2022-01-13 20:29:03 +01:00 committed by Jan De Dobbeleer
parent b1c427abaa
commit d45c51cccc
2 changed files with 17 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
src/segment_az.go
View file

@ -11,6 +11,7 @@ type az struct {
env Environment env Environment
AzureSubscription AzureSubscription
Origin string
} }
type AzureConfig struct { type AzureConfig struct {
@ -123,6 +124,7 @@ func (a *az) getAzureProfile() bool {
for _, subscription := range config.Subscriptions { for _, subscription := range config.Subscriptions {
if subscription.IsDefault { if subscription.IsDefault {
a.AzureSubscription = *subscription a.AzureSubscription = *subscription
a.Origin = "CLI"
return true return true
} }
} }
@ -151,5 +153,6 @@ func (a *az) getAzureRmContext() bool {
a.User = &AzureUser{ a.User = &AzureUser{
Name: defaultContext.Subscription.ExtendedProperties.Account, Name: defaultContext.Subscription.ExtendedProperties.Account,
} }
a.Origin = "PWSH"
return true return true
} }

14
src/segment_az_test.go
View file

@ -42,6 +42,20 @@ func TestAzSegment(t *testing.T) {
Template: "{{ .Burp }}", Template: "{{ .Burp }}",
HasPowerShell: true, HasPowerShell: true,
}, },
{
Case: "PWSH",
ExpectedEnabled: true,
ExpectedString: "PWSH",
Template: "{{ .Origin }}",
HasPowerShell: true,
},
{
Case: "CLI",
ExpectedEnabled: true,
ExpectedString: "CLI",
Template: "{{ .Origin }}",
HasCLI: true,
},
} }
for _, tc := range cases { for _, tc := range cases {