package emu.grasscutter.game.props;

import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.function.Consumer;

import static java.util.Map.entry;

import java.util.Arrays;
import java.util.stream.Stream;

import it.unimi.dsi.fastutil.ints.Int2ObjectMap;
import it.unimi.dsi.fastutil.ints.Int2ObjectOpenHashMap;
import lombok.Getter;

public enum FightProperty {
    FIGHT_PROP_NONE(0),
    FIGHT_PROP_BASE_HP(1),
    FIGHT_PROP_HP(2),
    FIGHT_PROP_HP_PERCENT(3),
    FIGHT_PROP_BASE_ATTACK(4),
    FIGHT_PROP_ATTACK(5),
    FIGHT_PROP_ATTACK_PERCENT(6),
    FIGHT_PROP_BASE_DEFENSE(7),
    FIGHT_PROP_DEFENSE(8),
    FIGHT_PROP_DEFENSE_PERCENT(9),
    FIGHT_PROP_BASE_SPEED(10),
    FIGHT_PROP_SPEED_PERCENT(11),
    FIGHT_PROP_HP_MP_PERCENT(12),
    FIGHT_PROP_ATTACK_MP_PERCENT(13),
    FIGHT_PROP_CRITICAL(20),
    FIGHT_PROP_ANTI_CRITICAL(21),
    FIGHT_PROP_CRITICAL_HURT(22),
    FIGHT_PROP_CHARGE_EFFICIENCY(23),
    FIGHT_PROP_ADD_HURT(24),
    FIGHT_PROP_SUB_HURT(25),
    FIGHT_PROP_HEAL_ADD(26),
    FIGHT_PROP_HEALED_ADD(27),
    FIGHT_PROP_ELEMENT_MASTERY(28),
    FIGHT_PROP_PHYSICAL_SUB_HURT(29),
    FIGHT_PROP_PHYSICAL_ADD_HURT(30),
    FIGHT_PROP_DEFENCE_IGNORE_RATIO(31),
    FIGHT_PROP_DEFENCE_IGNORE_DELTA(32),
    FIGHT_PROP_FIRE_ADD_HURT(40),
    FIGHT_PROP_ELEC_ADD_HURT(41),
    FIGHT_PROP_WATER_ADD_HURT(42),
    FIGHT_PROP_GRASS_ADD_HURT(43),
    FIGHT_PROP_WIND_ADD_HURT(44),
    FIGHT_PROP_ROCK_ADD_HURT(45),
    FIGHT_PROP_ICE_ADD_HURT(46),
    FIGHT_PROP_HIT_HEAD_ADD_HURT(47),
    FIGHT_PROP_FIRE_SUB_HURT(50),
    FIGHT_PROP_ELEC_SUB_HURT(51),
    FIGHT_PROP_WATER_SUB_HURT(52),
    FIGHT_PROP_GRASS_SUB_HURT(53),
    FIGHT_PROP_WIND_SUB_HURT(54),
    FIGHT_PROP_ROCK_SUB_HURT(55),
    FIGHT_PROP_ICE_SUB_HURT(56),
    FIGHT_PROP_EFFECT_HIT(60),
    FIGHT_PROP_EFFECT_RESIST(61),
    FIGHT_PROP_FREEZE_RESIST(62),
    FIGHT_PROP_TORPOR_RESIST(63),
    FIGHT_PROP_DIZZY_RESIST(64),
    FIGHT_PROP_FREEZE_SHORTEN(65),
    FIGHT_PROP_TORPOR_SHORTEN(66),
    FIGHT_PROP_DIZZY_SHORTEN(67),
    FIGHT_PROP_MAX_FIRE_ENERGY(70),
    FIGHT_PROP_MAX_ELEC_ENERGY(71),
    FIGHT_PROP_MAX_WATER_ENERGY(72),
    FIGHT_PROP_MAX_GRASS_ENERGY(73),
    FIGHT_PROP_MAX_WIND_ENERGY(74),
    FIGHT_PROP_MAX_ICE_ENERGY(75),
    FIGHT_PROP_MAX_ROCK_ENERGY(76),
    FIGHT_PROP_SKILL_CD_MINUS_RATIO(80),
    FIGHT_PROP_SHIELD_COST_MINUS_RATIO(81),
    FIGHT_PROP_CUR_FIRE_ENERGY(1000),
    FIGHT_PROP_CUR_ELEC_ENERGY(1001),
    FIGHT_PROP_CUR_WATER_ENERGY(1002),
    FIGHT_PROP_CUR_GRASS_ENERGY(1003),
    FIGHT_PROP_CUR_WIND_ENERGY(1004),
    FIGHT_PROP_CUR_ICE_ENERGY(1005),
    FIGHT_PROP_CUR_ROCK_ENERGY(1006),
    FIGHT_PROP_CUR_HP(1010),
    FIGHT_PROP_MAX_HP(2000),
    FIGHT_PROP_CUR_ATTACK(2001),
    FIGHT_PROP_CUR_DEFENSE(2002),
    FIGHT_PROP_CUR_SPEED(2003),
    FIGHT_PROP_NONEXTRA_ATTACK(3000),
    FIGHT_PROP_NONEXTRA_DEFENSE(3001),
    FIGHT_PROP_NONEXTRA_CRITICAL(3002),
    FIGHT_PROP_NONEXTRA_ANTI_CRITICAL(3003),
    FIGHT_PROP_NONEXTRA_CRITICAL_HURT(3004),
    FIGHT_PROP_NONEXTRA_CHARGE_EFFICIENCY(3005),
    FIGHT_PROP_NONEXTRA_ELEMENT_MASTERY(3006),
    FIGHT_PROP_NONEXTRA_PHYSICAL_SUB_HURT(3007),
    FIGHT_PROP_NONEXTRA_FIRE_ADD_HURT(3008),
    FIGHT_PROP_NONEXTRA_ELEC_ADD_HURT(3009),
    FIGHT_PROP_NONEXTRA_WATER_ADD_HURT(3010),
    FIGHT_PROP_NONEXTRA_GRASS_ADD_HURT(3011),
    FIGHT_PROP_NONEXTRA_WIND_ADD_HURT(3012),
    FIGHT_PROP_NONEXTRA_ROCK_ADD_HURT(3013),
    FIGHT_PROP_NONEXTRA_ICE_ADD_HURT(3014),
    FIGHT_PROP_NONEXTRA_FIRE_SUB_HURT(3015),
    FIGHT_PROP_NONEXTRA_ELEC_SUB_HURT(3016),
    FIGHT_PROP_NONEXTRA_WATER_SUB_HURT(3017),
    FIGHT_PROP_NONEXTRA_GRASS_SUB_HURT(3018),
    FIGHT_PROP_NONEXTRA_WIND_SUB_HURT(3019),
    FIGHT_PROP_NONEXTRA_ROCK_SUB_HURT(3020),
    FIGHT_PROP_NONEXTRA_ICE_SUB_HURT(3021),
    FIGHT_PROP_NONEXTRA_SKILL_CD_MINUS_RATIO(3022),
    FIGHT_PROP_NONEXTRA_SHIELD_COST_MINUS_RATIO(3023),
    FIGHT_PROP_NONEXTRA_PHYSICAL_ADD_HURT(3024);

