/***
    * 
    */
   package org.telscenter.pas.otrunk;
   
   import java.awt.BorderLayout;
   import java.awt.Color;
   import java.awt.Component;
   import java.awt.Dimension;
  import java.awt.event.ActionEvent;
  import java.awt.event.ActionListener;
  import java.awt.event.ComponentEvent;
  import java.awt.event.ComponentListener;
  import java.beans.beancontext.BeanContextServices;
  import java.io.BufferedReader;
  import java.io.ByteArrayInputStream;
  import java.io.ByteArrayOutputStream;
  import java.io.File;
  import java.io.FileInputStream;
  import java.io.FileReader;
  import java.io.InputStream;
  import java.io.StringReader;
  import java.net.URL;
  import java.util.HashMap;
  import java.util.Iterator;
  import java.util.List;
  import java.util.Map;
  import java.util.Timer;
  import java.util.TimerTask;
  import java.util.TooManyListenersException;
  import java.util.logging.Level;
  import java.util.logging.Logger;
  
  import javax.swing.BorderFactory;
  import javax.swing.JButton;
  import javax.swing.JLabel;
  import javax.swing.JOptionPane;
  import javax.swing.JPanel;
  
  import net.sf.sail.common.beansupport.IIconProvider;
  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.AnnotationService;
  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.apache.commons.io.IOUtils;
  import org.concord.data.state.OTDataStore;
  import org.concord.datagraph.state.OTDataAxis;
  import org.concord.datagraph.state.OTDataCollector;
  import org.concord.datagraph.state.OTDataGraphable;
  import org.concord.framework.otrunk.OTControllerService;
  import org.concord.framework.otrunk.OTObject;
  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.OTMLToXHTMLConverter;
  import org.concord.otrunk.datamodel.OTDatabase;
  import org.concord.otrunk.view.OTViewContainerPanel;
  import org.concord.otrunk.view.OTViewerHelper;
  import org.concord.otrunk.xml.XMLDatabase;
  import org.telscenter.pas.beans.IWorkReporter;
  import org.telscenter.pas.beans.PasStep;
  import org.telscenter.pas.otrunk.skeleton.OTGenericOTrunkStep;
  import org.telscenter.pas.service.NavigateAction;
  import org.xhtmlrenderer.simple.XHTMLPanel;
  
  import com.lowagie.text.pdf.PdfPCell;
  
  /***
   * @author scott
   *
   */
  public class OTrunkStep extends PasStep 
  	implements IIconProvider, SessionEventListener, IWorkReporter
  {
  	/***
  	 * Logger for this class
  	 */
  	private static final Logger logger = Logger.getLogger(OTrunkStep.class
  			.getName());
  	
  	/***
  	 * This is not intended to be serialized by the java object serializer
  	 * It is intended to be serialized by XMLEncoder
  	 */
  	private static final long serialVersionUID = 1L;
  
  	private OTViewerHelper viewerHelper;
  
  	private URL authoredDataURL;
  	private Rim workRim;
  
 	private String typeName = "Otrunk Step";
 	
 	private BeanIcons beanIcons;
 	
 	private ISock workSock;
     
 	private OTObject target;
 
 	private OTControllerService controllerService;
 	
 	private static HashMap<String, Class> supportedTypes = new HashMap<String, Class>();
 
 	private Class stepType;
 
 	private boolean needsSetup = false;
 	
 	private Timer timer;
 	
 	private boolean needsSockEntryInit = true;
 	
 	public OTrunkStep()
 	{
 		viewerHelper = new OTViewerHelper();
 		supportedTypes = new HashMap<String, Class>();
 	}
 	
 	public URL getAuthoredDataURL()
 	{
 		return authoredDataURL;
 	}
 
 	public void setAuthoredDataURL(URL url)
 	{
 		authoredDataURL = url;
 	}
 
 	public BeanIcons getBeanIcons()
 	{
 		return beanIcons;
 	}
 	
 	public void setBeanIcons(BeanIcons icons)
 	{
 		this.beanIcons = icons;
 	}
 	
 	public Component getComponent() {
 		return getComponent(false);
 	}	
 	
 	/* (non-Javadoc)
 	 * @see java.beans.beancontext.BeanContextChildComponentProxy#getComponent()
 	 */
 	public Component getComponent(boolean authoring) {
 		JPanel panel=new JPanel(new BorderLayout(5, 5));
 		OTViewContainerPanel otPanel = null;
 		if (needsSetup) {
 			try {
 			  setupStep();
 			} catch (Exception e) {
 				logger.log(Level.SEVERE, "Failed to create OTObject for step", e);
 			}
 		}
 		try {
 			OTObject root = target;
 
 			if(root == null){
 				OTDatabase otDB = viewerHelper.loadOTDatabase(getAuthoredDataURL());
 				viewerHelper.loadOTrunk(otDB, panel);
 
 				// load the user state
 				// get the last sock entry from the sock, get its string value
 				// turn the string into an input stream
 				Object lastEntry = null;
 				if(workSock != null && workSock.size() > 0){
 					lastEntry = workSock.peek();
 				}
 				if(lastEntry != null){
 					OTDatabase userDB = null;
 					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
 						userDB = viewerHelper.loadOTDatabase(reader, null);
 
 					} else if(lastEntry instanceof byte[]){
 						ByteArrayInputStream inStream = new ByteArrayInputStream((byte[])lastEntry); 
 
 						userDB = viewerHelper.loadOTDatabase(inStream, null);					
 					}
 
 					if(userDB != null){
 						viewerHelper.loadUserData(userDB, null);
 						if(userDB instanceof XMLDatabase ){
 							((XMLDatabase)userDB).setDirty(false);
 						}
 					} else {
 						viewerHelper.newAnonUserData();					
 					}
 				} else {
 					viewerHelper.newAnonUserData();
 				}
 
 
 				root = viewerHelper.getRootObject();
 				otPanel = viewerHelper.createViewContainerPanel();
 			} else {
 				OTViewContext viewContext = 
 					(OTViewContext) controllerService.getService(OTViewContext.class);
 
 				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());
 
 			panel.add(otPanel, BorderLayout.CENTER);
 
 			if(authoring == true) {
 				otPanel.setViewMode("authoring");
 			}
 			
 			otPanel.setCurrentObject(root);
 			
 			JButton save = new JButton("Save");
 			save.addActionListener(new ActionListener(){
 
 				public void actionPerformed(ActionEvent e) {
 					saveDataInternal();
 					JOptionPane.showMessageDialog(null, "Your data has been saved", "Saved", JOptionPane.INFORMATION_MESSAGE);
 				}				
 			});
 			
 			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);
 			
 			//Fix to prevent button from stretching across frame
 			JPanel savePanel = new JPanel();
 			savePanel.add(save);
 			panel.add(savePanel, BorderLayout.SOUTH);
 		} catch (Exception e) {
 			// TODO Auto-generated catch block
 			e.printStackTrace();
 		}
 
 		
 		return panel;
 	}
 
 	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;
 		}
 		
 		OTDatabase userDB = viewerHelper.getUserOtDB();
 
 		// If the database hasn't been modified, don't
 		// save anything
 		if(userDB instanceof XMLDatabase &&
 				!((XMLDatabase)userDB).isDirty()){
 			return;
 		}
 		
 		try {
 			if(workSock != null && userDB != null){
 				byte [] bytes = new byte [1];
 
 				// this is a hack to test this should be moved
 				// to the converter
 				//getWorkRim().setShape(bytes.getClass());
 
 				if(getWorkRim().getShape() == bytes.getClass()){
 					ByteArrayOutputStream out = new ByteArrayOutputStream();
 					viewerHelper.saveOTDatabase(userDB, out);
 					bytes = out.toByteArray();
 					workSock.add(bytes, replace);
 					if(userDB instanceof XMLDatabase){
 						((XMLDatabase)userDB).setDirty(false);
 					}
 				} else {
 					String userData = viewerHelper.saveOTDatabase(userDB);
 					workSock.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();
 		}										
 
 	}
 	
 	public Rim getWorkRim() {
 		return workRim;
 	}
 
 	public void setWorkRim(Rim workRim) {
 		this.workRim = workRim;
 	}
 
 
 	protected void consumeService(BeanContextServices bcs, Class<? extends Object>  serviceClass) {
 		if (serviceClass == AgentService.class) {
 			try {
 				final AgentService p = (AgentService) bcs.getService(this,
 						this, AgentService.class, this, this);
 				IAgent agent = p.getAgentsInRole(Role.RUN_WORKGROUP).getSingle();
 				if(workRim != null && agent != null){
 					workSock = p.getSock(workRim, agent);
 				}
 			} catch (TooManyListenersException e) {
 				e.printStackTrace();
 			} catch (MismatchedAgentSetSizeException e) {
 				e.printStackTrace();
 			} catch (UnsupportedRimShapeException e) {
 				e.printStackTrace();
 			}
 		}
 		if (serviceClass == SessionService.class) {
 			try {
 				SessionService sessionService = (SessionService) bcs.getService(this, this,
 						SessionService.class, this, this);
 				sessionService.addSessionEventListener(this);
 			} catch (TooManyListenersException e1) {
 				// TODO Auto-generated catch block
 				logger.severe("BeanContextServices, Class -  : exception: " + e1); //$NON-NLS-1$
 			}
 		}
 
 	}
 
 	public URL getIcon(int iconKind) {
 		BeanIcons icons = getBeanIcons();
 		if(icons == null) {
 			return null;
 		}
 		return icons.getIcon(iconKind);
 	}
 	
 	public static HashMap<String, Class> getSupportedTypes() {
 			supportedTypes.put("Quick Draw", org.concord.graph.util.state.OTDrawingTool.class);
 			supportedTypes.put("Data Graph", org.concord.datagraph.state.OTDataCollector.class);
 			supportedTypes.put("Concord Multi Tool", org.concord.otrunk.view.document.OTCompoundDoc.class);
 		return supportedTypes;
 	}
 	
 	public void setType(Class type) {
 		this.stepType = type;
 		this.needsSetup  = true;
 		
 		//set the type name for the authoring tool
 		for (Map.Entry<String, Class> entry : supportedTypes.entrySet()) {
 			if( entry.getValue().equals(type )) {
 				this.typeName = entry.getKey();
 			}
 		}
 	}
 	
 	public void setupStep() throws Exception {
 		// set the current step type
 		// get the OTObject which corresponds to this step
 		// set its target to a new object of the type specified
 		OTGenericOTrunkStep otStep = (OTGenericOTrunkStep) controllerService.getOTObject(this);
 		OTObject obj = otStep.getOTObjectService().createObject(stepType);
 		otStep.setTarget(obj);
 		this.setTarget(obj);
 		if (stepType == OTDataCollector.class) {
 			OTDataAxis yAxis = (OTDataAxis) otStep.getOTObjectService().createObject(OTDataAxis.class);
 			OTDataAxis xAxis = (OTDataAxis) otStep.getOTObjectService().createObject(OTDataAxis.class);
 			yAxis.setMin(0);
 			yAxis.setMax(100);
 			yAxis.setLabel("");
 			yAxis.setUnits("");
 			xAxis.setMin(0);
 			xAxis.setMax(100);
 			xAxis.setLabel("unknown");
 			xAxis.setUnits("");
 			((OTDataCollector)obj).setXDataAxis(xAxis);
 			((OTDataCollector)obj).setYDataAxis(yAxis);
 			OTDataGraphable graphable = (OTDataGraphable) otStep.getOTObjectService().createObject(OTDataGraphable.class);
 			graphable.setDataStore((OTDataStore) otStep.getOTObjectService().createObject(OTDataStore.class));
 			((OTDataCollector)obj).setSource(graphable);
 		}
 	}
     
     public String getType() {
         return typeName;
     }
     
 	public void stepExit() {
 		if (timer != null) {
 			timer.cancel();
 		}
 		saveDataInternal();
 	}
     
 	public void sessionInitiated(SessionEvent e) {
 	}
 
 	public void sessionStarted(SessionEvent e) {		
 	}
 
 	public void sessionStopped(SessionEvent e) {
 		stepExit();		
 	}
 
 	public List<JLabel> getCurrentAnswers() {
 		// TODO Auto-generated method stub
 		return null;
 	}
 
 	public List<String> getCurrentPrompts() {
 		// TODO Auto-generated method stub
 		return null;
 	}
 
 	public List<JPanel> getCurrentStepParts() {
 		// TODO Auto-generated method stub
 		return null;
 	}
 
 	public Map<String, String> getEntityToPromptMap() {
 		// TODO Auto-generated method stub
 		return null;
 	}
 
 	public Integer getPossibleScore() {
 		// TODO Auto-generated method stub
 		return null;
 	}
 
 	public JPanel getReportForLearner() {
 		/*
 		  
 		//the stuff Tony and Geoff were working on to implement show all work 
 		viewerHelper.
 		OTJComponentView objView = getOTJComponentView(viewerHelper.getRootObject(), "printing");
 
         OTXHTMLView xhtmlView = null;
         String bodyText = "";
         if(objView instanceof OTXHTMLView) {
             xhtmlView = (OTXHTMLView) objView;
         	bodyText = xhtmlView.getXHTMLText(viewerHelper.getRootObject());
         }
         System.out.println(bodyText);
         
 
 		*/
 		
         OTMLToXHTMLConverter oTMLToXHTMLConverter = null;
 		File outputFile = new File("output.txt");
 		
 		System.out.println("Initial outputFile length: " + outputFile.length());
 		OTObject otObject = null;
 		Thread oTMLToXHTMLConverterThread = null;
 		try {
 			/*
 			FileWriter fos = new FileWriter(outputFile);
 			fos.write("<?xml version=\"1.0\" encoding=\"UTF-8\"?>");
 			fos.close();
 			System.out.println("Intermediate outputFile length: " + outputFile.length());
 			
 			FileWriter fos2 = new FileWriter(outputFile);
 			fos2.write("abc");
 			fos2.close();
 			System.out.println("Intermediate outputFile length: " + outputFile.length());
 			*/
 			otObject = viewerHelper.getRootObject();
 			oTMLToXHTMLConverter = new OTMLToXHTMLConverter(viewerHelper.getViewFactory(), otObject);
 	        oTMLToXHTMLConverter.setXHTMLParams(outputFile, 1000, 1000);
 	        oTMLToXHTMLConverterThread = new Thread (oTMLToXHTMLConverter);
 	        oTMLToXHTMLConverterThread.start();
 	        //oTMLToXHTMLConverterThread.join();
 		} catch(Exception e) {
 			System.out.println("Exception: " + e);
 		}
 		
 		JLabel workLabel = new JLabel("The junk in yr trunk otrunk");
 		JPanel p = new JPanel();
 		p.add(workLabel);
 		
 		
 		final XHTMLPanel panel = new XHTMLPanel();
 
 		//panel.setAntiAliased(true);
 		//panel.setFontScalingFactor(3f);
 		panel.setOpaque(false);
 		panel.setBorder(BorderFactory.createMatteBorder(5, 4, 3, 2, Color.RED));
 		
 		//panel.setPreferredSize(new Dimension(600, 0));
 		panel.setPreferredSize(new Dimension());
 		panel.addComponentListener(new ComponentListener() {
 		
 			public void componentHidden(ComponentEvent e) {
 				// TODO Auto-generated method stub
 
 			}
 
 			public void componentMoved(ComponentEvent e) {
 				// TODO Auto-generated method stub
 
 			}
 
 			public void componentResized(ComponentEvent e) {
 				//panel.relayout();
 
 			}
 
 			public void componentShown(ComponentEvent e) {
 				// TODO Auto-generated method stub
 
 			}
 		});
 		
 		// System.out.println("HTML TO PRINT: " + blockInteraction.getPrompt());
 		//panel.setDocumentFromString("<?xml version="1.0" encoding="UTF-8"?>
 		String xhtmlString = "<?xml version=\"1.0\" encoding=\"UTF-8\"?><!DOCTYPE html PUBLIC \"-//W3C//DTD XHTML 1.0 Transitional//EN\" \"DTD/xhtml1-transitional.dtd\"><html xmlns=\"http://www.w3.org/1999/xhtml\" xml:lang=\"en\" lang=\"en\"><head><title> Transitional DTD XHTML Example </title></head><body bgcolor=\"#FFFFFF\" link=\"#000000\" text=\"red\"><p>This is a transitional XHTML example</p></body></html>";
 			
 		InputStream stream = null;
 		try {
 			//stream = IOUtils.toInputStream(blockInteraction.getPrompt());
 			stream = IOUtils.toInputStream(xhtmlString);
 			//panel.setDocument(stream, null);
 			
 			StringBuffer xhtml = new StringBuffer("<?xml version=\"1.0\" encoding=\"UTF-8\"?><!DOCTYPE html PUBLIC \"-//W3C//DTD XHTML 1.0 Transitional//EN\" \"DTD/xhtml1-transitional.dtd\"><html>");
 			
 			
 	        //StringBuffer fileData = new StringBuffer(1000);
 	        BufferedReader reader = new BufferedReader(
 	                new FileReader("output.txt"));
 	        char[] buf = new char[1024];
 	        int numRead=0;
 	        while((numRead=reader.read(buf)) != -1){
 	            xhtml.append(buf, 0, numRead);
 	        }
 	        reader.close();
 	        xhtml.append("</img></html>");
 	        
 	        int offset = xhtml.indexOf("<img");
 	        //xhtml.insert(offset + 4, " width='1000' height='1000'");
 	        
 	        stream = IOUtils.toInputStream(xhtml.toString());
 	        
 			System.out.println("Ending outputFile length: " + outputFile.length());
 			
 			FileInputStream fis = new FileInputStream(outputFile);
 			
 			FileReader fr = new FileReader(outputFile);
 
 			//DataInputStream dis = new DataInputStream(fis);
 			System.out.println(xhtml);
 			panel.setDocument(stream, null);
 			//panel.setDocument(fis, null);
 		} catch (NullPointerException e) {
 			e.printStackTrace();
 		} catch (Exception e) {
 			e.printStackTrace();
 		} finally {
 			IOUtils.closeQuietly(stream);
 		}
 //		panel.setPreferredSize(new Dimension(PROMPT_PANEL_WIDTH, PROMPT_PANEL_HEIGHT));
 //		panel.setMaximumSize(panel.getPreferredSize());
 		
 		
 		
 		return panel;
 	}
 
 	public JPanel getReportForLearner(NavigateAction navigateAction) {
 		// TODO Auto-generated method stub
 		return null;
 	}
 
 	public PdfPCell getReportForLearnerPDF() {
 		// TODO Auto-generated method stub
 		return null;
 	}
 
 	public JPanel getReportForLearnerWithAnnotations(
 			NavigateAction navigateAction) {
 		// TODO Auto-generated method stub
 		return null;
 	}
 
 	public JPanel getReportForLearnerWithAnnotations() {
 		// TODO Auto-generated method stub
 		return null;
 	}
 
 	public void initStepParts() {
 		// TODO Auto-generated method stub
 		
 	}
 
 	public void setPossibleScore(Integer possibleScore) {
 		// TODO Auto-generated method stub
 		
 	}
 
 	public PdfPCell getReportForLearnerPDF(AnnotationService annotationService) {
 		// TODO Auto-generated method stub
 		return null;
 	}
 
 	public void setTarget(OTObject target) {
 		this.target = target;		
 	}
 
 	public void setControllerService(OTControllerService controllerService) {
 		this.controllerService = controllerService;		
 	}
 
 }