/*
    Copyright (C) 2019-2020 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 "disk_writer_messages.h"
#include "copy_to_drive_job.h"
#include "compose.hpp"
#include "exceptions.h"
#include "dcpomatic_log.h"
#include <dcp/raw_convert.h>
#include <nanomsg/nn.h>
#include <unistd.h>
#include <fcntl.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <iostream>

#include "i18n.h"

using std::string;
using std::cout;
using std::min;
using boost::shared_ptr;
using boost::optional;
using dcp::raw_convert;

CopyToDriveJob::CopyToDriveJob (boost::filesystem::path dcp, Drive drive, Nanomsg& nanomsg)
	: Job (shared_ptr<Film>())
	, _dcp (dcp)
	, _drive (drive)
	, _nanomsg (nanomsg)
{

}

string
CopyToDriveJob::name () const
{
	return String::compose (_("Copying %1\nto %2"), _dcp.filename().string(), _drive.description());
}

string
CopyToDriveJob::json_name () const
{
	return N_("copy");
}

void
CopyToDriveJob::run ()
{
	LOG_DISK("Sending write request to disk writer for %1 %2", _dcp.string(), _drive.device());
	if (!_nanomsg.send(String::compose(DISK_WRITER_WRITE "\n%1\n%2\n", _dcp.string(), _drive.device()), 2000)) {
		throw CommunicationFailedError ();
	}

	enum State {
		SETUP,
		FORMAT,
		COPY,
		VERIFY
	} state = SETUP;

	while (true) {
		optional<string> s = _nanomsg.receive (10000);
		if (!s) {
			continue;
		}
		if (*s == DISK_WRITER_OK) {
			set_state (FINISHED_OK);
			return;
		} else if (*s == DISK_WRITER_ERROR) {
			optional<string> const m = _nanomsg.receive (500);
			optional<string> const n = _nanomsg.receive (500);
			throw CopyError (m.get_value_or("Unknown"), raw_convert<int>(n.get_value_or("0")));
		} else if (*s == DISK_WRITER_FORMATTING) {
			sub (_("Formatting drive"));
			set_progress_unknown ();
			state = FORMAT;
		} else if (*s == DISK_WRITER_COPY_PROGRESS) {
			if (state == FORMAT) {
				sub (_("Copying DCP"));
				state = COPY;
			}
			optional<string> progress = _nanomsg.receive (500);
			if (progress) {
				set_progress (raw_convert<float>(*progress));
			}
		} else if (*s == DISK_WRITER_VERIFY_PROGRESS) {
			if (state == COPY) {
				sub (_("Verifying copied files"));
				state = VERIFY;
			}
			optional<string> progress = _nanomsg.receive (500);
			if (progress) {
				set_progress (raw_convert<float>(*progress));
			}
		}
	}
}