/*
 * 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 (c) 2006 - 2013 Pentaho Corporation..  All rights reserved.
 */

package org.pentaho.reporting.engine.classic.core.modules.output.fast.csv;

import org.pentaho.reporting.engine.classic.core.InvalidReportStateException;
import org.pentaho.reporting.engine.classic.core.filter.MessageFormatSupport;
import org.pentaho.reporting.engine.classic.core.layout.model.LogicalPageBox;
import org.pentaho.reporting.engine.classic.core.layout.model.RenderBox;
import org.pentaho.reporting.engine.classic.core.layout.output.OutputProcessorMetaData;
import org.pentaho.reporting.engine.classic.core.modules.output.fast.template.FastExportTemplateProducer;
import org.pentaho.reporting.engine.classic.core.modules.output.fast.template.FormattedDataBuilder;
import org.pentaho.reporting.engine.classic.core.modules.output.fast.template.TemplatingOutputProcessor;
import org.pentaho.reporting.engine.classic.core.modules.output.table.base.SheetLayout;
import org.pentaho.reporting.engine.classic.core.modules.output.table.base.TableContentProducer;
import org.pentaho.reporting.engine.classic.core.modules.output.table.base.TableRectangle;
import org.pentaho.reporting.engine.classic.core.modules.output.table.csv.CSVTableModule;
import org.pentaho.reporting.engine.classic.core.util.InstanceID;
import org.pentaho.reporting.libraries.base.util.CSVQuoter;
import org.pentaho.reporting.libraries.fonts.encoding.EncodingRegistry;

import java.io.PrintWriter;
import java.io.StringWriter;
import java.util.HashMap;
import java.util.Locale;

public class CsvTemplateProducer implements FastExportTemplateProducer {
  private OutputProcessorMetaData metaData;
  private SheetLayout sheetLayout;
  private String encoding;
  private HashMap<InstanceID, String> idMapping;
  private String template;
  private CSVQuoter quoter;

  public CsvTemplateProducer( final OutputProcessorMetaData metaData, final SheetLayout sheetLayout,
      final String encoding ) {
    this.metaData = metaData;
    this.sheetLayout = sheetLayout;
    this.encoding = encoding;
    this.idMapping = new HashMap<InstanceID, String>();

    final String separator =
        metaData.getConfiguration().getConfigProperty( CSVTableModule.SEPARATOR, CSVTableModule.SEPARATOR_DEFAULT );
    if ( separator.length() == 0 ) {
      throw new IllegalArgumentException( "CSV separate cannot be an empty string." );
    }

    if ( this.encoding == null ) {
      this.encoding =
          metaData.getConfiguration().getConfigProperty(
              "org.pentaho.reporting.engine.classic.core.modules.output.table.csv.Encoding",
              EncodingRegistry.getPlatformDefaultEncoding() );
    }

    quoter = new CSVQuoter( separator.charAt( 0 ) );
  }

  public void produceTemplate( final LogicalPageBox pageBox ) {
    TableContentProducer contentProducer =
        TemplatingOutputProcessor.produceTableLayout( pageBox, sheetLayout, metaData );

    final SheetLayout sheetLayout = contentProducer.getSheetLayout();
    final int columnCount = contentProducer.getColumnCount();
    final int lastColumn = columnCount - 1;

    final int startRow = contentProducer.getFinishedRows();
    final int finishRow = contentProducer.getFilledRows();
    StringWriter swriter = new StringWriter();
    PrintWriter writer = new PrintWriter( swriter );

    for ( int row = startRow; row < finishRow; row++ ) {
      for ( short col = 0; col < columnCount; col++ ) {
        final RenderBox content = contentProducer.getContent( row, col );
        if ( content == null ) {
          writer.print( quoter.getSeparator() );
          continue;
        }

        if ( content.isCommited() == false ) {
          throw new InvalidReportStateException( "Uncommited content encountered" );
        }

        final long contentOffset = contentProducer.getContentOffset( row, col );
        final TableRectangle rectangle =
            sheetLayout.getTableBounds( content.getX(), content.getY() + contentOffset, content.getWidth(), content
                .getHeight(), null );
        if ( rectangle.isOrigin( col, row ) == false ) {
          // A spanned cell ..
          continue;
        }

        InstanceID instanceId = content.getNodeLayoutProperties().getInstanceId();
        String uuid = idMapping.get( instanceId );
        if ( uuid == null ) {
          uuid = String.valueOf( idMapping.size() );
          idMapping.put( instanceId, uuid );
        }

        final String formattedtext = "$(" + uuid + ")";
        writer.write( formattedtext );
        if ( col < lastColumn ) {
          writer.print( quoter.getSeparator() );
        }
        content.setFinishedTable( true );
      }

      writer.println();
    }

    writer.close();
    template = swriter.toString();
  }

  public CSVQuoter getQuoter() {
    return quoter;
  }

  public String getTemplate() {
    return template;
  }

  public FormattedDataBuilder createDataBuilder() {
    MessageFormatSupport messageFormatSupport = new MessageFormatSupport();
    messageFormatSupport.setLocale( Locale.ENGLISH );
    messageFormatSupport.setFormatString( getTemplate() );
    messageFormatSupport.setNullString( "" );
    return new CsvFormattedDataBuilder( idMapping, messageFormatSupport, getQuoter(), encoding );
  }
}