CsvDatasourceFactory.java

/*
 * Copyright (c) 2017 OBiBa. All rights reserved.
 *
 * This program and the accompanying materials
 * are made available under the terms of the GNU Public License v3.0.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

package org.obiba.magma.datasource.csv.support;

import java.io.File;
import java.util.ArrayList;
import java.util.List;

import javax.validation.constraints.NotNull;

import org.obiba.magma.AbstractDatasourceFactory;
import org.obiba.magma.Datasource;
import org.obiba.magma.ValueTable;
import org.obiba.magma.datasource.csv.CsvDatasource;

@SuppressWarnings("UnusedDeclaration")
public class CsvDatasourceFactory extends AbstractDatasourceFactory {

  private File bundle;

  private List<TableBundle> tables;

  private String characterSet;

  private String separator;

  private String quote;

  private int firstRow = 1;

  private boolean multilines = false;

  public void setBundle(File bundle) {
    this.bundle = bundle;
  }

  public void setTables(List<TableBundle> tables) {
    this.tables = tables;
  }

  public void setCharacterSet(String characterSet) {
    this.characterSet = characterSet;
  }

  public void setSeparator(String separator) {
    this.separator = separator;
  }

  public void setQuote(String quote) {
    this.quote = quote;
  }

  public void setFirstRow(int firstRow) {
    this.firstRow = firstRow;
  }

  public void setMultilines(boolean multilines) {
    this.multilines = multilines;
  }

  public CsvDatasourceFactory addTable(File tableDirectory) {
    if(tableDirectory != null && !hasTable(tableDirectory.getName())) {
      getTables().add(new TableBundle(tableDirectory));
    }
    return this;
  }

  public CsvDatasourceFactory addTable(String name, File data, String entityType) {
    if(name != null && !hasTable(name)) {
      getTables().add(new TableBundle(name, data, entityType));
    }
    return this;
  }

  public CsvDatasourceFactory addTable(String name, File variables, File data) {
    if(name != null && !hasTable(name)) {
      getTables().add(new TableBundle(name, variables, data));
    }
    return this;
  }

  public CsvDatasourceFactory addTable(ValueTable refTable, File data) {
    if(refTable.getName() != null && !hasTable(refTable.getName())) {
      getTables().add(new TableBundle(refTable, data));
    }
    return this;
  }

  public boolean hasTable(String name) {
    for(TableBundle table : getTables()) {
      if(table.getName().equals(name)) {
        return true;
      }
    }
    return false;
  }

  public List<TableBundle> getTables() {
    return tables == null ? tables = new ArrayList<>() : tables;
  }

  @NotNull
  @Override
  protected Datasource internalCreate() {
    CsvDatasource datasource = bundle != null && bundle.isDirectory()
        ? new CsvDatasource(getName(), bundle)
        : new CsvDatasource(getName());
    if(characterSet != null) {
      datasource.setCharacterSet(characterSet);
    }
    if(quote != null) {
      datasource.setQuote(Quote.fromString(quote));
    }
    if(separator != null) {
      datasource.setSeparator(Separator.fromString(separator));
    }
    datasource.setFirstRow(firstRow);
    datasource.setMultilines(multilines);

    for(TableBundle tableBundle : getTables()) {
      if(tableBundle.hasRefTable()) {
        datasource.addValueTable(tableBundle.getRefTable(), tableBundle.getData());
      } else if(tableBundle.hasVariables()) {
        datasource.addValueTable(tableBundle.getName(), tableBundle.getVariables(), tableBundle.getData());
      } else {
        datasource.addValueTable(tableBundle.getName(), tableBundle.getData(), tableBundle.getEntityType());
      }
    }

    return datasource;
  }
}