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
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
|
#!/usr/bin/env python
import string, sys, os.path, getopt
escapes = {}
escapes['\a'] = r'\a'
escapes['\b'] = r'\b'
escapes['\f'] = r'\f'
escapes['\n'] = r'\n'
escapes['\r'] = r'\r'
#escapes['\t'] = r'\t'
# Let's replace tabs with four spaces
# so the text looks nicely indented in the C source
escapes['\t'] = r' '
escapes['\v'] = r'\v'
# escapes['\''] = r'\''
escapes['\"'] = r'\"'
escapes['\\'] = r'\\'
def escape(c):
if c in escapes:
return (escapes[c], False)
elif c not in string.printable:
return ("\\x%x" % ord(c), True)
else:
return (c, False)
def stringize(line):
newline = '"'
was_escape = False
for c in line:
# Avoid an issue where characters after a hexadecimal escape are treated as part of that escape
# by adding two quotes
if was_escape and c in string.hexdigits:
newline += '""'
chars, was_escape = escape(c)
newline += chars
newline += '"'
return newline
class cfunc:
def __init__(self, type, name, lines):
self.type = type
self.name = name
self.lines = lines
def cdef(self):
result = ""
result += "static const char * const {0} = \n\t".format(self.cfunc_name())
result += '\n\t'.join(self.lines)
result += ';\n'
return result
def cfunc_name(self):
# Translate - and . to underscore
try: #Python 2
translator = string.maketrans('-.', '__')
munged_name = string.translate(self.name, translator)
except AttributeError: #Python 3
translator = "".maketrans('-.', '__')
munged_name = self.name.translate(translator)
return "{0}_{1}".format(self.type, munged_name)
TYPES = ['function', 'completion']
type_to_funcs = dict((t, []) for t in TYPES)
def usage(script_name):
print("Usage: {0} [--output output_directory] files...".format(script_name))
print("""Command options are:
--output directory\t\tThe directory to output the files
-h, --help\t\t\tShow this help message
""")
script_name = sys.argv[0]
try:
opts, file_paths = getopt.gnu_getopt(sys.argv[1:], 'h', ['output=', 'help'])
except getopt.GetoptError as err:
print(err.msg) # will print something like "option -a not recognized"
usage(script_name)
sys.exit(2)
output_directory = './'
for opt, value in opts:
if opt in ('--output',):
output_directory = value
elif opt in ('-h', '--help'):
usage(script_name)
sys.exit(0)
else:
assert False, "unhandled option"
for file in file_paths:
fd = open(file, 'r')
newlines = []
for line in fd:
newlines.append(stringize(line))
fd.close()
dirname = os.path.dirname(file)
# Try to figure out the file type (completion or function)
matches = [dir in dirname for dir in TYPES]
if matches.count(True) is not 1:
print("Cannot determine the type of the file at path {0}".format(file))
sys.exit(-1)
type = TYPES[matches.index(True)]
name = os.path.basename(file)
name, ext = os.path.splitext(name)
newfunc = cfunc(type, name, newlines)
type_to_funcs[type].append(newfunc)
# Sort our functions by name
for funcs in type_to_funcs.values():
funcs.sort(key=cfunc.cfunc_name)
# Output our header
fd = open(os.path.join(output_directory, 'builtin_scripts.h'), 'w')
fd.write('/* This file is generated by internalize_scripts.py */\n\n')
fd.write("""struct builtin_script_t {
const wchar_t *name;
const char *def;
};""")
fd.write('\n')
for type in TYPES:
funcs = type_to_funcs[type]
fd.write('\n')
fd.write('extern const struct builtin_script_t internal_{0}_scripts[{1}];'.format(type, len(funcs)))
fd.write('\n')
fd.close()
# Output the function definitions
fd = open(os.path.join(output_directory, 'builtin_scripts.cpp'), 'w')
fd.write('/* This file is generated by internalize_scripts.py */\n\n')
fd.write('#include "builtin_scripts.h"\n\n')
for type in TYPES:
for func in type_to_funcs[type]:
fd.write(func.cdef())
fd.write('\n')
# Output the refs
for type in TYPES:
funcs = type_to_funcs[type]
func_refs = ["{0}L{1}, {2}{3}".format("{", stringize(func.name), func.cfunc_name(), "}") for func in funcs]
fd.write('const struct builtin_script_t internal_{0}_scripts[{1}] =\n'.format(type, len(funcs)))
fd.write('{\n\t')
fd.write(',\n\t'.join(func_refs))
fd.write('\n};\n')
fd.close()
|
