summaryrefslogtreecommitdiff
path: root/Command/Assistant.hs
blob: ea8a87a3d8c2c2f670ab4c1f833b0ac1d4ed8eb2 (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
{- git-annex assistant
 -
 - Copyright 2012 Joey Hess <joey@kitenet.net>
 -
 - Licensed under the GNU GPL version 3 or higher.
 -}

module Command.Assistant where

import Common.Annex
import Command
import qualified Option
import qualified Command.Watch
import Init
import Locations.UserConfig

import System.Environment
import System.Posix.Directory

def :: [Command]
def = [noRepo checkAutoStart $ dontCheck repoExists $
	withOptions [Command.Watch.foregroundOption, Command.Watch.stopOption, autoStartOption] $ 
	command "assistant" paramNothing seek "automatically handle changes"]

autoStartOption :: Option
autoStartOption = Option.flag [] "autostart" "start in known repositories"

seek :: [CommandSeek]
seek = [withFlag Command.Watch.stopOption $ \stopdaemon ->
	withFlag Command.Watch.foregroundOption $ \foreground ->
	withFlag autoStartOption $ \autostart ->
	withNothing $ start foreground stopdaemon autostart]

start :: Bool -> Bool -> Bool -> CommandStart
start foreground stopdaemon autostart
	| autostart = do
		liftIO autoStart
		stop
	| otherwise = do
		ensureInitialized
		Command.Watch.start True foreground stopdaemon

{- Run outside a git repository. Check to see if any parameter is
 - --autostart and enter autostart mode. -}
checkAutoStart :: IO ()
checkAutoStart = ifM (elem "--autostart" <$> getArgs)
	( autoStart
	, error "Not in a git repository."
	) 

autoStart :: IO ()
autoStart = do
	autostartfile <- autoStartFile
	let nothing = error $ "Nothing listed in " ++ autostartfile
	ifM (doesFileExist autostartfile)
		( do
			dirs <- nub . lines <$> readFile autostartfile
			program <- readProgramFile
			when (null dirs) nothing
			forM_ dirs $ \d -> do
				putStrLn $ "git-annex autostart in " ++ d
				ifM (catchBoolIO $ go program d)
					( putStrLn "ok"
					, putStrLn "failed"
					)
		, nothing
		)
  where
	go program dir = do
		changeWorkingDirectory dir
		boolSystem program [Param "assistant"]