diff options
| -rw-r--r-- | src/Main.hs | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/src/Main.hs b/src/Main.hs index 4a96cf0..3af0d5f 100644 --- a/src/Main.hs +++ b/src/Main.hs @@ -34,6 +34,7 @@ data VTState = VTState { buffer :: Buffer , mode :: Mode , processCount :: Int + , count :: Int , register :: Char , registers :: Map Char String } @@ -53,6 +54,7 @@ main = do { mode = InsertMode , buffer = ("!while date; do sleep 1; done", "") , processCount = 0 + , count = 0 , register = defaultRegister , registers = Map.empty } @@ -148,6 +150,9 @@ data Command | Combine Command Command | Nop | RingBell + | AddCount Int + | MulCount Int + | SetCount Int | SetRegister Char instance Monoid Command where @@ -313,6 +318,10 @@ execCommand Nop = return () execCommand RingBell = liftIO ringBell +execCommand (AddCount i) = modify $ \q -> q { count = i + count q } +execCommand (MulCount i) = modify $ \q -> q { count = i * count q } +execCommand (SetCount i) = modify $ \q -> q { count = i } + execCommand (SetRegister c) = modify $ \q -> q { register = c } @@ -435,6 +444,7 @@ nmap = , ("\x0a", ExecuteInputBuffer <> ChangeMode InsertMode) , ("\x1b", RingBell) -- TODO cancel any unfinished commands ] + ++ (map (\i -> (show i, MulCount 10 <> AddCount i)) [0..9]) imap :: Keymap |