package com.eu.habbo.habbohotel.items.interactions.wired.selector;

import com.eu.habbo.habbohotel.gameclients.GameClient;
import com.eu.habbo.habbohotel.items.Item;
import com.eu.habbo.habbohotel.items.interactions.InteractionWiredEffect;
import com.eu.habbo.habbohotel.items.interactions.wired.WiredSettings;
import com.eu.habbo.habbohotel.rooms.Room;
import com.eu.habbo.habbohotel.rooms.RoomUnit;
import com.eu.habbo.habbohotel.users.Habbo;
import com.eu.habbo.habbohotel.wired.WiredEffectType;
import com.eu.habbo.habbohotel.wired.core.WiredContext;
import com.eu.habbo.habbohotel.wired.core.WiredManager;
import com.eu.habbo.messages.ServerMessage;

import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.LinkedHashSet;
import java.util.Set;
import java.util.stream.Collectors;

public class WiredEffectUsersGroup extends InteractionWiredEffect {
    private static final int GROUP_CURRENT_ROOM = 0;
    private static final int GROUP_SELECTED = 1;

    public static final WiredEffectType type = WiredEffectType.USERS_GROUP_SELECTOR;

    private int groupType = GROUP_CURRENT_ROOM;
    private int selectedGroupId = 0;
    private boolean filterExisting = false;
    private boolean invert = false;

    public WiredEffectUsersGroup(ResultSet set, Item baseItem) throws SQLException {
        super(set, baseItem);
    }

    public WiredEffectUsersGroup(int id, int userId, Item item, String extradata, int limitedStack, int limitedSells) {
        super(id, userId, item, extradata, limitedStack, limitedSells);
    }

    @Override
    public void execute(WiredContext ctx) {
        Room room = ctx.room();
        if (room == null) {
            return;
        }

        int targetGroupId = this.resolveTargetGroupId(room);
        Set<RoomUnit> result = new LinkedHashSet<>();

        if (targetGroupId > 0) {
            for (Habbo habbo : room.getHabbos()) {
                if (habbo == null || habbo.getRoomUnit() == null || habbo.getHabboStats() == null) {
                    continue;
                }

                if (habbo.getHabboStats().hasGuild(targetGroupId)) {
                    result.add(habbo.getRoomUnit());
                }
            }
        }

        Set<RoomUnit> availableUsers = room.getHabbos().stream()
                .filter(habbo -> habbo != null && habbo.getRoomUnit() != null)
                .map(Habbo::getRoomUnit)
                .collect(Collectors.toCollection(LinkedHashSet::new));
        result = this.applySelectorModifiers(result, availableUsers, ctx.targets().users(), this.filterExisting, this.invert);

        ctx.targets().setUsers(result);
    }

    @Override
    public boolean saveData(WiredSettings settings, GameClient gameClient) {
        int[] params = settings.getIntParams();

        this.groupType = (params.length > 0) ? this.normalizeGroupType(params[0]) : GROUP_CURRENT_ROOM;
        this.selectedGroupId = (params.length > 1) ? Math.max(0, params[1]) : 0;
        this.filterExisting = params.length > 2 && params[2] == 1;
        this.invert = params.length > 3 && params[3] == 1;
        this.setDelay(settings.getDelay());
        return true;
    }

    @Override
    public WiredEffectType getType() {
        return type;
    }

    @Override
    public boolean isSelector() {
        return true;
    }

    @Override
    public String getWiredData() {
        return WiredManager.getGson().toJson(new JsonData(this.groupType, this.selectedGroupId, this.filterExisting, this.invert, this.getDelay()));
    }

    @Override
    public void loadWiredData(ResultSet set, Room room) throws SQLException {
        this.onPickUp();

        String wiredData = set.getString("wired_data");
        if (wiredData == null || !wiredData.startsWith("{")) {
            return;
        }

        JsonData data = WiredManager.getGson().fromJson(wiredData, JsonData.class);
        if (data == null) {
            return;
        }

        this.groupType = this.normalizeGroupType(data.groupType);
        this.selectedGroupId = Math.max(0, data.selectedGroupId);
        this.filterExisting = data.filterExisting;
        this.invert = data.invert;
        this.setDelay(data.delay);
    }

    @Override
    public void onPickUp() {
        this.groupType = GROUP_CURRENT_ROOM;
        this.selectedGroupId = 0;
        this.filterExisting = false;
        this.invert = false;
        this.setDelay(0);
    }

    @Override
    public void serializeWiredData(ServerMessage message, Room room) {
        message.appendBoolean(false);
        message.appendInt(0);
        message.appendInt(0);
        message.appendInt(this.getBaseItem().getSpriteId());
        message.appendInt(this.getId());
        message.appendString("");
        message.appendInt(4);
        message.appendInt(this.groupType);
        message.appendInt(this.selectedGroupId);
        message.appendInt(this.filterExisting ? 1 : 0);
        message.appendInt(this.invert ? 1 : 0);
        message.appendInt(0);
        message.appendInt(this.getType().code);
        message.appendInt(this.getDelay());
        message.appendInt(0);
    }

    @Override
    public boolean execute(RoomUnit roomUnit, Room room, Object[] stuff) {
        return false;
    }

    private int resolveTargetGroupId(Room room) {
        if (this.groupType == GROUP_CURRENT_ROOM) {
            return room.getGuildId();
        }

        return this.selectedGroupId;
    }

    private int normalizeGroupType(int value) {
        return (value == GROUP_SELECTED) ? GROUP_SELECTED : GROUP_CURRENT_ROOM;
    }

    static class JsonData {
        int groupType;
        int selectedGroupId;
        boolean filterExisting;
        boolean invert;
        int delay;

        JsonData(int groupType, int selectedGroupId, boolean filterExisting, boolean invert, int delay) {
            this.groupType = groupType;
            this.selectedGroupId = selectedGroupId;
            this.filterExisting = filterExisting;
            this.invert = invert;
            this.delay = delay;
        }
    }
}
