Merge pull request #764 from haberman:staleness

PiperOrigin-RevId: 478052697
pull/13171/head
Copybara-Service 2 years ago
commit 9157381d73
  1. 3
      BUILD
  2. 97
      bazel/amalgamate.py
  3. 12
      bazel/build_defs.bzl

@ -985,6 +985,7 @@ upb_amalgamation(
":mini_table", ":mini_table",
":port", ":port",
":reflection", ":reflection",
":table_internal",
":upb", ":upb",
], ],
prefix = "php-", prefix = "php-",
@ -1019,10 +1020,12 @@ upb_amalgamation(
":mini_table", ":mini_table",
":port", ":port",
":reflection", ":reflection",
":table_internal",
":upb", ":upb",
], ],
prefix = "ruby-", prefix = "ruby-",
strip_import_prefix = ["src"], strip_import_prefix = ["src"],
visibility = ["@com_google_protobuf//ruby:__pkg__"],
) )
cc_library( cc_library(

@ -36,42 +36,36 @@ def parse_include(line):
return match.groups()[0] if match else None return match.groups()[0] if match else None
class Amalgamator: class Amalgamator:
def __init__(self, output_path, prefix): def __init__(self, h_out, c_out):
self.include_paths = ["."] self.include_paths = ["."]
self.included = set(["upb/port_def.inc", "upb/port_undef.inc"]) self.included = set()
self.output_h = open(output_path + prefix + "upb.h", "w") self.output_h = open(h_out, "w")
self.output_c = open(output_path + prefix + "upb.c", "w") self.output_c = open(c_out, "w")
self.h_out = h_out.split("/")[-1]
def amalgamate(self, h_files, c_files):
self.h_files = set(h_files)
self.output_c.write("/* Amalgamated source file */\n") self.output_c.write("/* Amalgamated source file */\n")
self.output_c.write('#include "%supb.h"\n' % (prefix)) self.output_c.write('#include "%s"\n' % (self.h_out))
if prefix == "ruby-": if self.h_out == "ruby-upb.h":
self.output_h.write("// Ruby is still using proto3 enum semantics for proto2\n") self.output_h.write("// Ruby is still using proto3 enum semantics for proto2\n")
self.output_h.write("#define UPB_DISABLE_PROTO2_ENUM_CHECKING\n") self.output_h.write("#define UPB_DISABLE_PROTO2_ENUM_CHECKING\n")
self.output_c.write(open("upb/port_def.inc").read())
self.output_h.write("/* Amalgamated source file */\n") self.output_h.write("/* Amalgamated source file */\n")
self.output_h.write(open("upb/port_def.inc").read())
def add_include_path(self, path): port_def = self._find_include_file("upb/port_def.inc")
self.include_paths.append(path) port_undef = self._find_include_file("upb/port_undef.inc")
self._process_file(port_def, self.output_h)
self._process_file(port_def, self.output_c)
def finish(self): for file in c_files:
self._add_header("upb/port_undef.inc") self._process_file(file, self.output_c)
self.add_src("upb/port_undef.inc")
self._process_file(port_undef, self.output_h)
self._process_file(port_undef, self.output_c)
def _process_file(self, infile_name, outfile): def _process_file(self, infile_name, outfile):
file = None lines = open(infile_name).readlines()
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)
lines = file.readlines()
has_copyright = lines[1].startswith(" * Copyright") has_copyright = lines[1].startswith(" * Copyright")
if has_copyright: if has_copyright:
@ -79,51 +73,52 @@ class Amalgamator:
lines.pop(0) lines.pop(0)
lines.pop(0) lines.pop(0)
lines.insert(0, "\n/** " + infile_name + " " + ("*" * 60) +"/");
for line in lines: for line in lines:
if not self._process_include(line, outfile): if not self._process_include(line):
outfile.write(line) outfile.write(line)
def _process_include(self, line, outfile): def _find_include_file(self, name):
for h_file in self.h_files:
if h_file.endswith(name):
return h_file
def _process_include(self, line):
include = parse_include(line) include = parse_include(line)
if not include: if not include:
return False return False
if not (include.startswith("upb") or include.startswith("google")): if not (include.startswith("upb") or include.startswith("google")):
return False return False
if include and (include.endswith("port_def.inc") or include.endswith("port_undef.inc")):
# Skip, we handle this separately
return True
if include.endswith("hpp"): if include.endswith("hpp"):
# Skip, we don't support the amalgamation from C++. # Skip, we don't support the amalgamation from C++.
return True return True
elif include in self.included:
return True
else: else:
# Include this upb header inline. # Include this upb header inline.
if include not in self.included: h_file = self._find_include_file(include)
if h_file:
self.h_files.remove(h_file)
self.included.add(include) self.included.add(include)
self._add_header(include) self._process_file(h_file, self.output_h)
return True return True
raise RuntimeError("Couldn't find include: " + include + ", h_files=" + repr(self.h_files))
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 ---- # ---- main ----
output_path = sys.argv[1] c_out = sys.argv[1]
prefix = sys.argv[2] h_out = sys.argv[2]
amalgamator = Amalgamator(output_path, prefix) amalgamator = Amalgamator(h_out, c_out)
files = [] c_files = []
h_files = []
for arg in sys.argv[3:]: for arg in sys.argv[3:]:
arg = arg.strip() arg = arg.strip()
if arg.startswith("-I"): if arg.endswith(".h") or arg.endswith(".inc"):
amalgamator.add_include_path(arg[2:]) h_files.append(arg)
elif arg.endswith(".h") or arg.endswith(".inc"):
pass
else: else:
files.append(arg) c_files.append(arg)
for filename in files:
amalgamator.add_src(filename)
amalgamator.finish() amalgamator.amalgamate(h_files, c_files)

@ -25,7 +25,6 @@
"""Internal rules for building upb.""" """Internal rules for building upb."""
load("@bazel_skylib//lib:paths.bzl", "paths")
load(":upb_proto_library.bzl", "GeneratedSrcsInfo") load(":upb_proto_library.bzl", "GeneratedSrcsInfo")
_DEFAULT_CPPOPTS = [] _DEFAULT_CPPOPTS = []
@ -95,13 +94,6 @@ def _get_real_roots(files):
roots[real_root] = True roots[real_root] = True
return roots.keys() return roots.keys()
def _get_includes(files, strip_import_prefix):
roots = _get_real_roots(files)
includes = ["-I" + root for root in roots]
for include in strip_import_prefix:
includes += ["-I" + paths.join(root, include) for root in roots]
return includes
def make_shell_script(name, contents, out): def make_shell_script(name, contents, out):
contents = contents.replace("$", "$$") contents = contents.replace("$", "$$")
native.genrule( native.genrule(
@ -140,11 +132,11 @@ def _upb_amalgamation(ctx):
inputs = [] inputs = []
for lib in ctx.attr.libs: for lib in ctx.attr.libs:
inputs += lib[SrcList].srcs inputs += lib[SrcList].srcs
srcs = [src for src in inputs if src.path.endswith("c")] srcs = [src for src in inputs if not src.path.endswith("hpp")]
ctx.actions.run( ctx.actions.run(
inputs = inputs, inputs = inputs,
outputs = ctx.outputs.outs, outputs = ctx.outputs.outs,
arguments = [ctx.bin_dir.path + "/", ctx.attr.prefix] + [f.path for f in srcs] + _get_includes(inputs, ctx.attr.strip_import_prefix), arguments = [f.path for f in ctx.outputs.outs] + [f.path for f in srcs],
progress_message = "Making amalgamation", progress_message = "Making amalgamation",
executable = ctx.executable._amalgamator, executable = ctx.executable._amalgamator,
) )

Loading…
Cancel
Save