Arcturus-Community/src/main/java/com/eu/habbo/habbohotel/users/HabboStats.java

package com.eu.habbo.habbohotel.users;

import com.eu.habbo.Emulator;
import com.eu.habbo.habbohotel.achievements.Achievement;
import com.eu.habbo.habbohotel.achievements.TalentTrackType;
import com.eu.habbo.habbohotel.catalog.CatalogItem;
import com.eu.habbo.habbohotel.rooms.RoomChatMessageBubbles;
import gnu.trove.list.array.TIntArrayList;
import gnu.trove.map.hash.THashMap;
import gnu.trove.stack.array.TIntArrayStack;

import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;

public class HabboStats implements Runnable
{
    private final int timeLoggedIn = Emulator.getIntUnixTimestamp();

    private Habbo habbo;

    public int achievementScore;
    public int respectPointsReceived;
    public int respectPointsGiven;
    public int respectPointsToGive;

    public int petRespectPointsToGive;

    public boolean blockFollowing;
    public boolean blockFriendRequests;
    public boolean blockRoomInvites;
    public boolean blockStaffAlerts;
    public boolean allowTrade;
    public boolean preferOldChat;
    public boolean blockCameraFollow;
    public RoomChatMessageBubbles chatColor;

    private int clubExpireTimestamp;

    public int volumeSystem;
    public int volumeFurni;
    public int volumeTrax;

    public int guild;
    public List<Integer> guilds;

    public String[] tags;

    public TIntArrayStack votedRooms;
    public int loginStreak;
    public int rentedItemId;
    public int rentedTimeEnd;
    public int hofPoints;
    public boolean ignorePets;
    public boolean ignoreBots;

    private final THashMap<Achievement, Integer> achievementProgress;
    private final THashMap<Achievement, Integer> achievementCache;
    private final THashMap<Integer, CatalogItem> recentPurchases;
    private final TIntArrayList favoriteRooms;
    public final TIntArrayList ignoredUsers;
    public final TIntArrayList secretRecipes;

    public int citizenshipLevel = -1;
    public int helpersLevel = -1;

    public final HabboNavigatorWindowSettings navigatorWindowSettings;
    public final THashMap<String, Object> cache;

    public long roomEnterTimestamp;
    public int chatCounter;
    public long lastChat;
    public long lastUsersSearched;
    public boolean nux = false;
    public boolean nuxReward = false;
    public int nuxStep = 1;

    private int muteEndTime = 0;
    public int mutedCount = 0;
    public boolean mutedBubbleTracker = false;

    public String changeNameChecked = "";
    public TIntArrayList calendarRewardsClaimed;

    public boolean allowNameChange = false;

