diff options
Diffstat (limited to 'gitlint/hooks.py')
-rw-r--r-- | gitlint/hooks.py | 62 |
1 files changed, 62 insertions, 0 deletions
diff --git a/gitlint/hooks.py b/gitlint/hooks.py new file mode 100644 index 0000000..fc4dc4e --- /dev/null +++ b/gitlint/hooks.py @@ -0,0 +1,62 @@ +import io +import shutil +import os +import stat + +from gitlint.utils import DEFAULT_ENCODING +from gitlint.git import git_hooks_dir + +COMMIT_MSG_HOOK_SRC_PATH = os.path.join(os.path.dirname(os.path.realpath(__file__)), "files", "commit-msg") +COMMIT_MSG_HOOK_DST_PATH = "commit-msg" +GITLINT_HOOK_IDENTIFIER = "### gitlint commit-msg hook start ###\n" + + +class GitHookInstallerError(Exception): + pass + + +class GitHookInstaller(object): + """ Utility class that provides methods for installing and uninstalling the gitlint commitmsg hook. """ + + @staticmethod + def commit_msg_hook_path(lint_config): + return os.path.join(git_hooks_dir(lint_config.target), COMMIT_MSG_HOOK_DST_PATH) + + @staticmethod + def _assert_git_repo(target): + """ Asserts that a given target directory is a git repository """ + hooks_dir = git_hooks_dir(target) + if not os.path.isdir(hooks_dir): + raise GitHookInstallerError(u"{0} is not a git repository.".format(target)) + + @staticmethod + def install_commit_msg_hook(lint_config): + GitHookInstaller._assert_git_repo(lint_config.target) + dest_path = GitHookInstaller.commit_msg_hook_path(lint_config) + if os.path.exists(dest_path): + raise GitHookInstallerError( + u"There is already a commit-msg hook file present in {0}.\n".format(dest_path) + + u"gitlint currently does not support appending to an existing commit-msg file.") + + # copy hook file + shutil.copy(COMMIT_MSG_HOOK_SRC_PATH, dest_path) + # make hook executable + st = os.stat(dest_path) + os.chmod(dest_path, st.st_mode | stat.S_IEXEC) + + @staticmethod + def uninstall_commit_msg_hook(lint_config): + GitHookInstaller._assert_git_repo(lint_config.target) + dest_path = GitHookInstaller.commit_msg_hook_path(lint_config) + if not os.path.exists(dest_path): + raise GitHookInstallerError(u"There is no commit-msg hook present in {0}.".format(dest_path)) + + with io.open(dest_path, encoding=DEFAULT_ENCODING) as fp: + lines = fp.readlines() + if len(lines) < 2 or lines[1] != GITLINT_HOOK_IDENTIFIER: + msg = u"The commit-msg hook in {0} was not installed by gitlint (or it was modified).\n" + \ + u"Uninstallation of 3th party or modified gitlint hooks is not supported." + raise GitHookInstallerError(msg.format(dest_path)) + + # If we are sure it's a gitlint hook, go ahead and remove it + os.remove(dest_path) |