summaryrefslogtreecommitdiff
path: root/Command/Reinject.hs
blob: e4abeef3c853065a745b6b15573dde0c8941db60 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
{- git-annex command
 -
 - Copyright 2011 Joey Hess <joey@kitenet.net>
 -
 - Licensed under the GNU GPL version 3 or higher.
 -}

module Command.Reinject where

import Common.Annex
import Command
import Logs.Location
import Annex.Content
import qualified Command.Fsck

def :: [Command]
def = [notDirect $ command "reinject" (paramPair "SRC" "DEST") seek
	SectionUtility "sets content of annexed file"]

seek :: [CommandSeek]
seek = [withWords start]

start :: [FilePath] -> CommandStart
start (src:dest:[])
	| src == dest = stop
	| otherwise =
		ifAnnexed src
			(error $ "cannot used annexed file as src: " ++ src)
			go
  where
	go = do
		showStart "reinject" dest
		next $ whenAnnexed (perform src) dest
start _ = error "specify a src file and a dest file"

perform :: FilePath -> FilePath -> (Key, Backend) -> CommandPerform
perform src _dest (key, backend) =
	{- Check the content before accepting it. -}
	ifM (Command.Fsck.checkKeySizeOr reject key src
		<&&> Command.Fsck.checkBackendOr reject backend key src)
		( do
			unlessM move $ error "mv failed!"
			next $ cleanup key
		, error "not reinjecting"
		)
  where
	-- the file might be on a different filesystem,
	-- so mv is used rather than simply calling
	-- moveToObjectDir; disk space is also
	-- checked this way.
	move = getViaTmp key $ \tmp ->
		liftIO $ boolSystem "mv" [File src, File tmp]
	reject = const $ return "wrong file?"

cleanup :: Key -> CommandCleanup
cleanup key = do
	logStatus key InfoPresent
	return True