remap with yarrnforge

2025-11-20 23:34:48 +00:00 · 2020-08-29 19:38:26 -05:00
parent fbcf26bdc9
commit 324519575c
566 changed files with 57853 additions and 0 deletions
--- a/src/main/java/dan200/computercraft/shared/TurtleUpgrades.java
+++ b/src/main/java/dan200/computercraft/shared/TurtleUpgrades.java
@@ -0,0 +1,177 @@
+/*
+ * This file is part of ComputerCraft - http://www.computercraft.info
+ * Copyright Daniel Ratcliffe, 2011-2020. Do not distribute without permission.
+ * Send enquiries to dratcliffe@gmail.com
+ */
+package dan200.computercraft.shared;
+
+import dan200.computercraft.ComputerCraft;
+import dan200.computercraft.api.turtle.ITurtleUpgrade;
+import dan200.computercraft.shared.computer.core.ComputerFamily;
+import dan200.computercraft.shared.util.InventoryUtil;
+import net.minecraft.item.ItemStack;
+import net.minecraftforge.fml.ModLoadingContext;
+
+import javax.annotation.Nonnull;
+import javax.annotation.Nullable;
+import java.util.*;
+import java.util.stream.Stream;
+
+public final class TurtleUpgrades
+{
+    private static class Wrapper
+    {
+        final ITurtleUpgrade upgrade;
+        final String id;
+        final String modId;
+        boolean enabled;
+
+        Wrapper( ITurtleUpgrade upgrade )
+        {
+            this.upgrade = upgrade;
+            this.id = upgrade.getUpgradeID().toString();
+            this.modId = ModLoadingContext.get().getActiveNamespace();
+            this.enabled = true;
+        }
+    }
+
+    private static ITurtleUpgrade[] vanilla;
+
+    private static final Map<String, ITurtleUpgrade> upgrades = new HashMap<>();
+    private static final IdentityHashMap<ITurtleUpgrade, Wrapper> wrappers = new IdentityHashMap<>();
+    private static boolean needsRebuild;
+
+    private TurtleUpgrades() {}
+
+    public static void register( @Nonnull ITurtleUpgrade upgrade )
+    {
+        Objects.requireNonNull( upgrade, "upgrade cannot be null" );
+        rebuild();
+
+        Wrapper wrapper = new Wrapper( upgrade );
+        String id = wrapper.id;
+        ITurtleUpgrade existing = upgrades.get( id );
+        if( existing != null )
+        {
+            throw new IllegalStateException( "Error registering '" + upgrade.getUnlocalisedAdjective() + " Turtle'. Upgrade ID '" + id + "' is already registered by '" + existing.getUnlocalisedAdjective() + " Turtle'" );
+        }
+
+        upgrades.put( id, upgrade );
+        wrappers.put( upgrade, wrapper );
+    }
+
+    @Nullable
+    public static ITurtleUpgrade get( String id )
+    {
+        rebuild();
+        return upgrades.get( id );
+    }
+
+    @Nullable
+    public static String getOwner( @Nonnull ITurtleUpgrade upgrade )
+    {
+        Wrapper wrapper = wrappers.get( upgrade );
+        return wrapper != null ? wrapper.modId : null;
+    }
+
+    public static ITurtleUpgrade get( @Nonnull ItemStack stack )
+    {
+        if( stack.isEmpty() ) return null;
+
+        for( Wrapper wrapper : wrappers.values() )
+        {
+            if( !wrapper.enabled ) continue;
+
+            ItemStack craftingStack = wrapper.upgrade.getCraftingItem();
+            if( !craftingStack.isEmpty() && InventoryUtil.areItemsSimilar( stack, craftingStack ) )
+            {
+                return wrapper.upgrade;
+            }
+        }
+
+        return null;
+    }
+
+    public static Stream<ITurtleUpgrade> getVanillaUpgrades()
+    {
+        if( vanilla == null )
+        {
+            vanilla = new ITurtleUpgrade[] {
+                // ComputerCraft upgrades
+                ComputerCraft.TurtleUpgrades.wirelessModemNormal,
+                ComputerCraft.TurtleUpgrades.wirelessModemAdvanced,
+                ComputerCraft.TurtleUpgrades.speaker,
+
+                // Vanilla Minecraft upgrades
+                ComputerCraft.TurtleUpgrades.diamondPickaxe,
+                ComputerCraft.TurtleUpgrades.diamondAxe,
+                ComputerCraft.TurtleUpgrades.diamondSword,
+                ComputerCraft.TurtleUpgrades.diamondShovel,
+                ComputerCraft.TurtleUpgrades.diamondHoe,
+                ComputerCraft.TurtleUpgrades.craftingTable,
+            };
+        }
+
+        return Arrays.stream( vanilla ).filter( x -> x != null && wrappers.get( x ).enabled );
+    }
+
+    public static Stream<ITurtleUpgrade> getUpgrades()
+    {
+        return wrappers.values().stream().filter( x -> x.enabled ).map( x -> x.upgrade );
+    }
+
+    public static boolean suitableForFamily( ComputerFamily family, ITurtleUpgrade upgrade )
+    {
+        return true;
+    }
+
+    /**
+     * Rebuild the cache of turtle upgrades. This is done before querying the cache or registering new upgrades.
+     */
+    private static void rebuild()
+    {
+        if( !needsRebuild ) return;
+
+        upgrades.clear();
+        for( Wrapper wrapper : wrappers.values() )
+        {
+            if( !wrapper.enabled ) continue;
+
+            ITurtleUpgrade existing = upgrades.get( wrapper.id );
+            if( existing != null )
+            {
+                ComputerCraft.log.error( "Error registering '" + wrapper.upgrade.getUnlocalisedAdjective() + " Turtle'." +
+                    " Upgrade ID '" + wrapper.id + "' is already registered by '" + existing.getUnlocalisedAdjective() + " Turtle'" );
+                continue;
+            }
+
+            upgrades.put( wrapper.id, wrapper.upgrade );
+        }
+
+        needsRebuild = false;
+    }
+
+    public static void enable( ITurtleUpgrade upgrade )
+    {
+        Wrapper wrapper = wrappers.get( upgrade );
+        if( wrapper.enabled ) return;
+
+        wrapper.enabled = true;
+        needsRebuild = true;
+    }
+
+    public static void disable( ITurtleUpgrade upgrade )
+    {
+        Wrapper wrapper = wrappers.get( upgrade );
+        if( !wrapper.enabled ) return;
+
+        wrapper.enabled = false;
+        upgrades.remove( wrapper.id );
+    }
+
+    public static void remove( ITurtleUpgrade upgrade )
+    {
+        wrappers.remove( upgrade );
+        needsRebuild = true;
+    }
+}