/**********************************************************************
 * Copyright (c) by Heiner Jostkleigrewe
 * 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 3 of the 
 * License, or (at your option) 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, 
 * see <http://www.gnu.org/licenses/>.
 * 
 * heiner@jverein.de
 * www.jverein.de
 **********************************************************************/

package de.jost_net.JVerein.io;

import java.io.FileWriter;
import java.util.ArrayList;
import java.util.Map;

import org.supercsv.cellprocessor.ift.CellProcessor;
import org.supercsv.io.CsvMapWriter;
import org.supercsv.io.ICsvMapWriter;
import org.supercsv.prefs.CsvPreference;

import de.jost_net.JVerein.Einstellungen;
import de.jost_net.JVerein.Variable.MitgliedMap;
import de.jost_net.JVerein.Variable.SollbuchungMap;
import de.jost_net.JVerein.io.Adressbuch.Adressaufbereitung;
import de.jost_net.JVerein.keys.VorlageTyp;
import de.jost_net.JVerein.rmi.Mitglied;
import de.jost_net.JVerein.rmi.Sollbuchung;
import de.jost_net.JVerein.util.VorlageUtil;
import de.willuhn.logging.Logger;
import de.willuhn.util.ProgressMonitor;

public class SollbuchungExportCSV extends SollbuchungExport
{

  private ArrayList<Sollbuchung> sollbuchungen = null;

  @Override
  public String getName()
  {
    return "Sollbuchungen CSV-Export";
  }

  @Override
  public IOFormat[] getIOFormats(Class<?> objectType)
  {
    if (objectType != Sollbuchung.class)
    {
      return null;
    }
    IOFormat f = new IOFormat()
    {

      @Override
      public String getName()
      {
        return SollbuchungExportCSV.this.getName();
      }

      /**
       * @see de.willuhn.jameica.hbci.io.IOFormat#getFileExtensions()
       */
      @Override
      public String[] getFileExtensions()
      {
        return new String[] { "*.csv" };
      }
    };
    return new IOFormat[] { f };
  }

  @Override
  protected void open()
  {
    sollbuchungen = new ArrayList<>();
  }

  @Override
  protected void startMitglied(Mitglied m)
  {
    //
  }

  @Override
  protected void endeMitglied()
  {
    //
  }

  @Override
  protected void add(Sollbuchung sollb)
  {
    sollbuchungen.add(sollb);
  }

  @Override
  protected void close(ProgressMonitor monitor)
  {
    try
    {
      ICsvMapWriter writer = new CsvMapWriter(new FileWriter(file),
          CsvPreference.EXCEL_NORTH_EUROPE_PREFERENCE);

      Mitglied m = null;
      Sollbuchung sollb = null;
      if (sollbuchungen.size() > 0)
      {
        sollb = sollbuchungen.get(0);
        m = sollb.getMitglied();
      }
      else
      {
        sollb = (Sollbuchung) Einstellungen.getDBService()
            .createObject(Sollbuchung.class, null);
        m = (Mitglied) Einstellungen.getDBService().createObject(Mitglied.class,
            null);
      }

      Map<String, Object> map = new MitgliedMap().getMap(m, null);
      map = new SollbuchungMap().getMap(sollb, map);
      String[] header = createHeader(map);
      Logger.debug("Header");
      for (String s : header)
      {
        Logger.debug(s);
      }
      CellProcessor[] processors = CellProcessors.createCellProcessors(map);

      writer.writeHeader(header);

      for (Sollbuchung sollbuch : sollbuchungen)
      {
        Map<String, Object> mp = new MitgliedMap()
            .getMap(sollbuch.getMitglied(), null);
        map = new SollbuchungMap().getMap(sollbuch, mp);
        writer.write(map, header, processors);
        monitor.log("Export: "
            + Adressaufbereitung.getNameVorname(sollbuch.getMitglied()));
      }
      writer.close();
    }
    catch (Exception e)
    {
      Logger.error("Fehler", e);
    }
  }

  private String[] createHeader(Map<String, Object> map)
  {
    return map.keySet().toArray(new String[0]);
  }

  @Override
  public String getDateiname(Object object)
  {
    return VorlageUtil.getName(VorlageTyp.SOLLBUCHUNGEN_DATEINAME, object)
        + ".csv";
  }
}
