minecraft-src/net/minecraft/server/players/StoredUserList.java

package net.minecraft.server.players;

import com.google.common.collect.Lists;
import com.google.common.collect.Maps;
import com.google.common.io.Files;
import com.google.gson.Gson;
import com.google.gson.GsonBuilder;
import com.google.gson.JsonArray;
import com.google.gson.JsonElement;
import com.google.gson.JsonObject;
import com.mojang.logging.LogUtils;
import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.File;
import java.io.IOException;
import java.nio.charset.StandardCharsets;
import java.util.Collection;
import java.util.List;
import java.util.Map;
import net.minecraft.Util;
import net.minecraft.util.GsonHelper;
import org.jetbrains.annotations.Nullable;
import org.slf4j.Logger;

public abstract class StoredUserList<K, V extends StoredUserEntry<K>> {
	private static final Logger LOGGER = LogUtils.getLogger();
	private static final Gson GSON = new GsonBuilder().setPrettyPrinting().create();
	private final File file;
	private final Map<String, V> map = Maps.<String, V>newHashMap();

	public StoredUserList(File file) {
		this.file = file;
	}

	public File getFile() {
		return this.file;
	}

	/**
	 * Adds an entry to the list
	 */
	public void add(V entry) {
		this.map.put(this.getKeyForUser(entry.getUser()), entry);

		try {
			this.save();
		} catch (IOException var3) {
			LOGGER.warn("Could not save the list after adding a user.", (Throwable)var3);
		}
	}

	@Nullable
	public V get(K obj) {
		this.removeExpired();
		return (V)this.map.get(this.getKeyForUser(obj));
	}

	public void remove(K user) {
		this.map.remove(this.getKeyForUser(user));

		try {
			this.save();
		} catch (IOException var3) {
			LOGGER.warn("Could not save the list after removing a user.", (Throwable)var3);
		}
	}

	public void remove(StoredUserEntry<K> entry) {
		this.remove(entry.getUser());
	}

	public String[] getUserList() {
		return (String[])this.map.keySet().toArray(new String[0]);
	}

	public boolean isEmpty() {
		return this.map.size() < 1;
	}

	/**
	 * Gets the key value for the given object
	 */
	protected String getKeyForUser(K obj) {
		return obj.toString();
	}

	protected boolean contains(K entry) {
		return this.map.containsKey(this.getKeyForUser(entry));
	}

	/**
	 * Removes expired bans from the list. See {@link BanEntry#hasBanExpired}
	 */
	private void removeExpired() {
		List<K> list = Lists.<K>newArrayList();

		for (V storedUserEntry : this.map.values()) {
			if (storedUserEntry.hasExpired()) {
				list.add(storedUserEntry.getUser());
			}
		}

		for (K object : list) {
			this.map.remove(this.getKeyForUser(object));
		}
	}

	protected abstract StoredUserEntry<K> createEntry(JsonObject entryData);

	public Collection<V> getEntries() {
		return this.map.values();
	}

	public void save() throws IOException {
		JsonArray jsonArray = new JsonArray();
		this.map.values().stream().map(storedUserEntry -> Util.make(new JsonObject(), storedUserEntry::serialize)).forEach(jsonArray::add);
		BufferedWriter bufferedWriter = Files.newWriter(this.file, StandardCharsets.UTF_8);

		try {
			GSON.toJson(jsonArray, GSON.newJsonWriter(bufferedWriter));
		} catch (Throwable var6) {
			if (bufferedWriter != null) {
				try {
					bufferedWriter.close();
				} catch (Throwable var5) {
					var6.addSuppressed(var5);
				}
			}

			throw var6;
		}

		if (bufferedWriter != null) {
			bufferedWriter.close();
		}
	}

	public void load() throws IOException {
		if (this.file.exists()) {
			BufferedReader bufferedReader = Files.newReader(this.file, StandardCharsets.UTF_8);

			label54: {
				try {
					this.map.clear();
					JsonArray jsonArray = GSON.fromJson(bufferedReader, JsonArray.class);
					if (jsonArray == null) {
						break label54;
					}

					for (JsonElement jsonElement : jsonArray) {
						JsonObject jsonObject = GsonHelper.convertToJsonObject(jsonElement, "entry");
						StoredUserEntry<K> storedUserEntry = this.createEntry(jsonObject);
						if (storedUserEntry.getUser() != null) {
							this.map.put(this.getKeyForUser(storedUserEntry.getUser()), storedUserEntry);
						}
					}
				} catch (Throwable var8) {
					if (bufferedReader != null) {
						try {
							bufferedReader.close();
						} catch (Throwable var7) {
							var8.addSuppressed(var7);
						}
					}

					throw var8;
				}

				if (bufferedReader != null) {
					bufferedReader.close();
				}

				return;
			}

			if (bufferedReader != null) {
				bufferedReader.close();
			}
		}
	}
}