profile.py

import os
import sys
import string
import datetime
import pytz
import socket
import json
import platform
import subprocess
import hashlib
import base64

PYTHON_VERSION = sys.version_info[0]
base_path = os.path.abspath(os.path.dirname(__file__))
base_dist_path = os.path.join(base_path, "dist")


def generate_app_profile():
    profile_template_file = os.path.join(base_path, "src/apps/profile-template.js")
    package_json_file = os.path.join(base_path, "package.json")

    with open(package_json_file) as f:
        package = json.loads(f.read())

    with open(profile_template_file) as f:
        profile_template = string.Template(f.read())

    app_name = package["config"]["app"]
    profile_name = os.path.join(base_path, "src/apps", "{}-profile.js".format(app_name))

    now = datetime.datetime.now(pytz.timezone("Australia/Perth"))

    package.update(package["config"])
    package["distributionType"] = sys.argv[1] if len(sys.argv) >= 2 else "unknown"

    vendor_file = os.path.join(base_dist_path, package["distributionType"], "vendor.js")

    if not os.path.exists(vendor_file):
        raise Exception("Vendor file({}) is missing".format(vendor_file))

    m = hashlib.md5()
    with open(vendor_file, "rb") as f:
        m.update(f.read())

    vendor_md5 = base64.urlsafe_b64encode(m.digest())
    # Python 2/3 compatibility:
    if PYTHON_VERSION == 3:
        vendor_md5 = vendor_md5.rstrip(b"=")
    else:
        vendor_md5 = vendor_md5.rstrip("=")

    package.update(
        {
            "build_datetime": now.strftime("%Y-%m-%d %H:%M:%S %Z(%z)"),
            "build_date": now.strftime("%Y-%m-%d %Z(%z)"),
            "build_time": now.strftime("%H-%M-%S %Z(%z)"),
            "build_platform": platform.system(),
            "build_host": socket.gethostname(),
            "vendor_md5": vendor_md5,
        }
    )

    # get the latest git commit.
    latest_commit = subprocess.check_output(["git", "log", "-n", "1"]).splitlines()
    commit_info = {}
    for line in latest_commit:
        for k, v in [
            (b"commit", "commit"),
            (b"Author:", "commit_author"),
            (b"Merge:", ""),
            (b"Date:", "commit_date"),
            (b"", "commit_message"),
        ]:
            if line[: len(k)] == k:
                if v and line[len(k) :].strip():
                    if v in commit_info:
                        commit_info[v] = "{}\\n{}".format(
                            commit_info[v], line[len(k) :].strip().decode("utf-8")
                        )
                    else:
                        commit_info[v] = line[len(k) :].strip().decode("utf-8")
                break
    if "commit" in commit_info:
        commit_info["commit"] = commit_info["commit"][:7]

    package.update(commit_info)

    # get the branch info
    branch = (
        [
            b
            for b in subprocess.check_output(["git", "branch"]).splitlines()
            if b.strip().startswith(b"*")
        ][0]
        .strip()[1:]
        .strip()
        .decode("utf-8")
    )

    package["repository_branch"] = branch

    # tranform value to json string
    for k, v in package.items():
        # Python 2/3 compatibility: forcibly convert byte characters back to a string.
        if isinstance(v, bytes):
            v = "".join(chr(i) for i in v)
        package[k] = json.dumps(v)

    for key, val in package.items():
        print("{}: {}".format(key, val))

    profile = profile_template.safe_substitute(package)

    print("\nPROFILE:")
    print(profile)

    with open(profile_name, "w") as f:
        f.write(profile)


if __name__ == "__main__":
    generate_app_profile()