@@ -1,3 +1,6 @@
1+{-# LANGUAGE LambdaCase #-}
2+{-# LANGUAGE MultiWayIf #-}
2-- Module : $Header$
5-- Module : $Header$
3-- Copyright : (c) 2015 Adam C. Foltzer
6-- Copyright : (c) 2015 Adam C. Foltzer
@@ -29,9 +32,16 @@
29-- > % cabal exec runhaskell Example.hs
32-- > % cabal exec runhaskell Example.hs
30-- > Example.hs: [panic master@2ae047ba5e4a6f0f3e705a43615363ac006099c1 (Mon Jan 11 11:50:59 2016 -0800) (14 commits in HEAD) (uncommitted files present)] oh no!
33-- > Example.hs: [panic master@2ae047ba5e4a6f0f3e705a43615363ac006099c1 (Mon Jan 11 11:50:59 2016 -0800) (14 commits in HEAD) (uncommitted files present)] oh no!
32−module Development.GitRev (gitHash, gitBranch, gitDirty, gitCommitCount, gitCommitDate) where
35+module Development.GitRev
34−import Control.Applicative
35import Control.Exception
45import Control.Exception
36import Control.Monad
46import Control.Monad
37import Data.Maybe
47import Data.Maybe
@@ -42,6 +52,9 @@ import System.Exit
42import System.FilePath
52import System.FilePath
43import System.Process
53import System.Process
45-- | Run git with the given arguments and no stdin, returning the
58-- | Run git with the given arguments and no stdin, returning the
46-- stdout output. If git isn't available or something goes wrong,
59-- stdout output. If git isn't available or something goes wrong,
47-- return the second argument.
60-- return the second argument.
@@ -53,7 +66,7 @@ runGit args def useIdx = do
53 if gitFound
66 if gitFound
55 -- a lot of bookkeeping to record the right dependencies
68 -- a lot of bookkeeping to record the right dependencies
56− pwd <- runIO getCurrentDirectory
69+ pwd <- runIO getDotGit
57 let hd = pwd </> ".git" </> "HEAD"
70 let hd = pwd </> ".git" </> "HEAD"
58 index = pwd </> ".git" </> "index"
71 index = pwd </> ".git" </> "index"
59 packedRefs = pwd </> ".git" </> "packed-refs"
72 packedRefs = pwd </> ".git" </> "packed-refs"
@@ -61,8 +74,7 @@ runGit args def useIdx = do
61 when hdExists $ do
74 when hdExists $ do
62 -- the HEAD file either contains the hash of a detached head
75 -- the HEAD file either contains the hash of a detached head
63 -- or a pointer to the file that contains the hash of the head
76 -- or a pointer to the file that contains the hash of the head
64− hdRef <- runIO $ readFile hd
77+ splitAt 5 `fmap` runIO (readFile hd) >>= \case
65− case splitAt 5 hdRef of
66 -- pointer to ref
78 -- pointer to ref
67 ("ref: ", relRef) -> do
79 ("ref: ", relRef) -> do
68 let ref = pwd </> ".git" </> relRef
80 let ref = pwd </> ".git" </> relRef
@@ -85,6 +97,36 @@ runGit args def useIdx = do
85 ExitFailure _ -> return def
97 ExitFailure _ -> return def
86 else return def
98 else return def
100+-- | Determine where our @.git@ directory is, in case we're in a
102+getDotGit :: IO FilePath
105+ let dotGit = pwd </> ".git"
106+ oops = return dotGit -- it's gonna fail, that's fine
107+ isDir <- doesDirectoryExist dotGit
108+ isFile <- doesFileExist dotGit
109+ if | isDir -> return dotGit
110+ | not isFile -> oops
112+ splitAt 8 `fmap` readFile dotGit >>= \case
113+ ("gitdir: ", relDir) -> do
114+ isRelDir <- doesDirectoryExist relDir
120+-- | Get the root directory of the Git repo.
121+getGitRoot :: IO FilePath
123+ pwd <- getCurrentDirectory
125+ readProcessWithExitCode "git" ["rev-parse", "--show-toplevel"] ""
127+ ExitSuccess -> return $ takeWhile (/= '\n') out
128+ ExitFailure _ -> return pwd -- later steps will fail, that's fine
88-- | Type to flag if the git index is used or not in a call to runGit
130-- | Type to flag if the git index is used or not in a call to runGit
89data IndexUsed = IdxUsed -- ^ The git index is used
131data IndexUsed = IdxUsed -- ^ The git index is used
90 | IdxNotUsed -- ^ The git index is /not/ used
132 | IdxNotUsed -- ^ The git index is /not/ used
@@ -103,6 +145,12 @@ gitBranch :: ExpQ
103gitBranch =
145gitBranch =
104 stringE =<< runGit ["rev-parse", "--abbrev-ref", "HEAD"] "UNKNOWN" IdxNotUsed
146 stringE =<< runGit ["rev-parse", "--abbrev-ref", "HEAD"] "UNKNOWN" IdxNotUsed
148+-- | Return the long git description for the current git commit, or
149+-- @UNKNOWN@ if not in a git repository.
152+ stringE =<< runGit ["describe", "--long", "--always"] "UNKNOWN" IdxNotUsed
106-- | Return @True@ if there are non-committed files present in the
154-- | Return @True@ if there are non-committed files present in the
107-- repository
155-- repository
108gitDirty :: ExpQ
156gitDirty :: ExpQ
@@ -112,6 +160,15 @@ gitDirty = do
112 "" -> conE falseName
160 "" -> conE falseName
113 _ -> conE trueName
161 _ -> conE trueName
163+-- | Return @True@ if there are non-commited changes to tracked files
164+-- present in the repository
165+gitDirtyTracked :: ExpQ
167+ output <- runGit ["status", "--porcelain","--untracked-files=no"] "" IdxUsed
169+ "" -> conE falseName
115-- | Return the number of commits in the current head
172-- | Return the number of commits in the current head
116gitCommitCount :: ExpQ
173gitCommitCount :: ExpQ
117gitCommitCount =
174gitCommitCount =