module System.Console.Haskeline.Backend.Win32(
win32Term,
win32TermStdin,
fileRunTerm
)where
import System.IO
import Foreign
import Foreign.C
import System.Win32 hiding (multiByteToWideChar)
import Graphics.Win32.Misc(getStdHandle, sTD_OUTPUT_HANDLE)
import Data.List(intercalate)
import Control.Concurrent hiding (throwTo)
import Data.Char(isPrint)
import Data.Maybe(mapMaybe)
import Control.Applicative
import Control.Monad
import System.Console.Haskeline.Key
import System.Console.Haskeline.Monads hiding (Handler)
import System.Console.Haskeline.LineState
import System.Console.Haskeline.Term
import System.Console.Haskeline.Backend.WCWidth
import Data.ByteString.Internal (createAndTrim)
import qualified Data.ByteString as B
#if defined(i386_HOST_ARCH)
# define WINDOWS_CCONV stdcall
#elif defined(x86_64_HOST_ARCH)
# define WINDOWS_CCONV ccall
#else
# error Unknown mingw32 arch
#endif
foreign import WINDOWS_CCONV "windows.h ReadConsoleInputW" c_ReadConsoleInput
:: HANDLE -> Ptr () -> DWORD -> Ptr DWORD -> IO Bool
foreign import WINDOWS_CCONV "windows.h WaitForSingleObject" c_WaitForSingleObject
:: HANDLE -> DWORD -> IO DWORD
foreign import WINDOWS_CCONV "windows.h GetNumberOfConsoleInputEvents"
c_GetNumberOfConsoleInputEvents :: HANDLE -> Ptr DWORD -> IO Bool
getNumberOfEvents :: HANDLE -> IO Int
getNumberOfEvents h = alloca $ \numEventsPtr -> do
failIfFalse_ "GetNumberOfConsoleInputEvents"
$ c_GetNumberOfConsoleInputEvents h numEventsPtr
fmap fromEnum $ peek numEventsPtr
getEvent :: HANDLE -> Chan Event -> IO Event
getEvent h = keyEventLoop (eventReader h)
eventReader :: HANDLE -> IO [Event]
eventReader h = do
let waitTime = 500
ret <- c_WaitForSingleObject h waitTime
yield
if ret /= (0)
then eventReader h
else do
es <- readEvents h
return $ mapMaybe processEvent es
consoleHandles :: MaybeT IO Handles
consoleHandles = do
h_in <- open "CONIN$"
h_out <- open "CONOUT$"
return Handles { hIn = h_in, hOut = h_out }
where
open file = handle (\(_::IOException) -> mzero) $ liftIO
$ createFile file (gENERIC_READ .|. gENERIC_WRITE)
(fILE_SHARE_READ .|. fILE_SHARE_WRITE) Nothing
oPEN_EXISTING 0 Nothing
processEvent :: InputEvent -> Maybe Event
processEvent KeyEvent {keyDown = True, unicodeChar = c, virtualKeyCode = vc,
controlKeyState = cstate}
= fmap (\e -> KeyInput [Key modifier' e]) $ keyFromCode vc `mplus` simpleKeyChar
where
simpleKeyChar = guard (c /= '\NUL') >> return (KeyChar c)
testMod ck = (cstate .&. ck) /= 0
modifier' = if hasMeta modifier && hasControl modifier
then noModifier {hasShift = hasShift modifier}
else modifier
modifier = Modifier {hasMeta = testMod ((1)
.|. (2))
,hasControl = testMod ((4)
.|. (8))
&& not (c > '\NUL' && c <= '\031')
,hasShift = testMod (16)
&& not (isPrint c)
}
processEvent WindowEvent = Just WindowResize
processEvent _ = Nothing
keyFromCode :: WORD -> Maybe BaseKey
keyFromCode (8) = Just Backspace
keyFromCode (37) = Just LeftKey
keyFromCode (39) = Just RightKey
keyFromCode (38) = Just UpKey
keyFromCode (40) = Just DownKey
keyFromCode (46) = Just Delete
keyFromCode (36) = Just Home
keyFromCode (35) = Just End
keyFromCode (33) = Just PageUp
keyFromCode (34) = Just PageDown
keyFromCode (13) = Just (KeyChar '\n')
keyFromCode _ = Nothing
data InputEvent = KeyEvent {keyDown :: BOOL,
repeatCount :: WORD,
virtualKeyCode :: WORD,
virtualScanCode :: WORD,
unicodeChar :: Char,
controlKeyState :: DWORD}
| WindowEvent
| OtherEvent
deriving Show
peekEvent :: Ptr () -> IO InputEvent
peekEvent pRecord = do
eventType :: WORD <- ((\hsc_ptr -> peekByteOff hsc_ptr 0)) pRecord
let eventPtr = ((\hsc_ptr -> hsc_ptr `plusPtr` 4)) pRecord
case eventType of
(1) -> getKeyEvent eventPtr
(4) -> return WindowEvent
_ -> return OtherEvent
readEvents :: HANDLE -> IO [InputEvent]
readEvents h = do
n <- getNumberOfEvents h
alloca $ \numEventsPtr ->
allocaBytes (n * (20)) $ \pRecord -> do
failIfFalse_ "ReadConsoleInput"
$ c_ReadConsoleInput h pRecord (toEnum n) numEventsPtr
numRead <- fmap fromEnum $ peek numEventsPtr
forM [0..toEnum numRead1] $ \i -> peekEvent
$ pRecord `plusPtr` (i * (20))
getKeyEvent :: Ptr () -> IO InputEvent
getKeyEvent p = do
kDown' <- ((\hsc_ptr -> peekByteOff hsc_ptr 0)) p
repeat' <- ((\hsc_ptr -> peekByteOff hsc_ptr 4)) p
keyCode <- ((\hsc_ptr -> peekByteOff hsc_ptr 6)) p
scanCode <- ((\hsc_ptr -> peekByteOff hsc_ptr 8)) p
char :: CWchar <- ((\hsc_ptr -> peekByteOff hsc_ptr 10)) p
state <- ((\hsc_ptr -> peekByteOff hsc_ptr 12)) p
return KeyEvent {keyDown = kDown',
repeatCount = repeat',
virtualKeyCode = keyCode,
virtualScanCode = scanCode,
unicodeChar = toEnum (fromEnum char),
controlKeyState = state}
data Coord = Coord {coordX, coordY :: Int}
deriving Show
instance Storable Coord where
sizeOf _ = ((4))
alignment _ = (2)
peek p = do
x :: CShort <- ((\hsc_ptr -> peekByteOff hsc_ptr 0)) p
y :: CShort <- ((\hsc_ptr -> peekByteOff hsc_ptr 2)) p
return Coord {coordX = fromEnum x, coordY = fromEnum y}
poke p c = do
((\hsc_ptr -> pokeByteOff hsc_ptr 0)) p (toEnum (coordX c) :: CShort)
((\hsc_ptr -> pokeByteOff hsc_ptr 2)) p (toEnum (coordY c) :: CShort)
foreign import ccall "haskeline_SetPosition"
c_SetPosition :: HANDLE -> Ptr Coord -> IO Bool
setPosition :: HANDLE -> Coord -> IO ()
setPosition h c = with c $ failIfFalse_ "SetConsoleCursorPosition"
. c_SetPosition h
foreign import WINDOWS_CCONV "windows.h GetConsoleScreenBufferInfo"
c_GetScreenBufferInfo :: HANDLE -> Ptr () -> IO Bool
getPosition :: HANDLE -> IO Coord
getPosition = withScreenBufferInfo $
((\hsc_ptr -> peekByteOff hsc_ptr 4))
withScreenBufferInfo :: (Ptr () -> IO a) -> HANDLE -> IO a
withScreenBufferInfo f h = allocaBytes ((22))
$ \infoPtr -> do
failIfFalse_ "GetConsoleScreenBufferInfo"
$ c_GetScreenBufferInfo h infoPtr
f infoPtr
getBufferSize :: HANDLE -> IO Layout
getBufferSize = withScreenBufferInfo $ \p -> do
c <- ((\hsc_ptr -> peekByteOff hsc_ptr 0)) p
return Layout {width = coordX c, height = coordY c}
foreign import WINDOWS_CCONV "windows.h WriteConsoleW" c_WriteConsoleW
:: HANDLE -> Ptr TCHAR -> DWORD -> Ptr DWORD -> Ptr () -> IO Bool
writeConsole :: HANDLE -> String -> IO ()
writeConsole _ "" = return ()
writeConsole h str = writeConsole' >> writeConsole h ys
where
(xs,ys) = splitAt limit str
limit = 20000
writeConsole'
= withArray (map (toEnum . fromEnum) xs)
$ \t_arr -> alloca $ \numWritten -> do
failIfFalse_ "WriteConsoleW"
$ c_WriteConsoleW h t_arr (toEnum $ length xs)
numWritten nullPtr
foreign import WINDOWS_CCONV "windows.h MessageBeep" c_messageBeep :: UINT -> IO Bool
messageBeep :: IO ()
messageBeep = c_messageBeep (1) >> return ()
foreign import WINDOWS_CCONV "windows.h GetConsoleMode" c_GetConsoleMode
:: HANDLE -> Ptr DWORD -> IO Bool
foreign import WINDOWS_CCONV "windows.h SetConsoleMode" c_SetConsoleMode
:: HANDLE -> DWORD -> IO Bool
withWindowMode :: MonadException m => Handles -> m a -> m a
withWindowMode hs f = do
let h = hIn hs
bracket (getConsoleMode h) (setConsoleMode h)
$ \m -> setConsoleMode h (m .|. (8)) >> f
where
getConsoleMode h = liftIO $ alloca $ \p -> do
failIfFalse_ "GetConsoleMode" $ c_GetConsoleMode h p
peek p
setConsoleMode h m = liftIO $ failIfFalse_ "SetConsoleMode" $ c_SetConsoleMode h m
data Handles = Handles { hIn, hOut :: HANDLE }
closeHandles :: Handles -> IO ()
closeHandles hs = closeHandle (hIn hs) >> closeHandle (hOut hs)
newtype Draw m a = Draw {runDraw :: ReaderT Handles m a}
deriving (Functor, Applicative, Monad, MonadIO, MonadException, MonadReader Handles)
type DrawM a = forall m . (MonadIO m, MonadReader Layout m) => Draw m a
instance MonadTrans Draw where
lift = Draw . lift
getPos :: MonadIO m => Draw m Coord
getPos = asks hOut >>= liftIO . getPosition
setPos :: Coord -> DrawM ()
setPos c = do
h <- asks hOut
maxY <- liftM (subtract 1) $ asks height
liftIO $ setPosition h c { coordY = max 0 $ min maxY $ coordY c }
printText :: MonadIO m => String -> Draw m ()
printText txt = do
h <- asks hOut
liftIO (writeConsole h txt)
printAfter :: [Grapheme] -> DrawM ()
printAfter gs = do
printText (graphemesToString gs)
movePosLeft gs
drawLineDiffWin :: LineChars -> LineChars -> DrawM ()
drawLineDiffWin (xs1,ys1) (xs2,ys2) = case matchInit xs1 xs2 of
([],[]) | ys1 == ys2 -> return ()
(xs1',[]) | xs1' ++ ys1 == ys2 -> movePosLeft xs1'
([],xs2') | ys1 == xs2' ++ ys2 -> movePosRight xs2'
(xs1',xs2') -> do
movePosLeft xs1'
let m = gsWidth xs1' + gsWidth ys1 (gsWidth xs2' + gsWidth ys2)
let deadText = stringToGraphemes $ replicate m ' '
printText (graphemesToString xs2')
printAfter (ys2 ++ deadText)
movePosRight, movePosLeft :: [Grapheme] -> DrawM ()
movePosRight str = do
p <- getPos
w <- asks width
setPos $ moveCoord w p str
where
moveCoord _ p [] = p
moveCoord w p cs = case splitAtWidth (w coordX p) cs of
(_,[],len) | len < w coordX p
-> Coord { coordY = coordY p,
coordX = coordX p + len
}
(_,cs',_)
-> moveCoord w Coord {
coordY = coordY p + 1,
coordX = 0
} cs'
movePosLeft str = do
p <- getPos
w <- asks width
setPos $ moveCoord w p str
where
moveCoord _ p [] = p
moveCoord w p cs = case splitAtWidth (coordX p) cs of
(_,[],len)
-> Coord { coordY = coordY p,
coordX = coordX p len
}
(_,_:cs',_)
-> moveCoord w Coord {
coordY = coordY p 1,
coordX = w1
} cs'
crlf :: String
crlf = "\r\n"
instance (MonadException m, MonadReader Layout m) => Term (Draw m) where
drawLineDiff (xs1,ys1) (xs2,ys2) = let
fixEsc = filter ((/= '\ESC') . baseChar)
in drawLineDiffWin (fixEsc xs1, fixEsc ys1) (fixEsc xs2, fixEsc ys2)
reposition _ _ = return ()
printLines [] = return ()
printLines ls = printText $ intercalate crlf ls ++ crlf
clearLayout = clearScreen
moveToNextLine s = do
movePosRight (snd s)
printText "\r\n"
ringBell True = liftIO messageBeep
ringBell False = return ()
win32TermStdin :: MaybeT IO RunTerm
win32TermStdin = do
liftIO (hIsTerminalDevice stdin) >>= guard
win32Term
win32Term :: MaybeT IO RunTerm
win32Term = do
hs <- consoleHandles
ch <- liftIO newChan
fileRT <- liftIO $ fileRunTerm stdin
return fileRT {
termOps = Left TermOps {
getLayout = getBufferSize (hOut hs)
, withGetEvent = withWindowMode hs
. win32WithEvent hs ch
, saveUnusedKeys = saveKeys ch
, evalTerm = EvalTerm (runReaderT' hs . runDraw)
(Draw . lift)
},
closeTerm = closeHandles hs
}
win32WithEvent :: MonadException m => Handles -> Chan Event
-> (m Event -> m a) -> m a
win32WithEvent h eventChan f = f $ liftIO $ getEvent (hIn h) eventChan
fileRunTerm :: Handle -> IO RunTerm
fileRunTerm h_in = do
putter <- putOut
cp <- getCodePage
return RunTerm {
closeTerm = return (),
putStrOut = putter,
wrapInterrupt = withCtrlCHandler,
termOps = Right FileOps
{ inputHandle = h_in
, wrapFileInput = hWithBinaryMode h_in
, getLocaleChar = getMultiByteChar cp h_in
, maybeReadNewline = hMaybeReadNewline h_in
, getLocaleLine = hGetLocaleLine h_in
>>= liftIO . codePageToUnicode cp
}
}
putOut :: IO (String -> IO ())
putOut = do
outIsTerm <- hIsTerminalDevice stdout
if outIsTerm
then do
h <- getStdHandle sTD_OUTPUT_HANDLE
return (writeConsole h)
else do
cp <- getCodePage
return $ \str -> unicodeToCodePage cp str >>= B.putStr >> hFlush stdout
type Handler = DWORD -> IO BOOL
foreign import ccall "wrapper" wrapHandler :: Handler -> IO (FunPtr Handler)
foreign import WINDOWS_CCONV "windows.h SetConsoleCtrlHandler" c_SetConsoleCtrlHandler
:: FunPtr Handler -> BOOL -> IO BOOL
withCtrlCHandler :: MonadException m => m a -> m a
withCtrlCHandler f = bracket (liftIO $ do
tid <- myThreadId
fp <- wrapHandler (handler tid)
_ <- c_SetConsoleCtrlHandler fp True
return fp)
(\fp -> liftIO $ c_SetConsoleCtrlHandler fp False)
(const f)
where
handler tid (0) = do
throwTo tid Interrupt
return True
handler _ _ = return False
foreign import WINDOWS_CCONV "WideCharToMultiByte" wideCharToMultiByte
:: CodePage -> DWORD -> LPCWSTR -> CInt -> LPCSTR -> CInt
-> LPCSTR -> LPBOOL -> IO CInt
unicodeToCodePage :: CodePage -> String -> IO B.ByteString
unicodeToCodePage cp wideStr = withCWStringLen wideStr $ \(wideBuff, wideLen) -> do
outSize <- wideCharToMultiByte cp 0 wideBuff (toEnum wideLen)
nullPtr 0 nullPtr nullPtr
createAndTrim (fromEnum outSize) $ \outBuff ->
fmap fromEnum $ wideCharToMultiByte cp 0 wideBuff (toEnum wideLen)
(castPtr outBuff) outSize nullPtr nullPtr
foreign import WINDOWS_CCONV "MultiByteToWideChar" multiByteToWideChar
:: CodePage -> DWORD -> LPCSTR -> CInt -> LPWSTR -> CInt -> IO CInt
codePageToUnicode :: CodePage -> B.ByteString -> IO String
codePageToUnicode cp bs = B.useAsCStringLen bs $ \(inBuff, inLen) -> do
outSize <- multiByteToWideChar cp 0 inBuff (toEnum inLen) nullPtr 0
allocaArray0 (fromEnum outSize) $ \outBuff -> do
outSize' <- multiByteToWideChar cp 0 inBuff (toEnum inLen) outBuff outSize
peekCWStringLen (outBuff, fromEnum outSize')
getCodePage :: IO CodePage
getCodePage = do
conCP <- getConsoleCP
if conCP > 0
then return conCP
else getACP
foreign import WINDOWS_CCONV "IsDBCSLeadByteEx" c_IsDBCSLeadByteEx
:: CodePage -> BYTE -> BOOL
getMultiByteChar :: CodePage -> Handle -> MaybeT IO Char
getMultiByteChar cp h = do
b1 <- hGetByte h
bs <- if c_IsDBCSLeadByteEx cp b1
then hGetByte h >>= \b2 -> return [b1,b2]
else return [b1]
cs <- liftIO $ codePageToUnicode cp (B.pack bs)
case cs of
[] -> getMultiByteChar cp h
(c:_) -> return c
getAttribute :: HANDLE -> IO WORD
getAttribute = withScreenBufferInfo $
((\hsc_ptr -> peekByteOff hsc_ptr 8))
fillConsoleChar :: HANDLE -> Char -> Int -> Coord -> IO ()
fillConsoleChar h c n start = with start $ \startPtr -> alloca $ \numWritten -> do
failIfFalse_ "FillConsoleOutputCharacter"
$ c_FillConsoleCharacter h (toEnum $ fromEnum c)
(toEnum n) startPtr numWritten
foreign import ccall "haskeline_FillConsoleCharacter" c_FillConsoleCharacter
:: HANDLE -> TCHAR -> DWORD -> Ptr Coord -> Ptr DWORD -> IO BOOL
fillConsoleAttribute :: HANDLE -> WORD -> Int -> Coord -> IO ()
fillConsoleAttribute h a n start = with start $ \startPtr -> alloca $ \numWritten -> do
failIfFalse_ "FillConsoleOutputAttribute"
$ c_FillConsoleAttribute h a
(toEnum n) startPtr numWritten
foreign import ccall "haskeline_FillConsoleAttribute" c_FillConsoleAttribute
:: HANDLE -> WORD -> DWORD -> Ptr Coord -> Ptr DWORD -> IO BOOL
clearScreen :: DrawM ()
clearScreen = do
lay <- ask
h <- asks hOut
let windowSize = width lay * height lay
let origin = Coord 0 0
attr <- liftIO $ getAttribute h
liftIO $ fillConsoleChar h ' ' windowSize origin
liftIO $ fillConsoleAttribute h attr windowSize origin
setPos origin