Try to improve ProcessingOutput type wrapping as well

Author: MaxNeedsSnacks

src/main/java/dev/latvian/mods/kubejs/create/KubeJSCreatePlugin.java

@@ -1,5 +1,6 @@
 package dev.latvian.mods.kubejs.create;
 
+import com.google.gson.JsonObject;
 import com.simibubi.create.Create;
 import com.simibubi.create.content.processing.recipe.ProcessingOutput;
 import com.simibubi.create.foundation.fluid.FluidIngredient;
@@ -16,6 +17,7 @@
 import dev.latvian.mods.kubejs.event.EventGroupRegistry;
 import dev.latvian.mods.kubejs.plugin.KubeJSPlugin;
 import dev.latvian.mods.kubejs.plugin.builtin.wrapper.ItemWrapper;
+import dev.latvian.mods.kubejs.plugin.builtin.wrapper.StringUtilsWrapper;
 import dev.latvian.mods.kubejs.recipe.component.RecipeComponentTypeRegistry;
 import dev.latvian.mods.kubejs.recipe.schema.RecipeSchemaRegistry;
 import dev.latvian.mods.kubejs.registry.BuilderTypeRegistry;
@@ -24,8 +26,15 @@
 import dev.latvian.mods.kubejs.script.TypeWrapperRegistry;
 import dev.latvian.mods.rhino.Context;
 import net.minecraft.core.registries.Registries;
+import net.minecraft.util.Mth;
+import net.minecraft.world.item.ItemStack;
+import net.minecraft.world.item.Items;
+import net.minecraft.world.level.ItemLike;
 import org.jetbrains.annotations.Nullable;
 
+import java.util.Map;
+import java.util.function.Function;
+
 public class KubeJSCreatePlugin implements KubeJSPlugin {
 	@Override
 	public void afterInit() {
@@ -58,13 +67,28 @@ public void registerTypeWrappers(TypeWrapperRegistry registry) {
 		registry.register(ProcessingOutput.class, KubeJSCreatePlugin::wrapProcessingOutput);
 	}
 
-	private static ProcessingOutput wrapProcessingOutput(Context cx, @Nullable Object from) {
-		if (from instanceof ProcessingOutput p) {
-			return p;
+	private static ProcessingOutput fromMapLike(Context cx, Object from, Function<String, Object> getter, boolean nested) {
+		var chance = (float) Mth.clamp(StringUtilsWrapper.parseDouble(getter.apply("chance"), 1.0), 0.0, 1.0);
+		if (nested) {
+			var output = ItemWrapper.wrap(cx, getter.apply("output"));
+			return new ProcessingOutput(output, chance);
+		} else {
+			return new ProcessingOutput(ItemWrapper.wrap(cx, from), chance);
 		}
+	}
 
-		var stack = ItemWrapper.wrap(cx, from);
-		return stack.isEmpty() ? ProcessingOutput.EMPTY : new ProcessingOutput(stack, 1F);
+	private static ProcessingOutput wrapProcessingOutput(Context cx, @Nullable Object from) {
+		return switch (from) {
+			case null -> ProcessingOutput.EMPTY;
+			case ProcessingOutput id -> id;
+			case ItemStack s -> s.isEmpty() ? ProcessingOutput.EMPTY : new ProcessingOutput(s, 1F);
+			case ItemLike i when i.asItem() == Items.AIR -> ProcessingOutput.EMPTY;
+			case ItemLike i -> new ProcessingOutput(i.asItem(), 1, 1F);
+			case JsonObject json when json.has("chance") -> fromMapLike(cx, json, json::get, json.has("output"));
+			case Map<?, ?> map when map.containsKey("chance") -> fromMapLike(cx, map, map::get, map.containsKey("output"));
+			// TODO: maybe a custom string-like type wrapper ("2x apple @ 20%" or something like that???)
+			default -> new ProcessingOutput(ItemWrapper.wrap(cx, from), 1F);
+		};
 	}
 
 	@Override