dnf-repo-0.5: src/YumRepoFile.hs
module YumRepoFile (
Mode(..),
SpecificChange(..),
readRepos,
RepoState,
ChangeEnable(Expire,UnExpire,Delete),
printAction,
selectRepo,
changeRepo,
saveRepo,
updateState,
expiring,
deleting
)
where
import Data.List.Extra (isPrefixOf, isInfixOf, isSuffixOf, nub, sort, sortOn,
stripInfix, trim)
import Data.Maybe (mapMaybe)
import SimpleCmd (error', warning)
import System.FilePath.Glob (compile, match)
type RepoState = (String,(Bool,FilePath))
data Mode = AddCopr String | AddKoji String
| EnableRepo String | DisableRepo String
| ExpireRepo String | ClearExpires
| DeleteRepo String
| Specific SpecificChange
deriving (Eq, Ord, Show)
modePattern :: Mode -> Maybe String
modePattern (AddCopr c) = Just c
modePattern (AddKoji k) = Just k
modePattern (EnableRepo r) = Just r
modePattern (DisableRepo r) = Just r
modePattern (ExpireRepo r) = Just r
modePattern ClearExpires = Nothing
modePattern (DeleteRepo r) = Just r
modePattern (Specific _) = Nothing
data SpecificChange = EnableModular | DisableModular
| EnableTesting | DisableTesting
| EnableDebuginfo | DisableDebuginfo
| EnableSource | DisableSource
deriving (Eq, Ord, Show)
repoSubstr :: SpecificChange -> String
repoSubstr EnableModular = "-modular"
repoSubstr DisableModular = "-modular"
repoSubstr EnableTesting = "-testing"
repoSubstr DisableTesting = "-testing"
repoSubstr EnableDebuginfo = "-debuginfo"
repoSubstr DisableDebuginfo = "-debuginfo"
repoSubstr EnableSource = "-source"
repoSubstr DisableSource = "-source"
data ChangeEnable = Disable String Bool
| Enable String Bool
| Expire String
| UnExpire
| Delete FilePath Bool
deriving (Eq,Ord,Show)
printAction :: Bool -> ChangeEnable -> IO ()
printAction save (Disable r s) =
if s
then putStrLn $
if save
then "disable " ++ quote r
else "with " ++ quote r ++ " disabled"
else warning $ quote r ++ " already disabled"
printAction save (Enable r s) =
if s
then putStrLn $
if save
then "enable " ++ quote r
else "with " ++ quote r ++ " enabled"
else warning $ quote r ++ " already enabled"
printAction _ (Expire r) =
putStrLn $ "expire " ++ quote r
printAction _ UnExpire =
putStrLn "unexpire:"
printAction _ (Delete f s) =
if s
then putStrLn $ "delete " ++ quote f
else warning $ quote f ++ " deletion skipped"
quote :: String -> String
quote s = '\'' : s ++ "'"
maybeRepoName :: ChangeEnable -> Maybe (ChangeEnable, String)
maybeRepoName d@(Disable r _) = Just (d, r)
maybeRepoName e@(Enable r _) = Just (e, r)
maybeRepoName x@(Expire r) = Just (x, r)
maybeRepoName UnExpire = Nothing
maybeRepoName (Delete _ _) = Nothing
changeRepo :: ChangeEnable -> [String]
changeRepo (Disable r True) = ["--disablerepo", r]
changeRepo (Enable r True) = ["--enablerepo", r]
changeRepo _ = []
saveRepo :: ChangeEnable -> [String]
saveRepo (Disable r True) = ["--disable", r]
saveRepo (Enable r True) = ["--enable", r]
saveRepo _ = []
expiring :: ChangeEnable -> Maybe String
expiring (Expire r) = Just r
expiring _ = Nothing
deleting :: ChangeEnable -> Maybe String
deleting (Delete r True) = Just r
deleting _ = Nothing
updateState :: [ChangeEnable] -> RepoState -> RepoState
updateState [] rs = rs
updateState (ce:ces) re@(repo,(enabled,file)) =
case ce of
-- FIXME can probably drop enable/not enabled
Disable r True | r == repo && enabled -> (repo,(False,file))
Enable r True | r == repo && not enabled -> (repo,(True,file))
_ -> updateState ces re
selectRepo :: Bool -> [RepoState] -> [Mode] -> [ChangeEnable]
selectRepo exact repostates modes =
nub $ foldr selectRepo' [] (nub (sort modes))
where
selectRepo' :: Mode -> [ChangeEnable] -> [ChangeEnable]
selectRepo' mode acc =
let results = nub $ mapMaybe (selectRepoMode mode acc) repostates
in
case results of
[] -> error' ("no match for repo pattern action: " ++ show mode)
[_] -> acc ++ results
_ ->
acc ++
case modePattern mode of
Nothing -> results
Just p ->
if isGlob p
then results
else
let actNames = mapMaybe maybeRepoName results
in
if null actNames
then results
else
let base = head $ sortOn (length . snd) actNames
in
if all ((snd base `isPrefixOf`) . snd) actNames
then [fst base]
else results
selectRepoMode :: Mode -> [ChangeEnable] -> RepoState
-> Maybe ChangeEnable
selectRepoMode mode acc (name,(enabled,file)) =
case mode of
AddCopr repo ->
maybeChange repo isSuffixOf (not enabled) (Enable name)
AddKoji repo ->
maybeChange repo isSuffixOf (not enabled) (Enable name)
EnableRepo pat ->
maybeChange pat matchesRepo (not enabled) (Enable name)
DisableRepo pat ->
maybeChange pat matchesRepo enabled (Disable name)
ExpireRepo pat ->
maybeChange pat matchesRepo True (const (Expire name))
ClearExpires -> Just UnExpire
DeleteRepo pat ->
maybeChange pat matchesRepo
(not enabled || error ("disable repo before deleting: " ++ name))
(Delete file)
Specific change ->
let substr = repoSubstr change in
if change `elem`
[EnableModular,EnableTesting,EnableDebuginfo,EnableSource]
then
maybeChange substr
(\p n -> p `isInfixOf` n &&
repoStatus acc (removeInfix substr name))
(not enabled) (Enable name)
else
maybeChange substr isInfixOf enabled (Disable name)
where
maybeChange :: String -> (String -> String -> Bool) -> Bool
-> (Bool -> ChangeEnable) -> Maybe ChangeEnable
maybeChange pat matcher state change =
if pat `matcher` name
then
if state
then Just $ change True
else
if isGlob pat
then Nothing
else Just $ change False
else Nothing
repoStatus :: [ChangeEnable] -> String -> Bool
repoStatus acc repo =
case lookup repo repostates of
Just (enabled,_) ->
enabled || Enable repo True `elem` acc
Nothing -> False
isGlob :: String -> Bool
isGlob pat = any (`elem` pat) "*?["
matchesRepo :: String -> String -> Bool
matchesRepo "" = error' "empty repo pattern"
matchesRepo pat
| isGlob pat = (match . compile) $
if exact
then pat
else
case ('^' == head pat,'$' == last pat) of
(True,True) -> init (tail pat)
(True,False) -> tail pat ++ ['*' | last pat /= '*']
(False,True) -> ['*' | head pat /= '*'] ++ init pat
_ -> ['*' | head pat /= '*'] ++ pat ++ ['*' | last pat /= '*']
| exact = (pat ==)
| otherwise =
case ('^' == head pat,'$' == last pat) of
(True,True) -> (init (tail pat) ==)
(True,False) -> (tail pat `isPrefixOf`)
(False,True) -> (init pat `isSuffixOf`)
_ -> (pat `isInfixOf`)
readRepos :: FilePath -> IO [RepoState]
readRepos file =
parseRepos file . lines <$> readFile file
-- parse ini
parseRepos :: FilePath -> [String] -> [RepoState]
parseRepos file ls =
case nextSection ls of
Nothing -> []
Just (section,rest) ->
let (enabled,more) =
case dropWhile (not . ("enabled=" `isPrefixOf`)) rest of
[] -> error' $ "no enabled field for " ++ section
(e:more') ->
case trim e of
"enabled=1" -> (True,more')
"enabled=0" -> (False,more')
_ -> error' $ "unknown enabled state " ++ e ++ " for " ++ section
in (section,(enabled,file)) : parseRepos file more
where
nextSection :: [String] -> Maybe (String,[String])
nextSection [] = Nothing
nextSection (l:ls') =
case l of
('[' : rest) ->
if last rest == ']'
then Just (init rest, ls')
else error' $ "bad section " ++ l ++ " in " ++ file
_ -> nextSection ls'
-- adapted from simple-cmd
removeInfix :: String -> String-> String
removeInfix inf orig =
maybe orig (uncurry (++)) $ stripInfix inf orig