com.inmethod.grid.toolbar.AddRecordsToolbar Maven / Gradle / Ivy

Go to download
package com.inmethod.grid.toolbar;

import java.io.*;

import org.apache.wicket.WicketRuntimeException;
import org.apache.wicket.ajax.AjaxRequestTarget;
import org.apache.wicket.ajax.markup.html.form.AjaxButton;
import org.apache.wicket.markup.html.form.Form;
import org.apache.wicket.model.IModel;
import org.apache.wicket.model.ResourceModel;

import com.inmethod.grid.IDataSource;
import com.inmethod.grid.datagrid.DataGrid;

/**
 * Toolbar that Adds a Button for Adding a new record to an existing DataGrid
 * @param 
 *            datasource model object type = grid type
 * @param 
 *            row/item model object type - must be serializable for "deep copy"
 *
 * @Author Tom Burton
 */
public class AddRecordsToolbar, T extends Serializable, S>
       extends AbstractToolbar
{
  private static final IModel ADD_BUTTON_MODEL =
             new ResourceModel("datagrid.add-new-item", "Add New Item");

  private DataGrid grid; //todo: should this only get from super?

  private final T defaultObject; //final makes sure the default doesn't change

  public AddRecordsToolbar(DataGrid grid, IModel model,
                           final T defaultObject)
  { this(grid, model, defaultObject, ADD_BUTTON_MODEL); }

  public AddRecordsToolbar(DataGrid datagrid, IModel model,
                           final T defaultObject,
                           IModel labelModel)
  {
    super(datagrid, labelModel);
    this.grid = datagrid;
    //ensure OutputMarkupId is set so the button will refresh properly
    datagrid.setOutputMarkupId(true);
    if (null != datagrid.getParent() )
    { datagrid.getParent().setOutputMarkupId(true); }
    this.defaultObject = defaultObject;
    Form form = new Form("addForm");
    add(form);
    AjaxButton addButton = new AjaxButton("add")
                           {
                              @Override
                              protected void onSubmit(AjaxRequestTarget target)
                              {
                                insert();
                                //target.add(findParent(DataGrid.class).getParent());
                              }
                           };
    addButton.setLabel(labelModel);
    form.add(addButton);
  }

  /** inserts data into the table */
  protected void insert()
  {
    long pre = 0;
    long post = 0;
    pre = grid.getTotalRowCount();
    grid.insertRow(getNewData());
    post = grid.getTotalRowCount();
    grid.update();
    //log.error("Pre: " + pre + " Post: " + post);
  }

  /** function to allow easy overrides for returning custom Data when adding
   *  new records to the table
   * @return the object to use for populating new rows of the data-view
   */
  protected T getNewData()
  {
    //if ( defaultObject == null ) { log.error("ERROR: defaultObject is null"); }
    if ( defaultObject == null )
    { //note: should this return null instead?
      throw new WicketRuntimeException("Can't deep copy a null object.");
    }
    Object obj;
    ObjectOutputStream oos;
    ByteArrayOutputStream byteArray;
    try
    {
      oos = new ObjectOutputStream(byteArray = new ByteArrayOutputStream());
      oos.writeObject(defaultObject);
      ObjectInputStream ois =
            new ObjectInputStream(new ByteArrayInputStream(byteArray.toByteArray()));
      obj = ois.readObject();
      return (T)obj;
    }
    catch(IOException ioe) { } //error writing output stream
    catch(ClassNotFoundException cnf ) {} //should never happen
    return defaultObject; //returns on Error or not serializable
  }

}