/*
    * Created on Apr 25, 2005, Copyright UC Regents
    */
   package org.telscenter.sailotrunk;
   
   import java.awt.Dimension;
   import java.awt.event.ActionEvent;
   import java.awt.event.ActionListener;
   import java.awt.event.WindowAdapter;
  import java.awt.event.WindowEvent;
  import java.beans.beancontext.BeanContextServices;
  import java.io.ByteArrayInputStream;
  import java.io.ByteArrayOutputStream;
  import java.io.File;
  import java.io.StringReader;
  import java.net.URL;
  import java.util.TimerTask;
  import java.util.TooManyListenersException;
  import java.util.logging.Logger;
  
  import javax.swing.AbstractAction;
  import javax.swing.Action;
  import javax.swing.JFileChooser;
  import javax.swing.JFrame;
  import javax.swing.JOptionPane;
  import javax.swing.JScrollPane;
  import javax.swing.JTextArea;
  import java.util.Timer;
  
  import net.sf.sail.common.beansupport.ITitleAware;
  import net.sf.sail.common.beansupport.SailBeanContextChildSupport;
  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.beans.service.SessionService;
  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 org.concord.framework.otrunk.OTObject;
  import org.concord.framework.otrunk.view.OTViewEntry;
  import org.concord.framework.otrunk.view.OTViewFactory;
  import org.concord.otrunk.OTMLToXHTMLConverter;
  import org.concord.otrunk.datamodel.OTDatabase;
  import org.concord.otrunk.view.OTConfig;
  import org.concord.otrunk.view.OTViewChild;
  import org.concord.otrunk.view.OTViewer;
  import org.concord.otrunk.view.OTViewerHelper;
  import org.concord.otrunk.xml.XMLDatabase;
  
  /***
   * Bean which should be the root bean of the curnit.   
   * It starts up the OTViewer and saves its data into a rim.
   * 
   * @author scytacki
   */
  public class SailOTViewer extends SailBeanContextChildSupport 
  	implements ITitleAware, OTPodAdapter
  {
  	private static final String EXPORT_HTML_DIALOG_TAG = "_dialog_";
  
  	public static final String SAILOTRUNK_VIEW_LEARNEROTML = "sailotrunk.learnerotml.edit";
  
  	public static final String SAILOTRUNK_OTMLURL = "sailotrunk.otmlurl";
  
  	public static final String SAILOTRUNK_EXPORT_HTML = "sailotrunk.export.html";
  	
  	static {
  		// I don't know if this will be soon enough
  		// but lets see
          System.setProperty("apple.laf.useScreenMenuBar","true");
  	}
  	
  	/***
  	 * Logger for this class
  	 */
  	private static final Logger logger = Logger.getLogger(SailOTViewer.class
  			.getName());
  
  	private static final long serialVersionUID = 1L;
  	
  	private URL authoredDataURL;
  
  	private AgentService agentService;
  	private SessionService sessionService;
  		
  	private Rim workRim;
  
  	private ISock workSock;
  
  	private boolean hideTree = false;
  	private int userMode = 0; 
  		
  	OTViewer viewer = null;
  	OTViewerHelper viewerHelper = new OTViewerHelper();
  	
 	private JTextArea xmlTextArea;
 	
 	SessionEventListener sessionListener = new SessionEventListener() {
 
 		private Timer timer;
 
 		public void sessionStarted (SessionEvent e) {
 		}
 
 		public void sessionInitiated (SessionEvent e) 
 		{
 			String boolStr = sessionService.getProperty(SAILOTRUNK_VIEW_LEARNEROTML, "false");
 			boolean showLearnerOtmlAsText = Boolean.parseBoolean(boolStr);
 			if (showLearnerOtmlAsText){
 				loadDataAsText();
 				return;
 			}
 			
 			String exportHtmlFolderStr = sessionService.getProperty(SAILOTRUNK_EXPORT_HTML, null);
 			if (exportHtmlFolderStr != null){
 				exportHtml(exportHtmlFolderStr);
 				return;
 			}
 			
 			
 			viewer = new OTViewer();
 
 			viewer.addService(OTPodAdapter.class, SailOTViewer.this);
 			
 			viewer.setUserMode(getUserMode());
 			
 			String authoredDataUrlStr = getAuthoredDataUrlStr();			
 			
 			viewer.init(authoredDataUrlStr);	
 
 			AbstractAction exitAction = new AbstractAction(){
 				/***
 				 * Not intended to be serialized, just added remove compile warning
 				 */
 				private static final long serialVersionUID = 1L;
 				
 				public void actionPerformed(ActionEvent e) {
 					if (timer != null) {
 						timer.cancel();
 					}
                     viewer.getViewContainerPanel().setCurrentObject(null);
 					saveDataInternal();
 					
 					getSessionService().userRequestsTermination();
 				}				
 			};
 			exitAction.putValue(Action.NAME, "Exit");		
 			
 			viewer.setExitAction(exitAction);
 			
 			viewer.updateMenuBar();
 			
 			if(viewer.getUserMode() == OTViewerHelper.NO_USER_MODE){
 				// if we are in no user mode then we don't need to read data in from the 
 				// sock or initialize the user data.
 				return;
 			}
 			
 			timer = new Timer();
 			long delay = 60*1000; // run every 1 minute
 			timer.scheduleAtFixedRate(new TimerTask() {
 				@Override
 				public void run() {
 					// System.err.println("Saving otrunk data...");
 					saveDataInternal();
 				}
 			}, delay, delay);
 			
 			XMLDatabase sailLearnerDB = getLearnerDatabase();
 			try {
 				if(sailLearnerDB != null){
 					viewer.loadUserDataDb(sailLearnerDB, null);
 					((XMLDatabase)sailLearnerDB).setDirty(false);				
 				} else {
 					viewer.newAnonUserData();
 				}
 			} catch (Exception e1) {
 				// TODO Auto-generated catch block
 				e1.printStackTrace();
 			}			
 		}
 
 		public void sessionStopped(SessionEvent e) {
 			saveDataInternal();
 		}
 
 	};
 
 	private boolean needsSockEntryInit = true;
 	
 	/***
 	 * This allows us to view the learner data sent from the SDS as a plain-text file,
 	 * which we can then edit by hand. Closing the file will then attempt to save the
 	 * modified OTML learner data back to the SDS as a new bundle.
 	 * 
 	 * Warning: No attempt is made to verify the XML, so no guarantees are given if
 	 * uploading of bad XML is attempted.
 	 * 
 	 * @author sfentress
 	 */
 	private void loadDataAsText(){
 		final ISock sock = getWorkSock();
 		
 		Object lastEntry = null;
 		if(sock != null && sock.size() > 0){
 			lastEntry = sock.peek();
 		}
 		
 		if (lastEntry == null){
 			System.err.println("Error loading data");
 			return;
 		}
 		
 		String dataString = null;
 		if (lastEntry instanceof String){
 			dataString = (String)lastEntry;
 		} else if(lastEntry instanceof byte[]){
 			dataString = new String((byte[])lastEntry);
 		}
 		
 		xmlTextArea = new JTextArea();
 		xmlTextArea.setLineWrap(true);
 		xmlTextArea.setWrapStyleWord(true);
 		
 		xmlTextArea.setText(dataString);
 		
 		final JFrame frame = new JFrame();
 		
 		frame.addWindowListener(new WindowAdapter() {
 			public void windowClosing(WindowEvent e)
 			{
 				int saveData = JOptionPane.showConfirmDialog(
 						frame, "Upload modified bundle?", "Confirm", JOptionPane.YES_NO_OPTION);
 				
 				if (saveData == JOptionPane.YES_OPTION){
 					getSessionService().userRequestsTermination();
 				}
 			}
 		});
 		
 		JScrollPane scrollPane = new JScrollPane();
 		scrollPane.setViewportView(xmlTextArea);
 		xmlTextArea.setCaretPosition(0);
 		frame.add(scrollPane);
 		frame.setPreferredSize(new Dimension(400,400));
 		frame.pack();
 		frame.setVisible(true);
 	}
 		
 	protected void exportHtml(String exportHtmlFolder) {		
 		try{
 			File fileToSave = null;
 			
 			if(EXPORT_HTML_DIALOG_TAG.equals(exportHtmlFolder)){
 				JFileChooser fileChooser = new JFileChooser();
 				int result = fileChooser.showSaveDialog(null);
 				switch(result){
 				case JFileChooser.APPROVE_OPTION:
 					fileToSave = fileChooser.getSelectedFile();
 				case JFileChooser.CANCEL_OPTION:
 				case JFileChooser.ERROR_OPTION:
 					break;
 				}
 			}
 			
 			viewerHelper.addService(OTPodAdapter.class, SailOTViewer.this);
 			
 			viewerHelper.setUserMode(getUserMode());
 					
 			String authoredDataUrlStr = getAuthoredDataUrlStr();
 
 			URL authoredContentURL = new URL(authoredDataUrlStr);
 			viewerHelper.loadAuthoredContentURL(authoredContentURL);
 			
 			if(viewerHelper.getUserMode() == OTViewerHelper.NO_USER_MODE){
 				// if we are in no user mode then we don't need to read data in from the 
 				// sock or initialize the user data, just save the html...
 			} else {
 				XMLDatabase learnerDatabase = getLearnerDatabase();
 				if(learnerDatabase != null){
 					viewerHelper.loadUserData(learnerDatabase, null);
 				}
 			}
 
 	    	OTObject rootObject = viewerHelper.getRootObject();
 	    	
 	    	OTViewEntry rootViewEntry = null;
 			if(rootObject instanceof OTViewChild){
 	    		rootViewEntry = ((OTViewChild)rootObject).getViewid();
 	    		rootObject = ((OTViewChild)rootObject).getObject();
 	    	}
 	    	OTViewFactory viewFactory = viewerHelper.getViewFactory();
 
 			final OTMLToXHTMLConverter conv = new OTMLToXHTMLConverter(viewFactory, rootObject, 
 				rootViewEntry,
 				OTConfig.getSystemPropertyViewMode());
 
 			if(fileToSave == null) {
 				String outputFolderStr = exportHtmlFolder;
 				if(outputFolderStr == null){
 					outputFolderStr = System.getProperty("user.home");
 				}
 				File outputFolder = new File(outputFolderStr);
 				outputFolder.mkdirs();
 				fileToSave = new File(outputFolder, "index.html");
 			}
 			
 			conv.setXHTMLParams(fileToSave, 800, 600);
 
 			Thread runner = new Thread(){
 				public void run() {
 					conv.run();
 					System.exit(0);
 				}
 				
 			};
 			
 			runner.start();
 		}
 		catch (Exception ex) {
 			ex.printStackTrace();
 		}
 	}
 
 
 	protected void registerDesiredServices(BeanContextServices bcs) {
 	}
 		
 	
 	/***
 	 * This method can be called more than once in the authoring runtime, when
 	 * the preview is shown more than once.
 	 * 
 	 * @param bcs
 	 * @param serviceClass
 	 * @param project
 	 */
 	@Override
 	protected void consumeService(BeanContextServices bcs, Class serviceClass) {
 		if (serviceClass == SessionService.class) {
 			try {
 				sessionService = (SessionService) bcs.getService(this, this,
 						SessionService.class, this, this);
 
 				sessionService.addSessionEventListener(sessionListener);
 			} catch (TooManyListenersException e1) {
 				// TODO Auto-generated catch block
 				logger
 						.severe("BeanContextServices, Class -  : exception: " + e1); //$NON-NLS-1$
 			}
 		}
 		if (serviceClass == AgentService.class)
 		{
 			try
 			{
 				final AgentService p = (AgentService) bcs.getService(this,
 						this, AgentService.class, this, this);
 				agentService = p;
 			}
 			catch (TooManyListenersException e)
 			{
 				e.printStackTrace();
 			}
 		}
 	}
 	
 	/***
 	 * @return the sessionService
 	 */
 	public SessionService getSessionService() {
 		return sessionService;
 	}
 
 	/***
 	 * @param sessionService the sessionService to set
 	 */
 	public void setSessionService(SessionService sessionService) {
 		this.sessionService = sessionService;
 	}
 
 	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;
 	}
 
 	public Rim getWorkRim() {
 		return workRim;
 	}
 
 	public void setWorkRim(Rim workRim) {
 		this.workRim = workRim;
 	}
 
 	public ISock getWorkSock() {
 		if (workSock == null) {
 			workSock = getRimSock(getWorkRim());
 		}
 		return workSock;
 	}
 
 
 	public URL getAuthoredDataURL()
 	{
 		return authoredDataURL;
 	}
 
 	public void setAuthoredDataURL(URL url)
 	{
 		authoredDataURL = url;
 	}
 	
 	public String getTitle() {
 		return "SailOTViewer title";
 	}
 
 
 	public void setTitle(String title) {
 	}
 
 
 	public boolean isHideTree() {
 		return hideTree;
 	}
 
 
 	public void setHideTree(boolean hideTree) {
 		this.hideTree = hideTree;
 	}
 
 
 	public int getUserMode() {
 		return userMode;
 	}
 
 
 	public void setUserMode(int userMode) {
 		this.userMode = userMode;
 	}	
 	
 	private void saveDataInternal() {
 		saveDataInternal(true);
 	}
 
 	private void saveDataInternal(boolean replace)
 	{
 		if (needsSockEntryInit ) {
 			// override replace so that a sockentry gets created in the current sessionbundle
 			replace = false;
 		}
 		
 		ISock sock = getWorkSock();
 		
 		// If we're viewing the straight otml, we don't need to
 		// use the userDB at all -- we can just upload the text.
 		String boolStr = sessionService.getProperty(SAILOTRUNK_VIEW_LEARNEROTML, "false");
 		boolean showLearnerOtmlAsText = Boolean.parseBoolean(boolStr);
 		if (showLearnerOtmlAsText){
 			if(getWorkRim().getShape() == byte[].class){
 				sock.add(xmlTextArea.getText().getBytes(), replace);
 			} else {
 				sock.add(xmlTextArea.getText(), replace);
 			}
 			return;
 		}
 		
 		OTDatabase userDB = viewer.getUserDataDb();
 
 		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){
 
 				// this is a hack to test this should be moved
 				// to the converter
 				//getWorkRim().setShape(bytes.getClass());
 				
 				if(getWorkRim().getShape() == byte[].class){
 					ByteArrayOutputStream out = new ByteArrayOutputStream();
 					viewerHelper.saveOTDatabase(userDB, out);
 					byte [] bytes = out.toByteArray();
 					sock.add(bytes, replace);
 					if(userDB instanceof XMLDatabase){
 						((XMLDatabase)userDB).setDirty(false);
 					}
 				} else {
 					String userData = viewerHelper.saveOTDatabase(userDB);
 					sock.add(userData, replace);
 				}
 				needsSockEntryInit = false;
 			} else {
 				System.err.println("No sock in OTrunkStep - cannot save data");
 			}
 			
 		} catch (Exception e1) {
 			// TODO Auto-generated catch block
 			e1.printStackTrace();
 		}										
 
 	}
 
 	/***
 	 * @see OTPodAdapter.registerBean()
 	 */
 	@SuppressWarnings("unchecked")
 	public void registerBean(Object bean) {
 		getBeanContext().add(bean);
 	}
 
 
 	private String getAuthoredDataUrlStr() {
 		String authoredDataUrlStr = getAuthoredDataURL().toExternalForm();
 		
 		// Figure out if there is a url being set through the configuration
 		String authoredDataConfigUrlStr = 
 			sessionService.getProperty(SAILOTRUNK_OTMLURL, null);
 		
 		// In the future we can load both the config url string
 		// and the interally set (pod) url string. So the data can 
 		// be split up.  For now we'll just replace the old string
 		// with the new one
 		if(authoredDataConfigUrlStr != null) {
 			authoredDataUrlStr = authoredDataConfigUrlStr;
 		}
 		return authoredDataUrlStr;
 	}
 
 	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 String) {
 				StringReader reader = new StringReader((String)lastEntry);
 
 				// FIXME need to pass in some context url this is needed
 				// for any referenced resources in the otml
 				return (XMLDatabase) viewerHelper.loadOTDatabase(reader, null);
 
 			} else if(lastEntry instanceof byte[]){
 				ByteArrayInputStream inStream = new ByteArrayInputStream((byte[])lastEntry); 
 
 				return (XMLDatabase) viewerHelper.loadOTDatabase(inStream, null);
 			}										
 
 		} catch (Exception exp) {
 			exp.printStackTrace();
 		}
 		
 		return null;
 	}
 	
 }