From 92b24d9790265f1ef689d6d8e8798b84ae9352a6 Mon Sep 17 00:00:00 2001
From: Josh Curl <josh@curl.me>
Date: Wed, 29 Jun 2016 00:52:38 -0700
Subject: [PATCH] Console switch warning based on running console and not
 enabled console

---
 cmd/control/console.go | 20 ++++++++++++--------
 1 file changed, 12 insertions(+), 8 deletions(-)

diff --git a/cmd/control/console.go b/cmd/control/console.go
index 84eb4c03..b9ead261 100644
--- a/cmd/control/console.go
+++ b/cmd/control/console.go
@@ -52,7 +52,7 @@ func consoleSwitch(c *cli.Context) error {
 	newConsole := c.Args()[0]
 
 	cfg := config.LoadConfig()
-	if newConsole == cfg.Rancher.Console {
+	if newConsole == currentConsole() {
 		log.Warnf("Console is already set to %s", newConsole)
 	}
 
@@ -127,13 +127,7 @@ func consoleList(c *cli.Context) error {
 	consoles = append(consoles, "default")
 	sort.Strings(consoles)
 
-	var currentConsole string
-	currentConsoleBytes, err := ioutil.ReadFile("/run/console-done")
-	if err == nil {
-		currentConsole = strings.TrimSpace(string(currentConsoleBytes))
-	} else {
-		log.Warnf("Failed to detect current console: %v", err)
-	}
+	currentConsole := currentConsole()
 
 	for _, console := range consoles {
 		if console == currentConsole {
@@ -147,3 +141,13 @@ func consoleList(c *cli.Context) error {
 
 	return nil
 }
+
+func currentConsole() (console string) {
+	consoleBytes, err := ioutil.ReadFile("/run/console-done")
+	if err == nil {
+		console = strings.TrimSpace(string(consoleBytes))
+	} else {
+		log.Warnf("Failed to detect current console: %v", err)
+	}
+	return
+}