1 files changed, 9 insertions, 9 deletions

diff --git a/whosocket.hs b/whosocket.hs
index 14d2a708..a8dcdff9 100644
--- a/whosocket.hs
+++ b/whosocket.hs
@@ -39,10 +39,8 @@ whosocket addr_str port_str = do
                         (Just addr_str) 
                         (Just port_str)
     let addr = head $ map addrAddress info
-    putStrLn $ "addr = "++show addr
-    putStrLn $ "unmap6mapped4 addr = "++show (unmap6mapped4 addr)
     r <- getLocalPeerCred' addr
-    -- putStrLn $ "r{"++show addr++"} = " ++ show r
+    putStrLn $ "r{"++show addr++"} = " ++ show r
 
     us <- UTmp.users
     let filterTTYs (_,tty,pid) = 
@@ -144,16 +142,18 @@ ttyToXorgs tty_pids = do
 
 identifyTTY tty_pids uid inode = do
     pid <- scanProc (show uid) (L.unpack inode)
-    -- putStrLn $ "scanProc --> "++show pid
+    putStrLn $ "scanProc --> "++show pid
     flip (maybe (return Nothing)) pid $ \(pid,ttydev) -> do
     tty <- ttyOrDisplay pid ttydev
-    -- putStrLn $ "users = " ++ show tty_pids
+    putStrLn $ "users = " ++ show tty_pids
     dts <- ttyToXorgs tty_pids
 
-    -- putStrLn $ "displays = " ++ show dts 
+    putStrLn $ "displays = " ++ show dts 
+    putStrLn $ "tty = " ++ show tty
     -- -- displays = [(":5",Chunk "tty7" Empty)]
 
-    let tty' = if take 3 tty=="tty" then Just (L.pack tty) else lookup tty dts
+    let tty' = if take 3 tty=="tty" then Just (L.pack tty) else lookup (parseTty tty) (map (first parseTty) dts)
     return tty'
-
-
+    where
+        parseTty :: String -> Float
+        parseTty = read .  tail . dropWhile (/=':')