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>
23 #include <dcp/encrypted_kdm.h>
24 #include <dcp/types.h>
27 #include "exceptions.h"
31 #include "safe_stringstream.h"
32 #include "quickmail.h"
37 using boost::shared_ptr;
41 ScreenKDM (shared_ptr<Screen> s, dcp::EncryptedKDM k)
46 shared_ptr<Screen> screen;
47 dcp::EncryptedKDM kdm;
51 kdm_filename (shared_ptr<const Film> film, ScreenKDM kdm)
53 return tidy_for_filename (film->name()) + "_" + tidy_for_filename (kdm.screen->cinema->name) + "_" + tidy_for_filename (kdm.screen->name) + ".kdm.xml";
58 shared_ptr<Cinema> cinema;
59 list<ScreenKDM> screen_kdms;
61 void make_zip_file (shared_ptr<const Film> film, boost::filesystem::path zip_file) const
64 struct zip* zip = zip_open (zip_file.string().c_str(), ZIP_CREATE | ZIP_EXCL, &error);
66 if (error == ZIP_ER_EXISTS) {
67 throw FileError ("ZIP file already exists", zip_file);
69 throw FileError ("could not create ZIP file", zip_file);
72 list<shared_ptr<string> > kdm_strings;
74 for (list<ScreenKDM>::const_iterator i = screen_kdms.begin(); i != screen_kdms.end(); ++i) {
75 shared_ptr<string> kdm (new string (i->kdm.as_xml ()));
76 kdm_strings.push_back (kdm);
78 struct zip_source* source = zip_source_buffer (zip, kdm->c_str(), kdm->length(), 0);
80 throw StringError ("could not create ZIP source");
83 if (zip_add (zip, kdm_filename (film, *i).c_str(), source) == -1) {
84 throw StringError ("failed to add KDM to ZIP archive");
88 if (zip_close (zip) == -1) {
89 throw StringError ("failed to close ZIP archive");
94 /* Not complete but sufficient for our purposes (we're using
95 ScreenKDM in a list where all the screens will be unique).
98 operator== (ScreenKDM const & a, ScreenKDM const & b)
100 return a.screen == b.screen;
103 static list<ScreenKDM>
105 shared_ptr<const Film> film,
106 list<shared_ptr<Screen> > screens,
107 boost::filesystem::path cpl,
110 dcp::Formulation formulation
113 list<dcp::EncryptedKDM> kdms = film->make_kdms (screens, cpl, from, to, formulation);
115 list<ScreenKDM> screen_kdms;
117 list<shared_ptr<Screen> >::iterator i = screens.begin ();
118 list<dcp::EncryptedKDM>::iterator j = kdms.begin ();
119 while (i != screens.end() && j != kdms.end ()) {
120 screen_kdms.push_back (ScreenKDM (*i, *j));
128 static list<CinemaKDMs>
130 shared_ptr<const Film> film,
131 list<shared_ptr<Screen> > screens,
132 boost::filesystem::path cpl,
135 dcp::Formulation formulation
138 list<ScreenKDM> screen_kdms = make_screen_kdms (film, screens, cpl, from, to, formulation);
139 list<CinemaKDMs> cinema_kdms;
141 while (!screen_kdms.empty ()) {
143 /* Get all the screens from a single cinema */
147 list<ScreenKDM>::iterator i = screen_kdms.begin ();
148 ck.cinema = i->screen->cinema;
149 ck.screen_kdms.push_back (*i);
150 list<ScreenKDM>::iterator j = i;
152 screen_kdms.remove (*j);
154 while (i != screen_kdms.end ()) {
155 if (i->screen->cinema == ck.cinema) {
156 ck.screen_kdms.push_back (*i);
157 list<ScreenKDM>::iterator j = i;
159 screen_kdms.remove (*j);
165 cinema_kdms.push_back (ck);
171 /** @param from KDM from time in local time.
172 * @param to KDM to time in local time.
176 shared_ptr<const Film> film,
177 list<shared_ptr<Screen> > screens,
178 boost::filesystem::path cpl,
181 dcp::Formulation formulation,
182 boost::filesystem::path directory
185 list<ScreenKDM> screen_kdms = make_screen_kdms (film, screens, cpl, from, to, formulation);
187 /* Write KDMs to the specified directory */
188 for (list<ScreenKDM>::iterator i = screen_kdms.begin(); i != screen_kdms.end(); ++i) {
189 boost::filesystem::path out = directory;
190 out /= kdm_filename (film, *i);
196 write_kdm_zip_files (
197 shared_ptr<const Film> film,
198 list<shared_ptr<Screen> > screens,
199 boost::filesystem::path cpl,
202 dcp::Formulation formulation,
203 boost::filesystem::path directory
206 list<CinemaKDMs> cinema_kdms = make_cinema_kdms (film, screens, cpl, from, to, formulation);
208 for (list<CinemaKDMs>::const_iterator i = cinema_kdms.begin(); i != cinema_kdms.end(); ++i) {
209 boost::filesystem::path path = directory;
210 path /= tidy_for_filename (i->cinema->name) + ".zip";
211 i->make_zip_file (film, path);
217 shared_ptr<const Film> film,
218 list<shared_ptr<Screen> > screens,
219 boost::filesystem::path cpl,
222 dcp::Formulation formulation
225 list<CinemaKDMs> cinema_kdms = make_cinema_kdms (film, screens, cpl, from, to, formulation);
227 for (list<CinemaKDMs>::const_iterator i = cinema_kdms.begin(); i != cinema_kdms.end(); ++i) {
229 boost::filesystem::path zip_file = boost::filesystem::temp_directory_path ();
230 zip_file /= boost::filesystem::unique_path().string() + ".zip";
231 i->make_zip_file (film, zip_file);
235 quickmail_initialize ();
237 SafeStringStream start;
238 start << from.date() << " " << from.time_of_day();
239 SafeStringStream end;
240 end << to.date() << " " << to.time_of_day();
242 string subject = Config::instance()->kdm_subject();
243 boost::algorithm::replace_all (subject, "$CPL_NAME", film->dcp_name ());
244 boost::algorithm::replace_all (subject, "$START_TIME", start.str ());
245 boost::algorithm::replace_all (subject, "$END_TIME", end.str ());
246 boost::algorithm::replace_all (subject, "$CINEMA_NAME", i->cinema->name);
247 quickmail mail = quickmail_create (Config::instance()->kdm_from().c_str(), subject.c_str ());
249 quickmail_add_to (mail, i->cinema->email.c_str ());
250 if (!Config::instance()->kdm_cc().empty ()) {
251 quickmail_add_cc (mail, Config::instance()->kdm_cc().c_str ());
253 if (!Config::instance()->kdm_bcc().empty ()) {
254 quickmail_add_bcc (mail, Config::instance()->kdm_bcc().c_str ());
257 quickmail_add_header (mail, "Content-Type: text/plain; charset=UTF-8");
259 string body = Config::instance()->kdm_email().c_str();
260 boost::algorithm::replace_all (body, "$CPL_NAME", film->dcp_name ());
261 boost::algorithm::replace_all (body, "$START_TIME", start.str ());
262 boost::algorithm::replace_all (body, "$END_TIME", end.str ());
263 boost::algorithm::replace_all (body, "$CINEMA_NAME", i->cinema->name);
265 SafeStringStream screens;
266 for (list<ScreenKDM>::const_iterator j = i->screen_kdms.begin(); j != i->screen_kdms.end(); ++j) {
267 screens << j->screen->name << ", ";
269 boost::algorithm::replace_all (body, "$SCREENS", screens.str().substr (0, screens.str().length() - 2));
271 quickmail_set_body (mail, body.c_str());
272 quickmail_add_attachment_file (mail, zip_file.string().c_str(), "application/zip");
274 int const port = Config::instance()->mail_user().empty() ? 25 : 587;
276 char const* error = quickmail_send (
278 Config::instance()->mail_server().c_str(),
280 Config::instance()->mail_user().c_str(),
281 Config::instance()->mail_password().c_str()
285 quickmail_destroy (mail);
286 throw KDMError (String::compose ("Failed to send KDM email (%1)", error));
288 quickmail_destroy (mail);