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
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
|
{-# OPTIONS_GHC -Wno-unrecognised-pragmas #-}
{-# HLINT ignore "Use for_" #-}
{-# HLINT ignore "Use <=<" #-}
module Daffm.Action.Core where
import Brick (suspendAndResume')
import qualified Brick.Widgets.List as L
import Control.Monad (void)
import Control.Monad.State (MonadIO (liftIO), MonadState, get, gets, modify, put)
import Daffm.State
import Daffm.Types (AppEvent, AppState (..), FileInfo (..), FilePathText, FileType (..))
import Data.Maybe (fromMaybe)
import qualified Data.Set as Set
import qualified Data.Text as Text
import qualified Data.Vector as Vec
import System.Directory (getHomeDirectory)
import qualified System.Exit as Proc
import System.FilePath (takeDirectory)
import qualified System.Process as Proc
modifyM :: (MonadState s m) => (s -> m s) -> m ()
modifyM f = get >>= f >>= put
loadDir :: FilePathText -> AppEvent ()
loadDir dir = do
modifyM (liftIO . (>>= filterInvalidSelections) . loadDirToState dir)
applySearch
reloadDir :: AppEvent ()
reloadDir = do
AppState {stateCwd} <- get
loadDir stateCwd
goBackToParentDir :: AppEvent ()
goBackToParentDir = do
dir <- gets (Text.pack . takeDirectory . Text.unpack . stateCwd)
loadDir dir
changeDir :: FilePathText -> AppEvent ()
changeDir = loadDir
goHome :: AppEvent ()
goHome = do
liftIO getHomeDirectory >>= changeDir . Text.pack
openSelectedFile :: AppEvent ()
openSelectedFile = do
currentFile >>= \case
Just (FileInfo {filePath, fileType = Directory}) -> loadDir filePath
Just _ -> do
opener <- gets (fromMaybe "echo '%F' | xargs -i xdg-open {}" . stateOpenerScript)
cmdSubstitutions opener >>= suspendAndRunShellCommand False
Nothing -> pure ()
shellCommand :: String -> IO Proc.ExitCode
shellCommand cmd = do
Proc.withCreateProcess
(Proc.shell cmd) {Proc.delegate_ctlc = True}
$ \_ _ _ p -> Proc.waitForProcess p
cmdSubstitutions :: Text.Text -> AppEvent Text.Text
cmdSubstitutions cmd = do
(AppState {stateFiles, stateCwd, stateFileSelections}) <- get
let file = maybe "" (filePath . snd) . L.listSelectedElement $ stateFiles
let escape = (\s -> "'" <> s <> "'") . Text.replace "'" "\\'"
let selections = Set.elems stateFileSelections
let selectionsOrCurrent = if Set.null stateFileSelections then [file] else selections
let subst =
Text.replace "%" (escape file)
. Text.replace "%d" (escape stateCwd)
. Text.replace "%s" (Text.unwords $ map escape selections)
. Text.replace "%S" (Text.dropWhileEnd (== '\n') $ Text.unlines selections)
. Text.replace "%f" (Text.unwords $ map escape selectionsOrCurrent)
. Text.replace "%F" (Text.dropWhileEnd (== '\n') $ Text.unlines selectionsOrCurrent)
pure . subst $ cmd
-- Suspend tui and run shell command
-- When waitForKey is true, it will prompt for a key press on success
-- When exit code is non-zero, it will print it and prompt for key press regardless of waitForKey
suspendAndRunShellCommand :: Bool -> Text.Text -> AppEvent ()
suspendAndRunShellCommand waitForKey cmd = do
suspendAndResume' $ do
exitCode <- shellCommand $ Text.unpack cmd
case exitCode of
Proc.ExitFailure code -> do
putStrLn $ "Process exited with " <> show code
putStrLn "Press any key to continue" >> void getChar
_ | waitForKey -> putStrLn "Press any key to continue" >> void getChar
_ -> pure ()
currentFile :: AppEvent (Maybe FileInfo)
currentFile = do
gets (fmap snd . L.listSelectedElement . stateFiles)
toggleCurrentFileSelection :: AppEvent ()
toggleCurrentFileSelection = do
currentFile >>= maybe (pure ()) (modify . toggleFileSelection . filePath)
moveCurrent 1
clearFileSelections :: AppEvent ()
clearFileSelections =
modify $ \s -> s {stateFileSelections = Set.empty}
moveCurrent :: Int -> AppEvent ()
moveCurrent count = do
files <- gets stateFiles
modify $ \s -> s {stateFiles = L.listMoveBy count files}
setSearchTerm :: Text.Text -> AppEvent ()
setSearchTerm "" = modify (\st -> st {stateSearchTerm = Nothing, stateSearchIndex = 0})
setSearchTerm term = modify (\st -> st {stateSearchTerm = Just term, stateSearchIndex = 0})
applySearch :: AppEvent ()
applySearch = get >>= apply
where
apply :: AppState -> AppEvent ()
apply (AppState {stateSearchTerm = Nothing}) =
modify
(\st -> st {stateSearchMatches = Vec.empty, stateSearchIndex = 0})
apply (AppState {stateSearchTerm = Just term, stateFiles}) = do
let search (_, FileInfo {fileName}) = Text.toLower term `Text.isInfixOf` Text.toLower fileName
let matches = Vec.map fst . Vec.filter search . Vec.indexed $ L.listElements stateFiles
modify
( \st ->
st
{ stateSearchMatches = matches,
stateSearchIndex = wrapSearchIndex st (stateSearchIndex st)
}
)
nextSearchMatch :: AppEvent ()
nextSearchMatch = do
st@(AppState {stateSearchMatches, stateFiles, stateSearchIndex}) <- get
let nextFiles =
if Vec.null stateSearchMatches
then stateFiles
else L.listMoveTo (stateSearchMatches Vec.! wrapSearchIndex st stateSearchIndex) stateFiles
modify (\st' -> st' {stateFiles = nextFiles})
wrapSearchIndex :: AppState -> Int -> Int
wrapSearchIndex (AppState {stateSearchMatches}) nextIndex =
let matchCount = length stateSearchMatches
in if
| nextIndex < 0 -> matchCount - 1
| nextIndex >= matchCount && matchCount /= 0 -> nextIndex `mod` matchCount
| otherwise -> nextIndex
updateSearchIndex :: (Int -> Int) -> AppEvent ()
updateSearchIndex upd =
modify (\st -> st {stateSearchIndex = wrapSearchIndex st $ upd $ stateSearchIndex st})
|
