LoginSystem/src/main/java/fr/xephi/authme/datasource/CacheDataSource.java

package fr.xephi.authme.datasource;

import fr.xephi.authme.AuthMe;
import fr.xephi.authme.Utils;
import fr.xephi.authme.cache.auth.PlayerAuth;
import fr.xephi.authme.cache.auth.PlayerCache;
import org.bukkit.entity.Player;

import java.util.List;
import java.util.concurrent.ConcurrentHashMap;

public class CacheDataSource implements DataSource {

    private DataSource source;
    public AuthMe plugin;
    private ConcurrentHashMap<String, PlayerAuth> cache = new ConcurrentHashMap<>();

    public CacheDataSource(AuthMe plugin, DataSource source) {
        this.plugin = plugin;
        this.source = source;
        /*
         * We need to load all players in cache ... It will took more time to
         * load the server, but it will be much easier to check for an
         * isAuthAvailable !
         */
        for (PlayerAuth auth : source.getAllAuths()) {
            cache.put(auth.getNickname().toLowerCase(), auth);
        }
    }

    @Override
    public synchronized boolean isAuthAvailable(String user) {
        return cache.containsKey(user.toLowerCase());
    }

    @Override
    public synchronized PlayerAuth getAuth(String user) {
        user = user.toLowerCase();
        if (cache.containsKey(user)) {
            return cache.get(user);
        } else {
            PlayerAuth auth = source.getAuth(user);
            if (auth != null)
                cache.put(user, auth);
            return auth;
        }
    }

    @Override
    public synchronized boolean saveAuth(PlayerAuth auth) {
        if (source.saveAuth(auth)) {
            cache.put(auth.getNickname(), auth);
            return true;
        }
        return false;
    }

    @Override
    public synchronized boolean updatePassword(PlayerAuth auth) {
        if (source.updatePassword(auth)) {
            if (cache.containsKey(auth.getNickname()))
                cache.get(auth.getNickname()).setHash(auth.getHash());
            return true;
        }
        return false;
    }

    @Override
    public boolean updateSession(PlayerAuth auth) {
        if (source.updateSession(auth)) {
            if (cache.containsKey(auth.getNickname())) {
                cache.get(auth.getNickname()).setIp(auth.getIp());
                cache.get(auth.getNickname()).setLastLogin(auth.getLastLogin());
                cache.get(auth.getNickname()).setRealName(auth.getRealName());
            }
            return true;
        }
        return false;
    }

    @Override
    public boolean updateQuitLoc(PlayerAuth auth) {
        if (source.updateQuitLoc(auth)) {
            if (cache.containsKey(auth.getNickname())) {
                cache.get(auth.getNickname()).setQuitLocX(auth.getQuitLocX());
                cache.get(auth.getNickname()).setQuitLocY(auth.getQuitLocY());
                cache.get(auth.getNickname()).setQuitLocZ(auth.getQuitLocZ());
                cache.get(auth.getNickname()).setWorld(auth.getWorld());
            }
            return true;
        }
        return false;
    }

    @Override
    public int getIps(String ip) {
        return source.getIps(ip);
    }

    @Override
    public int purgeDatabase(long until) {
        int cleared = source.purgeDatabase(until);
        if (cleared > 0) {
            for (PlayerAuth auth : cache.values()) {
                if (auth.getLastLogin() < until) {
                    cache.remove(auth.getNickname());
                }
            }
        }
        return cleared;
    }

    @Override
    public List<String> autoPurgeDatabase(long until) {
        List<String> cleared = source.autoPurgeDatabase(until);
        if (cleared.size() > 0) {
            for (PlayerAuth auth : cache.values()) {
                if (auth.getLastLogin() < until) {
                    cache.remove(auth.getNickname());
                }
            }
        }
        return cleared;
    }

    @Override
    public synchronized boolean removeAuth(String user) {
        if (source.removeAuth(user)) {
            cache.remove(user);
            return true;
        }
        return false;
    }

    @Override
    public synchronized void close() {
        source.close();
    }

    @Override
    public void reload() {
        cache.clear();
        source.reload();
        for (Player player : Utils.getOnlinePlayers()) {
            String user = player.getName().toLowerCase();
            if (PlayerCache.getInstance().isAuthenticated(user)) {
                PlayerAuth auth = source.getAuth(user);
                cache.put(user, auth);
            }
        }
    }

    @Override
    public synchronized boolean updateEmail(PlayerAuth auth) {
        if (source.updateEmail(auth)) {
            if (cache.containsKey(auth.getNickname()))
                cache.get(auth.getNickname()).setEmail(auth.getEmail());
            return true;
        }
        return false;
    }

    @Override
    public synchronized boolean updateSalt(PlayerAuth auth) {
        if (source.updateSalt(auth)) {
            if (cache.containsKey(auth.getNickname()))
                cache.get(auth.getNickname()).setSalt(auth.getSalt());
            return true;
        }
        return false;
    }

    @Override
    public synchronized List<String> getAllAuthsByName(PlayerAuth auth) {
        return source.getAllAuthsByName(auth);
    }

    @Override
    public synchronized List<String> getAllAuthsByIp(String ip) {
        return source.getAllAuthsByIp(ip);
    }

    @Override
    public synchronized List<String> getAllAuthsByEmail(String email) {
        return source.getAllAuthsByEmail(email);
    }

    @Override
    public synchronized void purgeBanned(List<String> banned) {
        source.purgeBanned(banned);
        for (PlayerAuth auth : cache.values()) {
            if (banned.contains(auth.getNickname())) {
                cache.remove(auth.getNickname());
            }
        }
    }

    @Override
    public DataSourceType getType() {
        return source.getType();
    }

    @Override
    public boolean isLogged(String user) {
        return source.isLogged(user.toLowerCase());
    }

    @Override
    public void setLogged(String user) {
        source.setLogged(user.toLowerCase());
    }

    @Override
    public void setUnlogged(String user) {
        source.setUnlogged(user.toLowerCase());
    }

    @Override
    public void purgeLogged() {
        source.purgeLogged();
    }

    @Override
    public int getAccountsRegistered() {
        return source.getAccountsRegistered();
    }

    @Override
    public void updateName(String oldone, String newone) {
        if (cache.containsKey(oldone)) {
            cache.put(newone, cache.get(oldone));
            cache.remove(oldone);
        }
        source.updateName(oldone, newone);
    }

    @Override
    public List<PlayerAuth> getAllAuths() {
        return source.getAllAuths();
    }

    @Override
    public List<PlayerAuth> getLoggedPlayers() {
        return source.getLoggedPlayers();
    }
}