aboutsummaryrefslogtreecommitdiffhomepage
path: root/src/compiler/ruby_plugin.cc
diff options
context:
space:
mode:
authorGravatar nnoble <nnoble@google.com>2014-12-10 16:31:01 -0800
committerGravatar Jan Tattermusch <jtattermusch@google.com>2014-12-11 15:11:44 -0800
commitebebb7e21c976b5887a24795cf1435bed58a2906 (patch)
tree6ff3d70312bb60e422b1b8621a6dd47de091f992 /src/compiler/ruby_plugin.cc
parent0005b778ee2b18026d3c47329ea61437c853de56 (diff)
First pass at making the compilers open-sourcable.
Change on 2014/12/10 by nnoble <nnoble@google.com> ------------- Created by MOE: http://code.google.com/p/moe-java MOE_MIGRATED_REVID=81829127
Diffstat (limited to 'src/compiler/ruby_plugin.cc')
-rw-r--r--src/compiler/ruby_plugin.cc76
1 files changed, 76 insertions, 0 deletions
diff --git a/src/compiler/ruby_plugin.cc b/src/compiler/ruby_plugin.cc
new file mode 100644
index 0000000000..73d1508ad0
--- /dev/null
+++ b/src/compiler/ruby_plugin.cc
@@ -0,0 +1,76 @@
+/*
+ *
+ * Copyright 2014, Google Inc.
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions are
+ * met:
+ *
+ * * Redistributions of source code must retain the above copyright
+ * notice, this list of conditions and the following disclaimer.
+ * * Redistributions in binary form must reproduce the above
+ * copyright notice, this list of conditions and the following disclaimer
+ * in the documentation and/or other materials provided with the
+ * distribution.
+ * * Neither the name of Google Inc. nor the names of its
+ * contributors may be used to endorse or promote products derived from
+ * this software without specific prior written permission.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+ * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
+ * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
+ * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
+ * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
+ * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
+ * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
+ * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
+ * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+ * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+ * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ *
+ */
+
+// Generates Ruby gRPC service interface out of Protobuf IDL.
+//
+// This is a Proto2 compiler plugin. See net/proto2/compiler/proto/plugin.proto
+// and net/proto2/compiler/public/plugin.h for more information on plugins.
+
+#include <memory>
+
+#include "src/compiler/ruby_generator.h"
+#include "src/compiler/ruby_generator_helpers-inl.h"
+#include <google/protobuf/compiler/code_generator.h>
+#include <google/protobuf/compiler/plugin.h>
+#include <google/protobuf/io/coded_stream.h>
+#include <google/protobuf/io/zero_copy_stream.h>
+#include <google/protobuf/descriptor.h>
+
+class RubyGrpcGenerator : public google::protobuf::compiler::CodeGenerator {
+ public:
+ RubyGrpcGenerator() {}
+ ~RubyGrpcGenerator() override {}
+
+ bool Generate(const google::protobuf::FileDescriptor* file,
+ const string& parameter,
+ google::protobuf::compiler::GeneratorContext* context,
+ string* error) const override {
+ // Get output file name.
+ string file_name;
+ if (!grpc_ruby_generator::ServicesFilename(file, &file_name)) {
+ return false;
+ }
+
+ std::unique_ptr<google::protobuf::io::ZeroCopyOutputStream> output(
+ context->OpenForInsert(file_name, "module_scope"));
+ google::protobuf::io::CodedOutputStream coded_out(output.get());
+ string code = grpc_ruby_generator::GetServices(file);
+ coded_out.WriteRaw(code.data(), code.size());
+ return true;
+ }
+};
+
+int main(int argc, char* argv[]) {
+ RubyGrpcGenerator generator;
+ return google::protobuf::compiler::PluginMain(argc, argv, &generator);
+}