src/main/java/dev/mayaqq/ygasi/mixin/AdvancementRewardsMixin.java


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33

package dev.mayaqq.ygasi.mixin;

import dev.mayaqq.ygasi.registry.ConfigRegistry;
import net.minecraft.advancement.AdvancementRewards;
import net.minecraft.server.network.ServerPlayerEntity;
import net.minecraft.util.Identifier;
import org.spongepowered.asm.mixin.Final;
import org.spongepowered.asm.mixin.Mixin;
import org.spongepowered.asm.mixin.Shadow;
import org.spongepowered.asm.mixin.injection.At;
import org.spongepowered.asm.mixin.injection.Inject;
import org.spongepowered.asm.mixin.injection.callback.CallbackInfo;

import static dev.mayaqq.ygasi.registry.StatRegistry.SKILL_POINTS;
import static dev.mayaqq.ygasi.registry.StatRegistry.SKILL_POINTS_TOTAL;


@Mixin(AdvancementRewards.class)
public class AdvancementRewardsMixin {

    @Shadow @Final private Identifier[] recipes;
    @Shadow @Final private int experience;

    @Inject(method = "apply", at = @At("HEAD"))
    private void apply(ServerPlayerEntity player, CallbackInfo ci) {
        if (recipes.length == 0 && experience != 1) {
            player.increaseStat(SKILL_POINTS, ConfigRegistry.CONFIG.pointsRewarded);
            player.increaseStat(SKILL_POINTS_TOTAL, ConfigRegistry.CONFIG.pointsRewarded);
        } else if (experience == 1) {
            player.addExperience(-1);
        }
    }
}