mirror of https://github.com/grpc/grpc.git
The C based gRPC (C++, Python, Ruby, Objective-C, PHP, C#)
https://grpc.io/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
82 lines
2.2 KiB
82 lines
2.2 KiB
5 years ago
|
#!/usr/bin/python
|
||
|
|
|
||
|
|
import sys
|
||
|
|
import re
|
||
|
|
import os
|
||
|
|
|
||
|
|
INCLUDE_RE = re.compile('^#include "([^"]*)"$')
|
||
|
|
|
||
|
|
def parse_include(line):
|
||
|
|
match = INCLUDE_RE.match(line)
|
||
|
|
return match.groups()[0] if match else None
|
||
|
|
|
||
|
|
class Amalgamator:
|
||
|
|
def __init__(self, output_path):
|
||
|
|
self.include_paths = ["."]
|
||
|
|
self.included = set(["upb/port_def.inc", "upb/port_undef.inc"])
|
||
|
|
self.output_h = open(output_path + "upb.h", "w")
|
||
|
|
self.output_c = open(output_path + "upb.c", "w")
|
||
|
|
|
||
|
|
self.output_c.write("/* Amalgamated source file */\n")
|
||
|
|
self.output_c.write('#include "upb.h"\n')
|
||
|
|
self.output_c.write(open("upb/port_def.inc").read())
|
||
|
|
|
||
|
|
self.output_h.write("/* Amalgamated source file */\n")
|
||
|
|
self.output_h.write('#include <stdint.h>')
|
||
|
|
self.output_h.write(open("upb/port_def.inc").read())
|
||
|
|
|
||
|
|
def add_include_path(self, path):
|
||
|
|
self.include_paths.append(path)
|
||
|
|
|
||
|
|
def finish(self):
|
||
|
|
self.output_c.write(open("upb/port_undef.inc").read())
|
||
|
|
self.output_h.write(open("upb/port_undef.inc").read())
|
||
|
|
|
||
|
|
def _process_file(self, infile_name, outfile):
|
||
|
|
file = None
|
||
|
|
for path in self.include_paths:
|
||
|
|
try:
|
||
|
|
full_path = os.path.join(path, infile_name)
|
||
|
|
file = open(full_path)
|
||
|
|
break
|
||
|
|
except IOError:
|
||
|
|
pass
|
||
|
|
if not file:
|
||
|
|
raise RuntimeError("Couldn't open file " + infile_name)
|
||
|
|
|
||
|
|
for line in file:
|
||
|
|
include = parse_include(line)
|
||
|
|
if include is not None and (include.startswith("upb") or
|
||
|
|
include.startswith("google")):
|
||
|
|
if include not in self.included:
|
||
|
|
self.included.add(include)
|
||
|
|
self._add_header(include)
|
||
|
|
else:
|
||
|
|
outfile.write(line)
|
||
|
|
|
||
|
|
def _add_header(self, filename):
|
||
|
|
self._process_file(filename, self.output_h)
|
||
|
|
|
||
|
|
def add_src(self, filename):
|
||
|
|
self._process_file(filename, self.output_c)
|
||
|
|
|
||
|
|
# ---- main ----
|
||
|
|
|
||
|
|
output_path = sys.argv[1]
|
||
|
|
amalgamator = Amalgamator(output_path)
|
||
|
|
files = []
|
||
|
|
|
||
|
|
for arg in sys.argv[2:]:
|
||
|
|
arg = arg.strip()
|
||
|
|
if arg.startswith("-I"):
|
||
|
|
amalgamator.add_include_path(arg[2:])
|
||
|
|
elif arg.endswith(".h") or arg.endswith(".inc"):
|
||
|
|
pass
|
||
|
|
else:
|
||
|
|
files.append(arg)
|
||
|
|
|
||
|
|
for filename in files:
|
||
|
|
amalgamator.add_src(filename)
|
||
|
|
|
||
|
|
amalgamator.finish()
|