From 423ad8e75362fa2dc5a31b426b2163a97cef0823 Mon Sep 17 00:00:00 2001
From: Bradley Cicenas <bradley@vektor.nyc>
Date: Wed, 1 Mar 2017 11:03:57 +1100
Subject: [PATCH] add stateMap for consistent state sorting

---
 sort.go | 18 ++++++++----------
 1 file changed, 8 insertions(+), 10 deletions(-)

diff --git a/sort.go b/sort.go
index 7cc31d4..95dae87 100644
--- a/sort.go
+++ b/sort.go
@@ -9,6 +9,13 @@ import (
 
 type sortMethod func(c1, c2 *Container) bool
 
+var stateMap = map[string]int{
+	"running": 3,
+	"paused":  2,
+	"exited":  1,
+	"created": 0,
+}
+
 var idSorter = func(c1, c2 *Container) bool { return c1.id < c2.id }
 var nameSorter = func(c1, c2 *Container) bool { return c1.name < c2.name }
 
@@ -50,16 +57,7 @@ var Sorters = map[string]sortMethod{
 		if c1.state == c2.state {
 			return nameSorter(c1, c2)
 		}
-		if c1.state == "running" {
-			return true
-		}
-		if c2.state == "running" {
-			return false
-		}
-		if c2.state == "paused" {
-			return false
-		}
-		return true
+		return stateMap[c1.state] > stateMap[c2.state]
 	},
 }