Merge pull request #415 from yurhasko/main

feat: segment-azure for displaying current active Azure subscription and resource group

Author: justjanne

README.md

@@ -250,12 +250,12 @@ Usage of powerline-go:
          (default "patched")
   -modules string
          The list of modules to load, separated by ','
-         (valid choices: aws, bzr, cwd, direnv, docker, docker-context, dotenv, duration, exit, fossil, gcp, git, gitlite, goenv, hg, host, jobs, kube, load, newline, nix-shell, node, perlbrew, perms, plenv, rbenv, root, rvm, shell-var, shenv, ssh, svn, termtitle, terraform-workspace, time, user, venv, vgo, vi-mode, wsl)
+         (valid choices: aws, bzr, cwd, direnv, docker, docker-context, dotenv, duration, exit, fossil, gcp, git, gitlite, goenv, hg, host, jobs, kube, load, newline, nix-shell, node, perlbrew, perms, plenv, rbenv, root, rvm, shell-var, shenv, ssh, svn, termtitle, terraform-workspace, time, user, venv, vgo, vi-mode, wsl, azure)
          Unrecognized modules will be invoked as 'powerline-go-MODULE' executable plugins and should output a (possibly empty) list of JSON objects that unmarshal to powerline-go's Segment structs.
          (default "venv,user,host,ssh,cwd,perms,git,hg,jobs,exit,root")
   -modules-right string
          The list of modules to load anchored to the right, for shells that support it, separated by ','
-         (valid choices: aws, bzr, cwd, direnv, docker, docker-context, dotenv, duration, exit, fossil, gcp, git, gitlite, goenv, hg, host, jobs, kube, load, newline, nix-shell, node, perlbrew, perms, plenv, rbenv, root, rvm, shell-var, shenv, ssh, svn, termtitle, terraform-workspace, time, user, venv, vgo, wsl)
+         (valid choices: aws, bzr, cwd, direnv, docker, docker-context, dotenv, duration, exit, fossil, gcp, git, gitlite, goenv, hg, host, jobs, kube, load, newline, nix-shell, node, perlbrew, perms, plenv, rbenv, root, rvm, shell-var, shenv, ssh, svn, termtitle, terraform-workspace, time, user, venv, vgo, wsl, azure)
          Unrecognized modules will be invoked as 'powerline-go-MODULE' executable plugins and should output a (possibly empty) list of JSON objects that unmarshal to powerline-go's Segment structs.
   -newline
          Show the prompt on a new line
@@ -268,7 +268,7 @@ Usage of powerline-go:
          Use '~' for your home dir. You may need to escape this character to avoid shell substitution.
   -priority string
          Segments sorted by priority, if not enough space exists, the least priorized segments are removed first. Separate with ','
-         (valid choices: aws, bzr, cwd, direnv, docker, docker-context, dotenv, duration, exit, fossil, gcp, git, gitlite, goenv, hg, host, jobs, kube, load, newline, nix-shell, node, perlbrew, perms, plenv, rbenv, root, rvm, shell-var, shenv, ssh, svn, termtitle, terraform-workspace, time, user, venv, vgo, vi-mode, wsl)
+         (valid choices: aws, bzr, cwd, direnv, docker, docker-context, dotenv, duration, exit, fossil, gcp, git, gitlite, goenv, hg, host, jobs, kube, load, newline, nix-shell, node, perlbrew, perms, plenv, rbenv, root, rvm, shell-var, shenv, ssh, svn, termtitle, terraform-workspace, time, user, venv, vgo, vi-mode, wsl, azure)
          (default "root,cwd,user,host,ssh,perms,git-branch,git-status,hg,jobs,exit,cwd-path")
   -shell string
          Set this to your shell type

args.go

