#! /usr/bin/env python3

####################################################################################################

from __future__ import print_function

import pygit2 as git
import os
import sys

####################################################################################################

if len(sys.argv) != 2:
    print("Give a repository path")
    sys.exit(1)

source_path = sys.argv[1]
source_path = os.path.realpath(os.path.expandvars(source_path))

print("Repository:", source_path)

####################################################################################################

def pretty_print_commit(commit):

    message_template = """
==================================================
Commit:
  hex: %s
  author: %s
  committer: %s
  parents: %s
  message: %s
"""

    print(message_template % (commit.hex,
                              commit.author.name,
                              commit.committer.name,
                              str([x.hex for x in commit.parents]),
                              commit.message,
                              ))

    # print(commit.tree)

####################################################################################################

# def walk_from_commit(commit):

#     pretty_print_commit(commit)
#     number_of_parents = len(commit.parents)
#     if number_of_parents == 1:
#         show_diff(commit.parents[0].tree, commit.tree)
#     elif number_of_parents > 1:
#         print("More than one parents")
#     for parent_commit in commit.parents:
#         walk_from_commit(parent_commit)

####################################################################################################

def walk_from_tree(tree):

    for entry in tree:
        print(entry.hex, entry.name, entry.attributes)
        obj = entry.to_object()
        if obj.type == git.GIT_OBJ_BLOB:
            print(entry.to_object().data)
        elif obj.type == git.GIT_OBJ_TREE:
            walk_from_tree(obj)

####################################################################################################

# translate_diff = {
#     git.GIT_DELTA_UNMODIFIED:'Unmodified',
#     git.GIT_DELTA_ADDED:'Added',
#     git.GIT_DELTA_DELETED:'Deleted',
#     git.GIT_DELTA_MODIFIED:'Modified',
#     git.GIT_DELTA_RENAMED:'Renamed',
#     git.GIT_DELTA_COPIED:'Copied',
#     git.GIT_DELTA_IGNORED:'Ignored',
#     git.GIT_DELTA_UNTRACKED:'Untracked',
#     }

# def show_diff(tree1, tree2):

#     diff = tree1.diff(tree2)
#     for file_diff in diff.changes['files']:
#         print(' '*2, translate_diff[file_diff[2]], file_diff[0])

####################################################################################################

repository_path = git.discover_repository(source_path)
repository = git.Repository(repository_path)

# head = repository.lookup_reference('HEAD').resolve()
# head = repository.head
# head_commit = repository[head.target]

# pretty_print_commit(head_commit)

# for commit in repository.walk(head_commit.id, git.GIT_SORT_TIME):
#     pretty_print_commit(commit)

# walk_from_commit(head_commit)
# walk_from_tree(head_commit.tree)

translate_status = {
 git.GIT_STATUS_CURRENT:'Current',
 git.GIT_STATUS_IGNORED:'Ignored',
 git.GIT_STATUS_INDEX_DELETED:'Index Deleted',
 git.GIT_STATUS_INDEX_MODIFIED:'Index Modified',
 git.GIT_STATUS_INDEX_NEW:'Index New',
 git.GIT_STATUS_WT_DELETED:'Wt Deleted',
 git.GIT_STATUS_WT_MODIFIED:'Wt Modified',
 git.GIT_STATUS_WT_NEW:'Wt New',
}

print("\nStatus:")
git_status = repository.status()
for filepath, status in git_status.items():
    print(translate_status.get(status, status), filepath)

print("\nDiff:")
#diff = repository.diff() # not stagged
#diff = repository.diff('HEAD', cached=True) # stagged
diff = repository.diff('HEAD') # all
for patch in diff:
    print(patch.old_file_path, patch.new_file_path, patch.status)
    
# print("\nDiff:")
# commit = repository.revparse_single('HEAD^')
# diff = repository.diff(head_commit, commit)
# for patch in diff:
#     print(patch.old_file_path, patch.new_file_path, patch.status)

####################################################################################################
#
# End
#
####################################################################################################