git-annex/Annex/HashObject.hs

{- git hash-object interface, with handle automatically stored in the Annex monad
 -
 - Copyright 2016 Joey Hess <id@joeyh.name>
 -
 - Licensed under the GNU AGPL version 3 or higher.
 -}

module Annex.HashObject (
	hashFile,
	hashBlob,
	hashObjectHandle,
	hashObjectStop,
) where

import Annex.Common
import qualified Git.HashObject
import qualified Annex
import Git.Types

hashObjectHandle :: Annex Git.HashObject.HashObjectHandle
hashObjectHandle = maybe startup return =<< Annex.getState Annex.hashobjecthandle
  where
	startup = do
		h <- inRepo $ Git.HashObject.hashObjectStart True
		Annex.changeState $ \s -> s { Annex.hashobjecthandle = Just h }
		return h

hashObjectStop :: Annex ()
hashObjectStop = maybe noop stop =<< Annex.getState Annex.hashobjecthandle
  where
	stop h = do
		liftIO $ Git.HashObject.hashObjectStop h
		Annex.changeState $ \s -> s { Annex.hashobjecthandle = Nothing }
		return ()

hashFile :: FilePath -> Annex Sha
hashFile f = do
	h <- hashObjectHandle
	liftIO $ Git.HashObject.hashFile h f

{- Note that the content will be written to a temp file.
 - So it may be faster to use Git.HashObject.hashObject for large
 - blob contents. -}
hashBlob :: Git.HashObject.HashableBlob b => b -> Annex Sha
hashBlob content = do
	h <- hashObjectHandle
	liftIO $ Git.HashObject.hashBlob h content