com.llamalad7.mixinextras.utils.TargetDecorations Maven / Gradle / Ivy

Go to download
package com.llamalad7.mixinextras.utils;

import org.spongepowered.asm.mixin.injection.struct.Target;

import java.util.HashMap;
import java.util.Map;
import java.util.WeakHashMap;
import java.util.function.Supplier;
import java.util.function.UnaryOperator;

public class TargetDecorations {
    private static final Map> IMPL = new WeakHashMap<>();

    public static boolean has(Target target, String key) {
        return IMPL.containsKey(target) && IMPL.get(target).containsKey(key);
    }

    @SuppressWarnings("unchecked")
    public static  T get(Target target, String key) {
        return IMPL.containsKey(target) ? (T) IMPL.get(target).get(key) : null;
    }

    @SuppressWarnings("unchecked")
    public static  T getOrPut(Target target, String key, Supplier supplier) {
        return (T) IMPL.computeIfAbsent(target, k -> new HashMap<>()).computeIfAbsent(key, k -> supplier.get());
    }

    public static void put(Target target, String key, Object value) {
        IMPL.computeIfAbsent(target, k -> new HashMap<>()).put(key, value);
    }

    public static  void modify(Target target, String key, UnaryOperator operator) {
        IMPL.computeIfAbsent(target, k -> new HashMap<>()).put(key, operator.apply(get(target, key)));
    }

    public static void remove(Target target, String key) {
        if (IMPL.containsKey(target)) {
            IMPL.get(target).remove(key);
        }
    }
}