    private HabboStats(ResultSet set, Habbo habbo) throws SQLException
    {
        this.cache = new THashMap<String, Object>(0);
        this.achievementProgress = new THashMap<Achievement, Integer>(0);
        this.achievementCache = new THashMap<Achievement, Integer>(0);
        this.recentPurchases = new THashMap<Integer, CatalogItem>(0);
        this.favoriteRooms = new TIntArrayList(0);
        this.ignoredUsers = new TIntArrayList(0);
        this.secretRecipes = new TIntArrayList(0);
        this.calendarRewardsClaimed = new TIntArrayList(0);

        this.habbo = habbo;

        this.achievementScore = set.getInt("achievement_score");
        this.respectPointsReceived = set.getInt("respects_received");
        this.respectPointsGiven = set.getInt("respects_given");
        this.petRespectPointsToGive = set.getInt("daily_pet_respect_points");
        this.respectPointsToGive = set.getInt("daily_respect_points");
        this.blockFollowing = set.getString("block_following").equals("1");
        this.blockFriendRequests = set.getString("block_friendrequests").equals("1");
        this.blockRoomInvites = set.getString("block_roominvites").equals("1");
        this.preferOldChat = set.getString("old_chat").equals("1");
        this.blockCameraFollow = set.getString("block_camera_follow").equals("1");
        this.guild = set.getInt("guild_id");
        this.guilds = new ArrayList<>();
        this.tags = set.getString("tags").split(";");
        this.allowTrade = set.getString("can_trade").equals("1");
        this.votedRooms = new TIntArrayStack();
        this.clubExpireTimestamp = set.getInt("club_expire_timestamp");
        this.loginStreak = set.getInt("login_streak");
        this.rentedItemId = set.getInt("rent_space_id");
        this.rentedTimeEnd = set.getInt("rent_space_endtime");
        this.volumeSystem = set.getInt("volume_system");
        this.volumeFurni = set.getInt("volume_furni");
        this.volumeTrax = set.getInt("volume_trax");
        this.chatColor = RoomChatMessageBubbles.getBubble(set.getInt("chat_color"));
        this.hofPoints = set.getInt("hof_points");
        this.blockStaffAlerts = set.getString("block_alerts").equals("1");
        this.citizenshipLevel = set.getInt("talent_track_citizenship_level");
        this.helpersLevel = set.getInt("talent_track_helpers_level");
        this.ignoreBots = set.getString("ignore_bots").equalsIgnoreCase("1");
        this.ignorePets = set.getString("ignore_pets").equalsIgnoreCase("1");
        this.nux = set.getString("nux").equals("1");
        this.muteEndTime = set.getInt("mute_end_timestamp");
        this.allowNameChange = set.getString("allow_name_change").equalsIgnoreCase("1");
        this.nuxReward = nux;

        try (PreparedStatement statement = set.getStatement().getConnection().prepareStatement("SELECT * FROM user_window_settings WHERE user_id = ? LIMIT 1"))
        {
            statement.setInt(1, this.habbo.getHabboInfo().getId());
            try (ResultSet nSet = statement.executeQuery())
            {
                if (nSet.next())
                {
                    this.navigatorWindowSettings = new HabboNavigatorWindowSettings(nSet);
                }
                else
                {
                    try (PreparedStatement stmt = statement.getConnection().prepareStatement("INSERT INTO user_window_settings (user_id) VALUES (?)"))
                    {
                        stmt.setInt(1, this.habbo.getHabboInfo().getId());
                        stmt.executeUpdate();
                    }

                    this.navigatorWindowSettings = new HabboNavigatorWindowSettings(habbo.getHabboInfo().getId());
                }
            }
        }

        try (PreparedStatement statement = set.getStatement().getConnection().prepareStatement("SELECT * FROM users_navigator_settings WHERE user_id = ?"))
        {
            statement.setInt(1, this.habbo.getHabboInfo().getId());
            try (ResultSet nSet = statement.executeQuery())
            {
                while (nSet.next())
                {
                    this.navigatorWindowSettings.addDisplayMode(nSet.getString("caption"), new HabboNavigatorPersonalDisplayMode(nSet));
                }
            }
        }

        try (PreparedStatement favoriteRoomsStatement = set.getStatement().getConnection().prepareStatement("SELECT * FROM users_favorite_rooms WHERE user_id = ?"))
        {
            favoriteRoomsStatement.setInt(1, this.habbo.getHabboInfo().getId());
            try (ResultSet favoriteSet = favoriteRoomsStatement.executeQuery())
            {
                while (favoriteSet.next())
                {
                    this.favoriteRooms.add(favoriteSet.getInt("room_id"));
                }
            }

        }

        try (PreparedStatement recipesStatement = set.getStatement().getConnection().prepareStatement("SELECT * FROM users_recipes WHERE user_id = ?"))
        {
            recipesStatement.setInt(1, this.habbo.getHabboInfo().getId());
            try (ResultSet recipeSet = recipesStatement.executeQuery())
            {
                while (recipeSet.next())
                {
                    this.secretRecipes.add(recipeSet.getInt("recipe"));
                }
            }
        }

        try (PreparedStatement calendarRewardsStatement = set.getStatement().getConnection().prepareStatement("SELECT * FROM calendar_rewards_claimed WHERE user_id = ?"))
        {
            calendarRewardsStatement.setInt(1, this.habbo.getHabboInfo().getId());
            try (ResultSet rewardSet = calendarRewardsStatement.executeQuery())
            {
                while (rewardSet.next())
                {
                    this.calendarRewardsClaimed.add(rewardSet.getInt("reward_id"));
                }
            }
        }
    }

