X-Git-Url: https://main.carlh.net/gitweb/?p=dcpomatic.git;a=blobdiff_plain;f=src%2Flib%2Fcinema_kdms.cc;h=94e83f8077de30baf735722c2dd696a4c2e3d608;hp=d7495e9ac2cab7217539071f96ad11038e5b3de1;hb=9262839360779de83b711f86151fcf56b86c3989;hpb=cd4daa051e12b29618da8e24712b337f50d869cc

diff --git a/src/lib/cinema_kdms.cc b/src/lib/cinema_kdms.cc
index d7495e9ac..94e83f807 100644
--- a/src/lib/cinema_kdms.cc
+++ b/src/lib/cinema_kdms.cc
@@ -1,19 +1,20 @@
 /*
     Copyright (C) 2013-2015 Carl Hetherington <cth@carlh.net>
 
-    This program is free software; you can redistribute it and/or modify
+    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.
 
-    This program is distributed in the hope that it will be useful,
+    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 this program; if not, write to the Free Software
-    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+    along with DCP-o-matic.  If not, see <http://www.gnu.org/licenses/>.
 
 */
 
@@ -26,7 +27,9 @@
 #include "emailer.h"
 #include "compose.hpp"
 #include "log.h"
-#include <zip.h>
+#include "zipper.h"
+#include "dcpomatic_log.h"
+#include "kdm_with_metadata.h"
 #include <boost/foreach.hpp>
 
 #include "i18n.h"
@@ -36,62 +39,47 @@ using std::cout;
 using std::string;
 using std::runtime_error;
 using boost::shared_ptr;
+using boost::function;
+using boost::optional;
 
 void
