summaryrefslogtreecommitdiff
path: root/Remote/Hook.hs
diff options
context:
space:
mode:
authorGravatar Joey Hess <joeyh@debian.org>2014-04-02 21:42:53 +0100
committerGravatar Joey Hess <joeyh@debian.org>2014-04-02 21:42:53 +0100
commit6da7cdf0fbf26f1faf7d5710e6ed488f1a4e9589 (patch)
tree7a903e2eca579335b7ce73d0220854e7a25c3bb9 /Remote/Hook.hs
git-annex (5.20140402) unstable; urgency=medium
* unannex, uninit: Avoid committing after every file is unannexed, for massive speedup. * --notify-finish switch will cause desktop notifications after each file upload/download/drop completes (using the dbus Desktop Notifications Specification) * --notify-start switch will show desktop notifications when each file upload/download starts. * webapp: Automatically install Nautilus integration scripts to get and drop files. * tahoe: Pass -d parameter before subcommand; putting it after the subcommand no longer works with tahoe-lafs version 1.10. (Thanks, Alberto Berti) * forget --drop-dead: Avoid removing the dead remote from the trust.log, so that if git remotes for it still exist anywhere, git annex info will still know it's dead and not show it. * git-annex-shell: Make configlist automatically initialize a remote git repository, as long as a git-annex branch has been pushed to it, to simplify setup of remote git repositories, including via gitolite. * add --include-dotfiles: New option, perhaps useful for backups. * Version 5.20140227 broke creation of glacier repositories, not including the datacenter and vault in their configuration. This bug is fixed, but glacier repositories set up with the broken version of git-annex need to have the datacenter and vault set in order to be usable. This can be done using git annex enableremote to add the missing settings. For details, see http://git-annex.branchable.com/bugs/problems_with_glacier/ * Added required content configuration. * assistant: Improve ssh authorized keys line generated in local pairing or for a remote ssh server to set environment variables in an alternative way that works with the non-POSIX fish shell, as well as POSIX shells. # imported from the archive
Diffstat (limited to 'Remote/Hook.hs')
-rw-r--r--Remote/Hook.hs159
1 files changed, 159 insertions, 0 deletions
diff --git a/Remote/Hook.hs b/Remote/Hook.hs
new file mode 100644
index 000000000..3735c228c
--- /dev/null
+++ b/Remote/Hook.hs
@@ -0,0 +1,159 @@
+{- A remote that provides hooks to run shell commands.
+ -
+ - Copyright 2011 Joey Hess <joey@kitenet.net>
+ -
+ - Licensed under the GNU GPL version 3 or higher.
+ -}
+
+module Remote.Hook (remote) where
+
+import qualified Data.ByteString.Lazy as L
+import qualified Data.Map as M
+
+import Common.Annex
+import Types.Remote
+import Types.Key
+import Types.Creds
+import qualified Git
+import Config
+import Config.Cost
+import Annex.Content
+import Annex.UUID
+import Remote.Helper.Special
+import Remote.Helper.Encryptable
+import Crypto
+import Utility.Metered
+import Utility.Env
+
+type Action = String
+type HookName = String
+
+remote :: RemoteType
+remote = RemoteType {
+ typename = "hook",
+ enumerate = findSpecialRemotes "hooktype",
+ generate = gen,
+ setup = hookSetup
+}
+
+gen :: Git.Repo -> UUID -> RemoteConfig -> RemoteGitConfig -> Annex (Maybe Remote)
+gen r u c gc = do
+ cst <- remoteCost gc expensiveRemoteCost
+ return $ Just $ encryptableRemote c
+ (storeEncrypted hooktype $ getGpgEncParams (c,gc))
+ (retrieveEncrypted hooktype)
+ Remote {
+ uuid = u,
+ cost = cst,
+ name = Git.repoDescribe r,
+ storeKey = store hooktype,
+ retrieveKeyFile = retrieve hooktype,
+ retrieveKeyFileCheap = retrieveCheap hooktype,
+ removeKey = remove hooktype,
+ hasKey = checkPresent r hooktype,
+ hasKeyCheap = False,
+ whereisKey = Nothing,
+ remoteFsck = Nothing,
+ repairRepo = Nothing,
+ config = c,
+ localpath = Nothing,
+ repo = r,
+ gitconfig = gc,
+ readonly = False,
+ availability = GloballyAvailable,
+ remotetype = remote
+ }
+ where
+ hooktype = fromMaybe (error "missing hooktype") $ remoteAnnexHookType gc
+
+hookSetup :: Maybe UUID -> Maybe CredPair -> RemoteConfig -> Annex (RemoteConfig, UUID)
+hookSetup mu _ c = do
+ u <- maybe (liftIO genUUID) return mu
+ let hooktype = fromMaybe (error "Specify hooktype=") $
+ M.lookup "hooktype" c
+ c' <- encryptionSetup c
+ gitConfigSpecialRemote u c' "hooktype" hooktype
+ return (c', u)
+
+hookEnv :: Action -> Key -> Maybe FilePath -> IO (Maybe [(String, String)])
+hookEnv action k f = Just <$> mergeenv (fileenv f ++ keyenv)
+ where
+ mergeenv l = addEntries l <$> getEnvironment
+ env s v = ("ANNEX_" ++ s, v)
+ keyenv = catMaybes
+ [ Just $ env "KEY" (key2file k)
+ , Just $ env "ACTION" action
+ , env "HASH_1" <$> headMaybe hashbits
+ , env "HASH_2" <$> headMaybe (drop 1 hashbits)
+ ]
+ fileenv Nothing = []
+ fileenv (Just file) = [env "FILE" file]
+ hashbits = map takeDirectory $ splitPath $ hashDirMixed k
+
+lookupHook :: HookName -> Action -> Annex (Maybe String)
+lookupHook hookname action = do
+ command <- getConfig (annexConfig hook) ""
+ if null command
+ then do
+ fallback <- getConfig (annexConfig hookfallback) ""
+ if null fallback
+ then do
+ warning $ "missing configuration for " ++ hook ++ " or " ++ hookfallback
+ return Nothing
+ else return $ Just fallback
+ else return $ Just command
+ where
+ hook = hookname ++ "-" ++ action ++ "-hook"
+ hookfallback = hookname ++ "-hook"
+
+runHook :: HookName -> Action -> Key -> Maybe FilePath -> Annex Bool -> Annex Bool
+runHook hook action k f a = maybe (return False) run =<< lookupHook hook action
+ where
+ run command = do
+ showOutput -- make way for hook output
+ ifM (liftIO $ boolSystemEnv "sh" [Param "-c", Param command] =<< hookEnv action k f)
+ ( a
+ , do
+ warning $ hook ++ " hook exited nonzero!"
+ return False
+ )
+
+store :: HookName -> Key -> AssociatedFile -> MeterUpdate -> Annex Bool
+store h k _f _p = sendAnnex k (void $ remove h k) $ \src ->
+ runHook h "store" k (Just src) $ return True
+
+storeEncrypted :: HookName -> [CommandParam] -> (Cipher, Key) -> Key -> MeterUpdate -> Annex Bool
+storeEncrypted h gpgOpts (cipher, enck) k _p = withTmp enck $ \tmp ->
+ sendAnnex k (void $ remove h enck) $ \src -> do
+ liftIO $ encrypt gpgOpts cipher (feedFile src) $
+ readBytes $ L.writeFile tmp
+ runHook h "store" enck (Just tmp) $ return True
+
+retrieve :: HookName -> Key -> AssociatedFile -> FilePath -> MeterUpdate -> Annex Bool
+retrieve h k _f d _p = runHook h "retrieve" k (Just d) $ return True
+
+retrieveCheap :: HookName -> Key -> FilePath -> Annex Bool
+retrieveCheap _ _ _ = return False
+
+retrieveEncrypted :: HookName -> (Cipher, Key) -> Key -> FilePath -> MeterUpdate -> Annex Bool
+retrieveEncrypted h (cipher, enck) _ f _p = withTmp enck $ \tmp ->
+ runHook h "retrieve" enck (Just tmp) $ liftIO $ catchBoolIO $ do
+ decrypt cipher (feedFile tmp) $
+ readBytes $ L.writeFile f
+ return True
+
+remove :: HookName -> Key -> Annex Bool
+remove h k = runHook h "remove" k Nothing $ return True
+
+checkPresent :: Git.Repo -> HookName -> Key -> Annex (Either String Bool)
+checkPresent r h k = do
+ showAction $ "checking " ++ Git.repoDescribe r
+ v <- lookupHook h action
+ liftIO $ catchMsgIO $ check v
+ where
+ action = "checkpresent"
+ findkey s = key2file k `elem` lines s
+ check Nothing = error $ action ++ " hook misconfigured"
+ check (Just hook) = do
+ env <- hookEnv action k Nothing
+ findkey <$> readProcessEnv "sh" ["-c", hook] env