/
generate_headers.py
115 lines (96 loc) · 3.53 KB
/
generate_headers.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
# vim: set ts=8 sts=2 sw=2 tw=99 et:
import re
import os, sys
import subprocess
argv = sys.argv[1:]
if len(argv) < 2:
sys.stderr.write('Usage: generate_headers.py <source_path> <output_folder>\n')
sys.exit(1)
SourceFolder = os.path.abspath(os.path.normpath(argv[0]))
OutputFolder = os.path.normpath(argv[1])
class FolderChanger:
def __init__(self, folder):
self.old = os.getcwd()
self.new = folder
def __enter__(self):
if self.new:
os.chdir(self.new)
def __exit__(self, type, value, traceback):
os.chdir(self.old)
def run_and_return(argv):
# Python 2.6 doesn't have check_output.
if hasattr(subprocess, 'check_output'):
text = subprocess.check_output(argv)
if str != bytes:
text = str(text, 'utf-8')
else:
p = subprocess.Popen(argv, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
output, ignored = p.communicate()
rval = p.poll()
if rval:
raise subprocess.CalledProcessError(rval, argv)
text = output.decode('utf8')
return text.strip()
def get_git_version():
revision_count = run_and_return(['git', 'rev-list', '--count', 'HEAD'])
revision_hash = run_and_return(['git', 'log', '--pretty=format:%h:%H', '-n', '1'])
shorthash, longhash = revision_hash.split(':')
return revision_count, shorthash, longhash
def output_version_headers():
with FolderChanger(SourceFolder):
count, shorthash, longhash = get_git_version()
with open(os.path.join(SourceFolder, 'product.version')) as fp:
contents = fp.read().rstrip()
m = re.match('(\d+)\.(\d+)\.(\d+)-?(.*)', contents)
if m == None:
raise Exception('Could not detremine product version')
major, minor, release, tag = m.groups()
product = "{0}.{1}.{2}.{3}".format(major, minor, release, count)
fullstring = product
if tag != "":
fullstring += "-{0}".format(tag)
with open(os.path.join(OutputFolder, 'amxmodx_version_auto.h'), 'w') as fp:
fp.write("""
#ifndef _AMXMODX_AUTO_VERSION_INFORMATION_H_
#define _AMXMODX_AUTO_VERSION_INFORMATION_H_
#define AMXX_BUILD_TAG \"{0}\"
#define AMXX_BUILD_CSET \"{1}\"
#define AMXX_BUILD_MAJOR \"{2}\"
#define AMXX_BUILD_MINOR \"{3}\"
#define AMXX_BUILD_RELEASE \"{4}\"
#define AMXX_BUILD_LOCAL_REV \"{6}\"
#define AMXX_BUILD_UNIQUEID AMXX_BUILD_LOCAL_REV \":\" AMXX_BUILD_CSET
#define AMXX_VERSION_STRING \"{5}\"
#define AMXX_VERSION_FILE {2},{3},{4},{6}
#endif // _AMXMODX_AUTO_VERSION_INFORMATION_H_
""".format(tag, shorthash, major, minor, release, fullstring, count))
version_num = int(major) * 100 + int(minor) * 10 + int(release)
with open(os.path.join(OutputFolder, 'amxmodx_version.inc'), 'w') as fp:
fp.write("""
#if defined _amxmodx_version_included
#endinput
#endif
#define _amxmodx_version_included
#define AMXX_VERSION_TAG \"{tag}\"
#define AMXX_VERSION_CSET \"{shorthash}\"
#define AMXX_VERSION_MAJOR \"{major}\"
#define AMXX_VERSION_MAJOR_NUM {major}
#define AMXX_VERSION_MINOR \"{minor}\"
#define AMXX_VERSION_MINOR_NUM {minor}
#define AMXX_VERSION_RELEASE \"{release}\"
#define AMXX_VERSION_LOCAL_REV_NUM {count}
#define AMXX_VERSION_LOCAL_REV \"{count}\"
#define AMXX_VERSION {major}.{minor}{release}
#define AMXX_VERSION_NUM {version_num}
stock const AMXX_VERSION_STR[] = "{fullstring}";
""".format(
tag = tag,
shorthash = shorthash,
major = major,
minor = minor,
release = release,
count = count,
version_num = version_num,
fullstring = fullstring
))
output_version_headers()