/*
 * Copyright (C) 2007-2023 Xagasoft, All rights reserved.
 *
 * This file is part of the libbu++ library and is released under the
 * terms of the license contained in the file LICENSE.
 */

#include "bu/csvwriter.h"
#include "bu/stream.h"

Bu::CsvWriter::CsvWriter( Bu::Stream &sOut, Bu::CsvWriter::Style eStyle ) :
    sOut( sOut )
{
    switch( eStyle )
    {
        case styleExcel:
            sEncode = Bu::slot( &encodeExcel );
            break;

        case styleC:
            sEncode = Bu::slot( &encodeExcel );
            break;
    }
}

Bu::CsvWriter::CsvWriter( Bu::Stream &sOut,
        Bu::CsvWriter::EncodeSignal sEncode ) :
    sOut( sOut ),
    sEncode( sEncode )
{
}

Bu::CsvWriter::~CsvWriter()
{
}

void Bu::CsvWriter::writeLine( const StrArray &aStrs )
{
    Bu::String sBuf;
    for( StrArray::const_iterator i = aStrs.begin(); i; i++ )
    {
        if( i != aStrs.begin() )
            sBuf += ",";
        sBuf += sEncode( *i );
    }
    sBuf += "\n";

    sOut.write( sBuf );
}

Bu::String Bu::CsvWriter::encodeExcel( const Bu::String &sIn )
{
    if( sIn.find('\"') || sIn.find(',') )
	{
		Bu::String sOut = "\"";
        for( Bu::String::const_iterator i = sIn.begin(); i; i++ )
        {
            if( *i == '\"' )
				sOut += "\"\"";
			else
				sOut += *i;
		}
		sOut += '\"';
        return sOut;
    }
    return sIn;
}

Bu::String Bu::CsvWriter::encodeC( const Bu::String &sIn )
{
    Bu::String sOut = "";
    for( Bu::String::const_iterator i = sIn.begin(); i; i++ )
    {
        if( *i == ',' )
            sOut += "\\,";
        else
            sOut += *i;
    }
    return sOut;
}