package com.sosnitzka.taiga.traits;

import net.minecraft.block.material.Material;
import net.minecraft.block.state.IBlockState;
import net.minecraft.entity.EntityLivingBase;
import net.minecraft.init.Blocks;
import net.minecraft.item.ItemStack;
import net.minecraft.util.math.BlockPos;
import net.minecraft.util.text.TextFormatting;
import net.minecraft.world.World;
import slimeknights.tconstruct.library.traits.AbstractTrait;
import slimeknights.tconstruct.library.utils.ToolHelper;

public class TraitOrganizing extends AbstractTrait {

    private static final float chance = 0.02f;

    public TraitOrganizing() {
        super("organizing", TextFormatting.GREEN);
    }

    @Override
    public void afterBlockBreak(ItemStack tool, World world, IBlockState state, BlockPos pos, EntityLivingBase player, boolean wasEffective) {
        if (!world.isRemote && (state.getMaterial() == Material.ROCK) && random.nextFloat() < chance) {
            world.setBlockState(pos, Blocks.LOG.getDefaultState());
            ToolHelper.healTool(tool, random.nextInt(5), player);
        }
    }
}