|
@@ -36,10 +36,12 @@
|
|
|
#include <google/protobuf/descriptor.pb.h>
|
|
|
#include <google/protobuf/io/printer.h>
|
|
|
#include <google/protobuf/io/zero_copy_stream.h>
|
|
|
+#include <google/protobuf/stubs/strutil.h>
|
|
|
|
|
|
#include <google/protobuf/compiler/csharp/csharp_generator.h>
|
|
|
-#include <google/protobuf/compiler/csharp/csharp_umbrella_class.h>
|
|
|
#include <google/protobuf/compiler/csharp/csharp_helpers.h>
|
|
|
+#include <google/protobuf/compiler/csharp/csharp_names.h>
|
|
|
+#include <google/protobuf/compiler/csharp/csharp_umbrella_class.h>
|
|
|
|
|
|
using google::protobuf::internal::scoped_ptr;
|
|
|
|
|
@@ -48,9 +50,39 @@ namespace protobuf {
|
|
|
namespace compiler {
|
|
|
namespace csharp {
|
|
|
|
|
|
-std::string GetOutputFile(const google::protobuf::FileDescriptor* file, const std::string file_extension)
|
|
|
-{
|
|
|
- return GetUmbrellaClassUnqualifiedName(file) + file_extension;
|
|
|
+std::string GetOutputFile(
|
|
|
+ const google::protobuf::FileDescriptor* file,
|
|
|
+ const std::string file_extension,
|
|
|
+ const bool generate_directories,
|
|
|
+ const std::string base_namespace,
|
|
|
+ string* error) {
|
|
|
+ string relative_filename = GetUmbrellaClassUnqualifiedName(file) + file_extension;
|
|
|
+ if (!generate_directories) {
|
|
|
+ return relative_filename;
|
|
|
+ }
|
|
|
+ string ns = GetFileNamespace(file);
|
|
|
+ string namespace_suffix = ns;
|
|
|
+ if (!base_namespace.empty()) {
|
|
|
+ // Check that the base_namespace is either equal to or a leading part of
|
|
|
+ // the file namespace. This isn't just a simple prefix; "Foo.B" shouldn't
|
|
|
+ // be regarded as a prefix of "Foo.Bar". The simplest option is to add "."
|
|
|
+ // to both.
|
|
|
+ string extended_ns = ns + ".";
|
|
|
+ if (extended_ns.find(base_namespace + ".") != 0) {
|
|
|
+ *error = "Namespace " + ns + " is not a prefix namespace of base namespace " + base_namespace;
|
|
|
+ return ""; // This will be ignored, because we've set an error.
|
|
|
+ }
|
|
|
+ namespace_suffix = ns.substr(base_namespace.length());
|
|
|
+ if (namespace_suffix.find(".") == 0) {
|
|
|
+ namespace_suffix = namespace_suffix.substr(1);
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ string namespace_dir = StringReplace(namespace_suffix, ".", "/", true);
|
|
|
+ if (!namespace_dir.empty()) {
|
|
|
+ namespace_dir += "/";
|
|
|
+ }
|
|
|
+ return namespace_dir + relative_filename;
|
|
|
}
|
|
|
|
|
|
void GenerateFile(const google::protobuf::FileDescriptor* file,
|
|
@@ -75,16 +107,26 @@ bool Generator::Generate(
|
|
|
}
|
|
|
|
|
|
std::string file_extension = ".cs";
|
|
|
+ std::string base_namespace = "";
|
|
|
+ bool generate_directories = false;
|
|
|
for (int i = 0; i < options.size(); i++) {
|
|
|
if (options[i].first == "file_extension") {
|
|
|
file_extension = options[i].second;
|
|
|
+ } else if (options[i].first == "base_namespace") {
|
|
|
+ base_namespace = options[i].second;
|
|
|
+ generate_directories = true;
|
|
|
} else {
|
|
|
*error = "Unknown generator option: " + options[i].first;
|
|
|
return false;
|
|
|
}
|
|
|
}
|
|
|
|
|
|
- std::string filename = GetOutputFile(file, file_extension);
|
|
|
+ string filename_error = "";
|
|
|
+ std::string filename = GetOutputFile(file, file_extension, generate_directories, base_namespace, &filename_error);
|
|
|
+ if (!filename_error.empty()) {
|
|
|
+ *error = filename_error;
|
|
|
+ return false;
|
|
|
+ }
|
|
|
scoped_ptr<io::ZeroCopyOutputStream> output(
|
|
|
generator_context->Open(filename));
|
|
|
io::Printer printer(output.get(), '$');
|