From f9b3470c35f9370b29f3db6cf26f8b3ec2f6d740 Mon Sep 17 00:00:00 2001
From: Daylin Morgan <me@dayl.in>
Date: Fri, 13 Sep 2024 14:46:00 -0500
Subject: [PATCH] feat: sort configured sessions

---
 src/config.nim  |  5 +++-
 src/project.nim | 69 ++++++++++++++++++++++++++++---------------------
 2 files changed, 44 insertions(+), 30 deletions(-)

diff --git a/src/config.nim b/src/config.nim
index 2d7dac8..5dabced 100644
--- a/src/config.nim
+++ b/src/config.nim
@@ -1,4 +1,4 @@
-import std/[os, streams, strformat, strutils]
+import std/[os, sequtils, streams, strformat, strutils, ]
 import yaml
 import term
 
@@ -10,6 +10,9 @@ type
   Session = object
     name*, dir*: string
 
+proc sessionNames*(tc: TsmConfig): seq[string] = 
+  tc.sessions.mapIt(it.name)
+
 proc loadConfigFile(): TsmConfig =
   let configPath = getEnv("TSM_CONFIG", getConfigDir() / "tsm" / "config.yml")
   try:
diff --git a/src/project.nim b/src/project.nim
index 3c498e6..6c08dfe 100644
--- a/src/project.nim
+++ b/src/project.nim
@@ -10,18 +10,16 @@ type
     open*: bool
     matched*: bool
 
-proc pathToName(path: string): string = splitPath(path)[1].replace(".", "_")
+proc pathToName(path: string): string =
+  splitPath(path)[1].replace(".", "_")
 
-proc newProject(
-  path: string,
-  open: bool,
-  name = "",
-  named: bool = false
-): Project =
+proc newProject(path: string, open: bool, name = "", named: bool = false): Project =
   result.location = path
   result.name =
-    if name != "": name
-   else: path.pathToName()
+    if name != "":
+      name
+    else:
+      path.pathToName()
   result.updated = getLastModificationTime(path)
   result.open = open
   result.named = named
@@ -37,7 +35,7 @@ proc getTsmDirs(): seq[string] =
   result = tsmDirs.split(":")
 
 proc findDuplicateProjects(
-  paths: seq[string], sessions: var HashSet[string]
+    paths: seq[string], sessions: var HashSet[string]
 ): seq[Project] =
   var candidates: Table[string, seq[string]]
   for p in paths:
@@ -45,30 +43,37 @@ proc findDuplicateProjects(
 
   let maxExtra = min(candidates.values.toSeq.mapIt(it.len))
   for i in 2..maxExtra:
-    let deduplicated = collect:
-      for path, pathSplit in candidates.pairs:
-        (name: pathSplit[^i..^1].joinPath, path: path)
+    let
+      deduplicated =
+        collect:
+          for path, pathSplit in candidates.pairs:
+            (name: pathSplit[^i..^1].joinPath, path: path)
     if deduplicated.mapIt(it[0]).toHashSet.len == candidates.len:
       for (name, path) in deduplicated:
         let open = name in sessions
         result.add newProject(path, open, name)
-        if open: sessions.excl name
+        if open:
+          sessions.excl name
       break
 
   if result.len == 0:
     termQuit "failed to deduplicate these paths:" & paths.join(", ")
 
+proc `<-`(candidates: var Table[string, seq[string]], path: string) =
+  let name = path.splitPath.tail
+  if candidates.hasKeyOrPut(name, @[path]):
+    candidates[name].add path
+
 proc findProjects*(open: bool = false): seq[Project] =
   let tsmConfig = loadTsmConfig()
   var candidates: Table[string, seq[string]]
   var sessions = tmux.sessions.toHashSet()
 
   for devDir in tsmConfig.dirs:
-    for path in walkDir(devDir):
-      if ({path.kind} * {pcFile, pcLinkToFile}).len > 0: continue
-      let name = path.path.splitPath.tail
-      if candidates.hasKeyOrPut(name, @[path.path]):
-        candidates[name].add path.path
+    for (kind, path) in walkDir(devDir):
+      if ({kind} * {pcFile, pcLinkToFile}).len > 0:
+        continue
+      candidates <- path
 
   for name, paths in candidates.pairs:
     if len(paths) == 1:
@@ -76,28 +81,35 @@ proc findProjects*(open: bool = false): seq[Project] =
         path = paths[0]
         open = path.pathToName in sessions
       result.add newProject(path, open)
-      if open: sessions.excl path.pathToName
+      if open:
+        sessions.excl path.pathToName
     else:
       result &= findDuplicateProjects(paths, sessions)
 
   for session in tsmConfig.sessions:
     if session.name notin sessions:
       result.add newProject(
-        path = session.dir,
-        open = false,
-        name = session.name,
-        named = true
+        path = session.dir, open = false, name = session.name, named = true
       )
 
   if open:
     result = result.filterIt(it.open)
+  let sessionNames = tsmConfig.sessionNames
 
-  # default order
-  # open -> configured -> mod time
-  result.sort do (x, y: Project) -> int:
+  # order open -> configured -> mod time
+  result.sort do(x, y: Project) -> int:
     result = cmp(y.open, x.open)
     if result == 0:
-      result = cmp(y.updated, x.updated)
+      result =
+        if y.name in sessionNames:
+          if x.name in sessionNames:
+            cmp(y.name, x.name)
+          else:
+            1
+        elif x.name in sessionNames:
+          -1
+        else:
+          cmp(y.updated, x.updated)
 
   if sessions.len > 0:
     result = sessions.toSeq().mapIt(newUnknownProject(it)) & result
@@ -107,4 +119,3 @@ proc findProjects*(open: bool = false): seq[Project] =
     termEcho "searched these directories: "
     echo getTsmDirs().mapIt("  " & it).join("\n")
     quit QuitFailure
-