|
|
@ -1,4 +1,4 @@ |
|
|
|
# Copyright 2016 gRPC authors. |
|
|
|
# Copyright 2020 The gRPC authors. |
|
|
|
# |
|
|
|
# |
|
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); |
|
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); |
|
|
|
# you may not use this file except in compliance with the License. |
|
|
|
# you may not use this file except in compliance with the License. |
|
|
@ -12,8 +12,6 @@ |
|
|
|
# See the License for the specific language governing permissions and |
|
|
|
# See the License for the specific language governing permissions and |
|
|
|
# limitations under the License. |
|
|
|
# limitations under the License. |
|
|
|
|
|
|
|
|
|
|
|
# TODO: Manually format. |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
from libc cimport stdlib |
|
|
|
from libc cimport stdlib |
|
|
|
from libcpp.vector cimport vector |
|
|
|
from libcpp.vector cimport vector |
|
|
|
from libcpp.utility cimport pair |
|
|
|
from libcpp.utility cimport pair |
|
|
@ -37,8 +35,16 @@ cdef extern from "grpc_tools/main.h" namespace "grpc_tools": |
|
|
|
string message |
|
|
|
string message |
|
|
|
|
|
|
|
|
|
|
|
int protoc_main(int argc, char *argv[]) |
|
|
|
int protoc_main(int argc, char *argv[]) |
|
|
|
int protoc_get_protos(char* protobuf_path, vector[string]* include_path, vector[pair[string, string]]* files_out, vector[cProtocError]* errors, vector[cProtocWarning]* wrnings) nogil except + |
|
|
|
int protoc_get_protos(char* protobuf_path, |
|
|
|
int protoc_get_services(char* protobuf_path, vector[string]* include_path, vector[pair[string, string]]* files_out, vector[cProtocError]* errors, vector[cProtocWarning]* wrnings) nogil except + |
|
|
|
vector[string]* include_path, |
|
|
|
|
|
|
|
vector[pair[string, string]]* files_out, |
|
|
|
|
|
|
|
vector[cProtocError]* errors, |
|
|
|
|
|
|
|
vector[cProtocWarning]* wrnings) nogil except + |
|
|
|
|
|
|
|
int protoc_get_services(char* protobuf_path, |
|
|
|
|
|
|
|
vector[string]* include_path, |
|
|
|
|
|
|
|
vector[pair[string, string]]* files_out, |
|
|
|
|
|
|
|
vector[cProtocError]* errors, |
|
|
|
|
|
|
|
vector[cProtocWarning]* wrnings) nogil except + |
|
|
|
|
|
|
|
|
|
|
|
def run_main(list args not None): |
|
|
|
def run_main(list args not None): |
|
|
|
cdef char **argv = <char **>stdlib.malloc(len(args)*sizeof(char *)) |
|
|
|
cdef char **argv = <char **>stdlib.malloc(len(args)*sizeof(char *)) |
|
|
@ -54,10 +60,12 @@ class ProtocError(Exception): |
|
|
|
self.message = message |
|
|
|
self.message = message |
|
|
|
|
|
|
|
|
|
|
|
def __repr__(self): |
|
|
|
def __repr__(self): |
|
|
|
return "ProtocError(filename=\"{}\", line={}, column={}, message=\"{}\")".format(self.filename, self.line, self.column, self.message) |
|
|
|
return "ProtocError(filename=\"{}\", line={}, column={}, message=\"{}\")".format( |
|
|
|
|
|
|
|
self.filename, self.line, self.column, self.message) |
|
|
|
|
|
|
|
|
|
|
|
def __str__(self): |
|
|
|
def __str__(self): |
|
|
|
return "{}:{}:{} error: {}".format(self.filename.decode("ascii"), self.line, self.column, self.message.decode("ascii")) |
|
|
|
return "{}:{}:{} error: {}".format(self.filename.decode("ascii"), |
|
|
|
|
|
|
|
self.line, self.column, self.message.decode("ascii")) |
|
|
|
|
|
|
|
|
|
|
|
class ProtocWarning(Warning): |
|
|
|
class ProtocWarning(Warning): |
|
|
|
def __init__(self, filename, line, column, message): |
|
|
|
def __init__(self, filename, line, column, message): |
|
|
@ -67,9 +75,9 @@ class ProtocWarning(Warning): |
|
|
|
self.message = message |
|
|
|
self.message = message |
|
|
|
|
|
|
|
|
|
|
|
def __repr__(self): |
|
|
|
def __repr__(self): |
|
|
|
return "ProtocWarning(filename=\"{}\", line={}, column={}, message=\"{}\")".format(self.filename, self.line, self.column, self.message) |
|
|
|
return "ProtocWarning(filename=\"{}\", line={}, column={}, message=\"{}\")".format( |
|
|
|
|
|
|
|
self.filename, self.line, self.column, self.message) |
|
|
|
|
|
|
|
|
|
|
|
# TODO: Maybe come up with something better than this |
|
|
|
|
|
|
|
__str__ = __repr__ |
|
|
|
__str__ = __repr__ |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@ -87,19 +95,20 @@ class ProtocErrors(Exception): |
|
|
|
return "\n".join(str(err) for err in self._errors) |
|
|
|
return "\n".join(str(err) for err in self._errors) |
|
|
|
|
|
|
|
|
|
|
|
cdef _c_protoc_error_to_protoc_error(cProtocError c_protoc_error): |
|
|
|
cdef _c_protoc_error_to_protoc_error(cProtocError c_protoc_error): |
|
|
|
return ProtocError(c_protoc_error.filename, c_protoc_error.line, c_protoc_error.column, c_protoc_error.message) |
|
|
|
return ProtocError(c_protoc_error.filename, c_protoc_error.line, |
|
|
|
|
|
|
|
c_protoc_error.column, c_protoc_error.message) |
|
|
|
|
|
|
|
|
|
|
|
cdef _c_protoc_warning_to_protoc_warning(cProtocWarning c_protoc_warning): |
|
|
|
cdef _c_protoc_warning_to_protoc_warning(cProtocWarning c_protoc_warning): |
|
|
|
return ProtocWarning(c_protoc_warning.filename, c_protoc_warning.line, c_protoc_warning.column, c_protoc_warning.message) |
|
|
|
return ProtocWarning(c_protoc_warning.filename, c_protoc_warning.line, |
|
|
|
|
|
|
|
c_protoc_warning.column, c_protoc_warning.message) |
|
|
|
|
|
|
|
|
|
|
|
cdef _handle_errors(int rc, vector[cProtocError]* errors, vector[cProtocWarning]* wrnings, bytes protobuf_path): |
|
|
|
cdef _handle_errors(int rc, vector[cProtocError]* errors, vector[cProtocWarning]* wrnings, bytes protobuf_path): |
|
|
|
for warning in dereference(wrnings): |
|
|
|
for warning in dereference(wrnings): |
|
|
|
warnings.warn(_c_protoc_warning_to_protoc_warning(warning)) |
|
|
|
warnings.warn(_c_protoc_warning_to_protoc_warning(warning)) |
|
|
|
if rc != 0: |
|
|
|
if rc != 0: |
|
|
|
if dereference(errors).size() != 0: |
|
|
|
if dereference(errors).size() != 0: |
|
|
|
py_errors = [_c_protoc_error_to_protoc_error(c_error) for c_error in dereference(errors)] |
|
|
|
py_errors = [_c_protoc_error_to_protoc_error(c_error) |
|
|
|
# TODO: Come up with a good system for printing multiple errors from |
|
|
|
for c_error in dereference(errors)] |
|
|
|
# protoc. |
|
|
|
|
|
|
|
raise ProtocErrors(py_errors) |
|
|
|
raise ProtocErrors(py_errors) |
|
|
|
raise Exception("An unknown error occurred while compiling {}".format(protobuf_path)) |
|
|
|
raise Exception("An unknown error occurred while compiling {}".format(protobuf_path)) |
|
|
|
|
|
|
|
|
|
|
|