//
// VMime library (http://www.vmime.org)
// Copyright (C) 2002 Vincent Richard <vincent@vmime.org>
//
// 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, write to the Free Software Foundation, Inc.,
// 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
//
// Linking this library statically or dynamically with other modules is making
// a combined work based on this library.  Thus, the terms and conditions of
// the GNU General Public License cover the whole combination.
//

#include "vmime/config.hpp"


#if VMIME_HAVE_FILESYSTEM_FEATURES


#include <fstream>
#include <sstream>

#include "vmime/fileAttachment.hpp"
#include "vmime/exception.hpp"

#include "vmime/streamContentHandler.hpp"
#include "vmime/utility/inputStreamPointerAdapter.hpp"

#include "vmime/contentDispositionField.hpp"

#include "vmime/platform.hpp"
#include "vmime/utility/file.hpp"


namespace vmime {


fileAttachment::fileAttachment(
	const string& filepath,
	const mediaType& type
) {

	m_type = type;

	setData(filepath);

	m_encoding = encoding::decide(m_data);
}


fileAttachment::fileAttachment(
	const string& filepath,
	const mediaType& type,
	const text& desc
) {

	m_type = type;
	m_desc = desc;

	setData(filepath);

	m_encoding = encoding::decide(m_data);
}


fileAttachment::fileAttachment(
	const string& filepath,
	const mediaType& type,
	const text& desc,
	const encoding& enc
) {

	m_type = type;
	m_desc = desc;

	setData(filepath);

	m_encoding = enc;
}


fileAttachment::fileAttachment(
	const shared_ptr <contentHandler>& cts,
	const word& filename,
	const mediaType& type) {

	if (!filename.isEmpty()) {
		m_fileInfo.setFilename(filename);
	}

	m_type = type;

	setData(cts);

	m_encoding = encoding::decide(m_data);
}


fileAttachment::fileAttachment(
	const shared_ptr <contentHandler>& cts,
	const word& filename,
	const mediaType& type,
	const text& desc
) {

	if (!filename.isEmpty()) {
		m_fileInfo.setFilename(filename);
	}

	m_type = type;
	m_desc = desc;

	setData(cts);

	m_encoding = encoding::decide(m_data);
}


fileAttachment::fileAttachment(
	const shared_ptr <contentHandler>& cts,
	const word& filename,
	const mediaType& type,
	const text& desc,
	const encoding& enc
) {

	if (!filename.isEmpty()) {
		m_fileInfo.setFilename(filename);
	}

	m_type = type;
	m_desc = desc;
	m_encoding = enc;

	setData(cts);
}


void fileAttachment::setData(const string& filepath) {

	shared_ptr <utility::fileSystemFactory> fsf = platform::getHandler()->getFileSystemFactory();
	utility::file::path path = fsf->stringToPath(filepath);

	shared_ptr <utility::file> file = fsf->create(path);

	if (!file->isFile()) {
		throw exceptions::open_file_error();
	}

	m_data = make_shared <streamContentHandler>(
		file->getFileReader()->getInputStream(), file->getLength()
	);

	m_fileInfo.setFilename(path.getLastComponent());
	m_fileInfo.setSize(file->getLength());
}


void fileAttachment::setData(const shared_ptr <contentHandler>& cts) {

	m_data = cts;

	m_fileInfo.setSize(cts->getLength());
}


void fileAttachment::generatePart(const shared_ptr <bodyPart>& part) const {

	defaultAttachment::generatePart(part);

	shared_ptr <contentDispositionField> cdf =
		dynamicCast <contentDispositionField>(part->getHeader()->ContentDisposition());

	if (m_fileInfo.hasSize()) cdf->setSize(utility::stringUtils::toString(m_fileInfo.getSize()));
	if (m_fileInfo.hasFilename() && !m_fileInfo.getFilename().isEmpty()) cdf->setFilename(m_fileInfo.getFilename());
	if (m_fileInfo.hasCreationDate()) cdf->setCreationDate(m_fileInfo.getCreationDate());
	if (m_fileInfo.hasModificationDate()) cdf->setModificationDate(m_fileInfo.getModificationDate());
	if (m_fileInfo.hasReadDate()) cdf->setReadDate(m_fileInfo.getReadDate());
}


const fileAttachment::fileInfo& fileAttachment::getFileInfo() const {

	return m_fileInfo;
}


fileAttachment::fileInfo& fileAttachment::getFileInfo() {

	return m_fileInfo;
}



//
// fileAttachment::fileInfo
//

fileAttachment::fileInfo::fileInfo()
	: m_filename(NULL),
	  m_size(NULL),
	  m_creationDate(NULL),
	  m_modifDate(NULL),
	  m_readDate(NULL) {

}


fileAttachment::fileInfo::~fileInfo()
{
	delete m_filename;
	delete m_size;
	delete m_creationDate;
	delete m_modifDate;
	delete m_readDate;
}

bool fileAttachment::fileInfo::hasFilename() const { return m_filename; }
const word& fileAttachment::fileInfo::getFilename() const { return *m_filename; }
void fileAttachment::fileInfo::setFilename(const string& name) { if (m_filename) { *m_filename = name; } else { m_filename = new word(name); } }
void fileAttachment::fileInfo::setFilename(const word& name) { if (m_filename) { *m_filename = name; } else { m_filename = new word(name); } }

bool fileAttachment::fileInfo::hasCreationDate() const { return m_creationDate; }
const datetime& fileAttachment::fileInfo::getCreationDate() const { return *m_creationDate; }
void fileAttachment::fileInfo::setCreationDate(const datetime& date) { if (m_creationDate) { *m_creationDate = date; } else { m_creationDate = new datetime(date); } }

bool fileAttachment::fileInfo::hasModificationDate() const { return m_modifDate; }
const datetime& fileAttachment::fileInfo::getModificationDate() const { return *m_modifDate; }
void fileAttachment::fileInfo::setModificationDate(const datetime& date) { if (m_modifDate) { *m_modifDate = date; } else { m_modifDate = new datetime(date); } }

bool fileAttachment::fileInfo::hasReadDate() const { return m_readDate; }
const datetime& fileAttachment::fileInfo::getReadDate() const { return *m_readDate; }
void fileAttachment::fileInfo::setReadDate(const datetime& date) { if (m_readDate) { *m_readDate = date; } else { m_readDate = new datetime(date); } }

bool fileAttachment::fileInfo::hasSize() const { return m_size; }
size_t fileAttachment::fileInfo::getSize() const { return *m_size; }
void fileAttachment::fileInfo::setSize(const size_t size) { if (m_size) { *m_size = size; } else { m_size = new size_t(size); } }


} // vmime


#endif // VMIME_HAVE_FILESYSTEM_FEATURES