2 Copyright (C) 2013 Carl Hetherington <cth@carlh.net>
4 This program is free software; you can redistribute it and/or modify
5 it under the terms of the GNU General Public License as published by
6 the Free Software Foundation; either version 2 of the License, or
7 (at your option) any later version.
9 This program is distributed in the hope that it will be useful,
10 but WITHOUT ANY WARRANTY; without even the implied warranty of
11 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 GNU General Public License for more details.
14 You should have received a copy of the GNU General Public License
15 along with this program; if not, write to the Free Software
16 Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
21 #include <boost/shared_ptr.hpp>
22 #include <quickmail.h>
24 #include <libdcp/kdm.h>
27 #include "exceptions.h"
34 using boost::shared_ptr;
38 ScreenKDM (shared_ptr<Screen> s, libdcp::KDM k)
43 shared_ptr<Screen> screen;
49 shared_ptr<Cinema> cinema;
50 list<ScreenKDM> screen_kdms;
52 void make_zip_file (boost::filesystem::path zip_file) const
55 struct zip* zip = zip_open (zip_file.string().c_str(), ZIP_CREATE | ZIP_EXCL, &error);
57 if (error == ZIP_ER_EXISTS) {
58 throw FileError ("ZIP file already exists", zip_file);
60 throw FileError ("could not create ZIP file", zip_file);
63 list<shared_ptr<string> > kdm_strings;
65 for (list<ScreenKDM>::const_iterator i = screen_kdms.begin(); i != screen_kdms.end(); ++i) {
66 shared_ptr<string> kdm (new string (i->kdm.as_xml ()));
67 kdm_strings.push_back (kdm);
69 struct zip_source* source = zip_source_buffer (zip, kdm->c_str(), kdm->length(), 0);
71 throw StringError ("could not create ZIP source");
74 string const name = tidy_for_filename (i->screen->cinema->name) + "_" +
75 tidy_for_filename (i->screen->name) + ".kdm.xml";
77 if (zip_add (zip, name.c_str(), source) == -1) {
78 throw StringError ("failed to add KDM to ZIP archive");
82 if (zip_close (zip) == -1) {
83 throw StringError ("failed to close ZIP archive");
88 /* Not complete but sufficient for our purposes (we're using
89 ScreenKDM in a list where all the screens will be unique).
92 operator== (ScreenKDM const & a, ScreenKDM const & b)
94 return a.screen == b.screen;
97 static list<ScreenKDM>
98 make_screen_kdms (shared_ptr<Film> film, list<shared_ptr<Screen> > screens, boost::posix_time::ptime from, boost::posix_time::ptime to)
100 list<libdcp::KDM> kdms = film->make_kdms (screens, from, to);
102 list<ScreenKDM> screen_kdms;
104 list<shared_ptr<Screen> >::iterator i = screens.begin ();
105 list<libdcp::KDM>::iterator j = kdms.begin ();
106 while (i != screens.end() && j != kdms.end ()) {
107 screen_kdms.push_back (ScreenKDM (*i, *j));
115 static list<CinemaKDMs>
116 make_cinema_kdms (shared_ptr<Film> film, list<shared_ptr<Screen> > screens, boost::posix_time::ptime from, boost::posix_time::ptime to)
118 list<ScreenKDM> screen_kdms = make_screen_kdms (film, screens, from, to);
119 list<CinemaKDMs> cinema_kdms;
121 while (!screen_kdms.empty ()) {
123 /* Get all the screens from a single cinema */
127 list<ScreenKDM>::iterator i = screen_kdms.begin ();
128 ck.cinema = i->screen->cinema;
129 ck.screen_kdms.push_back (*i);
130 list<ScreenKDM>::iterator j = i;
132 screen_kdms.remove (*j);
134 while (i != screen_kdms.end ()) {
135 if (i->screen->cinema == ck.cinema) {
136 ck.screen_kdms.push_back (*i);
137 list<ScreenKDM>::iterator j = i;
139 screen_kdms.remove (*j);
145 cinema_kdms.push_back (ck);
153 shared_ptr<Film> film, list<shared_ptr<Screen> > screens, boost::posix_time::ptime from, boost::posix_time::ptime to, boost::filesystem::path directory
156 list<ScreenKDM> screen_kdms = make_screen_kdms (film, screens, from, to);
158 /* Write KDMs to the specified directory */
159 for (list<ScreenKDM>::iterator i = screen_kdms.begin(); i != screen_kdms.end(); ++i) {
160 boost::filesystem::path out = directory;
161 out /= tidy_for_filename (i->screen->cinema->name) + "_" + tidy_for_filename (i->screen->name) + ".kdm.xml";
167 write_kdm_zip_files (
168 shared_ptr<Film> film, list<shared_ptr<Screen> > screens, boost::posix_time::ptime from, boost::posix_time::ptime to, boost::filesystem::path directory
171 list<CinemaKDMs> cinema_kdms = make_cinema_kdms (film, screens, from, to);
173 for (list<CinemaKDMs>::const_iterator i = cinema_kdms.begin(); i != cinema_kdms.end(); ++i) {
174 boost::filesystem::path path = directory;
175 path /= tidy_for_filename (i->cinema->name) + ".zip";
176 i->make_zip_file (path);
181 email_kdms (shared_ptr<Film> film, list<shared_ptr<Screen> > screens, boost::posix_time::ptime from, boost::posix_time::ptime to)
183 list<CinemaKDMs> cinema_kdms = make_cinema_kdms (film, screens, from, to);
185 for (list<CinemaKDMs>::const_iterator i = cinema_kdms.begin(); i != cinema_kdms.end(); ++i) {
187 boost::filesystem::path zip_file = boost::filesystem::temp_directory_path ();
188 zip_file /= boost::filesystem::unique_path().string() + ".zip";
189 i->make_zip_file (zip_file);
193 quickmail_initialize ();
194 quickmail mail = quickmail_create (Config::instance()->kdm_from().c_str(), "KDM delivery");
195 quickmail_add_to (mail, i->cinema->email.c_str ());
197 string body = Config::instance()->kdm_email().c_str();
198 boost::algorithm::replace_all (body, "$DCP_NAME", film->dcp_name ());
200 quickmail_set_body (mail, body.c_str());
201 quickmail_add_attachment_file (mail, zip_file.string().c_str());
202 char const* error = quickmail_send (mail, Config::instance()->mail_server().c_str(), 25, "", "");
204 quickmail_destroy (mail);
205 throw StringError (String::compose ("Failed to send KDM email (%1)", error));
207 quickmail_destroy (mail);