flipperzero-firmware/scripts/version.py

132 lines
4.0 KiB
Python

#!/usb/bin/env python3
from flipper.app import App
import subprocess
import os
import json
from datetime import date
class GitVersion:
def __init__(self, source_dir):
self.source_dir = source_dir
def get_version_info(self):
commit = self._exec_git("rev-parse --short HEAD") or "unknown"
dirty = False
try:
self._exec_git("diff --quiet")
except subprocess.CalledProcessError as e:
if e.returncode == 1:
dirty = True
# If WORKFLOW_BRANCH_OR_TAG is set in environment, is has precedence
# (set by CI)
branch = (
os.environ.get("WORKFLOW_BRANCH_OR_TAG", None)
or self._exec_git("rev-parse --abbrev-ref HEAD")
or "unknown"
)
branch_num = self._exec_git("rev-list --count HEAD") or "n/a"
try:
version = self._exec_git("describe --tags --abbrev=0 --exact-match")
except subprocess.CalledProcessError:
version = "unknown"
return {
"GIT_COMMIT": commit,
"GIT_BRANCH": branch,
"GIT_BRANCH_NUM": branch_num,
"VERSION": version,
"BUILD_DIRTY": dirty and 1 or 0,
}
def _exec_git(self, args):
cmd = ["git"]
cmd.extend(args.split(" "))
return (
subprocess.check_output(cmd, cwd=self.source_dir, stderr=subprocess.STDOUT)
.strip()
.decode()
)
class Main(App):
def init(self):
self.subparsers = self.parser.add_subparsers(help="sub-command help")
# generate
self.parser_generate = self.subparsers.add_parser(
"generate", help="Generate version header"
)
self.parser_generate.add_argument("-o", dest="outdir", required=True)
self.parser_generate.add_argument(
"-t",
dest="target",
type=int,
help="hardware target",
required=True,
)
self.parser_generate.add_argument("--dir", dest="sourcedir", required=True)
self.parser_generate.set_defaults(func=self.generate)
def generate(self):
current_info = GitVersion(self.args.sourcedir).get_version_info()
current_info.update(
{
"BUILD_DATE": date.today().strftime("%d-%m-%Y"),
"TARGET": self.args.target,
}
)
version_values = []
for key in current_info:
val = current_info[key]
if isinstance(val, str):
val = f'"{val}"'
version_values.append(f"#define {key} {val}")
new_version_info_fmt = "\n".join(version_values) + "\n"
current_version_info = None
version_header_name = os.path.join(self.args.outdir, "version.inc.h")
version_json_name = os.path.join(self.args.outdir, "version.json")
try:
with open(version_header_name, "r") as file:
current_version_info = file.read()
except EnvironmentError as e:
if self.args.debug:
print(e)
if current_version_info != new_version_info_fmt:
if self.args.debug:
print("old: ", current_version_info)
print("new: ", new_version_info_fmt)
with open(version_header_name, "w", newline="\n") as file:
file.write(new_version_info_fmt)
# os.utime("../lib/toolbox/version.c", None)
print("Version information updated")
else:
if self.args.debug:
print("Version information hasn't changed")
version_json = {
"firmware_build_date": current_info["BUILD_DATE"],
"firmware_commit": current_info["GIT_COMMIT"],
"firmware_branch": current_info["GIT_BRANCH"],
"firmware_target": current_info["TARGET"],
}
with open(version_json_name, "w", newline="\n") as file:
json.dump(version_json, file, indent=4)
return 0
if __name__ == "__main__":
Main()()