68 lines
1.7 KiB
Haskell
68 lines
1.7 KiB
Haskell
|
{- git-annex command
|
||
|
-
|
||
|
- Copyright 2012 Joey Hess <joey@kitenet.net>
|
||
|
-
|
||
|
- Licensed under the GNU GPL version 3 or higher.
|
||
|
-}
|
||
|
|
||
|
module Command.Direct where
|
||
|
|
||
|
import Common.Annex
|
||
|
import Command
|
||
|
import qualified Git
|
||
|
import qualified Git.Command
|
||
|
import qualified Git.LsFiles
|
||
|
import Config
|
||
|
import Annex.Content
|
||
|
import Annex.Content.Direct
|
||
|
|
||
|
def :: [Command]
|
||
|
def = [command "direct" paramNothing seek "switch repository to direct mode"]
|
||
|
|
||
|
seek :: [CommandSeek]
|
||
|
seek = [withNothing start]
|
||
|
|
||
|
start :: CommandStart
|
||
|
start = notBareRepo $
|
||
|
ifM isDirect
|
||
|
( stop , next perform )
|
||
|
|
||
|
perform :: CommandPerform
|
||
|
perform = do
|
||
|
showStart "commit" ""
|
||
|
showOutput
|
||
|
_ <- inRepo $ Git.Command.runBool "commit"
|
||
|
[Param "-a", Param "-m", Param "commit before switching to direct mode"]
|
||
|
top <- fromRepo Git.repoPath
|
||
|
(l, clean) <- inRepo $ Git.LsFiles.inRepo [top]
|
||
|
forM_ l go
|
||
|
void $ liftIO clean
|
||
|
next cleanup
|
||
|
where
|
||
|
{- Walk tree from top and move all present objects to the
|
||
|
- files that link to them, while updating direct mode mappings. -}
|
||
|
go = whenAnnexed $ \f (k, _) -> do
|
||
|
loc <- inRepo $ gitAnnexLocation k
|
||
|
createContentDir loc -- thaws directory too
|
||
|
locs <- filter (/= f) <$> addAssociatedFile k f
|
||
|
case locs of
|
||
|
[] -> whenM (liftIO $ doesFileExist loc) $ do
|
||
|
{- Move content from annex to direct file. -}
|
||
|
showStart "direct" f
|
||
|
updateCache k loc
|
||
|
thawContent loc
|
||
|
liftIO $ replaceFile f $ moveFile loc
|
||
|
showEndOk
|
||
|
(loc':_) -> do
|
||
|
{- Another direct file has the content, so
|
||
|
- hard link to it. -}
|
||
|
showStart "direct" f
|
||
|
liftIO $ replaceFile f $ createLink loc'
|
||
|
showEndOk
|
||
|
return Nothing
|
||
|
|
||
|
cleanup :: CommandCleanup
|
||
|
cleanup = do
|
||
|
setDirect True
|
||
|
return True
|