+++ /dev/null
-/*
- Copyright (C) 2013-2015 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 "exceptions.h"
-#include "cinema_kdms.h"
-#include "cinema.h"
-#include "screen.h"
-#include "config.h"
-#include "util.h"
-#include "emailer.h"
-#include "compose.hpp"
-#include "log.h"
-#include <zip.h>
-#include <boost/foreach.hpp>
-
-#include "i18n.h"
-
-using std::list;
-using std::cout;
-using std::string;
-using std::runtime_error;
-using boost::shared_ptr;
-using boost::function;
-
-void
-CinemaKDMs::make_zip_file (boost::filesystem::path zip_file, dcp::NameFormat name_format, dcp::NameFormat::Map name_values) const
-{
- int error;
- struct zip* zip = zip_open (zip_file.string().c_str(), ZIP_CREATE | ZIP_EXCL, &error);
- if (!zip) {
- if (error == ZIP_ER_EXISTS) {
- throw FileError ("ZIP file already exists", zip_file);
- }
- throw FileError ("could not create ZIP file", zip_file);
- }
-
- list<shared_ptr<string> > kdm_strings;
-
- name_values['c'] = cinema->name;
-
- BOOST_FOREACH (ScreenKDM const & i, screen_kdms) {
- shared_ptr<string> kdm (new string (i.kdm.as_xml ()));
- kdm_strings.push_back (kdm);
-
- struct zip_source* source = zip_source_buffer (zip, kdm->c_str(), kdm->length(), 0);
- if (!source) {
- throw runtime_error ("could not create ZIP source");
- }
-
- name_values['s'] = i.screen->name;
- string const name = name_format.get(name_values, ".xml");
- if (zip_add (zip, name.c_str(), source) == -1) {
- throw runtime_error ("failed to add KDM to ZIP archive");
- }
- }
-
- if (zip_close (zip) == -1) {
- throw runtime_error ("failed to close ZIP archive");
- }
-}
-
-/** Collect a list of ScreenKDMs into a list of CinemaKDMs so that each
- * CinemaKDM contains the KDMs for its cinema.
- */
-list<CinemaKDMs>
-CinemaKDMs::collect (list<ScreenKDM> screen_kdms)
-{
- list<CinemaKDMs> cinema_kdms;
-
- while (!screen_kdms.empty ()) {
-
- /* Get all the screens from a single cinema */
-
- CinemaKDMs ck;
-
- list<ScreenKDM>::iterator i = screen_kdms.begin ();
- ck.cinema = i->screen->cinema;
- ck.screen_kdms.push_back (*i);
- list<ScreenKDM>::iterator j = i;
- ++i;
- screen_kdms.remove (*j);
-
- while (i != screen_kdms.end ()) {
- if (i->screen->cinema == ck.cinema) {
- ck.screen_kdms.push_back (*i);
- list<ScreenKDM>::iterator j = i;
- ++i;
- screen_kdms.remove (*j);
- } else {
- ++i;
- }
- }
-
- cinema_kdms.push_back (ck);
- }
-
- return cinema_kdms;
-}
-
-/** Write one directory per cinema into another directory */
-int
-CinemaKDMs::write_directories (
- list<CinemaKDMs> cinema_kdms,
- boost::filesystem::path directory,
- dcp::NameFormat container_name_format,
- dcp::NameFormat filename_format,
- dcp::NameFormat::Map name_values,
- function<bool (boost::filesystem::path)> confirm_overwrite
- )
-{
- /* No specific screen */
- name_values['s'] = "";
-
- int written = 0;
-
- BOOST_FOREACH (CinemaKDMs const & i, cinema_kdms) {
- boost::filesystem::path path = directory;
- name_values['c'] = i.cinema->name;
- path /= container_name_format.get(name_values, "");
- if (!boost::filesystem::exists (path) || confirm_overwrite (path)) {
- boost::filesystem::create_directories (path);
- ScreenKDM::write_files (i.screen_kdms, path, filename_format, name_values, confirm_overwrite);
- }
- written += i.screen_kdms.size();
- }
-
- return written;
-}
-
-/** Write one ZIP file per cinema into a directory */
-int
-CinemaKDMs::write_zip_files (
- list<CinemaKDMs> cinema_kdms,
- boost::filesystem::path directory,
- dcp::NameFormat container_name_format,
- dcp::NameFormat filename_format,
- dcp::NameFormat::Map name_values,
- function<bool (boost::filesystem::path)> confirm_overwrite
- )
-{
- /* No specific screen */
- name_values['s'] = "";
-
- int written = 0;
-
- BOOST_FOREACH (CinemaKDMs const & i, cinema_kdms) {
- boost::filesystem::path path = directory;
- name_values['c'] = i.cinema->name;
- path /= container_name_format.get(name_values, ".zip");
- if (!boost::filesystem::exists (path) || confirm_overwrite (path)) {
- if (boost::filesystem::exists (path)) {
- /* Creating a new zip file over an existing one is an error */
- boost::filesystem::remove (path);
- }
- i.make_zip_file (path, filename_format, name_values);
- written += i.screen_kdms.size();
- }
- }
-
- return written;
-}
-
-/** Email one ZIP file per cinema to the cinema.
- * @param cinema_kdms KDMS to email.
- * @param name_format Format of filename to use.
- * @param name_values Values to substitute into \p name_format.
- * @param cpl_name Name of the CPL that the KDMs are for.
- * @param log Log to write email session transcript to, or 0.
- */
-void
-CinemaKDMs::email (
- list<CinemaKDMs> cinema_kdms,
- dcp::NameFormat name_format,
- dcp::NameFormat::Map name_values,
- string cpl_name,
- shared_ptr<Log> log
- )
-{
- Config* config = Config::instance ();
-
- if (config->mail_server().empty()) {
- throw NetworkError (_("No mail server configured in preferences"));
- }
-
- /* No specific screen */
- name_values['s'] = "";
-
- BOOST_FOREACH (CinemaKDMs const & i, cinema_kdms) {
-
- name_values['c'] = i.cinema->name;
-
- boost::filesystem::path zip_file = boost::filesystem::temp_directory_path ();
- zip_file /= boost::filesystem::unique_path().string() + ".zip";
- i.make_zip_file (zip_file, name_format, name_values);
-
- string subject = config->kdm_subject();
- boost::algorithm::replace_all (subject, "$CPL_NAME", cpl_name);
- boost::algorithm::replace_all (subject, "$START_TIME", name_values['b']);
- boost::algorithm::replace_all (subject, "$END_TIME", name_values['e']);
- boost::algorithm::replace_all (subject, "$CINEMA_NAME", i.cinema->name);
-
- string body = config->kdm_email().c_str();
- boost::algorithm::replace_all (body, "$CPL_NAME", cpl_name);
- boost::algorithm::replace_all (body, "$START_TIME", name_values['b']);
- boost::algorithm::replace_all (body, "$END_TIME", name_values['e']);
- boost::algorithm::replace_all (body, "$CINEMA_NAME", i.cinema->name);
-
- string screens;
- BOOST_FOREACH (ScreenKDM const & j, i.screen_kdms) {
- screens += j.screen->name + ", ";
- }
- boost::algorithm::replace_all (body, "$SCREENS", screens.substr (0, screens.length() - 2));
-
- Emailer email (config->kdm_from(), i.cinema->emails, subject, body);
-
- BOOST_FOREACH (string i, config->kdm_cc()) {
- email.add_cc (i);
- }
- if (!config->kdm_bcc().empty ()) {
- email.add_bcc (config->kdm_bcc ());
- }
-
- email.add_attachment (zip_file, name_format.get(name_values, ".zip"), "application/zip");
-
- Config* c = Config::instance ();
-
- try {
- email.send (c->mail_server(), c->mail_port(), c->mail_user(), c->mail_password());
- } catch (...) {
- if (log) {
- log->log ("Email content follows", LogEntry::TYPE_DEBUG_EMAIL);
- log->log (email.email(), LogEntry::TYPE_DEBUG_EMAIL);
- log->log ("Email session follows", LogEntry::TYPE_DEBUG_EMAIL);
- log->log (email.notes(), LogEntry::TYPE_DEBUG_EMAIL);
- }
- throw;
- }
-
- if (log) {
- log->log ("Email content follows", LogEntry::TYPE_DEBUG_EMAIL);
- log->log (email.email(), LogEntry::TYPE_DEBUG_EMAIL);
- log->log ("Email session follows", LogEntry::TYPE_DEBUG_EMAIL);
- log->log (email.notes(), LogEntry::TYPE_DEBUG_EMAIL);
- }
- }
-}