39 lines
		
	
	
	
		
			1.3 KiB
		
	
	
	
		
			Java
		
	
	
	
	
	
			
		
		
	
	
			39 lines
		
	
	
	
		
			1.3 KiB
		
	
	
	
		
			Java
		
	
	
	
	
	
| package net.minecraft.util.datafix.fixes;
 | |
| 
 | |
| import com.mojang.datafixers.DSL;
 | |
| import com.mojang.datafixers.DataFix;
 | |
| import com.mojang.datafixers.TypeRewriteRule;
 | |
| import com.mojang.datafixers.schemas.Schema;
 | |
| import com.mojang.serialization.Dynamic;
 | |
| import java.util.List;
 | |
| import java.util.Optional;
 | |
| 
 | |
| public class LevelLegacyWorldGenSettingsFix extends DataFix {
 | |
| 	private static final String WORLD_GEN_SETTINGS = "WorldGenSettings";
 | |
| 	private static final List<String> OLD_SETTINGS_KEYS = List.of(
 | |
| 		"RandomSeed", "generatorName", "generatorOptions", "generatorVersion", "legacy_custom_options", "MapFeatures", "BonusChest"
 | |
| 	);
 | |
| 
 | |
| 	public LevelLegacyWorldGenSettingsFix(Schema outputSchema) {
 | |
| 		super(outputSchema, false);
 | |
| 	}
 | |
| 
 | |
| 	@Override
 | |
| 	protected TypeRewriteRule makeRule() {
 | |
| 		return this.fixTypeEverywhereTyped(
 | |
| 			"LevelLegacyWorldGenSettingsFix", this.getInputSchema().getType(References.LEVEL), typed -> typed.update(DSL.remainderFinder(), dynamic -> {
 | |
| 				Dynamic<?> dynamic2 = dynamic.get("WorldGenSettings").orElseEmptyMap();
 | |
| 
 | |
| 				for (String string : OLD_SETTINGS_KEYS) {
 | |
| 					Optional<? extends Dynamic<?>> optional = dynamic.get(string).result();
 | |
| 					if (optional.isPresent()) {
 | |
| 						dynamic = dynamic.remove(string);
 | |
| 						dynamic2 = dynamic2.set(string, (Dynamic<?>)optional.get());
 | |
| 					}
 | |
| 				}
 | |
| 
 | |
| 				return dynamic.set("WorldGenSettings", dynamic2);
 | |
| 			})
 | |
| 		);
 | |
| 	}
 | |
| }
 |