    private final int id;
    private static final Int2ObjectMap<FightProperty> map = new Int2ObjectOpenHashMap<>();
    private static final Map<String, FightProperty> stringMap = new HashMap<>();

    public static final int[] fightProps = new int[] {1, 4, 7, 20, 21, 22, 23, 26, 27, 28, 29, 30, 40, 41, 42, 43, 44, 45, 46, 50, 51, 52, 53, 54, 55, 56, 2000, 2001, 2002, 2003, 1010};

    static {
        Stream.of(values()).forEach(e -> {
            map.put(e.getId(), e);
            stringMap.put(e.name(), e);
        });
    }

    private FightProperty(int id) {
        this.id = id;
    }

    public int getId() {
        return id;
    }

    public static FightProperty getPropById(int value) {
        return map.getOrDefault(value, FIGHT_PROP_NONE);
    }

    public static FightProperty getPropByName(String name) {
        return stringMap.getOrDefault(name, FIGHT_PROP_NONE);
    }

    public static FightProperty getPropByShortName(String name) {
        return shortNameMap.getOrDefault(name, FIGHT_PROP_NONE);
    }

    public static Set<String> getShortNames() {
        return shortNameMap.keySet();
    }

    // This was originally for relic properties so some names might not be applicable for e.g. setstats
    private static final Map<String, FightProperty> shortNameMap = Map.ofEntries(
        // Normal relic stats
        entry("hp", FIGHT_PROP_HP),
        entry("atk", FIGHT_PROP_ATTACK),
        entry("def", FIGHT_PROP_DEFENSE),
        entry("hp%", FIGHT_PROP_HP_PERCENT),
        entry("atk%", FIGHT_PROP_ATTACK_PERCENT),
        entry("def%", FIGHT_PROP_DEFENSE_PERCENT),
        entry("em", FIGHT_PROP_ELEMENT_MASTERY),
        entry("er", FIGHT_PROP_CHARGE_EFFICIENCY),
        entry("hb", FIGHT_PROP_HEAL_ADD),
        entry("heal", FIGHT_PROP_HEAL_ADD),
        entry("cd", FIGHT_PROP_CRITICAL_HURT),
        entry("cdmg", FIGHT_PROP_CRITICAL_HURT),
        entry("cr", FIGHT_PROP_CRITICAL),
        entry("crate", FIGHT_PROP_CRITICAL),
        entry("phys%", FIGHT_PROP_PHYSICAL_ADD_HURT),
        entry("dendro%", FIGHT_PROP_GRASS_ADD_HURT),
        entry("geo%", FIGHT_PROP_ROCK_ADD_HURT),
        entry("anemo%", FIGHT_PROP_WIND_ADD_HURT),
        entry("hydro%", FIGHT_PROP_WATER_ADD_HURT),
        entry("cryo%", FIGHT_PROP_ICE_ADD_HURT),
        entry("electro%", FIGHT_PROP_ELEC_ADD_HURT),
        entry("pyro%", FIGHT_PROP_FIRE_ADD_HURT),
        // Other stats
        entry("maxhp", FIGHT_PROP_MAX_HP),
        entry("dmg", FIGHT_PROP_ADD_HURT),  // This seems to get reset after attacks
        entry("cdr", FIGHT_PROP_SKILL_CD_MINUS_RATIO),
        entry("heali", FIGHT_PROP_HEALED_ADD),
        entry("shield", FIGHT_PROP_SHIELD_COST_MINUS_RATIO),
        entry("defi", FIGHT_PROP_DEFENCE_IGNORE_RATIO),
        entry("resall", FIGHT_PROP_SUB_HURT),  // This seems to get reset after attacks
        entry("resanemo", FIGHT_PROP_WIND_SUB_HURT),
        entry("rescryo", FIGHT_PROP_ICE_SUB_HURT),
        entry("resdendro", FIGHT_PROP_GRASS_SUB_HURT),
        entry("reselectro", FIGHT_PROP_ELEC_SUB_HURT),
        entry("resgeo", FIGHT_PROP_ROCK_SUB_HURT),
        entry("reshydro", FIGHT_PROP_WATER_SUB_HURT),
        entry("respyro", FIGHT_PROP_FIRE_SUB_HURT),
        entry("resphys", FIGHT_PROP_PHYSICAL_SUB_HURT)
    );

