module Propellor.Git where

import Utility.Process
import Utility.Exception
import Utility.Directory
import Utility.Misc
import Utility.PartialPrelude

import Data.Maybe
import Control.Applicative
import Prelude

getCurrentBranch :: IO String
getCurrentBranch = takeWhile (/= '\n')
	<$> readProcess "git" ["symbolic-ref", "--short", "HEAD"]

getCurrentBranchRef :: IO String
getCurrentBranchRef = takeWhile (/= '\n')
	<$> readProcess "git" ["symbolic-ref", "HEAD"]

getCurrentGitSha1 :: String -> IO String
getCurrentGitSha1 branchref = takeWhile (/= '\n')
	<$> readProcess "git" ["show-ref", "--hash", branchref]

hasOrigin :: IO Bool
hasOrigin = hasRemote "origin"

hasRemote :: String -> IO Bool
hasRemote remotename = catchDefaultIO False $ do
	rs <- lines <$> readProcess "git" ["remote"]
	return $ remotename `elem` rs

remoteUrl :: String -> IO (Maybe String)
remoteUrl remotename = catchDefaultIO Nothing $ headMaybe . lines
	<$> readProcess "git" ["config", "remote." ++ remotename ++ ".url"]

hasGitRepo :: IO Bool
hasGitRepo = doesFileExist ".git/HEAD"

type Version = [Int]

gitVersion :: IO Version
gitVersion = extract <$> readProcess "git" ["--version"]
  where
	extract s = case lines s of
		[] -> []
		(l:_) -> mapMaybe readish $ segment (== '.') $
			unwords $ drop 2 $ words l