package com.rs.grab;

import java.util.Queue;
import java.util.concurrent.ConcurrentLinkedQueue;

import com.rs.cache.Cache;
import com.rs.executor.GrabThread;
import com.rs.net.Session;

public class Grab {

	private ConcurrentLinkedQueue<ArchiveRequest> priorityArchives;
	private ConcurrentLinkedQueue<ArchiveRequest> nonPriorityArchives;

	private boolean initialized;
	private boolean isLoggedIn;

	private Session session;

	public Grab(Session session) {
		this.session = session;
	}

	public void addArchive(int index, int archive, boolean priority) {
		if (index == 255 && archive != 255) {
			if (archive < 0 || archive >= Cache.STORE.getIndexes().length || Cache.STORE.getIndexes()[archive] == null)
				return;
		} else if (index != 255) {
			if (index < 0 || index >= Cache.STORE.getIndexes().length || Cache.STORE.getIndexes()[index] == null || archive < 0 || !Cache.STORE.getIndexes()[index].archiveExists(archive))
				return;
		}

		if (!initialized) {
			finish();
			return;
		}

		Queue<ArchiveRequest> requestList = priority ? priorityArchives : nonPriorityArchives;
		if (requestList.size() >= 500) // in reality 100 prority 20 non
			return;

		ArchiveRequest req = new ArchiveRequest(index, archive);
		if (requestList.contains(req))
			return;

		synchronized (requestList) {
			requestList.add(req);
		}
	}

	public long processNext(long limit) {
		if (!session.getChannel().isConnected()) {
			finish();
			return 0;
		}
		long total = 0;

		if (priorityArchives.isEmpty()) {
			synchronized (nonPriorityArchives) {
				while (!nonPriorityArchives.isEmpty() && total < limit) {
					ArchiveRequest request = nonPriorityArchives.peek();
					int bytesSent = request.getBytesSent();
					if (session.getGrabPackets().sendCacheArchive(request, false))
						nonPriorityArchives.remove(request);
					total += request.getBytesSent() - bytesSent;
				}
			}
		} else {
			synchronized (priorityArchives) {
				while (!priorityArchives.isEmpty() && total < limit) {
					ArchiveRequest request = priorityArchives.peek();
					int bytesSent = request.getBytesSent();
					if (session.getGrabPackets().sendCacheArchive(request, true))
						priorityArchives.remove(request);
					total += request.getBytesSent() - bytesSent;
				}
			}
		}
		return total;
	}

	public void init() {
		if (initialized) {
			finish();
			return;
		}
		priorityArchives = new ConcurrentLinkedQueue<ArchiveRequest>();// Collections.synchronizedMap(new
		// LinkedHashMap<Long,
		// ArchiveRequest>(100));
		nonPriorityArchives = new ConcurrentLinkedQueue<ArchiveRequest>();// Collections.synchronizedMap(new
		// LinkedHashMap<Long,
		// ArchiveRequest>(20));
		initialized = true;
		GrabThread.add(this);
	}

	/*
	 * force close update server due to either error from client or hack attempt
	 */
	public void finish() {
		session.getChannel().close();
		if (!initialized)
			return;
		GrabThread.remove(this);
		priorityArchives = nonPriorityArchives = null;
		initialized = false;
	}

	public boolean isLoggedIn() {
		return isLoggedIn;
	}

	public void setLoggedIn(boolean clientLoggedIn) {
		if (!initialized) {
			finish();
			return;
		}
		this.isLoggedIn = clientLoggedIn;
	}

	public Session getSession() {
		return session;
	}

}