-CinemaKDMs::make_zip_file (string film_name, boost::filesystem::path zip_file) const
+make_zip_file (list<KDMWithMetadataPtr> kdms, boost::filesystem::path zip_file, dcp::NameFormat name_format, dcp::NameFormat::Map name_values)
 {
-	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;
-
-	BOOST_FOREACH (ScreenKDM const & i, screen_kdms) {
-		shared_ptr<string> kdm (new string (i.kdm.as_xml ()));
-		kdm_strings.push_back (kdm);
+	Zipper zipper (zip_file);
 
-		struct zip_source* source = zip_source_buffer (zip, kdm->c_str(), kdm->length(), 0);
-		if (!source) {
-			throw runtime_error ("could not create ZIP source");
-		}
-
-		if (zip_add (zip, i.filename(film_name).c_str(), source) == -1) {
-			throw runtime_error ("failed to add KDM to ZIP archive");
-		}
+	BOOST_FOREACH (KDMWithMetadataPtr i, kdms) {
+		name_values['i'] = i->kdm_id ();
+		string const name = careful_string_filter(name_format.get(name_values, ".xml"));
+		zipper.add (name, i->kdm_as_xml());
 	}
 
-	if (zip_close (zip) == -1) {
-		throw runtime_error ("failed to close ZIP archive");
-	}
+	zipper.close ();
 }
 
-list<CinemaKDMs>
-CinemaKDMs::collect (list<ScreenKDM> screen_kdms)
+/** Collect a list of KDMWithMetadatas into a list of list<KDMWithMetadata> so that each
+ *  CinemaKDM contains the KDMs for its cinema.
+ */
+list<list<KDMWithMetadataPtr> >
+collect (list<KDMWithMetadataPtr> screen_kdms)
 {
-	list<CinemaKDMs> cinema_kdms;
+	list<list<KDMWithMetadataPtr> > cinema_kdms;
 
 	while (!screen_kdms.empty ()) {
 
 		/* Get all the screens from a single cinema */
 
-		CinemaKDMs ck;
+		list<KDMWithMetadataPtr> ck;
 
-		list<ScreenKDM>::iterator i = screen_kdms.begin ();
-		ck.cinema = i->screen->cinema;
-		ck.screen_kdms.push_back (*i);
-		list<ScreenKDM>::iterator j = i;
+		list<KDMWithMetadataPtr>::iterator i = screen_kdms.begin ();
+		ck.push_back (*i);
+		list<KDMWithMetadataPtr>::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;
+			if ((*i)->cinema() == ck.front()->cinema()) {
+				ck.push_back (*i);
+				list<KDMWithMetadataPtr>::iterator j = i;
 				++i;
 				screen_kdms.remove (*j);
 			} else {
@@ -105,21 +93,81 @@ CinemaKDMs::collect (list<ScreenKDM> screen_kdms)
 	return cinema_kdms;
 }
 
-void
-CinemaKDMs::write_zip_files (string film_name, list<CinemaKDMs> cinema_kdms, boost::filesystem::path directory)
+/** Write one directory per cinema into another directory */
+int
+write_directories (
+	list<list<KDMWithMetadataPtr> > 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 (list<KDMWithMetadataPtr> const & i, cinema_kdms) {
+		boost::filesystem::path path = directory;
+		path /= container_name_format.get(name_values, "");
+		if (!boost::filesystem::exists (path) || confirm_overwrite (path)) {
+			boost::filesystem::create_directories (path);
+			write_files (i, path, filename_format, name_values, confirm_overwrite);
+		}
+		written += i.size();
+	}
+
+	return written;
+}
+
+/** Write one ZIP file per cinema into a directory */
+int
+write_zip_files (
+	list<list<KDMWithMetadataPtr> > 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
+	)
 {
-	BOOST_FOREACH (CinemaKDMs const & i, cinema_kdms) {
+	/* No specific screen */
+	name_values['s'] = "";
+
+	int written = 0;
+
+	BOOST_FOREACH (list<KDMWithMetadataPtr> const & i, cinema_kdms) {
 		boost::filesystem::path path = directory;
-		path /= tidy_for_filename (i.cinema->name) + ".zip";
-		i.make_zip_file (film_name, path);
+		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);
+			}
+			make_zip_file (i, path, filename_format, name_values);
+			written += i.size();
+		}
 	}
+
+	return written;
 }
 
-/** @param log Log to write email session transcript to, or 0 */
-/* XXX: should probably get from/to from the KDMs themselves */
+/** Email one ZIP file per cinema to the cinema.
+ *  @param cinema_kdms KDMS to email.
+ *  @param container_name_format Format of folder / ZIP to use.
+ *  @param filename_format Format of filenames to use.
+ *  @param name_values Values to substitute into \p container_name_format and \p filename_format.
+ *  @param cpl_name Name of the CPL that the KDMs are for.
+ */
 void
-CinemaKDMs::email (
-	string film_name, string cpl_name, list<CinemaKDMs> cinema_kdms, dcp::LocalTime from, dcp::LocalTime to, shared_ptr<Log> log
+email (
+	list<list<KDMWithMetadataPtr> > cinema_kdms,
+	dcp::NameFormat container_name_format,
+	dcp::NameFormat filename_format,
+	dcp::NameFormat::Map name_values,
+	string cpl_name
 	)
 {
 	Config* config = Config::instance ();
@@ -128,35 +176,42 @@ CinemaKDMs::email (
 		throw NetworkError (_("No mail server configured in preferences"));
 	}
 
-	BOOST_FOREACH (CinemaKDMs const & i, cinema_kdms) {
+	/* No specific screen */
+	name_values['s'] = "";
 
-		boost::filesystem::path zip_file = boost::filesystem::temp_directory_path ();
-		zip_file /= boost::filesystem::unique_path().string() + ".zip";
-		i.make_zip_file (film_name, zip_file);
+	BOOST_FOREACH (list<KDMWithMetadataPtr> const & i, cinema_kdms) {
+
+		if (i.front()->cinema()->emails.empty()) {
+			continue;
+		}
+
+		boost::filesystem::path zip_file = boost::filesystem::temp_directory_path() / boost::filesystem::unique_path();
+		boost::filesystem::create_directories (zip_file);
+		zip_file /= container_name_format.get(name_values, ".zip");
+		make_zip_file (i, zip_file, filename_format, name_values);
 
 		string subject = config->kdm_subject();
-		SafeStringStream start;
-		start << from.date() << " " << from.time_of_day();
-		SafeStringStream end;
-		end << to.date() << " " << to.time_of_day();
 		boost::algorithm::replace_all (subject, "$CPL_NAME", cpl_name);
-		boost::algorithm::replace_all (subject, "$START_TIME", start.str ());
-		boost::algorithm::replace_all (subject, "$END_TIME", end.str ());
-		boost::algorithm::replace_all (subject, "$CINEMA_NAME", i.cinema->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.front()->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", start.str ());
-		boost::algorithm::replace_all (body, "$END_TIME", end.str ());
-		boost::algorithm::replace_all (body, "$CINEMA_NAME", i.cinema->name);
-
-		SafeStringStream screens;
-		BOOST_FOREACH (ScreenKDM const & j, i.screen_kdms) {
-			screens << j.screen->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.front()->cinema()->name);
+
+		string screens;
+		BOOST_FOREACH (KDMWithMetadataPtr j, i) {
+			optional<string> screen_name = j->get('n');
+			if (screen_name) {
+				screens += *screen_name + ", ";
+			}
 		}
-		boost::algorithm::replace_all (body, "$SCREENS", screens.str().substr (0, screens.str().length() - 2));
+		boost::algorithm::replace_all (body, "$SCREENS", screens.substr (0, screens.length() - 2));
 
-		Emailer email (config->kdm_from(), i.cinema->emails, subject, body);
+		Emailer email (config->kdm_from(), i.front()->cinema()->emails, subject, body);
 
 		BOOST_FOREACH (string i, config->kdm_cc()) {
 			email.add_cc (i);
@@ -165,22 +220,26 @@ CinemaKDMs::email (
 			email.add_bcc (config->kdm_bcc ());
 		}
 
-		string const name = tidy_for_filename(i.cinema->name) + "_" + tidy_for_filename(film_name) + ".zip";
-		email.add_attachment (zip_file, name, "application/zip");
+		email.add_attachment (zip_file, container_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());
+			email.send (c->mail_server(), c->mail_port(), c->mail_protocol(), c->mail_user(), c->mail_password());
 		} catch (...) {
-			if (log) {
-				log->log (email.notes(), LogEntry::TYPE_DEBUG_EMAIL);
-			}
+			boost::filesystem::remove (zip_file);
+			dcpomatic_log->log ("Email content follows", LogEntry::TYPE_DEBUG_EMAIL);
+			dcpomatic_log->log (email.email(), LogEntry::TYPE_DEBUG_EMAIL);
+			dcpomatic_log->log ("Email session follows", LogEntry::TYPE_DEBUG_EMAIL);
+			dcpomatic_log->log (email.notes(), LogEntry::TYPE_DEBUG_EMAIL);
 			throw;
 		}
 
-		if (log) {
-			log->log (email.notes(), LogEntry::TYPE_DEBUG_EMAIL);
-		}
+		boost::filesystem::remove (zip_file);
+
+		dcpomatic_log->log ("Email content follows", LogEntry::TYPE_DEBUG_EMAIL);
+		dcpomatic_log->log (email.email(), LogEntry::TYPE_DEBUG_EMAIL);
+		dcpomatic_log->log ("Email session follows", LogEntry::TYPE_DEBUG_EMAIL);
+		dcpomatic_log->log (email.notes(), LogEntry::TYPE_DEBUG_EMAIL);
 	}
 }