| #!/usr/bin/python3 | |
| # TODO: option to make it to ask for a change type for each file being added | |
| # into the commit | |
| # TODO: Snippets | |
| from simple_term_menu import TerminalMenu | |
| from command import command | |
| from os.path import split | |
| import os | |
| from gitshort import get_staged | |
| import argparse | |
| COMMITMSG = ".commitmsg" | |
| EDITOR = "vim" | |
| def splitext(f): | |
| hidden = (f[0] == '.') | |
| res = list(f.split('.')) | |
| res = list(map(str.strip, res)) | |
| if hidden: | |
| res = ["%s%s" % (".", res[1])] + res[2:] | |
| return res | |
| def get_units(): | |
| staged = get_staged() | |
| units = set([splitext(s)[0] for s in staged]) # Echo.cpp, Echo.hpp -> Echo | |
| print(units) | |
| return units | |
| def get_modulemap(unit, modulemap=dict()): | |
| unit_initial = unit | |
| basename, unit = split(unit) | |
| basename = basename.split('/') | |
| basename = list(reversed(basename)) | |
| if len(basename) > 1: | |
| print(f"Unit {unit}. Select module") | |
| module = basename[TerminalMenu(basename).show()] | |
| elif len(basename) == 1: | |
| module = basename[0] | |
| else: | |
| module = "" | |
| if module not in modulemap.keys(): | |
| modulemap[module] = set() | |
| modulemap[module] = set.union(modulemap[module], {unit_initial}) | |
| return modulemap | |
| def get_commit_type(): | |
| print("Select commit type...") | |
| types = ["[i]Impl", "[e]Enh", "[b]Bug", "[c]Cascade", "[o]Opt", "[r]Ref", "[u]Build"] | |
| return types[TerminalMenu(types).show()][3:] | |
| def fmt(modulemap, commit_type, stem=False): | |
| modstrings = [] | |
| for k, v in modulemap.items(): | |
| if len(v) > 1 and stem: | |
| vals = "*" | |
| else: | |
| vals = ','.join(list(map(lambda p: split(p)[1], v))) | |
| modstrings.append(k + ":" + vals) | |
| return '[' + ' '.join(modstrings) + '] ' + commit_type + " | " | |
| def command_commit(prefix): | |
| with open(COMMITMSG, 'w') as f: | |
| f.write(prefix) | |
| command("vim " + COMMITMSG) | |
| command("git commit --file " + COMMITMSG) | |
| def _get_args(): | |
| parser = argparse.ArgumentParser() | |
| parser.add_argument("--sep", action="store_true", help="Push each dir:module pair as a separate commit under the same name") | |
| parser.add_argument("--stem", action="store_true", help="If true, module components will not be comma-enumerated" | |
| ", but replaced w/ * symbol, if there is more than 1 element in each module") | |
| p = parser.parse_args() | |
| return p | |
| def get_commit_message(): | |
| temp = ".tempmsg" | |
| with open(temp, 'w') as f: | |
| pass | |
| os.system("%s %s" % (EDITOR, temp)) | |
| with open(temp, 'r') as f: | |
| res = f.read() | |
| assert len(res.strip()) > 0 | |
| os.remove(temp) | |
| return res | |
| def main(): | |
| args = _get_args() | |
| try: | |
| modulemap = dict() | |
| units = get_units() | |
| for unit in units: | |
| modulemap = get_modulemap(unit, modulemap) | |
| print(modulemap) | |
| commit_type = get_commit_type() | |
| if not args.sep: | |
| formatted = fmt(modulemap, commit_type, args.stem) | |
| command_commit(formatted) | |
| else: | |
| command("git reset") | |
| commit_msg = get_commit_message() | |
| for k, vs in modulemap.items(): | |
| for v in vs: | |
| print("adding %s" % (str(v))) | |
| formatted = fmt({k: {v}}, commit_type, args.stem) | |
| command("git add %s.*" % v) | |
| command("git commit -m \'%s%s\'" % (formatted, commit_msg)) | |
| except Exception as e: | |
| print(e) | |
| finally: | |
| command("rm -f " + COMMITMSG) | |
| if __name__ == "__main__": | |
| main() |