diff options
| author | Carl Hetherington <cth@carlh.net> | 2015-10-07 11:47:27 +0100 |
|---|---|---|
| committer | Carl Hetherington <cth@carlh.net> | 2015-10-09 13:44:53 +0100 |
| commit | 4121af462c1e8b98866ca206a64678fcba96fde7 (patch) | |
| tree | 8970b97caf442ad6f54db897b87535ffe87cebbf /src/lib/kdm.cc | |
| parent | 7beb2cf613c7d310493fa39f9ef611b35f8f3d58 (diff) | |
Try to clean up KDM creation code a bit.
Diffstat (limited to 'src/lib/kdm.cc')
| -rw-r--r-- | src/lib/kdm.cc | 295 |
1 files changed, 0 insertions, 295 deletions
diff --git a/src/lib/kdm.cc b/src/lib/kdm.cc deleted file mode 100644 index f0ba8fb85..000000000 --- a/src/lib/kdm.cc +++ /dev/null @@ -1,295 +0,0 @@ -/* - Copyright (C) 2013-2015 Carl Hetherington <cth@carlh.net> - - This program 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, - 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. - -*/ - -#include "kdm.h" -#include "cinema.h" -#include "screen.h" -#include "exceptions.h" -#include "util.h" -#include "film.h" -#include "config.h" -#include "safe_stringstream.h" -#include "quickmail.h" -#include "compose.hpp" -#include <zip.h> -#include <dcp/encrypted_kdm.h> -#include <dcp/types.h> -#include <boost/shared_ptr.hpp> -#include <list> -#include <iostream> - -using std::list; -using std::string; -using std::cout; -using boost::shared_ptr; - -struct ScreenKDM -{ - ScreenKDM (shared_ptr<Screen> s, dcp::EncryptedKDM k) - : screen (s) - , kdm (k) - {} - - shared_ptr<Screen> screen; - dcp::EncryptedKDM kdm; -}; - -static string -kdm_filename (shared_ptr<const Film> film, ScreenKDM kdm) -{ - return tidy_for_filename (film->name()) + "_" + tidy_for_filename (kdm.screen->cinema->name) + "_" + tidy_for_filename (kdm.screen->name) + ".kdm.xml"; -} - -struct CinemaKDMs -{ - shared_ptr<Cinema> cinema; - list<ScreenKDM> screen_kdms; - - void make_zip_file (shared_ptr<const Film> film, boost::filesystem::path zip_file) 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; - - for (list<ScreenKDM>::const_iterator i = screen_kdms.begin(); i != screen_kdms.end(); ++i) { - 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 StringError ("could not create ZIP source"); - } - - if (zip_add (zip, kdm_filename (film, *i).c_str(), source) == -1) { - throw StringError ("failed to add KDM to ZIP archive"); - } - } - - if (zip_close (zip) == -1) { - throw StringError ("failed to close ZIP archive"); - } - } -}; - -/* Not complete but sufficient for our purposes (we're using - ScreenKDM in a list where all the screens will be unique). -*/ -bool -operator== (ScreenKDM const & a, ScreenKDM const & b) -{ - return a.screen == b.screen; -} - -static list<ScreenKDM> -make_screen_kdms ( - shared_ptr<const Film> film, - list<shared_ptr<Screen> > screens, - boost::filesystem::path cpl, - dcp::LocalTime from, - dcp::LocalTime to, - dcp::Formulation formulation - ) -{ - list<dcp::EncryptedKDM> kdms = film->make_kdms (screens, cpl, from, to, formulation); - - list<ScreenKDM> screen_kdms; - - list<shared_ptr<Screen> >::iterator i = screens.begin (); - list<dcp::EncryptedKDM>::iterator j = kdms.begin (); - while (i != screens.end() && j != kdms.end ()) { - screen_kdms.push_back (ScreenKDM (*i, *j)); - ++i; - ++j; - } - - return screen_kdms; -} - -static list<CinemaKDMs> -make_cinema_kdms ( - shared_ptr<const Film> film, - list<shared_ptr<Screen> > screens, - boost::filesystem::path cpl, - dcp::LocalTime from, - dcp::LocalTime to, - dcp::Formulation formulation - ) -{ - list<ScreenKDM> screen_kdms = make_screen_kdms (film, screens, cpl, from, to, formulation); - 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; -} - -/** @param from KDM from time in local time. - * @param to KDM to time in local time. - */ -void -write_kdm_files ( - shared_ptr<const Film> film, - list<shared_ptr<Screen> > screens, - boost::filesystem::path cpl, - dcp::LocalTime from, - dcp::LocalTime to, - dcp::Formulation formulation, - boost::filesystem::path directory - ) -{ - list<ScreenKDM> screen_kdms = make_screen_kdms (film, screens, cpl, from, to, formulation); - - /* Write KDMs to the specified directory */ - for (list<ScreenKDM>::iterator i = screen_kdms.begin(); i != screen_kdms.end(); ++i) { - boost::filesystem::path out = directory; - out /= kdm_filename (film, *i); - i->kdm.as_xml (out); - } -} - -void -write_kdm_zip_files ( - shared_ptr<const Film> film, - list<shared_ptr<Screen> > screens, - boost::filesystem::path cpl, - dcp::LocalTime from, - dcp::LocalTime to, - dcp::Formulation formulation, - boost::filesystem::path directory - ) -{ - list<CinemaKDMs> cinema_kdms = make_cinema_kdms (film, screens, cpl, from, to, formulation); - - for (list<CinemaKDMs>::const_iterator i = cinema_kdms.begin(); i != cinema_kdms.end(); ++i) { - boost::filesystem::path path = directory; - path /= tidy_for_filename (i->cinema->name) + ".zip"; - i->make_zip_file (film, path); - } -} - -void -email_kdms ( - shared_ptr<const Film> film, - list<shared_ptr<Screen> > screens, - boost::filesystem::path cpl, - dcp::LocalTime from, - dcp::LocalTime to, - dcp::Formulation formulation - ) -{ - list<CinemaKDMs> cinema_kdms = make_cinema_kdms (film, screens, cpl, from, to, formulation); - - for (list<CinemaKDMs>::const_iterator i = cinema_kdms.begin(); i != cinema_kdms.end(); ++i) { - - boost::filesystem::path zip_file = boost::filesystem::temp_directory_path (); - zip_file /= boost::filesystem::unique_path().string() + ".zip"; - i->make_zip_file (film, zip_file); - - /* Send email */ - - quickmail_initialize (); - - SafeStringStream start; - start << from.date() << " " << from.time_of_day(); - SafeStringStream end; - end << to.date() << " " << to.time_of_day(); - - string subject = Config::instance()->kdm_subject(); - boost::algorithm::replace_all (subject, "$CPL_NAME", film->dcp_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); - quickmail mail = quickmail_create (Config::instance()->kdm_from().c_str(), subject.c_str ()); - - quickmail_add_to (mail, i->cinema->email.c_str ()); - if (!Config::instance()->kdm_cc().empty ()) { - quickmail_add_cc (mail, Config::instance()->kdm_cc().c_str ()); - } - if (!Config::instance()->kdm_bcc().empty ()) { - quickmail_add_bcc (mail, Config::instance()->kdm_bcc().c_str ()); - } - - string body = Config::instance()->kdm_email().c_str(); - boost::algorithm::replace_all (body, "$CPL_NAME", film->dcp_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; - for (list<ScreenKDM>::const_iterator j = i->screen_kdms.begin(); j != i->screen_kdms.end(); ++j) { - screens << j->screen->name << ", "; - } - boost::algorithm::replace_all (body, "$SCREENS", screens.str().substr (0, screens.str().length() - 2)); - - quickmail_set_body (mail, body.c_str()); - quickmail_add_attachment_file (mail, zip_file.string().c_str(), "application/zip"); - - char const* error = quickmail_send ( - mail, - Config::instance()->mail_server().c_str(), - Config::instance()->mail_port(), - Config::instance()->mail_user().c_str(), - Config::instance()->mail_password().c_str() - ); - - if (error) { - quickmail_destroy (mail); - throw KDMError ( - String::compose ( - "Failed to send KDM email to %1 (%2)", - Config::instance()->mail_server(), - error - ) - ); - } - quickmail_destroy (mail); - } -} |