    @Override
    public void run()
    {
        try (Connection connection = Emulator.getDatabase().getDataSource().getConnection())
        {
            try (PreparedStatement statement = connection.prepareStatement("UPDATE users_settings SET achievement_score = ?, respects_received = ?, respects_given = ?, daily_respect_points = ?, block_following = ?, block_friendrequests = ?, online_time = online_time + ?, guild_id = ?, daily_pet_respect_points = ?, club_expire_timestamp = ?, login_streak = ?, rent_space_id = ?, rent_space_endtime = ?, volume_system = ?, volume_furni = ?, volume_trax = ?, block_roominvites = ?, old_chat = ?, block_camera_follow = ?, chat_color = ?, hof_points = ?, block_alerts = ?, talent_track_citizenship_level = ?, talent_track_helpers_level = ?, ignore_bots = ?, ignore_pets = ?, nux = ?, mute_end_timestamp = ?, allow_name_change = ? WHERE user_id = ? LIMIT 1"))
            {
                statement.setInt(1, this.achievementScore);
                statement.setInt(2, this.respectPointsReceived);
                statement.setInt(3, this.respectPointsGiven);
                statement.setInt(4, this.respectPointsToGive);
                statement.setString(5, this.blockFollowing ? "1" : "0");
                statement.setString(6, this.blockFriendRequests ? "1" : "0");
                statement.setInt(7, Emulator.getIntUnixTimestamp() - this.timeLoggedIn);
                statement.setInt(8, this.guild);
                statement.setInt(9, this.petRespectPointsToGive);
                statement.setInt(10, this.clubExpireTimestamp);
                statement.setInt(11, this.loginStreak);
                statement.setInt(12, this.rentedItemId);
                statement.setInt(13, this.rentedTimeEnd);
                statement.setInt(14, this.volumeSystem);
                statement.setInt(15, this.volumeFurni);
                statement.setInt(16, this.volumeTrax);
                statement.setString(17, this.blockRoomInvites ? "1" : "0");
                statement.setString(18, this.preferOldChat ? "1" : "0");
                statement.setString(19, this.blockCameraFollow ? "1" : "0");
                statement.setInt(20, this.chatColor.getType());
                statement.setInt(21, this.hofPoints);
                statement.setString(22, this.blockStaffAlerts ? "1" : "0");
                statement.setInt(23, this.citizenshipLevel);
                statement.setInt(24, this.helpersLevel);
                statement.setString(25, this.ignoreBots ? "1" : "0");
                statement.setString(26, this.ignorePets ? "1" : "0");
                statement.setString(27, this.nux ? "1" : "0");
                statement.setInt(28, this.muteEndTime);
                statement.setString(29, this.allowNameChange ? "1" : "0");
                statement.setInt(30, this.habbo.getHabboInfo().getId());
                statement.executeUpdate();
            }

            try (PreparedStatement statement = connection.prepareStatement("UPDATE user_window_settings SET x = ?, y = ?, width = ?, height = ?, open_searches = ? WHERE user_id = ? LIMIT 1"))
            {
                statement.setInt(1, this.navigatorWindowSettings.x);
                statement.setInt(2, this.navigatorWindowSettings.y);
                statement.setInt(3, this.navigatorWindowSettings.width);
                statement.setInt(4, this.navigatorWindowSettings.height);
                statement.setString(5, this.navigatorWindowSettings.openSearches ? "1" : "0");
                statement.setInt(6, this.habbo.getHabboInfo().getId());
                statement.executeUpdate();
            }

            this.navigatorWindowSettings.save(connection);
        }
        catch (SQLException e)
        {
            Emulator.getLogging().logSQLException(e);
        }
    }

    public void dispose()
    {
        this.run();
        this.habbo = null;
        this.recentPurchases.clear();
    }

    private static HabboStats createNewStats(Habbo habbo)
    {
        habbo.firstVisit = true;

        try (Connection connection = Emulator.getDatabase().getDataSource().getConnection(); PreparedStatement statement = connection.prepareStatement("INSERT INTO users_settings (user_id) VALUES (?)"))
        {
            statement.setInt(1, habbo.getHabboInfo().getId());
            statement.executeUpdate();
        }
        catch (SQLException e)
        {
            Emulator.getLogging().logSQLException(e);
        }

        return load(habbo);
    }

