mirror of
https://github.com/JanDeDobbeleer/oh-my-posh.git
synced 2024-11-10 04:54:03 -08:00
2c95d33c84
New segment `az` allows display of current Azure subscription name and ID with a configurable separator. Defaults to only display the name.
81 lines
1.6 KiB
Go
81 lines
1.6 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
)
|
|
|
|
type az struct {
|
|
props *properties
|
|
env environmentInfo
|
|
name string
|
|
id string
|
|
}
|
|
|
|
const (
|
|
//SubscriptionInfoSeparator is put between the name and ID
|
|
SubscriptionInfoSeparator Property = "info_separator"
|
|
//DisplaySubscriptionID hides or show the subscription GUID
|
|
DisplaySubscriptionID Property = "display_id"
|
|
//DisplaySubscriptionName hides or shows the subscription display name
|
|
DisplaySubscriptionName Property = "display_name"
|
|
)
|
|
|
|
func (a *az) string() string {
|
|
separator := ""
|
|
if a.idEnabled() && a.nameEnabled() {
|
|
separator = a.props.getString(SubscriptionInfoSeparator, " | ")
|
|
}
|
|
|
|
return fmt.Sprintf("%s%s%s", a.getName(), separator, a.getID())
|
|
}
|
|
|
|
func (a *az) init(props *properties, env environmentInfo) {
|
|
a.props = props
|
|
a.env = env
|
|
}
|
|
|
|
func (a *az) enabled() bool {
|
|
if (!a.idEnabled() && !a.nameEnabled()) || !a.env.hasCommand("az") {
|
|
return false
|
|
}
|
|
|
|
output := a.env.runCommand("az", "account", "show", "--query=[name,id]", "-o=tsv")
|
|
if output == "" {
|
|
return false
|
|
}
|
|
|
|
splittedOutput := strings.Split(output, "\n")
|
|
if len(splittedOutput) < 2 {
|
|
return false
|
|
}
|
|
|
|
a.name = strings.TrimSpace(splittedOutput[0])
|
|
a.id = strings.TrimSpace(splittedOutput[1])
|
|
return true
|
|
}
|
|
|
|
func (a *az) getID() string {
|
|
if !a.idEnabled() {
|
|
return ""
|
|
}
|
|
|
|
return a.id
|
|
}
|
|
|
|
func (a *az) getName() string {
|
|
if !a.nameEnabled() {
|
|
return ""
|
|
}
|
|
|
|
return a.name
|
|
}
|
|
|
|
func (a *az) idEnabled() bool {
|
|
return a.props.getBool(DisplaySubscriptionID, false)
|
|
}
|
|
|
|
func (a *az) nameEnabled() bool {
|
|
return a.props.getBool(DisplaySubscriptionName, true)
|
|
}
|