@@ -132,19 +132,19 @@ var args = arguments{
 		"modules",
 		strings.Join(defaults.Modules, ","),
 		commentsWithDefaults("The list of modules to load, separated by ','",
-			"(valid choices: aws, bzr, cwd, direnv, docker, docker-context, dotenv, duration, exit, fossil, gcp, git, gitlite, goenv, hg, host, jobs, kube, load, newline, nix-shell, node, perlbrew, perms, plenv, rbenv, root, rvm, shell-var, shenv, ssh, svn, termtitle, terraform-workspace, time, user, venv, vgo, vi-mode, wsl)",
+			"(valid choices: aws, bzr, cwd, direnv, docker, docker-context, dotenv, duration, exit, fossil, gcp, git, gitlite, goenv, hg, host, jobs, kube, load, newline, nix-shell, node, perlbrew, perms, plenv, rbenv, root, rvm, shell-var, shenv, ssh, svn, termtitle, terraform-workspace, time, user, venv, vgo, vi-mode, wsl, azure)",
 			"Unrecognized modules will be invoked as 'powerline-go-MODULE' executable plugins and should output a (possibly empty) list of JSON objects that unmarshal to powerline-go's Segment structs.")),
 	ModulesRight: flag.String(
 		"modules-right",
 		strings.Join(defaults.ModulesRight, ","),
 		comments("The list of modules to load anchored to the right, for shells that support it, separated by ','",
-			"(valid choices: aws, bzr, cwd, direnv, docker, docker-context, dotenv, duration, exit, fossil, gcp, git, gitlite, goenv, hg, host, jobs, kube, load, newline, nix-shell, node, perlbrew, perms, plenv, rbenv, root, rvm, shell-var, shenv, ssh, svn, termtitle, terraform-workspace, time, user, venv, vgo, wsl)",
+			"(valid choices: aws, bzr, cwd, direnv, docker, docker-context, dotenv, duration, exit, fossil, gcp, git, gitlite, goenv, hg, host, jobs, kube, load, newline, nix-shell, node, perlbrew, perms, plenv, rbenv, root, rvm, shell-var, shenv, ssh, svn, termtitle, terraform-workspace, time, user, venv, vgo, wsl, azure)",
 			"Unrecognized modules will be invoked as 'powerline-go-MODULE' executable plugins and should output a (possibly empty) list of JSON objects that unmarshal to powerline-go's Segment structs.")),
 	Priority: flag.String(
 		"priority",
 		strings.Join(defaults.Priority, ","),
 		commentsWithDefaults("Segments sorted by priority, if not enough space exists, the least priorized segments are removed first. Separate with ','",
-			"(valid choices: aws, bzr, cwd, direnv, docker, docker-context, dotenv, duration, exit, fossil, gcp, git, gitlite, goenv, hg, host, jobs, kube, load, newline, nix-shell, node, perlbrew, perms, plenv, rbenv, root, rvm, shell-var, shenv, ssh, svn, termtitle, terraform-workspace, time, user, venv, vgo, vi-mode, wsl)")),
+			"(valid choices: aws, bzr, cwd, direnv, docker, docker-context, dotenv, duration, exit, fossil, gcp, git, gitlite, goenv, hg, host, jobs, kube, load, newline, nix-shell, node, perlbrew, perms, plenv, rbenv, root, rvm, shell-var, shenv, ssh, svn, termtitle, terraform-workspace, time, user, venv, vgo, vi-mode, wsl, azure)")),
 	MaxWidthPercentage: flag.Int(
 		"max-width",
 		defaults.MaxWidthPercentage,

defaults.go

@@ -208,6 +208,9 @@ var defaults = Config{
 			DotEnvFg: 15, // white
 			DotEnvBg: 55, // purple
 
+			AzureFg: 15, // white
+			AzureBg: 26, // Azure blue
+
 			AWSFg: 15,  // white
 			AWSBg: 172, // AWS orange
 

main.go

@@ -78,6 +78,7 @@ var modules = map[string]func(*powerline) []pwl.Segment{
 	"docker":              segmentDocker,
 	"docker-context":      segmentDockerContext,
 	"dotenv":              segmentDotEnv,
+	"azure":               segmentAzure,
 	"duration":            segmentDuration,
 	"exit":                segmentExitCode,
 	"fossil":              segmentFossil,

segment-azure.go

@@ -0,0 +1,128 @@
+package main
+
+import (
+	"bytes"
+	"encoding/json"
+	"os"
+	"path/filepath"
+	"strings"
+
+	pwl "github.com/justjanne/powerline-go/powerline"
+)
+
+// utf8BOM is the UTF-8 byte order mark that Azure CLI may add to JSON files
+var utf8BOM = []byte{0xEF, 0xBB, 0xBF}
+
+type azureSubscription struct {
+	ID        string `json:"id"`
+	Name      string `json:"name"`
+	State     string `json:"state"`
+	IsDefault bool   `json:"isDefault"`
+}
+
+type azureProfile struct {
+	Subscriptions []azureSubscription `json:"subscriptions"`
+}
+
+func getAzureSubscription() string {
+	envSubID := os.Getenv("AZURE_SUBSCRIPTION_ID")
+
+	home, err := os.UserHomeDir()
+	if err != nil {
+		return envSubID
+	}
+
+	data, err := os.ReadFile(filepath.Join(home, ".azure", "azureProfile.json"))
+	if err != nil {
+		return envSubID
+	}
+
+	data = bytes.TrimPrefix(data, utf8BOM)
+
+	var profile azureProfile
+	if err := json.Unmarshal(data, &profile); err != nil {
+		return envSubID
+	}
+
+	if envSubID != "" {
+		for _, sub := range profile.Subscriptions {
+			if sub.ID == envSubID {
+				return sub.Name
+			}
+		}
+		return envSubID
+	}
+
+	var firstEnabled string
+	for _, sub := range profile.Subscriptions {
+		if sub.State != "Enabled" {
+			continue
+		}
+		if sub.IsDefault {
+			return sub.Name
+		}
+		if firstEnabled == "" {
+			firstEnabled = sub.Name
+		}
+	}
+
+	return firstEnabled
+}
+
+func getAzureResourceGroup() string {
+	if rg := os.Getenv("AZURE_DEFAULTS_GROUP"); rg != "" {
+		return rg
+	}
+
+	home, err := os.UserHomeDir()
+	if err != nil {
+		return ""
+	}
+
+	data, err := os.ReadFile(filepath.Join(home, ".azure", "config"))
+	if err != nil {
+		return ""
+	}
+
+	inDefaults := false
+	for _, line := range strings.Split(string(data), "\n") {
+		line = strings.TrimSpace(line)
+
+		if line == "[defaults]" {
+			inDefaults = true
+			continue
+		}
+
+		if strings.HasPrefix(line, "[") {
+			inDefaults = false
+			continue
+		}
+
+		if inDefaults && (strings.HasPrefix(line, "group ") || strings.HasPrefix(line, "group=")) {
+			if parts := strings.SplitN(line, "=", 2); len(parts) == 2 {
+				return strings.TrimSpace(parts[1])
+			}
+		}
+	}
+
+	return ""
+}
+
+func segmentAzure(p *powerline) []pwl.Segment {
+	subscription := getAzureSubscription()
+	if subscription == "" {
+		return []pwl.Segment{}
+	}
+
+	content := subscription
+	if rg := getAzureResourceGroup(); rg != "" {
+		content += " (" + rg + ")"
+	}
+
+	return []pwl.Segment{{
+		Name:       "azure",
+		Content:    content,
+		Foreground: p.theme.AzureFg,
+		Background: p.theme.AzureBg,
+	}}
+}

themes.go

@@ -78,6 +78,9 @@ type Theme struct {
 	DotEnvFg uint8
 	DotEnvBg uint8
 
+	AzureFg uint8
+	AzureBg uint8
+
 	AWSFg uint8
 	AWSBg uint8