Merge pull request #43 from kornilova-l/add-option-to-set-package-name

Add command line option for package name

Author: kornilova203

Main.cpp

@@ -15,7 +15,10 @@ int main(int argc, char *argv[]) {
                                              llvm::cl::cat(Category));
     llvm::cl::opt<std::string> ExcludePrefix("exclude-prefix",
                                              llvm::cl::cat(Category));
-
+    llvm::cl::opt<std::string> Package(
+        "package", llvm::cl::cat(Category),
+        llvm::cl::value_desc("package-name"),
+        llvm::cl::desc("Package name of generated Scala file"));
     clang::tooling::CommonOptionsParser op(argc, (const char **)argv, Category);
     clang::tooling::ClangTool Tool(op.getCompilations(),
                                    op.getSourcePathList());
@@ -41,6 +44,10 @@ int main(int argc, char *argv[]) {
 
     IR ir = actionFactory.getIntermediateRepresentation();
 
+    if (!Package.empty()) {
+        ir.setPackageName(Package.getValue());
+    }
+
     auto printLoc = PrintHeadersLocation.getValue();
     if (printLoc) {
         for (const auto &location : locations) {

ir/IR.cpp

@@ -37,6 +37,10 @@ bool IR::libObjEmpty() const {
 llvm::raw_ostream &operator<<(llvm::raw_ostream &s, const IR &ir) {
     assert(ir.generated); // typedefs were generated
 
+    if (!ir.packageName.empty()) {
+        s << "package " << ir.packageName << "\n\n";
+    }
+
     if (!ir.libObjEmpty() || !ir.enums.empty()) {
         s << "import scala.scalanative._\n"
           << "import scala.scalanative.native._\n"
@@ -194,3 +198,7 @@ bool IR::typeIsUsedOnlyInTypeDefs(std::string type) {
     return !(isTypeUsed(functions, type) || isTypeUsed(structs, type) ||
              isTypeUsed(unions, type));
 }
+
+void IR::setPackageName(std::string packageName) {
+    this->packageName = packageName;
+}

ir/IR.h

@@ -39,6 +39,8 @@ class IR {
 
     void generate(const std::string &excludePrefix);
 
+    void setPackageName(std::string packageName);
+
   private:
     /**
      * Generates type defs for enums, structs and unions
@@ -106,6 +108,7 @@ class IR {
     std::vector<Union> unions;
     std::vector<Enum> enums;
     bool generated = false; // generate type defs only once
+    std::string packageName;
 };
 
 #endif // SCALA_NATIVE_BINDGEN_INTERMEDIATEREPRESENTATION_H