SERVER-11095: Have getAllSourceFiles follow symlinks inside the modules directory

1 files changed, 13 insertions, 4 deletions

diff --git a/buildscripts/utils.py b/buildscripts/utils.py
index 63a4ba6c405..cecd479681b 100644
--- a/buildscripts/utils.py
+++ b/buildscripts/utils.py
@@ -15,12 +15,12 @@ import hashlib
 def getAllSourceFiles( arr=None , prefix="." ):
     if arr is None:
         arr = []
-    
+
     if not os.path.isdir( prefix ):
         # assume a file
         arr.append( prefix )
         return arr
-        
+
     for x in os.listdir( prefix ):
         if ( x.startswith( "." )
           or x.startswith( "pcre-" )
@@ -38,8 +38,18 @@ def getAllSourceFiles( arr=None , prefix="." ):
         if x.find("v8-3.25") != -1 or x.find("mozjs") != -1:
             continue
 
+        def isFollowableDir(prefix, full):
+            if not os.path.isdir(full):
+                return False
+            if not os.path.islink(full):
+                return True
+            # Follow softlinks in the modules directory (e.g: enterprise).
+            if os.path.split(prefix)[1] == "modules":
+                return True
+            return False;
+
         full = prefix + "/" + x
-        if os.path.isdir( full ) and not os.path.islink( full ):
+        if isFollowableDir(prefix, full):
             getAllSourceFiles( arr , full )
         else:
             if full.endswith( ".cpp" ) or full.endswith( ".h" ) or full.endswith( ".c" ):
@@ -250,4 +260,3 @@ def unicode_dammit(string, encoding='utf8'):
     #
     # name inpsired by BeautifulSoup's "UnicodeDammit"
     return string.decode(encoding, 'repr')
-