package net.sourceforge.squirrel_sql.plugins.dataimport.importer.csv;
/*
* Copyright (C) 2007 Thorsten Mürell
*
* This program is free software; you can redistribute it and/or
* modify it under the terms of the GNU General Public License
* as published by the Free Software Foundation; either version 2
* of the License, or any later version.
*
* 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 General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
*/
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStreamReader;
import java.text.DateFormat;
import java.text.ParseException;
import java.text.SimpleDateFormat;
import java.util.Date;
import javax.swing.JComponent;
import javax.swing.JOptionPane;
import net.sourceforge.squirrel_sql.fw.util.StringManager;
import net.sourceforge.squirrel_sql.fw.util.StringManagerFactory;
import net.sourceforge.squirrel_sql.plugins.dataimport.importer.IFileImporter;
import net.sourceforge.squirrel_sql.plugins.dataimport.importer.UnsupportedFormatException;
import com.csvreader.CsvReader;
/**
* This class implements the IFileImporter interface for reading CSV files.
*
* @author Thorsten Mürell
*/
public class CSVFileImporter implements IFileImporter {
private static final StringManager stringMgr =
StringManagerFactory.getStringManager(CSVFileImporter.class);
private CSVSettingsBean settings = null;
private File importFile = null;
private CsvReader reader = null;
/**
* The standard constructor
*
* @param importFile The import file
*/
public CSVFileImporter(File importFile) {
this.importFile = importFile;
this.settings = new CSVSettingsBean();
}
/*
* (non-Javadoc)
* @see net.sourceforge.squirrel_sql.plugins.dataimport.importer.IFileImporter#open()
*/
public boolean open() throws IOException {
reset();
return true;
}
/*
* (non-Javadoc)
* @see net.sourceforge.squirrel_sql.plugins.dataimport.importer.IFileImporter#close()
*/
public boolean close() throws IOException {
if (reader != null) {
reader.close();
}
return true;
}
/* (non-Javadoc)
* @see net.sourceforge.squirrel_sql.plugins.dataimport.importer.IFileImporter#getPreview(int)
*/
public String[][] getPreview(int noOfLines) throws IOException {
CsvReader csvReader = new CsvReader(new InputStreamReader(new FileInputStream(importFile), settings.getImportCharset()), settings.getSeperator());
String[][] data = new String[noOfLines][];
int row = 0;
int columns = -1;
while (csvReader.readRecord() && row < noOfLines) {
if (columns == -1) {
columns = csvReader.getColumnCount();
}
data[row] = new String[columns];
for (int i = 0; i < columns; i++) {
data[row][i] = csvReader.get(i);
}
row++;
}
csvReader.close();
String[][] outData = new String[row][];
for (int i = 0; i < row; i++) {
outData[i] = data[i];
}
return outData;
}
/*
* (non-Javadoc)
* @see net.sourceforge.squirrel_sql.plugins.dataimport.importer.IFileImporter#getRows()
*/
public int getRows() {
return -1;
}
/*
* (non-Javadoc)
* @see net.sourceforge.squirrel_sql.plugins.dataimport.importer.IFileImporter#next()
*/
public boolean next() throws IOException {
return reader.readRecord();
}
/*
* (non-Javadoc)
* @see net.sourceforge.squirrel_sql.plugins.dataimport.importer.IFileImporter#reset()
*/
public boolean reset() throws IOException {
if (reader != null) {
reader.close();
}
reader = new CsvReader(new InputStreamReader(new FileInputStream(importFile), settings.getImportCharset()), settings.getSeperator());
return true;
}
/*
* (non-Javadoc)
* @see net.sourceforge.squirrel_sql.plugins.dataimport.importer.IFileImporter#getString(int)
*/
public String getString(int column) throws IOException {
return reader.get(column);
}
/*
* (non-Javadoc)
* @see net.sourceforge.squirrel_sql.plugins.dataimport.importer.IFileImporter#getLong(int)
*/
public Long getLong(int column) throws IOException, UnsupportedFormatException {
try {
String longS = reader.get(column);
if (null == longS || 0 == longS.trim().length()){
return null;
}
return Long.parseLong(longS);
} catch (NumberFormatException nfe) {
throw new UnsupportedFormatException();
}
}
/*
* (non-Javadoc)
* @see net.sourceforge.squirrel_sql.plugins.dataimport.importer.IFileImporter#getInt(int)
*/
public Integer getInt(int column) throws IOException, UnsupportedFormatException {
try {
String intS = reader.get(column);
if (null == intS || 0 == intS.trim().length()){
return null;
}
return Integer.parseInt(intS);
} catch (NumberFormatException nfe) {
throw new UnsupportedFormatException(nfe);
}
}
/*
* (non-Javadoc)
* @see net.sourceforge.squirrel_sql.plugins.dataimport.importer.IFileImporter#getDate(int)
*/
public Date getDate(int column) throws IOException, UnsupportedFormatException {
Date d = null;
try {
DateFormat f = new SimpleDateFormat(settings.getDateFormat());
String dateString = reader.get(column);
// we allow the return of null values if the the reader returns
// an empty String or a null String
if (null != dateString && dateString.trim().length() > 0) {
d = f.parse(dateString);
}
} catch (IllegalArgumentException e) {
//i18n[CSVFileImporter.invalidDateFormat=Invalid date format given]
JOptionPane.showMessageDialog(null, stringMgr.getString("CSVFileImporter.invalidDateFormat"));
throw new UnsupportedFormatException();
} catch (ParseException pe) {
throw new UnsupportedFormatException();
}
return d;
}
/*
* (non-Javadoc)
* @see net.sourceforge.squirrel_sql.plugins.dataimport.importer.IFileImporter#getConfigurationPanel()
*/
public JComponent getConfigurationPanel() {
return new CSVSettingsPanel(settings);
}
}