#!/usr/bin/env python

import re

import subprocess

import sys

OK_C = "\033[92m"

WARN_C = "\033[93m"

FAIL_C = "\033[91m"

END_C = "\033[0m"

JIRA_REGEX = "(SPII|PCRM|NEMS|KF|IGWAY|NRL|APM|APMSPII|SPY3|SPINECLI)\-(\d+)" # Example match: SPII-12345

FEATURE_REGEX = (

"feature\/\w+\-(\w+\-\d+)\-" # Example match: feature/das-SPII-12345-foo

)

OLD_FEATURE_REGEX = "feature\/\w+\-(\d+)\-" # Example match: feature/das-12345-foo

sys.stdin = open("/dev/tty") # Ensure that input can be taken by hook

MSG_FILE = sys.argv[1] # commit-msg hook provides file name as first arg

def append_to_commit_message(jira_id):

print("Appended {0} to commit message.".format(jira_id))

with open(MSG_FILE, "a") as f:

f.write("{0}".format(jira_id))

def prepend_to_commit_message(jira_id):

print("Prepended {} to commit message.".format(jira_id))

with open(MSG_FILE, "r") as f:

text = f.read()

text = "{} {}".format(jira_id, text)

with open(MSG_FILE, "w") as f:

f.write(text)

# Check if JIRA ID already exists in commit message

print("I am doing something")

with open(MSG_FILE, "r") as f:

msg = ""

for line in f:

if not line.startswith("#"):

msg += line.rstrip()

if msg == "":

# No commit message let Git fail the commot

sys.exit(0)

commitMatch = re.search(JIRA_REGEX, msg)

if commitMatch:

print(

"JIRA {0}{1}-{2}{3} referenced by commit message.".format(

OK_C, commitMatch.group(1), commitMatch.group(2), END_C

)

)

sys.exit(0)

print(WARN_C + "No JIRA ID referenced by commit message." + END_C)

# Check if KEY-n+ can be used from branch name

branch = subprocess.check_output("git rev-parse --abbrev-ref HEAD", shell=True)

branchMatch = re.search(FEATURE_REGEX, str(branch))

if branchMatch:

response = input(

"Use {0}{1}{2} from branch name? (y/n) ".format(

OK_C, branchMatch.group(1), END_C

)

)

if response.strip().lower().startswith("y"):

prepend_to_commit_message(branchMatch.group(1))

sys.exit(0)

oldBranchMatch = re.search(OLD_FEATURE_REGEX, str(branch))

if oldBranchMatch:

jira_id = "SPII-{}".format(oldBranchMatch.group(1))

response = input(

"Use {0}{1}{2} from branch name? (y/n) ".format(OK_C, jira_id, END_C)

)

if response.strip().lower().startswith("y"):

prepend_to_commit_message(jira_id)

sys.exit(0)

# All else fails ask what they want to use

jira_id = input("Please enter the JIRA ID (for example SPII-1234 or NEMS-123): ")

if not re.search(JIRA_REGEX, jira_id):

print(FAIL_C + "Did not provide valid JIRA ID. Commit Aborted." + END_C)

sys.exit(1)

else:

prepend_to_commit_message(jira_id)

sys.exit(0)