1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
|
{-# LANGUAGE OverloadedStrings #-}
import Control.Monad (void)
import Data.Monoid
import qualified Data.Text as T
import qualified Graphics.Vty as V
import Lens.Micro ((&), (.~), (^.))
import System.Directory (getHomeDirectory)
import System.FilePath ((</>))
import Brick.AttrMap
import Brick.Main
import Brick.Markup (markup, (@?))
import Brick.Types
import Brick.Util (fg, on)
import Brick.Widgets.Border as B
import Brick.Widgets.Center as C
import Brick.Widgets.Core
import Data.Text.Markup ((@@))
import Brick.Widgets.DefnList
import Data.Maybe (isJust)
import Presets
import Types.Preset
import Types.State
drawUI :: St -> [Widget ()]
drawUI st = [ui]
where
ui = vBox [ hBorderWithLabel (str "[ Status ]")
, vLimit 3 $ C.center $ status
, hBorderWithLabel (str "[ Current settings ]")
, padAll 1 $ C.center $ settingsBox
, hBorderWithLabel (str "[ Presets ]")
, padAll 1 $ C.center $ presetsBox
, hBorderWithLabel (str "[ Actions ]")
, vLimit 5 $ C.center $ actionsBox
]
status = str "Ready to scan first page"
settingsBox = defnList AlignRight Nothing
[ ("run OCRmyPDF", if st^.stOCR then "yes" else "no")
, ("colour data", show $ st^.stColour)
, ("page size", show $ st^.stPaper)
, ("DPI", show $ st^.stDPI)
, ("output format", show $ st^.stOutFormat)
, ("output dir", st^.stOutdir)
]
presetsBox = defnList AlignLeft
(Just $ V.withStyle V.currentAttr V.bold)
(map (\(Preset k desc _) -> ([k], desc)) presets)
actionsBox = defnList AlignLeft
(Just $ V.withStyle V.currentAttr V.bold) $
(if st^.stOutFormat == PDF
then (if isJust $ st^.stScanningSession
then [ ("SPC", "scan next page")
, ("RET", "scan final page")
, ("q", "declare last scanned page was the final page")
]
else [ ("SPC", "scan first page of multi-page document")
, ("RET", "scan single page document")
, ("q", "quit sscan")
]
)
else [ ("SPC", "scan page")
, ("q", "quit sscan")
]
)
handleQ :: St -> EventM () (Next St)
handleQ st = undefined
handleRET :: St -> EventM () (Next St)
handleRET st = undefined
handleSPC :: St -> EventM () (Next St)
handleSPC st = undefined
handleHotKey :: St -> Char -> EventM () (Next St)
handleHotKey st 'q' = handleQ st
handleHotKey st ' ' = handleSPC st
handleHotKey st 'o' = continue $
if isJust $ st^.stScanningSession
then st
else st & stOCR .~ (not $ st^.stOCR)
handleHotKey st 'c' = continue $
if isJust $ st^.stScanningSession
then st
else st & stColour .~ (cycleColour $ st^.stColour)
handleHotKey st 'p' = continue $
if isJust $ st^.stScanningSession
then st
else st & stPaper .~ (cyclePaper $ st^.stPaper)
handleHotKey st c = continue $ if isJust $ st^.stScanningSession
then st
else case lookupPreset c of
Just (Preset _ _ f) -> f st
_ -> st
appEvent :: St -> BrickEvent () e -> EventM () (Next St)
appEvent st (VtyEvent e) =
case e of
V.EvKey (V.KEnter) [] -> handleRET st
V.EvKey (V.KChar c) [] -> handleHotKey st c
_ -> continue st
appEvent st _ = continue st
theApp :: App St e ()
theApp =
App { appDraw = drawUI
, appChooseCursor = neverShowCursor
, appHandleEvent = appEvent
, appStartEvent = return
, appAttrMap = const $ attrMap V.defAttr []
}
main = do
home <- getHomeDirectory
papersize <- init <$> readFile "/etc/papersize"
let paper = if papersize == "letter" then Letter else A4
initialState = St
{ _stScanningSession = Nothing
, _stOCR = True
, _stColour = Greyscale
, _stPaper = paper
, _stDefaultPaper = paper
, _stDPI = 300
, _stOutFormat = PDF
, _stOutdir = home </> "tmp"
}
void $ defaultMain theApp initialState
|
