package com.sbs.jpm.client;

import java.util.Date;

import com.google.gwt.core.client.EntryPoint;
import com.google.gwt.user.client.ui.RootPanel;
import com.smartgwt.client.data.DSRequest;
import com.smartgwt.client.data.DSResponse;
import com.smartgwt.client.data.DataSource;
import com.smartgwt.client.data.DataSourceField;
import com.smartgwt.client.rpc.RPCResponse;
import com.smartgwt.client.types.DSDataFormat;
import com.smartgwt.client.types.DSOperationType;
import com.smartgwt.client.types.DSProtocol;
import com.smartgwt.client.types.FieldType;
import com.smartgwt.client.util.SC;
import com.smartgwt.client.widgets.Button;
import com.smartgwt.client.widgets.events.ClickEvent;
import com.smartgwt.client.widgets.events.ClickHandler;
import com.smartgwt.client.widgets.form.fields.TextItem;
import com.smartgwt.client.widgets.layout.VLayout;
import com.smartgwt.client.widgets.tree.TreeGrid;
import com.smartgwt.client.widgets.tree.TreeGridField;
import com.smartgwt.client.widgets.tree.TreeNode;
import com.smartgwt.client.widgets.tree.events.DataArrivedEvent;
import com.smartgwt.client.widgets.tree.events.DataArrivedHandler;

public class TestCase implements EntryPoint {

	private WBSPanel wbsPanel;
	private TaskDataSource dataSource;

	public void onModuleLoad() {

		dataSource = new TaskDataSource();

		wbsPanel = new WBSPanel();
		wbsPanel.setWidth100();
		wbsPanel.setHeight100();

		RootPanel.get().add(wbsPanel);
		
		SC.showConsole();

	}

	private class WBSPanel extends VLayout {

		private Button addToolbarButton;

		private TreeGrid tasksTreeGrid;
		private TreeGridField nameTreeGridField;

		public WBSPanel() {

			addToolbarButton = new Button();
			addToolbarButton.setTitle("Add");
			addToolbarButton.addClickHandler(new ClickHandler() {
				@Override
				public void onClick(ClickEvent event) {
					addTask();
				}
			});

			nameTreeGridField = new TreeGridField();
			nameTreeGridField.setName(TaskDataSource.NAME_FIELD);
			nameTreeGridField.setTitle("Task Name");
			nameTreeGridField.setTreeField(true);

			TextItem nameTextItem = new TextItem();
			nameTextItem.setName(TaskDataSource.NAME_FIELD);
			nameTextItem.setTitle("Project name");

			tasksTreeGrid = new TreeGrid();
			tasksTreeGrid.setWidth100();
			tasksTreeGrid.setHeight(500);
			tasksTreeGrid.setTreeRootValue(1);
			tasksTreeGrid.setDataSource(dataSource);
			tasksTreeGrid.setLoadDataOnDemand(false);
			tasksTreeGrid.setAutoFetchData(true);
			tasksTreeGrid.setFields(nameTreeGridField);
			tasksTreeGrid.addDataArrivedHandler(new DataArrivedHandler() {
				@Override
				public void onDataArrived(DataArrivedEvent event) {
					tasksTreeGrid.getTree().openAll();
				}
			});
			this.setMembers(addToolbarButton, tasksTreeGrid);
		}

		private void addTask() {
			dataSource.updateCache();
		}
	}

	private class TaskTreeNode extends TreeNode {

		public TaskTreeNode(String taskId, String reportsTo, String name) {
			setTaskId(taskId);
			setReportsTo(reportsTo);
			setName(name);
			setPojo();
		}

		public void setTaskId(String value) {
			setAttribute(TaskDataSource.UNIQUE_ID_FIELD, value);
		}

		public void setReportsTo(String value) {
			setAttribute(TaskDataSource.PARENT_UNIQUE_ID, value);
		}

		public void setName(String name) {
			setAttribute(TaskDataSource.NAME_FIELD, name);
		}

		public void setPojo() {
			setAttributeAsJavaObject(TaskDataSource.POJO_FIELD, new Pojo());
		}
	}

	private class TaskDataSource extends GwtRpcDataSource {
		private static final String DS_UNIQUE_ID = "_TaskDataSource";

