refactor identifyTTY to match better how it will be used when integrated

1 files changed, 9 insertions, 9 deletions

diff --git a/whosocket.hs b/whosocket.hs
index 420b707a..97912eff 100644
--- a/whosocket.hs
+++ b/whosocket.hs
@@ -41,8 +41,15 @@ whosocket addr_str port_str = do
     r <- getLocalPeerCred' addr
     -- putStrLn $ "r{"++show addr++"} = " ++ show r
 
+    us <- UTmp.users
+    let filterTTYs (_,tty,pid) = 
+            if L.take 3 tty == "tty"
+              then Just (tty,pid)
+              else Nothing
+        tty_pids = mapMaybe filterTTYs us
+
     tty <- maybe (return Nothing)
-                 (uncurry identifyTTY)
+                 (uncurry $ identifyTTY tty_pids)
                  r
     putStrLn $ "uid = " ++ show (fmap fst r)
     L.putStrLn $ "tty = " <++?> tty
@@ -132,18 +139,11 @@ ttyToXorgs tty_pids = do
     let dts = catMaybes dts'
     return dts
 
-identifyTTY uid inode = do
+identifyTTY tty_pids uid inode = do
     pid <- scanProc (show uid) (L.unpack inode)
     -- putStrLn $ "scanProc --> "++show pid
     flip (maybe (return Nothing)) pid $ \(pid,ttydev) -> do
     tty <- ttyOrDisplay pid ttydev
-
-    us <- users
-    let filterTTYs (_,tty,pid) = 
-            if L.take 3 tty == "tty"
-              then Just (tty,pid)
-              else Nothing
-        tty_pids = mapMaybe filterTTYs us
     -- putStrLn $ "users = " ++ show tty_pids
     dts <- ttyToXorgs tty_pids