1 | /* |
2 | * |
3 | * Copyright 2015 gRPC authors. |
4 | * |
5 | * Licensed under the Apache License, Version 2.0 (the "License"); |
6 | * you may not use this file except in compliance with the License. |
7 | * You may obtain a copy of the License at |
8 | * |
9 | * http://www.apache.org/licenses/LICENSE-2.0 |
10 | * |
11 | * Unless required by applicable law or agreed to in writing, software |
12 | * distributed under the License is distributed on an "AS IS" BASIS, |
13 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
14 | * See the License for the specific language governing permissions and |
15 | * limitations under the License. |
16 | * |
17 | */ |
18 | |
19 | #ifndef GRPC_INTERNAL_COMPILER_PYTHON_GENERATOR_H |
20 | #define GRPC_INTERNAL_COMPILER_PYTHON_GENERATOR_H |
21 | |
22 | #include <utility> |
23 | #include <vector> |
24 | |
25 | #include "src/compiler/config.h" |
26 | #include "src/compiler/schema_interface.h" |
27 | |
28 | namespace grpc_python_generator { |
29 | |
30 | // Data pertaining to configuration of the generator with respect to anything |
31 | // that may be used internally at Google. |
32 | struct GeneratorConfiguration { |
33 | GeneratorConfiguration(); |
34 | grpc::string grpc_package_root; |
35 | // TODO(https://github.com/grpc/grpc/issues/8622): Drop this. |
36 | grpc::string beta_package_root; |
37 | // TODO(https://github.com/google/protobuf/issues/888): Drop this. |
38 | grpc::string import_prefix; |
39 | std::vector<grpc::string> prefixes_to_filter; |
40 | }; |
41 | |
42 | class PythonGrpcGenerator : public grpc::protobuf::compiler::CodeGenerator { |
43 | public: |
44 | PythonGrpcGenerator(const GeneratorConfiguration& config); |
45 | ~PythonGrpcGenerator(); |
46 | |
47 | bool Generate(const grpc::protobuf::FileDescriptor* file, |
48 | const grpc::string& parameter, |
49 | grpc::protobuf::compiler::GeneratorContext* context, |
50 | grpc::string* error) const; |
51 | |
52 | private: |
53 | GeneratorConfiguration config_; |
54 | }; |
55 | |
56 | } // namespace grpc_python_generator |
57 | |
58 | #endif // GRPC_INTERNAL_COMPILER_PYTHON_GENERATOR_H |
59 | |