108 lines
3.6 KiB
Haskell
108 lines
3.6 KiB
Haskell
{- Git configuration
|
|
-
|
|
- Copyright 2011 Joey Hess <joey@kitenet.net>
|
|
-
|
|
- Licensed under the GNU GPL version 3 or higher.
|
|
-}
|
|
|
|
module Config where
|
|
|
|
import Common.Annex
|
|
import qualified Git
|
|
import qualified Git.Config
|
|
import qualified Git.Command
|
|
import qualified Annex
|
|
import qualified Build.SysConfig
|
|
import Utility.DataUnits
|
|
|
|
type ConfigKey = String
|
|
|
|
{- Changes a git config setting in both internal state and .git/config -}
|
|
setConfig :: ConfigKey -> String -> Annex ()
|
|
setConfig k value = do
|
|
inRepo $ Git.Command.run "config" [Param k, Param value]
|
|
-- re-read git config and update the repo's state
|
|
newg <- inRepo Git.Config.read
|
|
Annex.changeState $ \s -> s { Annex.repo = newg }
|
|
|
|
{- Looks up a per-remote config setting in git config.
|
|
- Failing that, tries looking for a global config option. -}
|
|
getConfig :: Git.Repo -> ConfigKey -> String -> Annex String
|
|
getConfig r key def = do
|
|
def' <- fromRepo $ Git.Config.get ("annex." ++ key) def
|
|
fromRepo $ Git.Config.get (remoteConfig r key) def'
|
|
|
|
{- A per-remote config setting in git config. -}
|
|
remoteConfig :: Git.Repo -> ConfigKey -> String
|
|
remoteConfig r key = "remote." ++ fromMaybe "" (Git.remoteName r) ++ ".annex-" ++ key
|
|
|
|
{- Calculates cost for a remote. Either the default, or as configured
|
|
- by remote.<name>.annex-cost, or if remote.<name>.annex-cost-command
|
|
- is set and prints a number, that is used. -}
|
|
remoteCost :: Git.Repo -> Int -> Annex Int
|
|
remoteCost r def = do
|
|
cmd <- getConfig r "cost-command" ""
|
|
(fromMaybe def . readish) <$>
|
|
if not $ null cmd
|
|
then liftIO $ snd <$> pipeFrom "sh" ["-c", cmd]
|
|
else getConfig r "cost" ""
|
|
|
|
cheapRemoteCost :: Int
|
|
cheapRemoteCost = 100
|
|
semiCheapRemoteCost :: Int
|
|
semiCheapRemoteCost = 110
|
|
expensiveRemoteCost :: Int
|
|
expensiveRemoteCost = 200
|
|
|
|
{- Adjusts a remote's cost to reflect it being encrypted. -}
|
|
encryptedRemoteCostAdj :: Int
|
|
encryptedRemoteCostAdj = 50
|
|
|
|
{- Make sure the remote cost numbers work out. -}
|
|
prop_cost_sane :: Bool
|
|
prop_cost_sane = False `notElem`
|
|
[ expensiveRemoteCost > 0
|
|
, cheapRemoteCost < semiCheapRemoteCost
|
|
, semiCheapRemoteCost < expensiveRemoteCost
|
|
, cheapRemoteCost + encryptedRemoteCostAdj > semiCheapRemoteCost
|
|
, cheapRemoteCost + encryptedRemoteCostAdj < expensiveRemoteCost
|
|
, semiCheapRemoteCost + encryptedRemoteCostAdj < expensiveRemoteCost
|
|
]
|
|
|
|
{- Checks if a repo should be ignored. -}
|
|
repoNotIgnored :: Git.Repo -> Annex Bool
|
|
repoNotIgnored r = not . fromMaybe False . Git.configTrue <$> getConfig r "ignore" ""
|
|
|
|
{- If a value is specified, it is used; otherwise the default is looked up
|
|
- in git config. forcenumcopies overrides everything. -}
|
|
getNumCopies :: Maybe Int -> Annex Int
|
|
getNumCopies v = perhaps (use v) =<< Annex.getState Annex.forcenumcopies
|
|
where
|
|
use (Just n) = return n
|
|
use Nothing = perhaps (return 1) =<<
|
|
readish <$> fromRepo (Git.Config.get config "1")
|
|
perhaps fallback = maybe fallback (return . id)
|
|
config = "annex.numcopies"
|
|
|
|
{- Gets the trust level set for a remote in git config. -}
|
|
getTrustLevel :: Git.Repo -> Annex (Maybe String)
|
|
getTrustLevel r = fromRepo $ Git.Config.getMaybe $ remoteConfig r "trustlevel"
|
|
|
|
{- Gets annex.diskreserve setting. -}
|
|
getDiskReserve :: Bool -> Annex Integer
|
|
getDiskReserve sanitycheck = do
|
|
g <- gitRepo
|
|
r <- getConfig g "diskreserve" ""
|
|
when sanitycheck $ check r
|
|
return $ fromMaybe megabyte $ readSize dataUnits r
|
|
where
|
|
megabyte = 1000000
|
|
check r
|
|
| not (null r) && not cancheck = do
|
|
unlessM (Annex.getState Annex.force) $
|
|
error $ "You have configured a diskreserve of "
|
|
++ r ++
|
|
" but disk space checking is not working"
|
|
return ()
|
|
| otherwise = return ()
|
|
cancheck = Build.SysConfig.statfs_sanity_checked == Just True
|