package com.earth2me.essentials.spawn;

import com.earth2me.essentials.IEssentialsModule;
import com.earth2me.essentials.settings.Spawns;
import com.earth2me.essentials.storage.AsyncStorageObjectHolder;
import net.ess3.api.IEssentials;
import java.io.File;
import java.util.HashMap;
import java.util.Locale;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentMap;
import org.bukkit.Location;
import org.bukkit.World;


public class SpawnStorage extends AsyncStorageObjectHolder<Spawns> implements IEssentialsModule
{
	private ConcurrentMap<String, Location> spawns;

	public SpawnStorage(final IEssentials ess)
	{
		super(ess, Spawns.class);
		reloadConfig();
	}

	@Override
	public final void reloadConfig()
	{
		super.reloadConfig();
		spawns = new ConcurrentHashMap<String, Location>();
	}

	@Override
	public File getStorageFile()
	{
		return new File(ess.getDataFolder(), "spawn.yml");
	}

	@Override
	public void finishRead()
	{
	}

	@Override
	public void finishWrite()
	{
	}

	public void setSpawn(final Location loc, final String group)
	{
		acquireWriteLock();
		try
		{
			if (getData().getSpawns() == null)
			{
				getData().setSpawns(new HashMap<String, Location>());
			}
			getData().getSpawns().put(group.toLowerCase(Locale.ENGLISH), loc);
		}
		finally
		{
			unlock();
		}

		spawns.clear();

		if ("default".equalsIgnoreCase(group))
		{
			loc.getWorld().setSpawnLocation(loc.getBlockX(), loc.getBlockY(), loc.getBlockZ());
		}
	}

	public Location getSpawn(final String group)
	{
		Location spawnLocation;
		if (spawns.containsKey(group))
		{
			spawnLocation = spawns.get(group);
		}
		else
		{
			acquireReadLock();
			try
			{
				if (getData().getSpawns() == null || group == null)
				{
					return getWorldSpawn();
				}
				final Map<String, Location> spawnMap = getData().getSpawns();
				String groupName = group.toLowerCase(Locale.ENGLISH);
				if (!spawnMap.containsKey(groupName))
				{
					groupName = "default";
				}
				if (!spawnMap.containsKey(groupName))
				{
					spawnLocation = getWorldSpawn();
				}
				else
				{
					spawnLocation = spawnMap.get(groupName);
				}
			}
			finally
			{
				unlock();
			}

			spawns.put(group, spawnLocation);
		}
		return spawnLocation;
	}

	private Location getWorldSpawn()
	{
		for (World world : ess.getServer().getWorlds())
		{
			if (world.getEnvironment() != World.Environment.NORMAL)
			{
				continue;
			}
			return world.getSpawnLocation();
		}
		return ess.getServer().getWorlds().get(0).getSpawnLocation();
	}
}