From 768a1f96c9d7a8dc1bb6e68d94d117d3e7409a69 Mon Sep 17 00:00:00 2001 From: tv Date: Sun, 28 Dec 2014 22:52:47 +0100 Subject: add Scanner.hGetKey --- src/Scanner.hs | 59 +++++++++++++++++++++++++++++++++------------------------- 1 file changed, 34 insertions(+), 25 deletions(-) diff --git a/src/Scanner.hs b/src/Scanner.hs index 1f8eb5c..3c96a6e 100644 --- a/src/Scanner.hs +++ b/src/Scanner.hs @@ -4,6 +4,7 @@ module Scanner ( getKey + , hGetKey ) where import Prelude hiding ((/)) @@ -17,12 +18,17 @@ import System.IO -- high level interface + getKey :: IO String -getKey = do - _ <- hLookAhead stdin -- wait for input - ((_, raw_s), _) <- runScanner scan +getKey = hGetKey stdin + +hGetKey :: Handle -> IO String +hGetKey h = do + _ <- hLookAhead h -- wait for input + ((_, raw_s), _) <- runScanner $ scan h return $ map toChar raw_s + type P = C type I = C @@ -80,42 +86,43 @@ scan, scanESC, scanCS :: , MonadError ScanError m , MonadState ScanState m , MonadWriter ScanLog m - ) => m () - + ) + => Handle + -> m () -scan = do - c <- liftIO $ hGetC stdin +scan h = do + c <- liftIO $ hGetC h tell [c] case () of _ - | c == 01/11 -> scanESC + | c == 01/11 -> scanESC h | otherwise -> return () -scanESC = do - mb_c <- liftIO $ hWaitGetC timeout stdin +scanESC h = do + mb_c <- liftIO $ hWaitGetC timeout h whenJust mb_c $ \ c -> do tell [c] case () of _ | c == 05/11 -> -- CSI - scanCS + scanCS h | c == 01/11 -> -- XXX M-F1 and other crazy chords may cause -- \ESC\ESC... on wu, so we just recurse here... - scanESC + scanESC h | c == 04/15 -> -- XXX Non-CSI SS3 - one $ between (04/00) (07/14) + one h $ between (04/00) (07/14) | otherwise -> return () -scanCS = do - zeroOrMore $ between (03/00) (03/15) -- parameter bytes - zeroOrMore $ between (02/00) (02/15) -- intermediate bytes - one $ between (04/00) (07/14) -- final byte +scanCS h = do + zeroOrMore h $ between (03/00) (03/15) -- parameter bytes + zeroOrMore h $ between (02/00) (02/15) -- intermediate bytes + one h $ between (04/00) (07/14) -- final byte between :: C -> C -> (C -> Bool) @@ -128,21 +135,23 @@ zeroOrMore, one :: , MonadError ScanError m , MonadState ScanState m , MonadWriter ScanLog m - ) => (C -> Bool) -> m () - + ) + => Handle + -> (C -> Bool) + -> m () -zeroOrMore p = do - mb_c <- liftIO $ hWaitLookAheadC timeout stdin +zeroOrMore h p = do + mb_c <- liftIO $ hWaitLookAheadC timeout h whenJust mb_c $ \ c -> when (p c) $ do - _ <- liftIO $ hGetC stdin -- drop + _ <- liftIO $ hGetC h -- drop tell [c] modify $ \q -> q { buffer = buffer q ++ [c] } - zeroOrMore p + zeroOrMore h p -one p = do - mb_c <- liftIO $ hWaitLookAheadC timeout stdin +one h p = do + mb_c <- liftIO $ hWaitLookAheadC timeout h whenJust mb_c $ \ c -> do if p c then do -- cgit v1.2.3