anvillib/common/src/main/java/net/anvilcraft/anvillib/cosmetics/CosmeticModel.java

package net.anvilcraft.anvillib.cosmetics;

import net.anvilcraft.anvillib.mixin.accessor.AnimatedGeoModelAccessor;
import net.minecraft.util.Identifier;
import software.bernie.geckolib3.core.IAnimatable;
import software.bernie.geckolib3.core.builder.Animation;
import software.bernie.geckolib3.file.AnimationFile;
import software.bernie.geckolib3.geo.exception.GeckoLibException;
import software.bernie.geckolib3.geo.render.built.GeoBone;
import software.bernie.geckolib3.geo.render.built.GeoModel;
import software.bernie.geckolib3.model.AnimatedGeoModel;
import software.bernie.geckolib3.resource.GeckoLibCache;

public class CosmeticModel extends AnimatedGeoModel<CosmeticItem> {
    @Override
    public Identifier getAnimationFileLocation(CosmeticItem animatable) {
        return animatable.getCosmetic().getAnimationFileLocation();
    }

    @Override
    public Identifier getModelLocation(CosmeticItem animatable) {
        return animatable.getCosmetic().getModelLocation();
    }

    @Override
    public Identifier getTextureLocation(CosmeticItem animatable) {
        return animatable.getCosmetic().getTextureLocation();
    }

    @Override
	public Animation getAnimation(String name, IAnimatable animatable) {
        Identifier location = ((CosmeticItem)animatable).getCosmetic().getAnimationFileLocation();
		AnimationFile animation = CosmeticsManager.getAnimations(location);
        if (animation == null) {
            animation = GeckoLibCache.getInstance().getAnimations().get(location);
        }
        
		if (animation == null) {
			throw new GeckoLibException(location,
					"Could not find animation file. Please double check name.");
		}

		return animation.getAnimation(name);
	}

    @Override
    public GeoModel getModel(Identifier location) {
        GeoModel model = CosmeticsManager.getModel(location);
        if (model == null) {
            model = GeckoLibCache.getInstance().getGeoModels().get(location);
        }

        AnimatedGeoModelAccessor accessor = (AnimatedGeoModelAccessor)this;
        if (model != accessor.getCurrentModel()) {
			accessor.getAnimationProcessor().clearModelRendererList();
			accessor.setCurrentModel(model);

			for (GeoBone bone : model.topLevelBones) {
				registerBone(bone);
			}
		}

		return model;
    }
}