Add hostname support

1 files changed, 18 insertions, 0 deletions

diff --git a/statusbar.go b/statusbar.go
index 29d6846..cc810bd 100644
--- a/statusbar.go
+++ b/statusbar.go
@@ -36,6 +36,7 @@ type Config struct {
 	Music MusicConfig
 	Clock ClockConfig
 	Battery BatteryConfig
+	Hostname HostnameConfig
 }
 
 type IPConfig struct {
@@ -68,6 +69,10 @@ type BatteryConfig struct {
 	SubLabel string `toml:"sublabel"`
 }
 
+type HostnameConfig struct {
+	Label string
+}
+
 func diskSizes(config MountConfig) string {
 	var b strings.Builder
 	var statfs syscall.Statfs_t
@@ -302,6 +307,17 @@ func buildBattery(config BatteryConfig) string {
 	return b.String()
 }
 
+func buildHostname(config HostnameConfig) string {
+	var b strings.Builder
+	b.WriteString(config.Label)
+	hostname, err := os.Hostname()
+	if err != nil {
+		panic(err)
+	}
+	b.WriteString(hostname)
+	return b.String()
+}
+
 func getDefaultConfigPath() string {
 	return "/etc/statusbar.conf"
 }
@@ -351,6 +367,8 @@ func main() {
 				b.WriteString(buildMusic(config.Music))
 			case "battery":
 				b.WriteString(buildBattery(config.Battery))
+			case "hostname":
+				b.WriteString(buildHostname(config.Hostname))
 			default:
 			}
 			if i < len(config.Sections) - 1 {