Add proper expiration and sort settings

1 files changed, 26 insertions, 1 deletions

diff --git a/cmd/corvid/main.go b/cmd/corvid/main.go
index 53c84aa..7fffbd1 100644
--- a/cmd/corvid/main.go
+++ b/cmd/corvid/main.go
@@ -39,10 +39,35 @@ func main() {
 }
 
 func server() {
-	srv.Start()
+	defaultExpiration := getEnvInt("CORVID_DEFAULT_EXPIRATION", 5000)
+	sortDirection := 1
+
+	switch os.Getenv("CORVID_SORT_DIRECTION") {
+	case "NEWEST_FIRST":
+		sortDirection = 1
+	case "OLDEST_FIRST":
+		sortDirection = -1
+	}
+
+	srv.Start(defaultExpiration, sortDirection)
 	select {}
 }
 
+func getEnvInt(key string, fallback int) int {
+	str := os.Getenv(key)
+
+	if len(str) == 0 {
+		return fallback
+	}
+
+	value, err := strconv.Atoi(str)
+	if err != nil {
+		return fallback
+	}
+
+	return value
+}
+
 func call(name string, args ...interface{}) error {
 	const CORVID_DBUS_OBJECT = "/sh/cxl/Corvid"
 	const CORVID_DBUS_NAME = "sh.cxl.Corvid"