summaryrefslogtreecommitdiff
path: root/Command/Sync.hs
blob: d519943432746c7a2cff16ca601158dc0eddbe3c (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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
{- git-annex command
 -
 - Copyright 2011 Joey Hess <joey@kitenet.net>
 - Copyright 2011 Joachim Breitner <mail@joachim-breitner.de>
 -
 - Licensed under the GNU GPL version 3 or higher.
 -}

{-# LANGUAGE BangPatterns #-}

module Command.Sync where

import Common.Annex
import Command
import qualified Remote
import qualified Annex.Branch
import qualified Git.Command
import qualified Git.Config
import qualified Git.Ref
import qualified Git

import qualified Data.ByteString.Lazy.Char8 as L

def :: [Command]
def = [command "sync" (paramOptional (paramRepeating paramRemote))
	[seek] "synchronize local repository with remote repositories"]

-- syncing involves several operations, any of which can independantly fail
seek :: CommandSeek
seek args = do
	!branch <- currentBranch
	remotes <- syncRemotes branch args
	showStart "syncing" $ "branch " ++ Git.Ref.describe branch ++ " with remote repositories " ++ intercalate "," (map Remote.name remotes)
	showOutput
	return $
		[ commit
		, mergeLocal branch
		] ++
		[ fetch remote | remote <- remotes ] ++
		[ mergeRemote remote branch | remote <- remotes ] ++
		[ mergeAnnex ] ++
		[ pushLocal branch ] ++
		[ pushRemote remote branch | remote <- remotes ]

syncRemotes :: Git.Ref -> [String] -> Annex [Remote.Remote Annex]
syncRemotes branch [] = defaultSyncRemotes branch
syncRemotes _ rs = mapM Remote.byName rs

defaultSyncRemotes :: Git.Ref -> Annex [Remote.Remote Annex]
defaultSyncRemotes branch = 
	mapM Remote.byName
		=<< process . L.unpack <$> inRepo showref
	where 
		syncbranch = Git.Ref $ "refs/heads/synced/" ++ Git.Ref.describe branch
		showref = Git.Command.pipeRead
			[Param "show-ref", Param (Git.Ref.describe syncbranch)]
		process = map getRemoteName . filter isRemote .
			map getBranchName . lines
	        isRemote r = "refs/remotes/" `isPrefixOf` r
		getBranchName = snd . separate (== ' ')
		getRemoteName = fst . separate (== '/') . snd . separate (== '/') . snd . separate (== '/')

commit :: CommandStart
commit = do
	showStart "commit" ""
	next $ next $ do
		showOutput
		-- Commit will fail when the tree is clean, so ignore failure.
		_ <- inRepo $ Git.Command.runBool "commit"
			[Param "-a", Param "-m", Param "git-annex automatic sync"]
		return True

mergeLocal :: Git.Ref -> CommandStart
mergeLocal branch = mergeFromIfExists $ Git.Ref $
	"refs/heads/synced/" ++ Git.Ref.describe branch

pushLocal :: Git.Ref -> CommandStart
pushLocal branch = do
	let syncBranch = Git.Ref $ "refs/heads/synced/" ++ Git.Ref.describe branch
	ex <- inRepo $ Git.Ref.exists syncBranch
	if ex then do
		showStart "updateing" $
			Git.Ref.describe syncBranch ++
			" to the state of " ++ Git.Ref.describe branch ++ "..."
		next $ next $
			inRepo $ Git.Command.runBool "branch"
				[Param "-f", Param (Git.Ref.describe syncBranch)]
	else
		return Nothing

mergeFromIfExists :: Git.Ref -> CommandStart
mergeFromIfExists fromBranch = do
	ex <- inRepo $ Git.Ref.exists fromBranch
	if ex then do
		showStart "merging" $ Git.Ref.describe fromBranch ++ "..."
		next $ next $
			inRepo $ Git.Command.runBool "merge" [Param (show fromBranch)]
	else do
		showNote $ Git.Ref.describe fromBranch ++ " does not exist, not merging."
		showOutput
		return Nothing

fetch :: Remote.Remote Annex -> CommandStart
fetch remote = do
	showStart "fetching from" (Remote.name remote)
	next $ next $ do
		showOutput
		checkRemote remote
		inRepo $ Git.Command.runBool "fetch" [Param (Remote.name remote)]

mergeRemote :: Remote.Remote Annex -> Git.Ref -> CommandStart
mergeRemote remote branch = mergeFromIfExists $ Git.Ref $
	"refs/remotes/" ++ Remote.name remote ++ 
	"/synced/" ++ Git.Ref.describe branch

pushRemote :: Remote.Remote Annex -> Git.Ref -> CommandStart
pushRemote remote branch = do
	showStart "pushing to" (Remote.name remote)
	let syncbranch = Git.Ref $ "refs/heads/synced/" ++
		Git.Ref.describe branch
	let syncbranchRemote = Git.Ref $ "refs/remotes/" ++
		Remote.name remote ++ "/" ++ Git.Ref.describe syncbranch
	let refspec = Git.Ref.describe branch ++ ":" ++ Git.Ref.describe syncbranch
	ex <- inRepo $ Git.Ref.exists syncbranchRemote
	next $ next $ do
		showOutput
		inRepo $ Git.Command.runBool "push" $
			[ Param (Remote.name remote)
			, Param (Git.Ref.describe Annex.Branch.name) ] ++ 
			[ Param refspec | ex ]

currentBranch :: Annex Git.Ref
currentBranch = Git.Ref . firstLine . L.unpack <$>
	inRepo (Git.Command.pipeRead [Param "symbolic-ref", Param "HEAD"])

checkRemote :: Remote.Remote Annex -> Annex ()
checkRemote remote = do
	remoteurl <- fromRepo $
		Git.Config.get ("remote." ++ Remote.name remote ++ ".url") ""
	when (null remoteurl) $
		error $ "No url is configured for the remote: " ++ Remote.name remote

mergeAnnex :: CommandStart
mergeAnnex = next $ next $ do
	Annex.Branch.forceUpdate
	return True