From 32c66e15063c221a00425a9fff07d7f0da6fe478 Mon Sep 17 00:00:00 2001
From: Martin Kraemer <martin@apache.org>
Date: Thu, 9 Nov 2006 00:06:36 +0000
Subject: [PATCH] Thrift Include Path Specification

Summary:
Add support for include path specification in thrift.  Using a -I <dir> option you can build a list of search paths to resolve thrift includes.

Reviewed By: mcslee

Revert Plan: revertible


git-svn-id: https://svn.apache.org/repos/asf/incubator/thrift/trunk@664868 13f79535-47bb-0310-9956-ffa450edef68
---
 compiler/cpp/src/main.cc | 66 +++++++++++++++++++++++++++++++---------
 1 file changed, 51 insertions(+), 15 deletions(-)

diff --git a/compiler/cpp/src/main.cc b/compiler/cpp/src/main.cc
index aee89ae3..6d29e7f6 100644
--- a/compiler/cpp/src/main.cc
+++ b/compiler/cpp/src/main.cc
@@ -77,6 +77,11 @@ string g_curdir;
  */
 string g_curpath;
 
+/**
+ * Search path for inclusions
+ */
+vector <string> g_incl_searchpath;
+
 /**
  * Global debug state
  */
@@ -226,23 +231,43 @@ string directory_name(string filename) {
  */
 string include_file(string filename) {
   // Absolute path? Just try that
-  if (filename[0] != '/') {
-    filename = g_curdir + "/" + filename;
-  }
-
-  // Realpath!
-  char rp[PATH_MAX];
-  if (realpath(filename.c_str(), rp) == NULL) {
-    pwarning(0, "Cannot open include file %s\n", filename.c_str());
-    return std::string();
-  }
+  if (filename[0] == '/') {
+    // Realpath!
+    char rp[PATH_MAX];
+    if (realpath(filename.c_str(), rp) == NULL) {
+      pwarning(0, "Cannot open include file %s\n", filename.c_str());
+      return std::string();
+    }
   
-  // Stat this files
-  struct stat finfo;
-  if (stat(rp, &finfo) == 0) {
-    return rp;
+    // Stat this files
+    struct stat finfo;
+    if (stat(rp, &finfo) == 0) {
+      return rp;
+    }
+  } else { // relative path, start searching
+    // new search path with current dir global
+    vector<string> sp = g_incl_searchpath;
+    sp.insert(sp.begin(), g_curdir);
+   
+    // iterate through paths
+    vector<string>::iterator it;
+    for (it = sp.begin(); it != sp.end(); it++) {
+      string sfilename = *(it) + "/" + filename;
+      
+      // Realpath!
+      char rp[PATH_MAX];
+      if (realpath(sfilename.c_str(), rp) == NULL) {
+        continue;
+      }
+  
+      // Stat this files
+      struct stat finfo;
+      if (stat(rp, &finfo) == 0) {
+        return rp;
+      }
+    }
   }
-
+  
   // Uh oh
   pwarning(0, "Could not find include file %s\n", filename.c_str());
   return std::string();
@@ -259,6 +284,8 @@ void usage() {
   fprintf(stderr, "  --php        Generate PHP output files\n");
   fprintf(stderr, "  --phpi       Generate PHP inlined files\n");
   fprintf(stderr, "  --py         Generate Python output files\n");
+  fprintf(stderr, "  -I dir       Add a directory to the list of directories \n");
+  fprintf(stderr, "               searched for include directives\n");
   fprintf(stderr, "  --nowarn     Suppress all compiler warnings (BAD!)\n");
   fprintf(stderr, "  --strict     Strict compiler warnings on\n");
   fprintf(stderr, "  --v[erbose]  Verbose mode\n");
@@ -419,6 +446,15 @@ int main(int argc, char** argv) {
         gen_phpi = true;
       } else if (strcmp(arg, "--py") == 0) {
         gen_py = true;
+      } else if (strcmp(arg, "-I") == 0) {
+        // An argument of "-I\ asdf" is invalid and has unknown results
+        arg = argv[++i];
+
+        if (arg == NULL) {
+          fprintf(stderr, "!!! Missing Include directory");
+          usage();
+        }
+        g_incl_searchpath.push_back(arg);
       } else {
         fprintf(stderr, "!!! Unrecognized option: %s\n", arg);
         usage();
-- 
2.17.1