    public static HabboStats load(Habbo habbo)
    {
        HabboStats stats = null;
        try (Connection connection = Emulator.getDatabase().getDataSource().getConnection())
        {
            try(PreparedStatement statement = connection.prepareStatement("SELECT * FROM users_settings WHERE user_id = ? LIMIT 1"))
            {
                statement.setInt(1, habbo.getHabboInfo().getId());
                try (ResultSet set = statement.executeQuery())
                {
                    set.first();
                    if (set.getRow() != 0)
                    {
                        stats = new HabboStats(set, habbo);
                    }
                    else
                    {
                        stats = createNewStats(habbo);
                    }
                }
            }

            if(stats != null)
            {
                try (PreparedStatement statement = connection.prepareStatement("SELECT guild_id FROM guilds_members WHERE user_id = ? AND level_id < 3 LIMIT 100"))
                {
                    statement.setInt(1, habbo.getHabboInfo().getId());
                    try (ResultSet set = statement.executeQuery())
                    {

                        int i = 0;
                        while (set.next())
                        {
                            stats.guilds.add(set.getInt("guild_id"));
                            i++;
                        }
                    }
                }

                Collections.sort(stats.guilds);

                try (PreparedStatement statement = connection.prepareStatement("SELECT room_id FROM room_votes WHERE user_id = ?"))
                {
                    statement.setInt(1, habbo.getHabboInfo().getId());
                    try (ResultSet set = statement.executeQuery())
                    {
                        while (set.next())
                        {
                            stats.votedRooms.push(set.getInt("room_id"));
                        }
                    }
                }

                try (PreparedStatement statement = connection.prepareStatement("SELECT * FROM users_achievements WHERE user_id = ?"))
                {
                    statement.setInt(1, habbo.getHabboInfo().getId());
                    try (ResultSet set = statement.executeQuery())
                    {
                        while (set.next())
                        {
                            Achievement achievement = Emulator.getGameEnvironment().getAchievementManager().getAchievement(set.getString("achievement_name"));

                            if (achievement != null)
                            {
                                stats.achievementProgress.put(achievement, set.getInt("progress"));
                            }
                        }
                    }
                }
            }
        }
        catch (SQLException e)
        {
            Emulator.getLogging().logSQLException(e);
        }

        return stats;
    }

    public void addGuild(int guildId)
    {
        if (!this.guilds.contains(guildId))
        {
            this.guilds.add(guildId);
        }
    }

    public void removeGuild(int guildId)
    {
        this.guilds.remove((Integer)guildId);
    }

    public boolean hasGuild(int guildId)
    {
        for(int i : this.guilds)
        {
            if(i == guildId)
                return true;
        }

        return false;
    }

    public int getAchievementScore()
    {
        return this.achievementScore;
    }

    public void addAchievementScore(int achievementScore)
    {
        this.achievementScore += achievementScore;
    }

    public int getAchievementProgress(Achievement achievement)
    {
        if(this.achievementProgress.containsKey(achievement))
            return this.achievementProgress.get(achievement);

        return -1;
    }

    public void setProgress(Achievement achievement, int progress)
    {
        this.achievementProgress.put(achievement, progress);
    }

    public int getRentedTimeEnd()
    {
        return this.rentedTimeEnd;
    }

    public void setRentedTimeEnd(int rentedTimeEnd)
    {
        this.rentedTimeEnd = rentedTimeEnd;
    }

    public int getRentedItemId()
    {
        return this.rentedItemId;
    }

    public void setRentedItemId(int rentedItemId)
    {
        this.rentedItemId = rentedItemId;
    }

    public boolean canRentSpace()
    {
        return this.rentedTimeEnd < Emulator.getIntUnixTimestamp();
    }

    public int getClubExpireTimestamp()
    {
        return this.clubExpireTimestamp;
    }

    public void setClubExpireTimestamp(int clubExpireTimestamp)
    {
        this.clubExpireTimestamp = clubExpireTimestamp;
    }

    public boolean hasActiveClub()
    {
        return this.clubExpireTimestamp > Emulator.getIntUnixTimestamp();
    }