		public static final String PARENT_UNIQUE_ID = "parentUniqueId"
				+ DS_UNIQUE_ID;
		public static final String UNIQUE_ID_FIELD = "uniqueIdField"
				+ DS_UNIQUE_ID;
		public static final String NAME_FIELD = "nameField" + DS_UNIQUE_ID;
		public static final String POJO_FIELD = "pojoField" + DS_UNIQUE_ID;

		private TaskDataSource() {

			DataSourceField uniqueId = new DataSourceField();
			uniqueId.setName(UNIQUE_ID_FIELD);
			uniqueId.setPrimaryKey(true);
			uniqueId.setHidden(true);
			uniqueId.setType(FieldType.INTEGER);

			DataSourceField reportsToField = new DataSourceField();
			reportsToField.setName(PARENT_UNIQUE_ID);
			reportsToField.setForeignKey(DS_UNIQUE_ID + "." + UNIQUE_ID_FIELD);
			reportsToField.setHidden(true);
			reportsToField.setType(FieldType.INTEGER);

			this.setID(DS_UNIQUE_ID);
			this.setFields(uniqueId, reportsToField);
			this.setCacheAllData(true);
			this.setCacheMaxAge(3600);
		}

		@Override
		protected void executeFetch(final String requestId, DSRequest request,
				final DSResponse response) {

			TaskTreeNode root = new TaskTreeNode("root", "1", "Project 1");
			TaskTreeNode node2 = new TaskTreeNode("t1", "root", "Task 1");
			TaskTreeNode node3 = new TaskTreeNode("t2", "root", "Task 2");
			TaskTreeNode node4 = new TaskTreeNode("t3", "root", "Task 3");
			TaskTreeNode node5 = new TaskTreeNode("t4", "root", "Task 4");

			response.setData(new TreeNode[] { root, node2, node3, node4, node5 });
			response.setStatus(RPCResponse.STATUS_SUCCESS);
			response.setInvalidateCache(true);
			processResponse(requestId, response);
		}

		@Override
		protected void executeAdd(final String requestId, DSRequest request,
				final DSResponse response) {
		}

		@Override
		protected void executeUpdate(final String requestId,
				final DSRequest request, final DSResponse response) {
		}

		@Override
		protected void executeRemove(final String requestId, DSRequest request,
				final DSResponse response) {
		}

		public void updateCache() {
			TreeNode newTaskt = new TreeNode();
			newTaskt.setAttribute(TaskDataSource.NAME_FIELD,
					"New task from server");
			newTaskt.setAttribute(TaskDataSource.PARENT_UNIQUE_ID, "root");
			newTaskt.setAttribute(TaskDataSource.UNIQUE_ID_FIELD,
					new Date().getTime() + "");
			// again this setter introduces a bug in development mode
			// but only when calling updateCaches()
			newTaskt.setAttributeAsJavaObject(TaskDataSource.POJO_FIELD,
					new Pojo());

			DSResponse dsResponse = new DSResponse();
			dsResponse.setData(newTaskt);
			dsResponse.setOperationType(DSOperationType.ADD);
			updateCaches(dsResponse);
		}
	}

	private abstract class GwtRpcDataSource extends DataSource {

		public GwtRpcDataSource() {
			setDataProtocol(DSProtocol.CLIENTCUSTOM);
			setDataFormat(DSDataFormat.CUSTOM);
			setClientOnly(false);
		}

		@Override
		protected Object transformRequest(DSRequest request) {
			String requestId = request.getRequestId();
			DSResponse response = new DSResponse();
			response.setAttribute("clientContext",
					request.getAttributeAsObject("clientContext"));
			response.setStatus(0);

			switch (request.getOperationType()) {
			case FETCH:
				executeFetch(requestId, request, response);
				break;
			case ADD:
				executeAdd(requestId, request, response);
				break;
			case UPDATE:
				executeUpdate(requestId, request, response);
				break;
			case REMOVE:
				executeRemove(requestId, request, response);
				break;
			default:
				break;
			}
			return request.getData();
		}

		protected abstract void executeFetch(String requestId,
				DSRequest request, DSResponse response);

		protected abstract void executeAdd(String requestId, DSRequest request,
				DSResponse response);

		protected abstract void executeUpdate(String requestId,
				DSRequest request, DSResponse response);

		protected abstract void executeRemove(String requestId,
				DSRequest request, DSResponse response);

	}

	private class Pojo {

		public Pojo() {
		}

	}
}
