package com.schedule_i_weed.data; import net.minecraft.entity.effect.StatusEffect; import net.minecraft.entity.effect.StatusEffects; import net.minecraft.nbt.NbtCompound; import java.util.ArrayList; import java.util.List; public class StrainData { private StrainType baseType; private List ingredients; private int quality; private int price; private String customName; private List effects; public StrainData(StrainType baseType) { this.baseType = baseType; this.ingredients = new ArrayList<>(); this.quality = 100; this.price = baseType.getBasePrice(); this.customName = ""; this.effects = getDefaultEffects(baseType); } private List getDefaultEffects(StrainType type) { List defaultEffects = new ArrayList<>(); switch (type) { case OG_KUSH: defaultEffects.add(StatusEffects.REGENERATION); defaultEffects.add(StatusEffects.RESISTANCE); break; case SOUR_DIESEL: defaultEffects.add(StatusEffects.SPEED); defaultEffects.add(StatusEffects.HASTE); break; case GREEN_CRACK: defaultEffects.add(StatusEffects.JUMP_BOOST); defaultEffects.add(StatusEffects.SPEED); break; case GRANDDADDY_PURPLE: defaultEffects.add(StatusEffects.STRENGTH); defaultEffects.add(StatusEffects.NIGHT_VISION); break; } return defaultEffects; } public void addIngredient(String ingredient) { this.ingredients.add(ingredient); recalculateStats(); } private void recalculateStats() { int priceModifier = 0; List newEffects = new ArrayList<>(getDefaultEffects(baseType)); for (String ingredient : ingredients) { switch (ingredient) { case "donut": priceModifier += 50; quality += 5; newEffects.add(StatusEffects.SATURATION); break; case "banana": priceModifier += 30; quality += 10; newEffects.add(StatusEffects.REGENERATION); break; case "cola_can": priceModifier += 40; quality -= 5; newEffects.add(StatusEffects.SPEED); break; case "mouthwash": priceModifier += 80; quality -= 15; newEffects.add(StatusEffects.NAUSEA); break; case "viagra_pill": priceModifier += 120; quality += 15; newEffects.add(StatusEffects.STRENGTH); break; case "flu_medicine": priceModifier += 70; quality += 8; newEffects.add(StatusEffects.RESISTANCE); break; case "energy_drink": priceModifier += 90; quality -= 10; newEffects.add(StatusEffects.SPEED); newEffects.add(StatusEffects.HASTE); break; case "motor_oil": priceModifier += 150; quality -= 30; newEffects.add(StatusEffects.POISON); newEffects.add(StatusEffects.NAUSEA); break; case "battery": priceModifier += 100; quality -= 20; newEffects.add(StatusEffects.GLOWING); break; case "paracetamol": priceModifier += 60; quality += 12; newEffects.add(StatusEffects.REGENERATION); break; case "coffee_beans": priceModifier += 45; quality += 7; newEffects.add(StatusEffects.SPEED); break; case "bleach": priceModifier += 200; quality -= 40; newEffects.add(StatusEffects.WITHER); newEffects.add(StatusEffects.BLINDNESS); break; } } this.price = baseType.getBasePrice() + priceModifier; this.effects = newEffects; if (quality > 100) quality = 100; if (quality < 0) quality = 0; } public NbtCompound toNbt() { NbtCompound nbt = new NbtCompound(); nbt.putString("BaseType", baseType.getName()); nbt.putInt("Quality", quality); nbt.putInt("Price", price); nbt.putString("CustomName", customName); NbtCompound ingredientsNbt = new NbtCompound(); for (int i = 0; i < ingredients.size(); i++) { ingredientsNbt.putString("Ingredient" + i, ingredients.get(i)); } nbt.put("Ingredients", ingredientsNbt); return nbt; } public static StrainData fromNbt(NbtCompound nbt) { StrainType type = StrainType.fromName(nbt.getString("BaseType")); StrainData data = new StrainData(type); data.quality = nbt.getInt("Quality"); data.price = nbt.getInt("Price"); data.customName = nbt.getString("CustomName"); NbtCompound ingredientsNbt = nbt.getCompound("Ingredients"); int i = 0; while (ingredientsNbt.contains("Ingredient" + i)) { data.ingredients.add(ingredientsNbt.getString("Ingredient" + i)); i++; } data.recalculateStats(); return data; } public StrainType getBaseType() { return baseType; } public List getIngredients() { return ingredients; } public int getQuality() { return quality; } public int getPrice() { return price; } public String getCustomName() { return customName; } public void setCustomName(String name) { this.customName = name; } public List getEffects() { return effects; } }