    public THashMap<Achievement, Integer> getAchievementProgress()
    {
        return this.achievementProgress;
    }

    public THashMap<Achievement, Integer> getAchievementCache()
    {
        return this.achievementCache;
    }

    public void addPurchase(CatalogItem item)
    {
        if(!this.recentPurchases.containsKey(item.getId()))
        {
            this.recentPurchases.put(item.getId(), item);
        }
    }

    public THashMap<Integer, CatalogItem> getRecentPurchases()
    {
        return this.recentPurchases;
    }

    public void disposeRecentPurchases()
    {
        this.recentPurchases.clear();
    }

    public boolean addFavoriteRoom(int roomId)
    {
        if (this.favoriteRooms.contains(roomId))
            return false;

        if (Emulator.getConfig().getInt("hotel.rooms.max.favorite") <= this.favoriteRooms.size())
            return false;

        try (Connection connection = Emulator.getDatabase().getDataSource().getConnection(); PreparedStatement statement = connection.prepareStatement("INSERT INTO users_favorite_rooms (user_id, room_id) VALUES (?, ?)");)
        {
            statement.setInt(1, this.habbo.getHabboInfo().getId());
            statement.setInt(2, roomId);
            statement.execute();
        }
        catch (SQLException e)
        {
            Emulator.getLogging().logSQLException(e);
        }

        this.favoriteRooms.add(roomId);
        return true;
    }

    public void removeFavoriteRoom(int roomId)
    {
        if (this.favoriteRooms.remove(roomId))
        {
            try (Connection connection = Emulator.getDatabase().getDataSource().getConnection(); PreparedStatement statement = connection.prepareStatement("DELETE FROM users_favorite_rooms WHERE user_id = ? AND room_id = ? LIMIT 1"))
            {
                statement.setInt(1, this.habbo.getHabboInfo().getId());
                statement.setInt(2, roomId);
                statement.execute();
            }
            catch (SQLException e)
            {
                Emulator.getLogging().logSQLException(e);
            }
        }
    }

    public boolean hasFavoriteRoom(int roomId)
    {
        return this.favoriteRooms.contains(roomId);
    }

    public TIntArrayList getFavoriteRooms()
    {
        return this.favoriteRooms;
    }

    public boolean hasRecipe(int id)
    {
        return this.secretRecipes.contains(id);
    }

    public boolean addRecipe(int id)
    {
        if (this.secretRecipes.contains(id))
            return false;

        try (Connection connection = Emulator.getDatabase().getDataSource().getConnection(); PreparedStatement statement = connection.prepareStatement("INSERT INTO users_recipes (user_id, recipe) VALUES (?, ?)"))
        {
            statement.setInt(1, this.habbo.getHabboInfo().getId());
            statement.setInt(2, id);
            statement.execute();
        }
        catch (SQLException e)
        {
            Emulator.getLogging().logSQLException(e);
        }

        this.secretRecipes.add(id);
        return true;
    }

    public int talentTrackLevel(TalentTrackType type)
    {
        if (type == TalentTrackType.CITIZENSHIP)
            return this.citizenshipLevel;
        else if (type == TalentTrackType.HELPER)
            return this.helpersLevel;

        return -1;
    }

    public void setTalentLevel(TalentTrackType type, int level)
    {
        if (type == TalentTrackType.CITIZENSHIP)
            this.citizenshipLevel = level;
        else if (type == TalentTrackType.HELPER)
            this.helpersLevel = level;
    }

    public int getMuteEndTime()
    {
        return this.muteEndTime;
    }

    public int addMuteTime(int seconds)
    {
        if (remainingMuteTime() == 0)
        {
            this.muteEndTime = Emulator.getIntUnixTimestamp();
        }

        this.mutedBubbleTracker = true;
        this.muteEndTime += seconds;

        return remainingMuteTime();
    }

    public int remainingMuteTime()
    {
        return Math.max(0, this.muteEndTime - Emulator.getIntUnixTimestamp());
    }

    public boolean allowTalk()
    {
        return this.remainingMuteTime() == 0;
    }

    public void unMute()
    {
        this.muteEndTime = 0;
        this.mutedBubbleTracker = false;
    }
}