Merge master.
[dcpomatic.git] / src / lib / kdm.cc
index 3503306e74060c181d93dfb2abd73e81a851c522..a2ed1be73498dd924752f998fa154e479ae18cb9 100644 (file)
@@ -21,7 +21,8 @@
 #include <boost/shared_ptr.hpp>
 #include <quickmail.h>
 #include <zip.h>
-#include <libdcp/kdm.h>
+#include <dcp/encrypted_kdm.h>
+#include <dcp/types.h>
 #include "kdm.h"
 #include "cinema.h"
 #include "exceptions.h"
 
 using std::list;
 using std::string;
+using std::stringstream;
+using std::cout;
 using boost::shared_ptr;
 
 struct ScreenKDM
 {
-       ScreenKDM (shared_ptr<Screen> s, libdcp::KDM k)
+       ScreenKDM (shared_ptr<Screen> s, dcp::EncryptedKDM k)
                : screen (s)
                , kdm (k)
        {}
        
        shared_ptr<Screen> screen;
-       libdcp::KDM kdm;
+       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 (boost::filesystem::path zip_file) const
+       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);
@@ -71,10 +80,7 @@ struct CinemaKDMs
                                throw StringError ("could not create ZIP source");
                        }
                        
-                       string const name = tidy_for_filename (i->screen->cinema->name) + "_" +
-                               tidy_for_filename (i->screen->name) + ".kdm.xml";
-                       
-                       if (zip_add (zip, name.c_str(), source) == -1) {
+                       if (zip_add (zip, kdm_filename (film, *i).c_str(), source) == -1) {
                                throw StringError ("failed to add KDM to ZIP archive");
                        }
                }
@@ -95,14 +101,21 @@ operator== (ScreenKDM const & a, ScreenKDM const & b)
 }
 
 static list<ScreenKDM>
-make_screen_kdms (shared_ptr<Film> film, list<shared_ptr<Screen> > screens, boost::posix_time::ptime from, boost::posix_time::ptime to)
+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<libdcp::KDM> kdms = film->make_kdms (screens, from, to);
+       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<libdcp::KDM>::iterator j = kdms.begin ();
+       list<dcp::EncryptedKDM>::iterator j = kdms.begin ();
        while (i != screens.end() && j != kdms.end ()) {
                screen_kdms.push_back (ScreenKDM (*i, *j));
                ++i;
@@ -113,9 +126,16 @@ make_screen_kdms (shared_ptr<Film> film, list<shared_ptr<Screen> > screens, boos
 }
 
 static list<CinemaKDMs>
-make_cinema_kdms (shared_ptr<Film> film, list<shared_ptr<Screen> > screens, boost::posix_time::ptime from, boost::posix_time::ptime to)
+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, from, to);
+       list<ScreenKDM> screen_kdms = make_screen_kdms (film, screens, cpl, from, to, formulation);
        list<CinemaKDMs> cinema_kdms;
 
        while (!screen_kdms.empty ()) {
@@ -148,61 +168,120 @@ make_cinema_kdms (shared_ptr<Film> film, list<shared_ptr<Screen> > screens, boos
        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<Film> film, list<shared_ptr<Screen> > screens, boost::posix_time::ptime from, boost::posix_time::ptime to, boost::filesystem::path directory
+       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, from, to);
+       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 /= tidy_for_filename (i->screen->cinema->name) + "_" + tidy_for_filename (i->screen->name) + ".kdm.xml";
+               out /= kdm_filename (film, *i);
                i->kdm.as_xml (out);
        }
 }
 
 void
 write_kdm_zip_files (
-       shared_ptr<Film> film, list<shared_ptr<Screen> > screens, boost::posix_time::ptime from, boost::posix_time::ptime to, boost::filesystem::path directory
+       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, from, to);
+       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 (path);
+               i->make_zip_file (film, path);
        }
 }
 
 void
-email_kdms (shared_ptr<Film> film, list<shared_ptr<Screen> > screens, boost::posix_time::ptime from, boost::posix_time::ptime to)
+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, from, to);
+       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 (zip_file);
+               i->make_zip_file (film, zip_file);
                
                /* Send email */
                
                quickmail_initialize ();
-               quickmail mail = quickmail_create (Config::instance()->kdm_from().c_str(), "KDM delivery");
+
+               stringstream start;
+               start << from.date() << " " << from.time_of_day();
+               stringstream 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, "$DCP_NAME", film->dcp_name ());
+               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);
                
+               stringstream 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());
-               char const* error = quickmail_send (mail, Config::instance()->mail_server().c_str(), 25, "", "");
+               quickmail_add_attachment_file (mail, zip_file.string().c_str(), "application/zip");
+
+               int const port = Config::instance()->mail_user().empty() ? 25 : 587;
+
+               char const* error = quickmail_send (
+                       mail,
+                       Config::instance()->mail_server().c_str(),
+                       port,
+                       Config::instance()->mail_user().c_str(),
+                       Config::instance()->mail_password().c_str()
+                       );
+               
                if (error) {
                        quickmail_destroy (mail);
-                       throw StringError (String::compose ("Failed to send KDM email (%1)", error));
+                       throw KDMError (String::compose ("Failed to send KDM email (%1)", error));
                }
                quickmail_destroy (mail);
        }