#!/bin/python3
import os
from os import environ as env
import sys
import shutil
import subprocess

binary_filename = os.path.abspath(sys.argv[1])
master_repository_directory = os.path.abspath(sys.argv[2])
data_repository_directory = os.path.abspath(sys.argv[3])
directory = sys.argv[4]

os.chdir(master_repository_directory)

filename = os.path.basename(binary_filename)
commit_hash = subprocess.check_output(["git", "rev-parse", "--verify", "--short", "HEAD"]).decode("utf-8").strip()
commit_subject = subprocess.check_output(["git", "log", "-1", "--pretty=format:'%s'"]).decode("utf-8").strip()

is_tag = env["GITHUB_EVENT_NAME"] == "push" and env["GITHUB_REF"].startswith("refs/tags")
if not is_tag:
    is_pr = env["GITHUB_REF"].startswith("refs/pull")
    if is_pr:
        # Pull Request - prN (pr1)
        middle = "pr" + env["GITHUB_REF"].split("/")[2]
    else:
        # Latest commit - short hash (20f2448)
        middle = commit_hash
    filename_split = filename.split("-")
    filename = "-".join([*filename_split[:2], middle, *filename_split[2:]])

# Set author info to the latest commit author
author_name = subprocess.check_output(["git", "log", "-1", "--pretty=format:'%an'"]).decode("utf-8")
author_email = subprocess.check_output(["git", "log", "-1", "--pretty=format:'%ae'"]).decode("utf-8")

# Move file
os.chdir(data_repository_directory)
os.makedirs(directory, exist_ok=True)
shutil.copy(binary_filename, os.path.join(directory, filename))

# Push changes
subprocess.check_call(["git", "config", "user.name", author_name])
subprocess.check_call(["git", "config", "user.email", author_email])
subprocess.check_call(["git", "add", os.path.join(directory, filename)])
subprocess.check_call(["git", "commit", "-m", f"Add binary for {commit_hash}: \"{commit_subject}\""])
subprocess.check_call(["git", "push"])