mudhorn/oh-my-posh
1
0
Fork 0
mirror of https://github.com/JanDeDobbeleer/oh-my-posh.git synced 2025-01-26 02:21:25 -08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

fix(az): allow color templates

Browse source
This commit is contained in:
Jan De Dobbeleer 2021-11-09 19:52:02 +01:00 committed by Jan De Dobbeleer
parent 2d25c59c00
commit 35e0a6d42c
1 changed files with 18 additions and 26 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

44
src/segment_az.go
View file

@ -8,7 +8,15 @@ import (
type az struct {
props *properties
env environmentInfo
AZ *AzureAccount
EnvironmentName string `json:"environmentName"`
HomeTenantID string `json:"homeTenantId"`
ID string `json:"id"`
IsDefault bool `json:"isDefault"`
Name string `json:"name"`
State string `json:"state"`
TenantID string `json:"tenantId"`
User *AzureUser `json:"user"`
}
const (
@ -22,25 +30,14 @@ type AzureUser struct {
Name string `json:"name"`
}
type AzureAccount struct {
EnvironmentName string `json:"environmentName"`
HomeTenantID string `json:"homeTenantId"`
ID string `json:"id"`
IsDefault bool `json:"isDefault"`
Name string `json:"name"`
State string `json:"state"`
TenantID string `json:"tenantId"`
User *AzureUser `json:"user"`
}
func (a *az) string() string {
if a.AZ != nil && a.AZ.Name == updateMessage {
if a != nil && a.Name == updateMessage {
return updateMessage
}
segmentTemplate := a.props.getString(SegmentTemplate, "{{.Name}}")
template := &textTemplate{
Template: segmentTemplate,
Context: a.AZ,
Context: a,
Env: a.env,
}
text, err := template.render()
@ -73,13 +70,11 @@ func (a *az) getFromEnvVars() bool {
return false
}
a.AZ = &AzureAccount{
EnvironmentName: environmentName,
Name: accountName,
ID: id,
User: &AzureUser{
Name: userName,
},
a.EnvironmentName = environmentName
a.Name = accountName
a.ID = id
a.User = &AzureUser{
Name: userName,
}
return true
@ -99,13 +94,10 @@ func (a *az) getFromAzCli() bool {
if strings.Contains(output, updateConsentNeeded) {
a.props.foreground = updateForeground
a.props.background = updateBackground
a.AZ = &AzureAccount{
Name: updateMessage,
}
a.Name = updateMessage
return true
}
a.AZ = &AzureAccount{}
err := json.Unmarshal([]byte(output), a.AZ)
err := json.Unmarshal([]byte(output), a)
return err == nil
}