    private static final List<FightProperty> flatProps = Arrays.asList(
        FIGHT_PROP_BASE_HP, FIGHT_PROP_HP, FIGHT_PROP_BASE_ATTACK, FIGHT_PROP_ATTACK, FIGHT_PROP_BASE_DEFENSE,
        FIGHT_PROP_DEFENSE, FIGHT_PROP_HEALED_ADD, FIGHT_PROP_CUR_FIRE_ENERGY, FIGHT_PROP_CUR_ELEC_ENERGY,
        FIGHT_PROP_CUR_WATER_ENERGY, FIGHT_PROP_CUR_GRASS_ENERGY, FIGHT_PROP_CUR_WIND_ENERGY, FIGHT_PROP_CUR_ICE_ENERGY,
        FIGHT_PROP_CUR_ROCK_ENERGY, FIGHT_PROP_CUR_HP, FIGHT_PROP_MAX_HP, FIGHT_PROP_CUR_ATTACK, FIGHT_PROP_CUR_DEFENSE);

    @Getter
    public static class CompoundProperty {
        private FightProperty result;
        private FightProperty base;
        private FightProperty percent;
        private FightProperty flat;

        public CompoundProperty(FightProperty result, FightProperty base, FightProperty percent, FightProperty flat) {
            this.result = result;
            this.base = base;
            this.percent = percent;
            this.flat = flat;
        }
    }

    private static Map<FightProperty, CompoundProperty> compoundProperties = Map.ofEntries(
        entry(FIGHT_PROP_MAX_HP, new CompoundProperty(FIGHT_PROP_MAX_HP, FIGHT_PROP_BASE_HP, FIGHT_PROP_HP_PERCENT, FIGHT_PROP_HP)),
        entry(FIGHT_PROP_CUR_ATTACK, new CompoundProperty(FIGHT_PROP_CUR_ATTACK, FIGHT_PROP_BASE_ATTACK, FIGHT_PROP_ATTACK_PERCENT, FIGHT_PROP_ATTACK)),
        entry(FIGHT_PROP_CUR_DEFENSE, new CompoundProperty(FIGHT_PROP_CUR_DEFENSE, FIGHT_PROP_BASE_DEFENSE, FIGHT_PROP_DEFENSE_PERCENT, FIGHT_PROP_DEFENSE))
    );

    public static CompoundProperty getCompoundProperty(FightProperty result) {
        return compoundProperties.get(result);
    }

    public static void forEachCompoundProperty(Consumer<CompoundProperty> consumer) {
        compoundProperties.values().forEach(consumer);
    }

    public static boolean isPercentage(FightProperty prop) {
        return !flatProps.contains(prop);
    }
}
