/*
    Copyright (C) 2020-2021 Carl Hetherington <cth@carlh.net>

    This file is part of DCP-o-matic.

    DCP-o-matic 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 2 of the License, or
    (at your option) any later version.

    DCP-o-matic 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 DCP-o-matic.  If not, see <http://www.gnu.org/licenses/>.

*/


#include "zipper.h"
#include "exceptions.h"
#include "dcpomatic_assert.h"
#include <zip.h>
#include <boost/filesystem.hpp>
#include <stdexcept>


using std::string;
using std::runtime_error;
using std::shared_ptr;


Zipper::Zipper (boost::filesystem::path file)
{
	int error;
	_zip = zip_open (file.string().c_str(), ZIP_CREATE | ZIP_EXCL, &error);
	if (!_zip) {
		if (error == ZIP_ER_EXISTS) {
			throw FileError ("ZIP file already exists", file);
		}
		throw FileError ("could not create ZIP file", file);
	}
}


void
Zipper::add (string name, string content)
{
	shared_ptr<string> copy(new string(content));
	_store.push_back (copy);

	auto source = zip_source_buffer (_zip, copy->c_str(), copy->length(), 0);
	if (!source) {
		throw runtime_error ("could not create ZIP source");
	}

	if (zip_add(_zip, name.c_str(), source) == -1) {
		throw runtime_error ("failed to add data to ZIP archive");
	}
}


void
Zipper::close ()
{
	if (zip_close(_zip) == -1) {
		throw runtime_error ("failed to close ZIP archive");
	}
	_zip = 0;
}


Zipper::~Zipper ()
{
	if (_zip) {
		zip_close(_zip);
	}
}