FileImportController.java

/*
 * This program is free software; you can redistribute it and/or modify it under the 
 * terms of the GNU Lesser General Public License, version 2.1 as published by the Free Software 
 * Foundation.
 *
 * You should have received a copy of the GNU Lesser General Public License along with this 
 * program; if not, you can obtain a copy at http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html 
 * or from the Free Software Foundation, Inc., 
 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; 
 * without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
 * See the GNU Lesser General Public License for more details.
 *
 * Copyright 2008 - 2009 Pentaho Corporation.  All rights reserved.
 *
 *
 * Created June, 2010
 * @author Ezequiel Cuellar
 */
package org.pentaho.platform.dataaccess.datasource.wizard.controllers;

import org.pentaho.platform.dataaccess.datasource.wizard.DatasourceMessages;
import org.pentaho.platform.dataaccess.datasource.wizard.models.DatasourceModel;
import org.pentaho.platform.dataaccess.datasource.wizard.models.ModelInfo;
import org.pentaho.ui.xul.binding.BindingConvertor;
import org.pentaho.ui.xul.binding.BindingFactory;
import org.pentaho.ui.xul.components.XulButton;
import org.pentaho.ui.xul.components.XulFileUpload;
import org.pentaho.ui.xul.components.XulLabel;
import org.pentaho.ui.xul.containers.XulDialog;
import org.pentaho.ui.xul.gwt.GwtXulDomContainer;
import org.pentaho.ui.xul.gwt.GwtXulRunner;
import org.pentaho.ui.xul.gwt.binding.GwtBindingFactory;
import org.pentaho.ui.xul.gwt.util.AsyncXulLoader;
import org.pentaho.ui.xul.gwt.util.IXulLoaderCallback;
import org.pentaho.ui.xul.impl.AbstractXulEventHandler;
import org.pentaho.ui.xul.stereotype.Bindable;

import com.google.gwt.core.client.GWT;

//TODO: move to the CSV datasource package
public class FileImportController extends AbstractXulEventHandler  {

  private XulDialog datasourceDialog;
  
  private XulDialog waitingDialog;

  private XulFileUpload fileUpload;

  private XulDialog errorDialog;

  private XulLabel errorLabel;

  private XulLabel waitingLabel;
  
  private XulButton okButton;
  
  private BindingFactory bf;

  private DatasourceMessages messages;

  private DatasourceModel datasourceModel;

  public FileImportController(DatasourceModel datasourceModel, DatasourceMessages messages) {
    this.datasourceModel = datasourceModel;
    setDatasourceMessages(messages);

  }

  @Bindable
  public void init() {
    bf = new GwtBindingFactory(document);
    
    fileUpload = (XulFileUpload) document.getElementById("fileUpload"); //$NON-NLS-1$
    datasourceDialog = (XulDialog) document.getElementById("fileImportEditorWindow");//$NON-NLS-1$
    waitingDialog = (XulDialog) document.getElementById("waitingDialog"); //$NON-NLS-1$
    waitingLabel = (XulLabel) document.getElementById("waitingDialogLabel"); //$NON-NLS-1$
    errorLabel = (XulLabel) document.getElementById("errorLabel"); //$NON-NLS-1$
    errorDialog = (XulDialog) document.getElementById("errorDialog"); //$NON-NLS-1$
    
    okButton = (XulButton) document.getElementById("okButton"); //$NON-NLS-1$

    BindingConvertor<String, Boolean> isDisabledConvertor = new BindingConvertor<String, Boolean>() {

      public Boolean sourceToTarget(String aValue) {
       return (aValue == null || "".equals(aValue));
      }

      public String targetToSource(Boolean aValue) {
        return null;
      }
    };
    
    bf.createBinding("fileUpload", "selectedFile", "okButton", "disabled", isDisabledConvertor);//$NON-NLS-1$ //$NON-NLS-2$ //$NON-NLS-3$ //$NON-NLS-4$

  }

  public String getName() {
    return "fileImportController"; //$NON-NLS-1$
  }

  @Bindable
  public void submitCsv() {
    String fileName = datasourceModel.getModelInfo().getFileInfo().getTmpFilename();
    if(fileName != null && fileName.endsWith(".tmp")) {  //$NON-NLS-1$
      fileName = fileName.substring(0, fileName.lastIndexOf(".tmp")); //$NON-NLS-1$
    }
    
    fileUpload.addParameter("file_name", fileName == null ? "" : fileName); //$NON-NLS-1$ //$NON-NLS-2$
    fileUpload.addParameter("mark_temporary", "true"); //$NON-NLS-1$ //$NON-NLS-2$
    // specify that we want any compressed files to be unpacked
    fileUpload.addParameter("unzip", "true"); //$NON-NLS-1$ //$NON-NLS-2$
    showWaitingDialog();
    fileUpload.submit();
  }
  
  public void closeWaitingDialog() {
    waitingDialog.hide();
  }
  public void showWaitingDialog() {
    waitingLabel.setValue(messages.getString("fileImportDialog.UPLOADING", fileUpload.getSeletedFile())); //$NON-NLS-1$
    waitingDialog.show();
  }

  @Bindable
  public void showFileImportDialog() {
    fileUpload.setSelectedFile("");
    datasourceDialog.show();
  }

  @Bindable
  public void close() {
    datasourceDialog.hide();
  }

  @Bindable
  public void uploadSuccess(String uploadedFile) {
    closeWaitingDialog();
    close();
    String selectedFile = this.fileUpload.getSeletedFile();
    
    if( uploadedFile.indexOf("\n") != -1 ) {
      // uploadedFile is newline-separated list of file names
      // for now we only support a single file
      showErroDialog(messages.getString("fileImportDialog.COMPRESSED_TOO_MANY_FILES"));
			return;
    }
    if(selectedFile.endsWith(".zip") || selectedFile.endsWith(".tgz") || selectedFile.endsWith(".tar")) {
      // check to see what kind of file was extracted from the compressed upload
      if(!uploadedFile.endsWith(".csv.tmp") && !uploadedFile.endsWith(".txt.tmp") ) {
        showErroDialog(messages.getString("fileImportDialog.COMPRESSED_NO_CSV"));
        return;
      }
    }

  if(selectedFile.endsWith(".csv") || selectedFile.endsWith(".txt")
        || selectedFile.endsWith(".zip") || selectedFile.endsWith(".tgz") || selectedFile.endsWith(".tar")) {
		datasourceModel.getModelInfo().getFileInfo().setTmpFilename(uploadedFile);
		datasourceModel.getModelInfo().getFileInfo().setFriendlyFilename(selectedFile);
	} else {
		showErroDialog(messages.getString("fileImportDialog.INVALID_FILE"));
	}
  }

  @Bindable
  public void uploadFailure(Throwable t) {
    closeWaitingDialog();
    close();
    datasourceModel.getModelInfo().getFileInfo().setTmpFilename("");
    datasourceModel.getModelInfo().getFileInfo().setFriendlyFilename("");
    showErroDialog(t.getMessage());
  }
  
  public void showErroDialog(String error) {
    errorLabel.setValue(error);
    errorDialog.show();
  }

  @Bindable
  public void closeErrorDialog() {
    errorDialog.hide();
  }

  
  public void setDatasourceMessages( DatasourceMessages datasourceMessages ) {
    this.messages = datasourceMessages;
  }


}