package org.telscenter.sailotrunk;
   
   import java.awt.BorderLayout;
   import java.awt.Component;
   import java.io.ByteArrayInputStream;
   import java.io.ByteArrayOutputStream;
   import java.io.File;
   import java.io.FileNotFoundException;
   import java.io.FileOutputStream;
  import java.net.URL;
  
  import javax.swing.JDialog;
  import javax.swing.JPanel;
  
  import net.sf.sail.core.beans.Pod;
  import net.sf.sail.core.beans.SessionContext;
  import net.sf.sail.core.beans.event.SessionEvent;
  import net.sf.sail.core.beans.event.SessionEventListener;
  import net.sf.sail.core.beans.service.AgentService;
  import net.sf.sail.core.curnit.ICurnit;
  import net.sf.sail.core.entity.IAgent;
  import net.sf.sail.core.entity.ISock;
  import net.sf.sail.core.entity.MismatchedAgentSetSizeException;
  import net.sf.sail.core.entity.Rim;
  import net.sf.sail.core.entity.Role;
  import net.sf.sail.core.entity.UnsupportedRimShapeException;
  import net.sf.sail.core.session.SessionDataService;
  import net.sf.sail.core.uuid.CurnitUuid;
  import net.sf.sail.core.uuid.PodUuid;
  
  import org.concord.framework.otrunk.OTControllerService;
  import org.concord.framework.otrunk.OTID;
  import org.concord.framework.otrunk.OTObject;
  import org.concord.framework.otrunk.OTrunk;
  import org.concord.framework.otrunk.view.OTFrameManager;
  import org.concord.framework.otrunk.view.OTViewContext;
  import org.concord.framework.otrunk.view.OTViewFactory;
  import org.concord.otrunk.OTrunkImpl;
  import org.concord.otrunk.datamodel.OTDatabase;
  import org.concord.otrunk.datamodel.OTUUID;
  import org.concord.otrunk.view.OTViewContainerPanel;
  import org.concord.otrunk.view.OTViewerHelper;
  import org.concord.otrunk.xml.XMLDatabase;
  
  public class OTrunkCurnit implements ICurnit 
  {
  	private CurnitUuid curnitId;
  	private OTDatabase otDb;
  	private OTrunk otrunk;
  	private URL otmlUrl;
  	private OTViewerHelper otViewerHelper;
  	private Object rootBean;
  	private OTObject otrunkRoot;
  	private OTControllerService otControllerService;
  	private SessionContext sessionContext;
  	private Pod pod;
  	private Rim<byte []> rim;
  	private AgentService agentService;
  	private ISock workSock;
  	private boolean authoring = false;
  
  	private final int PREVIEW_WINDOW_WIDTH = 800;
  	private final int PREVIEW_WINDOW_HEIGHT = 600;
  	
  	public OTrunkCurnit(URL otmlUrl)
  	{
  		this.otmlUrl = otmlUrl;
  		otViewerHelper = new OTViewerHelper();
  		
  	}
  
  	/***
  	 * This returns a CurnitUUID
  	 * 
  	 * @see net.sf.sail.core.curnit.ICurnit#getCurnitId()
  	 */
  	public CurnitUuid getCurnitId() {
  		if(curnitId != null){
  			return curnitId;
  		}
  		
  		try {
  			OTDatabase otDb = getOTDatabase();
  			if(otDb == null){
  				System.err.println("No valid ot database");
  				return null;
  			}
  
  			OTID dbId = otDb.getDatabaseId();			
  			if(!(dbId instanceof OTUUID)){
  				System.err.println("OTDatabase doesn't have a uuid");
  				return null;
  			}
  			
  			curnitId = new CurnitUuid(((OTUUID)dbId).asByteArray());
  			return curnitId;
  		} catch (Exception e) {
  			// TODO Auto-generated catch block
  			e.printStackTrace();
 		}
 		
 		return null; 
 	}
 
 	public Object getRootBean() {
 		if(rootBean != null){
 			return rootBean;
 		}
 
 		rootBean = otControllerService.getRealObject(otrunkRoot);
 		return rootBean;
 	}
 
 	public void initialize() 
 	{
 		try {
 			OTDatabase otDb = getOTDatabase();
 			otrunk = new OTrunkImpl(otDb);
 			otViewerHelper.initOTrunk(otrunk);
 			
 			otrunkRoot = otrunk.getRoot();
 			
 			// FIXME need to setup the learner database.   
 			// In the SailOTViewer this is done by having the 
 			// root object add a session listener. 
 			pod = new Pod();
 
 			if(otrunkRoot.getGlobalId() instanceof OTUUID){
 				pod.setPodId(new PodUuid(otrunkRoot.getGlobalId().toExternalForm()));
 			} else if(otDb.getDatabaseId() instanceof OTUUID){
 				pod.setPodId(new PodUuid(otDb.getDatabaseId().toExternalForm()));				
 			} else {
 				throw new IllegalStateException("The otml file loaded in needs an id on the top" +
 						"level otrunk element or root object");
 			}
 
 			rim = new Rim<byte []>();
 			pod.add(rim);
 			rim.setName("ot.learner.data");
 			rim.setShape(byte [].class);
 
 			if(!authoring){
 				SessionDataService sessionDataService = sessionContext.getSessionDataService();
 				agentService = sessionDataService.getPersistenceService(sessionContext);
 
 				XMLDatabase sailLearnerDB = getLearnerDatabase();
 				try {
 					if(sailLearnerDB != null){
 						otViewerHelper.loadUserData(sailLearnerDB, null);
 						((XMLDatabase)sailLearnerDB).setDirty(false);				
 					} else {
 						otViewerHelper.newAnonUserData();
 					}
 					
 					otrunkRoot = otrunk.getUserRuntimeObject(otrunkRoot, otViewerHelper.getCurrentUser());
 				} catch (Exception e1) {
 					// TODO Auto-generated catch block
 					e1.printStackTrace();
 				}
 				
 				sessionContext.addSessionEventListener(new SessionEventListener(){
 
 					public void sessionInitiated(SessionEvent e) {						
 					}
 
 					public void sessionStarted(SessionEvent e) {
 					}
 
 					public void sessionStopped(SessionEvent e) {
 						saveDataInternal();
 					}					
 				});
 			}
 
 			
 			otControllerService = otrunkRoot.getOTObjectService().createControllerService();
 			OTViewFactory viewFactory = otViewerHelper.getViewFactory();
 			if(viewFactory != null){
 				otControllerService.addService(OTViewContext.class, viewFactory.getViewContext());
 			}
 			
 		} catch (Exception e) {
 			// TODO Auto-generated catch block
 			e.printStackTrace();
 		}
 		
 		
 		// TODO Auto-generated method stub
 
 	}
 
 	public OTDatabase getOTDatabase() throws Exception
 	{
 		if(otDb != null){
 			return otDb;
 		}
 		
 		if(otmlUrl.getProtocol().startsWith("html")){
 			otDb = otViewerHelper.loadOTDatabaseXML(otmlUrl);
 		} else {
 			otDb = otViewerHelper.loadOTDatabase(otmlUrl);
 		}
 		
 		return otDb;
 	}
 	
 	public void previewCurnit(Component component) throws Exception {
 		if( otDb == null)
 			throw new Exception("Curnit has not been initialized");
 		
 		otControllerService.saveRealObject(rootBean, otrunkRoot);
 		
 		//load the ot data from the db to the frame
 		otViewerHelper.loadOTrunk(otDb, component);
 		
 		//the popup window that will display the project preview
 		JDialog popup = new JDialog();
 
 		
 		//section of code below sets up the view to display the project preview
 		OTViewContext viewContext = 
 			(OTViewContext) otControllerService.getService(OTViewContext.class);
 
 		JPanel panel=new JPanel(new BorderLayout(5, 5));
 		OTViewContainerPanel otPanel = null;
 		OTFrameManager frameManager = 
 			(OTFrameManager) viewContext.getViewService(OTFrameManager.class);
 		otPanel = new OTViewContainerPanel(frameManager);
 		otPanel.setTopLevelContainer(true);
 		OTViewFactory viewFactory = viewContext.createChildViewFactory();
 		otPanel.setOTViewFactory(viewFactory);
 		
 		panel.setLayout(new BorderLayout());
 		otPanel.setCurrentObject(otrunkRoot);
 		panel.add(otPanel, BorderLayout.CENTER);
 
 		//add the project preview to the popup window
 		popup.setSize(PREVIEW_WINDOW_WIDTH, PREVIEW_WINDOW_HEIGHT);
 		popup.add(panel);
 		popup.setVisible(true);
 		
 	}
 	
 	public void saveCurnit(File file) throws FileNotFoundException, Exception {
 		
 		if( otDb == null)
 			throw new Exception("Curnit has not been initialized");
 		
 		otControllerService.saveRealObject(rootBean, otrunkRoot);
 		otViewerHelper.saveOTDatabase(otDb, new FileOutputStream(file));
 		
 	}
 
 	public void setSessionContext(SessionContext sessionContext) {
 		this.sessionContext = sessionContext;
 	}
 	
 	public Rim getWorkRim() {
 		return rim;
 	}
 
 	public ISock getWorkSock() {
 		if (workSock == null) {
 			workSock = getRimSock(getWorkRim());
 		}
 		return workSock;
 	}
 	
 	protected ISock getRimSock(Rim rim)
 	{
 		ISock sock = null;
 		if (rim != null)
 		{
 			try {
 				IAgent agent = agentService.getAgentsInRole(Role.RUN_WORKGROUP).getSingle();
 				sock = agentService.getSock(rim, agent);
 			} catch (MismatchedAgentSetSizeException e) {
 				// TODO Auto-generated catch block
 				e.printStackTrace();
 			} catch (UnsupportedRimShapeException e) {
 				// TODO Auto-generated catch block
 				e.printStackTrace();
 			}
 		}
 		return sock;
 	}
 
 	private XMLDatabase getLearnerDatabase() {
 		
 		ISock sock = getWorkSock();
 		
 		Object lastEntry = null;
 		if(sock != null && sock.size() > 0){
 			lastEntry = sock.peek();
 		}
 
 		if(lastEntry == null){
 			return null;
 		}
 		
 		try {
 			if(lastEntry instanceof byte[]){
 				ByteArrayInputStream inStream = new ByteArrayInputStream((byte[])lastEntry); 
 
 				return (XMLDatabase) otViewerHelper.loadOTDatabase(inStream, null);
 			}										
 
 		} catch (Exception exp) {
 			exp.printStackTrace();
 		}
 		
 		return null;
 	}
 
 	private void saveDataInternal()
 	{
 		ISock sock = getWorkSock();
 		
 		OTDatabase userDB = otViewerHelper.getUserOtDB();
 
 		if(userDB == null){
 			// no user database was created 
 			return;
 		}
 		
 		// If the database hasn't been modified, don't
 		// save anything
 		if(userDB instanceof XMLDatabase &&
 				!((XMLDatabase)userDB).isDirty()){
 			return;
 		}
 		
 		try {
 			if(sock != null && userDB != null){
 				ByteArrayOutputStream out = new ByteArrayOutputStream();
 				otViewerHelper.saveOTDatabase(userDB, out);
 				byte [] bytes = out.toByteArray();
 				sock.add(bytes);
 				if(userDB instanceof XMLDatabase){
 					((XMLDatabase)userDB).setDirty(false);
 				}
 			} else {
 				System.err.println("No sock in OTrunkStep - cannot save data");
 			}
 			
 		} catch (Exception e1) {
 			// TODO Auto-generated catch block
 			e1.printStackTrace();
 		}										
 
 	}
 	
 	/***
 	 * This is false by default
 	 * 
 	 * @param mode
 	 */
 	public void setAuthoring(boolean mode)
 	{
 		authoring  = mode;
 	}
 }