ServerListModel.java

/*
 * This file is part of Spoutcraft.
 *
 * Copyright (c) 2011 SpoutcraftDev <http://spoutcraft.org/>
 * Spoutcraft is licensed under the GNU Lesser General Public License.
 *
 * Spoutcraft is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * Spoutcraft is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
package org.spoutcraft.client.gui.server;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.UnsupportedEncodingException;
import java.net.MalformedURLException;
import java.net.URL;
import java.net.URLDecoder;
import java.util.ArrayList;
import java.util.Collections;
import java.util.HashMap;
import java.util.LinkedList;
import java.util.List;

import gnu.trove.map.hash.TIntObjectHashMap;
import org.yaml.snakeyaml.Yaml;

import org.spoutcraft.api.gui.GenericListView;
import org.spoutcraft.api.gui.ListWidgetItem;
import org.spoutcraft.api.gui.Orientation;
import org.spoutcraft.client.SpoutClient;
import org.spoutcraft.client.gui.database.AbstractAPIModel;

public class ServerListModel extends AbstractAPIModel {
	protected TIntObjectHashMap<ServerDataBaseEntry> dbEntries = new TIntObjectHashMap<ServerDataBaseEntry>();
	protected List<String> countries = new LinkedList<String>();

	public ServerListModel() {
		API = "http://servers.spout.org/api2.php";
		//refreshAPIData(API + "?random", 0, true);		
		//loadCountries();
	}

	public String getAPI() {
		return API;
	}

	private void loadCountries() {
		new Thread() {
			public void run() {
				//long start = System.currentTimeMillis();
				URL url1;
				try {
					url1 = new URL(API + "?countries");
				} catch (MalformedURLException e) {
					return;
				}
				//System.out.println("Loading " + url1.toString());
				BufferedReader reader;
				try {
					reader = new BufferedReader(new InputStreamReader(url1.openStream()));
				} catch (IOException e1) {
					System.out.println(e1.getClass().toString());
					return;
				}
				Yaml yaml = new Yaml();
				ArrayList<String> yamlObj = (ArrayList<String>) yaml.load(reader);
				//System.out.println("Loaded in " + (System.currentTimeMillis() - start) + " ms");
				synchronized (countries) {
					countries.clear();
					for (String c:yamlObj) {
						if (!c.trim().isEmpty()) {
							countries.add(c);
						}
					}
				}
				try {
					reader.close();
				} catch (IOException e) {
					return;
				}
			}
		}.start();
	}

	public String getDefaultUrl() {
		return API + "?featured";
	}

	/**
	 * Processes loaded API data.
	 * @param clear
	 */
	protected void refreshList(boolean clear) {
		if (clear) {
			lastPage = 0;
			entries.clear();
			for (GenericListView view:getViews()) {
				view.setScrollPosition(Orientation.VERTICAL, 0);
			}
		}
		for (Object item:apiData) {
			try {
				HashMap<String, Object> hash = (HashMap<String, Object>) item;
				String name = URLDecoder.decode((String) hash.get("name"), "UTF-8");
				name = name.replaceAll("\\&\\;", "&");
				int uid = Integer.valueOf((String)hash.get("uniqueid"));
				int port = Integer.valueOf((String)hash.get("port"));
				String adress = (String) hash.get("ip");
				byte accessType = Byte.valueOf((String) hash.get("whitelist"));
				String country = (String) hash.get("country");
				String version = URLDecoder.decode((String) hash.get("mcversion"), "UTF-8");
				ServerItem server = new ServerItem(name, adress, port, uid, version);
				server.setFavorite(false);
				server.setCountry(country);
				server.setAccessType(accessType);
				entries.add(server);
			} catch(UnsupportedEncodingException e) {}
			catch(Exception e2) { continue; }
		}
		update();
	}

	public ServerDataBaseEntry getServerDBEntry(int uid) {
		if (dbEntries.contains(uid)) {
			return dbEntries.get(uid);
		} else {
			return null;
		}
	}

	public List<ServerItem> getServers() {
		ArrayList<ServerItem> servers = new ArrayList<ServerItem>();
		for (ListWidgetItem item:entries) {
			if (item instanceof ServerItem) {
				servers.add((ServerItem)item);
			}
		}
		return servers;
	}

	public List<String> getCountries() {
		return Collections.unmodifiableList(countries);
	}
}