package org.whatsitcalled.webflange.webapp;
   
   import java.io.IOException;
   import java.io.InputStream;
   import java.util.Set;
   
   import org.apache.wicket.AttributeModifier;
   import org.apache.wicket.Page;
   import org.apache.wicket.ajax.AjaxRequestTarget;
  import org.apache.wicket.ajax.markup.html.AjaxLink;
  import org.apache.wicket.behavior.SimpleAttributeModifier;
  import org.apache.wicket.extensions.ajax.markup.html.form.upload.UploadProgressBar;
  import org.apache.wicket.extensions.ajax.markup.html.modal.ModalWindow;
  import org.apache.wicket.markup.html.WebPage;
  import org.apache.wicket.markup.html.basic.Label;
  import org.apache.wicket.markup.html.form.Form;
  import org.apache.wicket.markup.html.form.RequiredTextField;
  import org.apache.wicket.markup.html.form.upload.FileUpload;
  import org.apache.wicket.markup.html.form.upload.FileUploadField;
  import org.apache.wicket.markup.html.link.Link;
  import org.apache.wicket.markup.html.panel.Panel;
  import org.apache.wicket.markup.repeater.Item;
  import org.apache.wicket.markup.repeater.data.DataView;
  import org.apache.wicket.model.AbstractReadOnlyModel;
  import org.apache.wicket.model.CompoundPropertyModel;
  import org.apache.wicket.model.IModel;
  import org.apache.wicket.util.lang.Bytes;
  import org.whatsitcalled.webflange.ResourceFactory;
  import org.whatsitcalled.webflange.model.LoadTest;
  import org.whatsitcalled.webflange.model.Script;
  import org.whatsitcalled.webflange.service.ScriptServiceException;
  import org.whatsitcalled.webflange.webapp.model.ScriptDataProvider;
  
  public class ScriptPanel extends Panel {
  	AddScriptForm scriptForm;
  
  	public ScriptPanel(String id) {
  		super(id);
  		add(new Link("newScript") {
  			public void onClick() {
  				scriptForm.setVisible(true);
  				setScript(new Script());
  			}
  		});
  		add(new Label("selectHdr", "Action"));
  		add(new Label("scriptHdr", "Script Name"));
  
  		add(new DataView("scripts", new ScriptDataProvider()) {
  			protected void populateItem(final Item item) {
  				final Script script = (Script) item.getModelObject();
  				// Edit Script stuff
  				item.add(new Link("edit") {
  					public void onClick() {
  						scriptForm.setVisible(true);
  						setScript((Script) getParent().getModelObject());
  					}
  				});
  				// View Script Modal Window stuff
  				final ModalWindow viewScript = new ModalWindow("viewScript");
  				viewScript.setTitle(script.getName());
  				viewScript.setCookieName("viewScript");
  				viewScript.setPageMapName("viewScript");
  
  				viewScript.setPageCreator(new ModalWindow.PageCreator() {
  					public Page createPage() {
  						return new ScriptViewPage(script, viewScript);
  					}
  				});
  
  				item.add(viewScript);
  
  				item.add(new AjaxLink("view") {
  					public void onClick(AjaxRequestTarget target) {
  						viewScript.show(target);
  					}
  				});
  
  				// Delete Function
  				Link deleteLink = new Link("delete") {
  					public void onClick() {
  						Script script = (Script) getParent().getModelObject();
  						ResourceFactory.getScriptDAO().removeScript(
  								script.getId());
  					}
  				};
  				//Check for associated LoadTests, and don't allow delete if it is associated
  				Set<LoadTest> tests = script.getLoadTests();
  				String testString = "\\n"; 
  				for (LoadTest test:tests) {
  					testString = testString + test.getName() + "\\n";
  				}
  				
  				if (tests != null && tests.size() > 0) {
  					deleteLink.add(new SimpleAttributeModifier("onclick",
  							"alert('This script cannot be deleted, as it is being used by the following LoadTests: " + testString + "'); return false;"));
  					
  				} else {
  				deleteLink.add(new SimpleAttributeModifier("onclick",
  						"return confirm('Delete script " + script.getName()
 								+ "?');"));
 				}
 				
 				item.add(deleteLink);
 
 				item.add(new Label("name", script.getName()));
 				item.add(new AttributeModifier("class", true,
 						new AbstractReadOnlyModel() {
 							public Object getObject() {
 								return (item.getIndex() % 2 == 1) ? "even"
 										: "odd";
 							}
 						}));
 			}
 		});
 
 		// add the Test Form
 		scriptForm = new AddScriptForm("addScriptForm",
 				new CompoundPropertyModel(getScript()));
 		scriptForm.setVisible(false);
 		add(scriptForm);
 
 	}
 
 	private Script getScript() {
 		Script script = (Script) getModelObject();
 		if (script == null) {
 			script = new Script();
 			setScript(script);
 		}
 		return script;
 	}
 
 	private void setScript(Script script) {
 		setModel(new CompoundPropertyModel(script));
 	}
 
 	public void onModelChanged() {
 		if ((scriptForm != null) && this.getModel() != null) {
 			scriptForm.setModel(this.getModel());
 		}
 	}
 
 	public final class AddScriptForm extends Form {
 		private FileUploadField fileUploadField;
 
 		/**
 		 * Constructor
 		 * 
 		 * @param id
 		 *            The name of this component
 		 */
 		public AddScriptForm(final String id, IModel model) {
 			// Check if we have a Script in the parents modelobject
 			// Construct form with no validation listener
 			super(id, new CompoundPropertyModel(getScript()));
 			add(new RequiredTextField("name"));
 			// set this form to multipart mode (allways needed for uploads!)
 			setMultiPart(true);
 
 			// Add one file input field
 			add(fileUploadField = new FileUploadField("fileInput"));
 			add(new UploadProgressBar("progress", this));
 
 			setMaxSize(Bytes.megabytes(10));
 
 		}
 
 		/**
 		 * Show the resulting valid edit
 		 */
 		public final void onSubmit() {
 			final Script script = (Script) getModelObject();
 			final FileUpload upload = fileUploadField.getFileUpload();
 			InputStream stream = null;
 			if (upload != null) {
 				try {
 					stream = upload.getInputStream();
 				} catch (IOException e) {
 					// TODO Auto-generated catch block
 					e.printStackTrace();
 				}
 			}
 			try {
 				ResourceFactory.getScriptService().saveScript(script, stream);
 			} catch (ScriptServiceException e) {
 				// TODO Auto-generated catch block
 				e.printStackTrace();
 			}
 			this.setModelObject(new CompoundPropertyModel(new Script()));
 			scriptForm.setVisible(false);
 		}
 	}
 
 }