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 std::string grpc_package_root; 35 // TODO(https://github.com/grpc/grpc/issues/8622): Drop this. 36 std::string beta_package_root; 37 // TODO(https://github.com/google/protobuf/issues/888): Drop this. 38 std::string import_prefix; 39 std::vector<std::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 uint64_t GetSupportedFeatures() const override; 48 49 bool Generate(const grpc::protobuf::FileDescriptor* file, 50 const std::string& parameter, 51 grpc::protobuf::compiler::GeneratorContext* context, 52 std::string* error) const override; 53 54 private: 55 GeneratorConfiguration config_; 56 }; 57 58 } // namespace grpc_python_generator 59 60 #endif // GRPC_INTERNAL_COMPILER_PYTHON_GENERATOR_H 61