f97c783283
This new approach allows filtering out checks from the default set that are not appropriate for a command, rather than having to list every check that is appropriate. It also reduces some boilerplate. Haskell does not define Eq for functions, so I had to go a long way around with each check having a unique id. Meh.
82 lines
2 KiB
Haskell
82 lines
2 KiB
Haskell
{- git-annex command
|
|
-
|
|
- Copyright 2010 Joey Hess <joey@kitenet.net>
|
|
-
|
|
- Licensed under the GNU GPL version 3 or higher.
|
|
-}
|
|
|
|
module Command.Get where
|
|
|
|
import Common.Annex
|
|
import Command
|
|
import qualified Annex
|
|
import qualified Remote
|
|
import Annex.Content
|
|
import qualified Command.Move
|
|
|
|
def :: [Command]
|
|
def = [dontCheck fromOpt $ command "get" paramPaths seek
|
|
"make content of annexed files available"]
|
|
|
|
seek :: [CommandSeek]
|
|
seek = [withNumCopies start]
|
|
|
|
start :: FilePath -> Maybe Int -> CommandStart
|
|
start file numcopies = isAnnexed file $ \(key, _) -> do
|
|
inannex <- inAnnex key
|
|
if inannex
|
|
then stop
|
|
else autoCopies key (<) numcopies $ do
|
|
showStart "get" file
|
|
from <- Annex.getState Annex.fromremote
|
|
case from of
|
|
Nothing -> next $ perform key
|
|
Just name -> do
|
|
-- get --from = copy --from
|
|
src <- Remote.byName name
|
|
next $ Command.Move.fromPerform src False key
|
|
|
|
perform :: Key -> CommandPerform
|
|
perform key = do
|
|
ok <- getViaTmp key (getKeyFile key)
|
|
if ok
|
|
then next $ return True -- no cleanup needed
|
|
else stop
|
|
|
|
{- Try to find a copy of the file in one of the remotes,
|
|
- and copy it to here. -}
|
|
getKeyFile :: Key -> FilePath -> Annex Bool
|
|
getKeyFile key file = do
|
|
remotes <- Remote.keyPossibilities key
|
|
if null remotes
|
|
then do
|
|
showNote "not available"
|
|
Remote.showLocations key []
|
|
return False
|
|
else trycopy remotes remotes
|
|
where
|
|
trycopy full [] = do
|
|
Remote.showTriedRemotes full
|
|
Remote.showLocations key []
|
|
return False
|
|
trycopy full (r:rs) = do
|
|
probablythere <- probablyPresent r
|
|
if probablythere
|
|
then docopy r (trycopy full rs)
|
|
else trycopy full rs
|
|
-- This check is to avoid an ugly message if a remote is a
|
|
-- drive that is not mounted.
|
|
probablyPresent r =
|
|
if Remote.hasKeyCheap r
|
|
then do
|
|
res <- Remote.hasKey r key
|
|
case res of
|
|
Right b -> return b
|
|
Left _ -> return False
|
|
else return True
|
|
docopy r continue = do
|
|
showAction $ "from " ++ Remote.name r
|
|
copied <- Remote.retrieveKeyFile r key file
|
|
if copied
|
|
then return True
|
|
else continue
|