package carpet.script.api;

import carpet.fakes.SpawnHelperInnerInterface;
import carpet.script.CarpetContext;
import carpet.script.Expression;
import carpet.script.LazyValue;
import carpet.script.exception.InternalExpressionException;
import carpet.script.utils.SystemInfo;
import carpet.script.value.ListValue;
import carpet.script.value.MapValue;
import carpet.script.value.NumericValue;
import carpet.script.value.StringValue;
import carpet.script.value.Value;
import carpet.utils.SpawnReporter;
import it.unimi.dsi.fastutil.objects.Object2IntMap;
import net.minecraft.entity.SpawnGroup;
import net.minecraft.server.world.ServerWorld;
import net.minecraft.world.SpawnHelper;

import java.util.HashMap;
import java.util.Locale;
import java.util.Map;

public class Monitoring {

    public static void apply(Expression expression)
    {
        expression.addLazyFunction("system_info", -1, (c, t, lv) ->
        {
            if (lv.size() == 0)
            {
                Value ret = SystemInfo.getAll((CarpetContext) c);
                return (cc, tt) -> ret;
            }
            if (lv.size() == 1) {
                String what = lv.get(0).evalValue(c).getString();
                Value res = SystemInfo.get(what, (CarpetContext) c);
                if (res == null) throw new InternalExpressionException("Unknown option for 'system_info': " + what);
                return (cc, tt) -> res;
            }
            throw new InternalExpressionException("'system_info' requires one or no parameters");
        });
        // game processed snooper functions
        expression.addLazyFunction("get_mob_counts", -1, (c, t, lv) ->
        {
            CarpetContext cc = (CarpetContext)c;
            ServerWorld world = cc.s.getWorld();
            SpawnHelper.Info info = world.getChunkManager().getSpawnInfo();
            if (info == null) return LazyValue.NULL;
            Object2IntMap<SpawnGroup> mobcounts = info.getGroupToCount();
            int chunks = ((SpawnHelperInnerInterface)info).cmGetChunkCount();
            Value retVal;
            if (lv.size() == 0)
            {
                Map<Value, Value> retDict = new HashMap<>();
                for (SpawnGroup category: mobcounts.keySet())
                {
                    int currentCap = (int)(category.getCapacity() * chunks / SpawnReporter.currentMagicNumber()); // MAGIC_NUMBER
                    retDict.put(
                            new StringValue(category.asString().toLowerCase(Locale.ROOT)),
                            ListValue.of(
                                    new NumericValue(mobcounts.getInt(category)),
                                    new NumericValue(currentCap))
                    );
                }
                retVal = MapValue.wrap(retDict);
            }
            else
            {
                String catString = lv.get(0).evalValue(c).getString();
                SpawnGroup cat = SpawnGroup.byName(catString.toLowerCase(Locale.ROOT));
                if (cat == null) throw new InternalExpressionException("Unreconized mob category: "+catString);
                retVal = ListValue.of(
                        new NumericValue(mobcounts.getInt(cat)),
                        new NumericValue((int)(cat.getCapacity() * chunks / SpawnReporter.currentMagicNumber()))
                );
            }
            return (_c, _t) -> retVal;
        });
    }
}