2011-07-05 18:31:46 -04:00
|
|
|
{- git-annex key/value backends
|
2010-10-27 16:53:54 -04:00
|
|
|
-
|
|
|
|
- Copyright 2010 Joey Hess <joey@kitenet.net>
|
|
|
|
-
|
|
|
|
- Licensed under the GNU GPL version 3 or higher.
|
|
|
|
-}
|
2010-10-10 13:47:04 -04:00
|
|
|
|
2010-10-11 17:52:46 -04:00
|
|
|
module Backend (
|
2011-09-15 16:57:02 -04:00
|
|
|
BackendFile,
|
2010-10-21 16:30:16 -04:00
|
|
|
list,
|
2011-07-05 18:31:46 -04:00
|
|
|
orderedList,
|
|
|
|
genKey,
|
2010-11-01 17:50:37 -04:00
|
|
|
lookupFile,
|
2010-11-15 18:04:19 -04:00
|
|
|
chooseBackends,
|
2011-03-16 01:23:20 -04:00
|
|
|
lookupBackendName,
|
|
|
|
maybeLookupBackendName
|
2010-10-11 17:52:46 -04:00
|
|
|
) where
|
2010-10-10 13:47:04 -04:00
|
|
|
|
2010-11-22 17:51:55 -04:00
|
|
|
import System.IO.Error (try)
|
2010-10-13 00:42:46 -04:00
|
|
|
import System.Posix.Files
|
2010-10-16 16:20:49 -04:00
|
|
|
|
2011-10-05 16:02:51 -04:00
|
|
|
import Common.Annex
|
2011-12-13 15:05:07 -04:00
|
|
|
import qualified Git.Config
|
2011-12-13 15:22:43 -04:00
|
|
|
import qualified Git.CheckAttr
|
2010-10-14 03:18:11 -04:00
|
|
|
import qualified Annex
|
2011-06-01 21:56:04 -04:00
|
|
|
import Types.Key
|
|
|
|
import qualified Types.Backend as B
|
2011-07-05 18:31:46 -04:00
|
|
|
|
|
|
|
-- When adding a new backend, import it here and add it to the list.
|
|
|
|
import qualified Backend.SHA
|
2011-11-04 15:21:45 -04:00
|
|
|
import qualified Backend.WORM
|
2011-08-06 14:57:22 -04:00
|
|
|
import qualified Backend.URL
|
2011-07-05 18:31:46 -04:00
|
|
|
|
2011-12-31 04:11:39 -04:00
|
|
|
list :: [Backend]
|
2011-11-04 15:21:45 -04:00
|
|
|
list = Backend.SHA.backends ++ Backend.WORM.backends ++ Backend.URL.backends
|
2010-10-14 15:58:53 -04:00
|
|
|
|
|
|
|
{- List of backends in the order to try them when storing a new key. -}
|
2011-12-31 04:11:39 -04:00
|
|
|
orderedList :: Annex [Backend]
|
2011-07-05 18:31:46 -04:00
|
|
|
orderedList = do
|
2011-01-25 21:49:04 -04:00
|
|
|
l <- Annex.getState Annex.backends -- list is cached here
|
2010-11-22 15:46:57 -04:00
|
|
|
if not $ null l
|
2010-10-14 15:58:53 -04:00
|
|
|
then return l
|
2011-10-10 17:37:44 -04:00
|
|
|
else handle =<< Annex.getState Annex.forcebackend
|
2011-01-26 00:17:38 -04:00
|
|
|
where
|
2011-10-10 17:37:44 -04:00
|
|
|
handle Nothing = standard
|
|
|
|
handle (Just "") = standard
|
|
|
|
handle (Just name) = do
|
|
|
|
l' <- (lookupBackendName name :) <$> standard
|
|
|
|
Annex.changeState $ \s -> s { Annex.backends = l' }
|
2010-10-21 16:30:16 -04:00
|
|
|
return l'
|
2011-12-13 15:05:07 -04:00
|
|
|
standard = fromRepo $ parseBackendList . Git.Config.get "annex.backends" ""
|
2011-10-10 17:37:44 -04:00
|
|
|
parseBackendList [] = list
|
|
|
|
parseBackendList s = map lookupBackendName $ words s
|
2010-10-17 11:47:36 -04:00
|
|
|
|
2011-07-05 18:31:46 -04:00
|
|
|
{- Generates a key for a file, trying each backend in turn until one
|
|
|
|
- accepts it. -}
|
2011-12-31 04:11:39 -04:00
|
|
|
genKey :: FilePath -> Maybe Backend -> Annex (Maybe (Key, Backend))
|
2011-07-05 18:31:46 -04:00
|
|
|
genKey file trybackend = do
|
|
|
|
bs <- orderedList
|
2011-08-06 12:50:20 -04:00
|
|
|
let bs' = maybe bs (: bs) trybackend
|
2011-07-05 18:31:46 -04:00
|
|
|
genKey' bs' file
|
2011-12-31 04:11:39 -04:00
|
|
|
genKey' :: [Backend] -> FilePath -> Annex (Maybe (Key, Backend))
|
2011-07-05 18:31:46 -04:00
|
|
|
genKey' [] _ = return Nothing
|
|
|
|
genKey' (b:bs) file = do
|
|
|
|
r <- (B.getKey b) file
|
|
|
|
case r of
|
|
|
|
Nothing -> genKey' bs file
|
2011-12-06 13:02:50 -04:00
|
|
|
Just k -> return $ Just (makesane k, b)
|
|
|
|
where
|
|
|
|
-- keyNames should not contain newline characters.
|
|
|
|
makesane k = k { keyName = map fixbadchar (keyName k) }
|
|
|
|
fixbadchar c
|
|
|
|
| c == '\n' = '_'
|
|
|
|
| otherwise = c
|
2011-03-23 17:57:10 -04:00
|
|
|
|
2010-10-13 03:20:05 -04:00
|
|
|
{- Looks up the key and backend corresponding to an annexed file,
|
2010-10-13 00:42:46 -04:00
|
|
|
- by examining what the file symlinks to. -}
|
2011-12-31 04:11:39 -04:00
|
|
|
lookupFile :: FilePath -> Annex (Maybe (Key, Backend))
|
2010-10-13 03:20:05 -04:00
|
|
|
lookupFile file = do
|
2010-10-31 18:04:34 -04:00
|
|
|
tl <- liftIO $ try getsymlink
|
|
|
|
case tl of
|
2010-10-31 14:39:53 -04:00
|
|
|
Left _ -> return Nothing
|
2011-07-05 18:31:46 -04:00
|
|
|
Right l -> makekey l
|
2010-10-31 18:04:34 -04:00
|
|
|
where
|
2011-09-28 16:43:10 -04:00
|
|
|
getsymlink = takeFileName <$> readSymbolicLink file
|
2011-07-05 18:31:46 -04:00
|
|
|
makekey l = maybe (return Nothing) (makeret l) (fileKey l)
|
2011-10-10 17:37:44 -04:00
|
|
|
makeret l k = let bname = keyBackendName k in
|
2011-07-05 18:31:46 -04:00
|
|
|
case maybeLookupBackendName bname of
|
2011-10-10 17:37:44 -04:00
|
|
|
Just backend -> return $ Just (k, backend)
|
|
|
|
Nothing -> do
|
|
|
|
when (isLinkToAnnex l) $ warning $
|
|
|
|
"skipping " ++ file ++
|
|
|
|
" (unknown backend " ++
|
|
|
|
bname ++ ")"
|
|
|
|
return Nothing
|
2010-11-01 17:50:37 -04:00
|
|
|
|
2011-12-31 04:11:39 -04:00
|
|
|
type BackendFile = (Maybe Backend, FilePath)
|
2011-09-15 16:57:02 -04:00
|
|
|
|
2010-11-01 17:50:37 -04:00
|
|
|
{- Looks up the backends that should be used for each file in a list.
|
|
|
|
- That can be configured on a per-file basis in the gitattributes file.
|
|
|
|
-}
|
2011-09-15 16:57:02 -04:00
|
|
|
chooseBackends :: [FilePath] -> Annex [BackendFile]
|
2011-11-08 15:34:10 -04:00
|
|
|
chooseBackends fs = Annex.getState Annex.forcebackend >>= go
|
|
|
|
where
|
|
|
|
go Nothing = do
|
2011-12-13 15:22:43 -04:00
|
|
|
pairs <- inRepo $ Git.CheckAttr.lookup "annex.backend" fs
|
2011-11-08 15:34:10 -04:00
|
|
|
return $ map (\(f,b) -> (maybeLookupBackendName b, f)) pairs
|
|
|
|
go (Just _) = do
|
2011-07-05 18:31:46 -04:00
|
|
|
l <- orderedList
|
2011-12-15 18:11:42 -04:00
|
|
|
return $ map (\f -> (Just $ Prelude.head l, f)) fs
|
2011-03-23 02:10:59 -04:00
|
|
|
|
2011-07-05 18:31:46 -04:00
|
|
|
{- Looks up a backend by name. May fail if unknown. -}
|
2011-12-31 04:11:39 -04:00
|
|
|
lookupBackendName :: String -> Backend
|
2011-07-15 03:12:05 -04:00
|
|
|
lookupBackendName s = fromMaybe unknown $ maybeLookupBackendName s
|
2011-07-05 18:31:46 -04:00
|
|
|
where
|
|
|
|
unknown = error $ "unknown backend " ++ s
|
2011-12-31 04:11:39 -04:00
|
|
|
maybeLookupBackendName :: String -> Maybe Backend
|
2011-12-15 18:11:42 -04:00
|
|
|
maybeLookupBackendName s = headMaybe matches
|
2011-10-10 17:37:44 -04:00
|
|
|
where
|
|
|
|
matches = filter (\b -> s == B.name b) list
|