Arcturus-Community/src/main/java/com/eu/habbo/habbohotel/achievements/Achievement.java

package com.eu.habbo.habbohotel.achievements;

import gnu.trove.map.hash.THashMap;

import java.sql.ResultSet;
import java.sql.SQLException;

public class Achievement
{

    public final int id;


    public final String name;


    public final AchievementCategories category;


    public final THashMap<Integer, AchievementLevel> levels;


    public Achievement(ResultSet set) throws SQLException
    {
        this.levels = new THashMap<>();

        this.id = set.getInt("id");
        this.name = set.getString("name");
        this.category = AchievementCategories.valueOf(set.getString("category").toUpperCase());

        this.addLevel(new AchievementLevel(set));
    }


    public void addLevel(AchievementLevel level)
    {
        synchronized (this.levels)
        {
            this.levels.put(level.level, level);
        }
    }


    public AchievementLevel getLevelForProgress(int progress)
    {
        AchievementLevel l = null;
        if (progress > 0)
        {
            for (AchievementLevel level : this.levels.values())
            {
                if (progress >= level.progress)
                {
                    if (l != null)
                    {
                        if (l.level > level.level)
                        {
                            continue;
                        }
                    }

                    l = level;
                }
            }
        }
        return l;
    }


    public AchievementLevel getNextLevel(int currentLevel)
    {
        AchievementLevel l = null;

        for(AchievementLevel level : this.levels.values())
        {
            if(level.level == (currentLevel + 1))
                return level;
        }

        return null;
    }

    public AchievementLevel firstLevel()
    {
        return this.levels.get(1);
    }
}