Line data Source code
1 : // Protocol Buffers - Google's data interchange format
2 : // Copyright 2008 Google Inc. All rights reserved.
3 : // https://developers.google.com/protocol-buffers/
4 : //
5 : // Redistribution and use in source and binary forms, with or without
6 : // modification, are permitted provided that the following conditions are
7 : // met:
8 : //
9 : // * Redistributions of source code must retain the above copyright
10 : // notice, this list of conditions and the following disclaimer.
11 : // * Redistributions in binary form must reproduce the above
12 : // copyright notice, this list of conditions and the following disclaimer
13 : // in the documentation and/or other materials provided with the
14 : // distribution.
15 : // * Neither the name of Google Inc. nor the names of its
16 : // contributors may be used to endorse or promote products derived from
17 : // this software without specific prior written permission.
18 : //
19 : // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
20 : // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
21 : // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
22 : // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
23 : // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
24 : // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
25 : // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
26 : // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
27 : // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
28 : // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
29 : // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
30 :
31 : #include <sstream>
32 :
33 : #include <google/protobuf/compiler/code_generator.h>
34 : #include <google/protobuf/compiler/plugin.h>
35 : #include <google/protobuf/descriptor.h>
36 : #include <google/protobuf/descriptor.pb.h>
37 : #include <google/protobuf/io/printer.h>
38 : #include <google/protobuf/io/zero_copy_stream.h>
39 :
40 : #include <google/protobuf/compiler/csharp/csharp_generator.h>
41 : #include <google/protobuf/compiler/csharp/csharp_umbrella_class.h>
42 : #include <google/protobuf/compiler/csharp/csharp_helpers.h>
43 :
44 : using google::protobuf::internal::scoped_ptr;
45 :
46 : namespace google {
47 : namespace protobuf {
48 : namespace compiler {
49 : namespace csharp {
50 :
51 0 : std::string GetOutputFile(const google::protobuf::FileDescriptor* file, const std::string file_extension)
52 : {
53 0 : return GetUmbrellaClassUnqualifiedName(file) + file_extension;
54 : }
55 :
56 0 : void GenerateFile(const google::protobuf::FileDescriptor* file,
57 : io::Printer* printer) {
58 0 : UmbrellaClassGenerator umbrellaGenerator(file);
59 0 : umbrellaGenerator.Generate(printer);
60 0 : }
61 :
62 0 : bool Generator::Generate(
63 0 : const FileDescriptor* file,
64 : const string& parameter,
65 : GeneratorContext* generator_context,
66 : string* error) const {
67 :
68 : vector<pair<string, string> > options;
69 0 : ParseGeneratorParameter(parameter, &options);
70 :
71 : // We only support proto3 - but we make an exception for descriptor.proto.
72 0 : if (file->syntax() != FileDescriptor::SYNTAX_PROTO3 && !IsDescriptorProto(file)) {
73 : *error = "C# code generation only supports proto3 syntax";
74 : return false;
75 : }
76 :
77 0 : std::string file_extension = ".cs";
78 0 : for (int i = 0; i < options.size(); i++) {
79 0 : if (options[i].first == "file_extension") {
80 0 : file_extension = options[i].second;
81 : } else {
82 0 : *error = "Unknown generator option: " + options[i].first;
83 0 : return false;
84 : }
85 : }
86 :
87 0 : std::string filename = GetOutputFile(file, file_extension);
88 : scoped_ptr<io::ZeroCopyOutputStream> output(
89 0 : generator_context->Open(filename));
90 0 : io::Printer printer(output.get(), '$');
91 :
92 0 : GenerateFile(file, &printer);
93 :
94 0 : return true;
95 : }
96 :
97 : } // namespace csharp
98 : } // namespace compiler
99 : } // namespace protobuf
100 : } // namespace google
|