Lots of #include <iostream>s for Arch.
[dcpomatic.git] / src / lib / kdm.cc
index 793a3fa0eb5c0cea99d0edc60424742490ea3fae..f0ba8fb85022ac8080594767eb5b61cf127cc39d 100644 (file)
@@ -1,5 +1,5 @@
 /*
-    Copyright (C) 2013 Carl Hetherington <cth@carlh.net>
+    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
 
 */
 
-#include <list>
-#include <boost/shared_ptr.hpp>
-#include <quickmail.h>
-#include <zip.h>
-#include <dcp/encrypted_kdm.h>
 #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::stringstream;
+using std::cout;
 using boost::shared_ptr;
 
 struct ScreenKDM
@@ -40,7 +45,7 @@ struct ScreenKDM
                : screen (s)
                , kdm (k)
        {}
-       
+
        shared_ptr<Screen> screen;
        dcp::EncryptedKDM kdm;
 };
@@ -66,23 +71,23 @@ struct CinemaKDMs
                        }
                        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");
                }
@@ -102,15 +107,16 @@ static list<ScreenKDM>
 make_screen_kdms (
        shared_ptr<const Film> film,
        list<shared_ptr<Screen> > screens,
-       boost::filesystem::path dcp,
+       boost::filesystem::path cpl,
        dcp::LocalTime from,
-       dcp::LocalTime to
+       dcp::LocalTime to,
+       dcp::Formulation formulation
        )
 {
-       list<dcp::EncryptedKDM> kdms = film->make_kdms (screens, dcp, 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<dcp::EncryptedKDM>::iterator j = kdms.begin ();
        while (i != screens.end() && j != kdms.end ()) {
@@ -126,27 +132,28 @@ static list<CinemaKDMs>
 make_cinema_kdms (
        shared_ptr<const Film> film,
        list<shared_ptr<Screen> > screens,
-       boost::filesystem::path dcp,
+       boost::filesystem::path cpl,
        dcp::LocalTime from,
-       dcp::LocalTime to
+       dcp::LocalTime to,
+       dcp::Formulation formulation
        )
 {
-       list<ScreenKDM> screen_kdms = make_screen_kdms (film, screens, dcp, from, to);
+       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);
@@ -164,17 +171,21 @@ make_cinema_kdms (
        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 dcp,
+       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, dcp, 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) {
@@ -188,13 +199,14 @@ void
 write_kdm_zip_files (
        shared_ptr<const Film> film,
        list<shared_ptr<Screen> > screens,
-       boost::filesystem::path dcp,
+       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, dcp, 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;
@@ -207,50 +219,76 @@ void
 email_kdms (
        shared_ptr<const Film> film,
        list<shared_ptr<Screen> > screens,
-       boost::filesystem::path dcp,
+       boost::filesystem::path cpl,
        dcp::LocalTime from,
-       dcp::LocalTime to
+       dcp::LocalTime to,
+       dcp::Formulation formulation
        )
 {
-       list<CinemaKDMs> cinema_kdms = make_cinema_kdms (film, screens, dcp, 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 (film, zip_file);
-               
+
                /* Send email */
-               
+
                quickmail_initialize ();
-               quickmail mail = quickmail_create (Config::instance()->kdm_from().c_str(), "KDM delivery");
+
+               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 ());
-               stringstream start;
-               start << from.date() << " " << from.time_of_day();
                boost::algorithm::replace_all (body, "$START_TIME", start.str ());
-               stringstream end;
-               end << to.date() << " " << to.time_of_day();
                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");
 
-               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_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 (%1)", error));
+                       throw KDMError (
+                               String::compose (
+                                       "Failed to send KDM email to %1 (%2)",
+                                       Config::instance()->mail_server(),
+                                       error
+                                       )
+                               );
                }
                quickmail_destroy (mail);
        }