SourceFile remover

Author: PhoenixVX

gui/src/main/java/org/mcphackers/mcp/main/MainGUI.java

@@ -58,7 +58,7 @@ public class MainGUI extends MCP {
 	public static final TaskMode[] TASKS = {TaskMode.DECOMPILE, TaskMode.RECOMPILE, TaskMode.REOBFUSCATE, TaskMode.BUILD, TaskMode.UPDATE_MD5, TaskMode.CREATE_PATCH};
 	public static final String[] TABS = {"task.decompile", "task.recompile", "task.reobfuscate", "task.build", "options.running"};
 	public static final TaskParameter[][] TAB_PARAMETERS = {
-			{TaskParameter.PATCHES, TaskParameter.FERNFLOWER_OPTIONS, TaskParameter.IGNORED_PACKAGES, TaskParameter.OUTPUT_SRC, TaskParameter.DECOMPILE_RESOURCES, TaskParameter.GUESS_GENERICS, TaskParameter.STRIP_GENERICS},
+			{TaskParameter.PATCHES, TaskParameter.FERNFLOWER_OPTIONS, TaskParameter.IGNORED_PACKAGES, TaskParameter.OUTPUT_SRC, TaskParameter.DECOMPILE_RESOURCES, TaskParameter.GUESS_GENERICS, TaskParameter.STRIP_GENERICS, TaskParameter.STRIP_SOURCE_FILE},
 			{TaskParameter.SOURCE_VERSION, TaskParameter.TARGET_VERSION, TaskParameter.JAVA_HOME}, {TaskParameter.OBFUSCATION, TaskParameter.SRG_OBFUSCATION, TaskParameter.EXCLUDED_CLASSES},
 			{TaskParameter.FULL_BUILD}, {TaskParameter.RUN_BUILD, TaskParameter.RUN_ARGS, TaskParameter.GAME_ARGS}
 	};

src/main/java/org/mcphackers/mcp/tasks/TaskReobfuscate.java

@@ -17,6 +17,7 @@
 import org.mcphackers.mcp.MCPPaths;
 import org.mcphackers.mcp.tasks.mode.TaskParameter;
 import org.mcphackers.mcp.tools.FileUtil;
+import org.mcphackers.mcp.tools.injector.SourceFileTransformer;
 import org.mcphackers.mcp.tools.mappings.MappingUtil;
 import org.mcphackers.rdi.injector.data.ClassStorage;
 import org.mcphackers.rdi.injector.data.Mappings;
@@ -59,6 +60,7 @@ protected Stage[] setStages() {
 
 	private void reobfuscate() throws IOException {
 		final Path reobfBin = MCPPaths.get(mcp, BIN, side);
+		final boolean stripSourceFile = mcp.getOptions().getBooleanParameter(TaskParameter.STRIP_SOURCE_FILE);
 
 		Side[] sides = side == Side.MERGED ? new Side[]{Side.CLIENT, Side.SERVER} : new Side[]{side};
 
@@ -80,6 +82,9 @@ private void reobfuscate() throws IOException {
 			if (mappings != null) {
 				injector.applyMappings(mappings);
 			}
+			if (stripSourceFile) {
+				injector.addTransform(SourceFileTransformer::removeSourceFileAttributes);
+			}
 			injector.transform();
 			new ClassStorageWriter(injector.getStorage(), ClassWriter.COMPUTE_MAXS).write(Files.newOutputStream(reobfJar));
 

src/main/java/org/mcphackers/mcp/tasks/mode/TaskParameter.java

@@ -34,7 +34,8 @@ public enum TaskParameter {
 	DECOMPILE_RESOURCES("resources", Boolean.class, false),
 	GUESS_GENERICS("generics", Boolean.class, false),
 	STRIP_GENERICS("stripgenerics", Boolean.class, false),
-	OUTPUT_SRC("outputsrc", Boolean.class, true);
+	OUTPUT_SRC("outputsrc", Boolean.class, true),
+	STRIP_SOURCE_FILE("stripsourcefile", Boolean.class, true);
 
 	public static final TaskParameter[] VALUES = TaskParameter.values();
 

src/main/java/org/mcphackers/mcp/tools/injector/SourceFileTransformer.java

@@ -0,0 +1,15 @@
+package org.mcphackers.mcp.tools.injector;
+
+import org.mcphackers.rdi.injector.data.ClassStorage;
+import org.mcphackers.rdi.injector.transform.Injection;
+import org.objectweb.asm.tree.ClassNode;
+
+public abstract class SourceFileTransformer implements Injection {
+	public abstract void transform(ClassStorage storage);
+
+	public static void removeSourceFileAttributes(ClassStorage storage) {
+		for (ClassNode clazz : storage.getClasses()) {
+			clazz.sourceFile = null;
+		}
+	}
+}

src/main/resources/lang/en_US.lang

@@ -123,6 +123,7 @@ task.param.resources = Keep resources
 task.param.generics = Guess generics
 task.param.stripgenerics = Strip generics
 task.param.outputsrc = Output source
+task.param.stripsourcefile = Strip SourceFile attribute
 
 tasks.success = Finished successfully!
 tasks.warning = Finished with warnings!