RandomSec/src/main/java/com/metaweb/gridworks/history/History.java

package com.metaweb.gridworks.history;

import java.io.Serializable;
import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;
import java.util.Properties;

import org.json.JSONException;
import org.json.JSONWriter;

import com.metaweb.gridworks.Jsonizable;
import com.metaweb.gridworks.ProjectManager;
import com.metaweb.gridworks.model.Project;

public class History implements Serializable, Jsonizable {
	private static final long serialVersionUID = -1529783362243627391L;
	
	protected long				 _projectID;
	protected List<HistoryEntry> _pastEntries;
	protected List<HistoryEntry> _futureEntries;
	
	public History(Project project) {
		_projectID = project.id;
		_pastEntries = new ArrayList<HistoryEntry>();
		_futureEntries = new ArrayList<HistoryEntry>();
	}
	
	public void addEntry(HistoryEntry entry) {
		for (HistoryEntry entry2 : _futureEntries) {
			entry2.delete();
		}
		_futureEntries.clear();
		
		_pastEntries.add(entry);
		
		entry.apply(ProjectManager.singleton.getProject(_projectID));
		setModified();
	}
	
	protected void setModified() {
		ProjectManager.singleton.getProjectMetadata(_projectID).updateModified();
	}
	
	public List<HistoryEntry> getLastPastEntries(int count) {
		if (count <= 0) {
			return new LinkedList<HistoryEntry>(_pastEntries);
		} else {
			return _pastEntries.subList(Math.max(_pastEntries.size() - count, 0), _pastEntries.size());
		}
	}
	
	public void undoRedo(long lastDoneEntryID) {
		if (lastDoneEntryID == 0) {
			undo(_pastEntries.size());
		} else {
			for (int i = 0; i < _pastEntries.size(); i++) {
				if (_pastEntries.get(i).id == lastDoneEntryID) {
					undo(_pastEntries.size() - i - 1);
					return;
				}
			}
			
			for (int i = 0; i < _futureEntries.size(); i++) {
				if (_futureEntries.get(i).id == lastDoneEntryID) {
					redo(i + 1);
					return;
				}
			}
		}
	}
	
	protected HistoryEntry getEntry(long entryID) {
		for (int i = 0; i < _pastEntries.size(); i++) {
			if (_pastEntries.get(i).id == entryID) {
				return _pastEntries.get(i);
			}
		}
		
		for (int i = 0; i < _futureEntries.size(); i++) {
			if (_futureEntries.get(i).id == entryID) {
				return _futureEntries.get(i);
			}
		}
		return null;
	}
	
	protected void undo(int times) {
		Project project = ProjectManager.singleton.getProject(_projectID);
		
		while (times > 0 && _pastEntries.size() > 0) {
			HistoryEntry entry = _pastEntries.remove(_pastEntries.size() - 1);
			times--;
			
			entry.revert(project);
			
			_futureEntries.add(0, entry);
		}
		setModified();
	}
	
	protected void redo(int times) {
		Project project = ProjectManager.singleton.getProject(_projectID);
		
		while (times > 0 && _futureEntries.size() > 0) {
			HistoryEntry entry = _futureEntries.remove(0);
			times--;
			
			entry.apply(project);
			
			_pastEntries.add(entry);
		}
		setModified();
	}
	
	public void write(JSONWriter writer, Properties options)
			throws JSONException {
		
		writer.object();
		
		writer.key("past"); writer.array();
		for (HistoryEntry entry : _pastEntries) {
			entry.write(writer, options);
		}
		writer.endArray();
		
		writer.key("future"); writer.array();
		for (HistoryEntry entry : _futureEntries) {
			entry.write(writer, options);
		}
		writer.endArray();
		
		writer.endObject();
	}
}