support exclude source path in relocation

.gitignore

@@ -15,5 +15,6 @@ yarn-error.log
 src/docs/.vuepress/dist/
 .DS_Store
 jd-gui.cfg
+lib/
 bin/
 .vscode/

src/main/groovy/com/github/jengelman/gradle/plugins/shadow/impl/RelocatorRemapper.groovy

@@ -24,6 +24,7 @@ import com.github.jengelman.gradle.plugins.shadow.relocation.RelocateClassContex
 import com.github.jengelman.gradle.plugins.shadow.relocation.RelocatePathContext
 import com.github.jengelman.gradle.plugins.shadow.relocation.Relocator
 import com.github.jengelman.gradle.plugins.shadow.tasks.ShadowCopyAction.RelativeArchivePath
+import groovy.util.logging.Slf4j
 import org.objectweb.asm.commons.Remapper
 
 import java.util.regex.Matcher
@@ -34,12 +35,14 @@ import java.util.regex.Pattern
  *
  * @author John Engelman
  */
+@Slf4j
 class RelocatorRemapper extends Remapper {
 
     private final Pattern classPattern = Pattern.compile("(\\[*)?L(.+)")
 
     List<Relocator> relocators
     ShadowStats stats
+    public String currentFilePath
 
     RelocatorRemapper(List<Relocator> relocators, ShadowStats stats) {
         this.relocators = relocators
@@ -65,7 +68,13 @@ class RelocatorRemapper extends Remapper {
                 name = m.group(2)
             }
 
-            for (Relocator r : relocators) {
+            for (Relocator r : relocators)  {
+                boolean canRelocate = r.canRelocateSourceFile(currentFilePath)
+                if (!canRelocate) {
+                    String info = String.format("skipped %s as src path %s excluded", r, currentFilePath)
+                    log.debug(info)
+                    continue
+                }
                 if (r.canRelocateClass(name)) {
                     RelocateClassContext classContext = RelocateClassContext.builder().className(name).stats(stats).build()
                     value = prefix + r.relocateClass(classContext) + suffix
@@ -97,6 +106,13 @@ class RelocatorRemapper extends Remapper {
         }
 
         for (Relocator r : relocators) {
+            boolean canRelocate = r.canRelocateSourceFile(currentFilePath)
+            if (!canRelocate) {
+                String info = String.format("skipped %s as src path %s excluded", r, currentFilePath)
+                log.debug(info)
+                continue
+            }
+
             if (r.canRelocatePath(name)) {
                 RelocatePathContext pathContext = RelocatePathContext.builder().path(name).stats(stats).build()
                 value = prefix + r.relocatePath(pathContext) + suffix

src/main/groovy/com/github/jengelman/gradle/plugins/shadow/relocation/Relocator.groovy

@@ -37,4 +37,6 @@ interface Relocator {
     String relocateClass(RelocateClassContext context)
 
     String applyToSourceContent(String sourceContent)
+
+    boolean canRelocateSourceFile(String sourceFilePath)
 }

src/main/groovy/com/github/jengelman/gradle/plugins/shadow/relocation/SimpleRelocator.groovy

@@ -46,6 +46,8 @@ class SimpleRelocator implements Relocator {
     private final Set<String> includes
 
     private final Set<String> excludes
+
+    private final Set<String> excludeSources
 
     private final boolean rawString
 
@@ -54,11 +56,11 @@ class SimpleRelocator implements Relocator {
     }
 
     SimpleRelocator(String patt, String shadedPattern, List<String> includes, List<String> excludes) {
-        this(patt, shadedPattern, includes, excludes, false)
+        this(patt, shadedPattern, includes, excludes, [], false)
     }
 
     SimpleRelocator(String patt, String shadedPattern, List<String> includes, List<String> excludes,
-                           boolean rawString) {
+                    List<String> excludeSources, boolean rawString) {
         this.rawString = rawString
 
         if (rawString) {
@@ -87,6 +89,7 @@ class SimpleRelocator implements Relocator {
 
         this.includes = normalizePatterns(includes)
         this.excludes = normalizePatterns(excludes)
+        this.excludeSources = normalizePatterns(excludeSources)
     }
 
     SimpleRelocator include(String pattern) {
@@ -99,6 +102,11 @@ class SimpleRelocator implements Relocator {
         return this
     }
 
+    SimpleRelocator excludeSource(String pattern) {
+        this.excludeSources.addAll normalizePatterns([pattern])
+        return this
+    }
+
     private static Set<String> normalizePatterns(Collection<String> patterns) {
         Set<String> normalized = null
 
@@ -149,6 +157,17 @@ class SimpleRelocator implements Relocator {
         return false
     }
 
+    private boolean isExcludedSource(String srcPath) {
+        if (srcPath != null && excludeSources != null && !excludeSources.isEmpty()) {
+            for (String excludeSource : excludeSources) {
+                if (srcPath.startsWith(excludeSource)) {
+                    return true
+                }
+            }
+        }
+        return false
+    }
+
     boolean canRelocatePath(String path) {
         if (rawString) {
             return Pattern.compile(pathPattern).matcher(path).find()
@@ -182,6 +201,10 @@ class SimpleRelocator implements Relocator {
                 canRelocatePath(className.replace('.', '/'))
     }
 
+    boolean canRelocateSourceFile(String sourceFilePath) {
+        return !isExcludedSource(sourceFilePath)
+    }
+
     String relocatePath(RelocatePathContext context) {
         String path = context.path
         context.stats.relocate(pathPattern, shadedPathPattern)
@@ -242,4 +265,14 @@ class SimpleRelocator implements Relocator {
     boolean getRawString() {
         return rawString
     }
+
+    @Input
+    Set<String> getExcludeSources() {
+        return excludeSources
+    }
+
+    @Override
+    public String toString() {
+        return String.format("Relocate[%s->%s][%s]", pattern, shadedPattern, excludeSources)
+    }
 }

src/main/groovy/com/github/jengelman/gradle/plugins/shadow/tasks/ShadowCopyAction.groovy

@@ -347,6 +347,7 @@ class ShadowCopyAction implements CopyAction {
             // that use the constant pool to determine the dependencies of a class.
             ClassWriter cw = new ClassWriter(0)
 
+            remapper.currentFilePath = path
             ClassVisitor cv = new ClassRemapper(cw, remapper)
 
             try {

src/test/groovy/com/github/jengelman/gradle/plugins/shadow/relocation/SimpleRelocatorTest.groovy

@@ -42,6 +42,21 @@ class SimpleRelocatorTest extends TestCase {
       stats = new ShadowStats()
     }
 
+    void testIsExcludedSource() {
+        SimpleRelocator relocator
+
+        println("hello")
+        relocator = new SimpleRelocator("org.foo", null, null, null)
+        relocator.excludeSource "org/apache/iceberg/spark/parquet"
+        relocator.excludeSource "org/apache/spark/sql/execution/datasources/parquet"
+        println(relocator)
+
+        assertEquals(false, relocator.canRelocateSourceFile("org/apache/iceberg/spark/parquet/SparkNativeParquet.class"))
+        assertEquals(false, relocator.canRelocateSourceFile("org/apache/iceberg/spark/parquet/SparkNativeParquet\$.class"))
+        assertEquals(false, relocator.canRelocateSourceFile("org/apache/spark/sql/execution/datasources/parquet/v1.class"))
+        assertEquals(true, relocator.canRelocateSourceFile("org/foo/Class.class"))
+    }
+
     void testCanRelocatePath() {
         SimpleRelocator relocator