*   This file is part of Skript.
 *  Skript is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation, either version 3 of the License, or
 *  (at your option) any later version.
 *  Skript is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  GNU General Public License for more details.
 *  You should have received a copy of the GNU General Public License
 *  along with Skript.  If not, see <http://www.gnu.org/licenses/>.
 * Copyright 2011-2017 Peter G├╝ttinger and contributors
package ch.njol.skript.expressions;

import java.util.stream.Stream;

import org.bukkit.enchantments.Enchantment;
import org.bukkit.event.Event;
import org.eclipse.jdt.annotation.Nullable;

import ch.njol.skript.Skript;
import ch.njol.skript.aliases.ItemType;
import ch.njol.skript.classes.Changer.ChangeMode;
import ch.njol.skript.doc.Description;
import ch.njol.skript.doc.Examples;
import ch.njol.skript.doc.Name;
import ch.njol.skript.doc.Since;
import ch.njol.skript.lang.Expression;
import ch.njol.skript.lang.ExpressionType;
import ch.njol.skript.lang.SkriptParser.ParseResult;
import ch.njol.skript.lang.util.SimpleExpression;
import ch.njol.skript.util.EnchantmentType;
import ch.njol.util.Kleenean;
import ch.njol.util.coll.CollectionUtils;

@Name("Enchantment Level")
@Description("The level of a particular <a href='classes.html#enchantment'>enchantment</a> on an item.")
@Examples({"player's tool is a sword of sharpness:",
		"	message \"You have a sword of sharpness %level of sharpness of the player's tool% equipped\""})
public class ExprEnchantmentLevel extends SimpleExpression<Integer> {
	static {
		Skript.registerExpression(ExprEnchantmentLevel.class, Integer.class, ExpressionType.PROPERTY,
				"[the] [enchant[ment]] level[s] of %enchantments% (on|of) %itemtypes%",
				"[the] %enchantments% [enchant[ment]] level[s] (on|of) %itemtypes%",
				"%itemtypes%'[s] %enchantments% [enchant[ment]] level[s]",
				"%itemtypes%'[s] [enchant[ment]] level[s] of %enchantments%");
	private Expression<ItemType> items;
	private Expression<Enchantment> enchants;
	@SuppressWarnings({"unchecked", "null"})
	public boolean init(Expression<?>[] exprs, int matchedPattern, Kleenean isDelayed, ParseResult parseResult) {
		int i = matchedPattern < 2 ? 1 : 0;
		items = (Expression<ItemType>) exprs[i];
		enchants = (Expression<Enchantment>) exprs[i ^ 1];
		return true;
	protected Integer[] get(final Event e) {
		Enchantment[] enchantments = enchants.getArray(e);
		return Stream.of(items.getArray(e))
			.filter(enchantment -> CollectionUtils.contains(enchantments, enchantment.getType()))
	public boolean isSingle() {
		return items.isSingle() && enchants.isSingle();
	public Class<? extends Integer> getReturnType() {
		return Integer.class;
	public Class<?>[] acceptChange(ChangeMode mode) {
		switch (mode) {
			case SET:
			case REMOVE:
			case ADD:
				return CollectionUtils.array(Number.class);
				return null;
	public void change(Event e, @Nullable Object[] delta, ChangeMode mode) {
		if (delta == null)
		ItemType[] source = items.getArray(e);
		Enchantment[] enchantments = enchants.getArray(e);
		int newLevel = ((Number) delta[0]).intValue();
		for (ItemType item : source) {
			if (!item.hasAnyEnchantments(enchantments))
			EnchantmentType[] enchants = item.getEnchantmentTypes();
			assert enchants != null; // Can't be null at this point due to the above check
			for (EnchantmentType enchant : enchants) {
				Enchantment type = enchant.getType();
				int changed = newLevel;
				assert type != null;
				if (mode == ChangeMode.ADD)
					changed = Math.max(0, enchant.getLevel() + changed);
				else if (mode == ChangeMode.REMOVE)
					changed = Math.max(0, enchant.getLevel() - changed);
				if (changed > 0)
					item.addEnchantments(new EnchantmentType(type, newLevel));
	public String toString(final @Nullable Event e, final boolean debug) {
		return "the level of " + enchants.toString(e, debug) + " of " + items.toString(e, debug);