1fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// Protocol Buffers - Google's data interchange format 2fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// Copyright 2008 Google Inc. All rights reserved. 3fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// http://code.google.com/p/protobuf/ 4fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// 5fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// Redistribution and use in source and binary forms, with or without 6fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// modification, are permitted provided that the following conditions are 7fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// met: 8fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// 9fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// * Redistributions of source code must retain the above copyright 10fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// notice, this list of conditions and the following disclaimer. 11fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// * Redistributions in binary form must reproduce the above 12fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// copyright notice, this list of conditions and the following disclaimer 13fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// in the documentation and/or other materials provided with the 14fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// distribution. 15fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// * Neither the name of Google Inc. nor the names of its 16fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// contributors may be used to endorse or promote products derived from 17fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// this software without specific prior written permission. 18fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// 19fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 20fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 21fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 22fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 23fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 24fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 25fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 26fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 27fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 28fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 29fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 30fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 31fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// Author: kenton@google.com (Kenton Varda) 32fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// Based on original Protocol Buffers design by 33fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// Sanjay Ghemawat, Jeff Dean, and others. 34fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 35fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville#include <google/protobuf/compiler/java/java_generator.h> 36fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville#include <google/protobuf/compiler/java/java_file.h> 37fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville#include <google/protobuf/compiler/java/java_helpers.h> 38fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville#include <google/protobuf/io/printer.h> 39fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville#include <google/protobuf/io/zero_copy_stream.h> 40fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville#include <google/protobuf/descriptor.pb.h> 41fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville#include <google/protobuf/stubs/strutil.h> 42fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 43fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillenamespace google { 44fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillenamespace protobuf { 45fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillenamespace compiler { 46fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillenamespace java { 47fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 48d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville 49fbaaef999ba563838ebd00874ed8a1c01fbf286dWink SavilleJavaGenerator::JavaGenerator() {} 50fbaaef999ba563838ebd00874ed8a1c01fbf286dWink SavilleJavaGenerator::~JavaGenerator() {} 51fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 52fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillebool JavaGenerator::Generate(const FileDescriptor* file, 53fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville const string& parameter, 54fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville OutputDirectory* output_directory, 55fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville string* error) const { 56fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville vector<pair<string, string> > options; 57fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville ParseGeneratorParameter(parameter, &options); 58fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 59fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville // ----------------------------------------------------------------- 60fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville // parse generator options 61fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 62fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville // Name a file where we will write a list of generated file names, one 63fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville // per line. 64fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville string output_list_file; 65fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 66fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville for (int i = 0; i < options.size(); i++) { 67fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville if (options[i].first == "output_list_file") { 68fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville output_list_file = options[i].second; 69fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville } else { 70fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville *error = "Unknown generator option: " + options[i].first; 71fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville return false; 72fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville } 73fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville } 74fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 75fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 76fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville // ----------------------------------------------------------------- 77fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 78fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 79fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville FileGenerator file_generator(file); 80fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville if (!file_generator.Validate(error)) { 81fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville return false; 82fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville } 83fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 84fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville string package_dir = 85fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville StringReplace(file_generator.java_package(), ".", "/", true); 86fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville if (!package_dir.empty()) package_dir += "/"; 87fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 88fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville vector<string> all_files; 89fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 90fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville string java_filename = package_dir; 91fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville java_filename += file_generator.classname(); 92fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville java_filename += ".java"; 93fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville all_files.push_back(java_filename); 94fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 95fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville // Generate main java file. 96fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville scoped_ptr<io::ZeroCopyOutputStream> output( 97fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville output_directory->Open(java_filename)); 98fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville io::Printer printer(output.get(), '$'); 99fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville file_generator.Generate(&printer); 100fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 101fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville // Generate sibling files. 102fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville file_generator.GenerateSiblings(package_dir, output_directory, &all_files); 103fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 104fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville // Generate output list if requested. 105fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville if (!output_list_file.empty()) { 106fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville // Generate output list. This is just a simple text file placed in a 107fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville // deterministic location which lists the .java files being generated. 108fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville scoped_ptr<io::ZeroCopyOutputStream> srclist_raw_output( 109fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville output_directory->Open(output_list_file)); 110fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville io::Printer srclist_printer(srclist_raw_output.get(), '$'); 111fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville for (int i = 0; i < all_files.size(); i++) { 112fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville srclist_printer.Print("$filename$\n", "filename", all_files[i]); 113fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville } 114fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville } 115fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 116fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville return true; 117fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville} 118fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 119fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville} // namespace java 120fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville} // namespace compiler 121fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville} // namespace protobuf 122fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville} // namespace google 123