Replace CinemaKDMs class with a list<KDMWithMetadata>
[dcpomatic.git] / src / lib / cinema_kdms.cc
index 7952b7ab22e3f22e3b75f7bb03ea084791234721..94e83f8077de30baf735722c2dd696a4c2e3d608 100644 (file)
@@ -27,8 +27,9 @@
 #include "emailer.h"
 #include "compose.hpp"
 #include "log.h"
+#include "zipper.h"
 #include "dcpomatic_log.h"
-#include <zip.h>
+#include "kdm_with_metadata.h"
 #include <boost/foreach.hpp>
 
 #include "i18n.h"
@@ -39,70 +40,46 @@ using std::string;
 using std::runtime_error;
 using boost::shared_ptr;
 using boost::function;
+using boost::optional;
 
 void
-CinemaKDMs::make_zip_file (boost::filesystem::path zip_file, dcp::NameFormat name_format, dcp::NameFormat::Map name_values) const
+make_zip_file (list<KDMWithMetadataPtr> kdms, boost::filesystem::path zip_file, dcp::NameFormat name_format, dcp::NameFormat::Map name_values)
 {
-       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;
+       Zipper zipper (zip_file);
 
-       name_values['c'] = cinema->name;
-
-       BOOST_FOREACH (ScreenKDM const & i, screen_kdms) {
-               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 runtime_error ("could not create ZIP source");
-               }
-
-               name_values['s'] = i.screen->name;
-               name_values['i'] = i.kdm.id ();
+       BOOST_FOREACH (KDMWithMetadataPtr i, kdms) {
+               name_values['i'] = i->kdm_id ();
                string const name = careful_string_filter(name_format.get(name_values, ".xml"));
-               if (zip_add (zip, name.c_str(), source) == -1) {
-                       throw runtime_error ("failed to add KDM to ZIP archive");
-               }
+               zipper.add (name, i->kdm_as_xml());
        }
 
-       if (zip_close (zip) == -1) {
-               throw runtime_error ("failed to close ZIP archive");
-       }
+       zipper.close ();
 }
 
-/** Collect a list of ScreenKDMs into a list of CinemaKDMs so that each
+/** Collect a list of KDMWithMetadatas into a list of list<KDMWithMetadata> so that each
  *  CinemaKDM contains the KDMs for its cinema.
  */
-list<CinemaKDMs>
-CinemaKDMs::collect (list<ScreenKDM> screen_kdms)
+list<list<KDMWithMetadataPtr> >
+collect (list<KDMWithMetadataPtr> screen_kdms)
 {
-       list<CinemaKDMs> cinema_kdms;
+       list<list<KDMWithMetadataPtr> > cinema_kdms;
 
        while (!screen_kdms.empty ()) {
 
                /* Get all the screens from a single cinema */
 
-               CinemaKDMs ck;
+               list<KDMWithMetadataPtr> ck;
 
-               list<ScreenKDM>::iterator i = screen_kdms.begin ();
-               ck.cinema = i->screen->cinema;
-               ck.screen_kdms.push_back (*i);
-               list<ScreenKDM>::iterator j = i;
+               list<KDMWithMetadataPtr>::iterator i = screen_kdms.begin ();
+               ck.push_back (*i);
+               list<KDMWithMetadataPtr>::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;
+                       if ((*i)->cinema() == ck.front()->cinema()) {
+                               ck.push_back (*i);
+                               list<KDMWithMetadataPtr>::iterator j = i;
                                ++i;
                                screen_kdms.remove (*j);
                        } else {
@@ -118,8 +95,8 @@ CinemaKDMs::collect (list<ScreenKDM> screen_kdms)
 
 /** Write one directory per cinema into another directory */
 int
-CinemaKDMs::write_directories (
-       list<CinemaKDMs> cinema_kdms,
+write_directories (
+       list<list<KDMWithMetadataPtr> > cinema_kdms,
        boost::filesystem::path directory,
        dcp::NameFormat container_name_format,
        dcp::NameFormat filename_format,
@@ -132,15 +109,14 @@ CinemaKDMs::write_directories (
 
        int written = 0;
 
-       BOOST_FOREACH (CinemaKDMs const & i, cinema_kdms) {
+       BOOST_FOREACH (list<KDMWithMetadataPtr> const & i, cinema_kdms) {
                boost::filesystem::path path = directory;
-               name_values['c'] = i.cinema->name;
                path /= container_name_format.get(name_values, "");
                if (!boost::filesystem::exists (path) || confirm_overwrite (path)) {
                        boost::filesystem::create_directories (path);
-                       ScreenKDM::write_files (i.screen_kdms, path, filename_format, name_values, confirm_overwrite);
+                       write_files (i, path, filename_format, name_values, confirm_overwrite);
                }
-               written += i.screen_kdms.size();
+               written += i.size();
        }
 
        return written;
@@ -148,8 +124,8 @@ CinemaKDMs::write_directories (
 
 /** Write one ZIP file per cinema into a directory */
 int
-CinemaKDMs::write_zip_files (
-       list<CinemaKDMs> cinema_kdms,
+write_zip_files (
+       list<list<KDMWithMetadataPtr> > cinema_kdms,
        boost::filesystem::path directory,
        dcp::NameFormat container_name_format,
        dcp::NameFormat filename_format,
@@ -162,17 +138,16 @@ CinemaKDMs::write_zip_files (
 
        int written = 0;
 
-       BOOST_FOREACH (CinemaKDMs const & i, cinema_kdms) {
+       BOOST_FOREACH (list<KDMWithMetadataPtr> const & i, cinema_kdms) {
                boost::filesystem::path path = directory;
-               name_values['c'] = i.cinema->name;
                path /= container_name_format.get(name_values, ".zip");
                if (!boost::filesystem::exists (path) || confirm_overwrite (path)) {
                        if (boost::filesystem::exists (path)) {
                                /* Creating a new zip file over an existing one is an error */
                                boost::filesystem::remove (path);
                        }
-                       i.make_zip_file (path, filename_format, name_values);
-                       written += i.screen_kdms.size();
+                       make_zip_file (i, path, filename_format, name_values);
+                       written += i.size();
                }
        }
 
@@ -187,8 +162,8 @@ CinemaKDMs::write_zip_files (
  *  @param cpl_name Name of the CPL that the KDMs are for.
  */
 void
-CinemaKDMs::email (
-       list<CinemaKDMs> cinema_kdms,
+email (
+       list<list<KDMWithMetadataPtr> > cinema_kdms,
        dcp::NameFormat container_name_format,
        dcp::NameFormat filename_format,
        dcp::NameFormat::Map name_values,
@@ -204,38 +179,39 @@ CinemaKDMs::email (
        /* No specific screen */
        name_values['s'] = "";
 
-       BOOST_FOREACH (CinemaKDMs const & i, cinema_kdms) {
+       BOOST_FOREACH (list<KDMWithMetadataPtr> const & i, cinema_kdms) {
 
-               if (i.cinema->emails.empty()) {
+               if (i.front()->cinema()->emails.empty()) {
                        continue;
                }
 
-               name_values['c'] = i.cinema->name;
-
                boost::filesystem::path zip_file = boost::filesystem::temp_directory_path() / boost::filesystem::unique_path();
                boost::filesystem::create_directories (zip_file);
                zip_file /= container_name_format.get(name_values, ".zip");
-               i.make_zip_file (zip_file, filename_format, name_values);
+               make_zip_file (i, zip_file, filename_format, name_values);
 
                string subject = config->kdm_subject();
                boost::algorithm::replace_all (subject, "$CPL_NAME", cpl_name);
                boost::algorithm::replace_all (subject, "$START_TIME", name_values['b']);
                boost::algorithm::replace_all (subject, "$END_TIME", name_values['e']);
-               boost::algorithm::replace_all (subject, "$CINEMA_NAME", i.cinema->name);
+               boost::algorithm::replace_all (subject, "$CINEMA_NAME", i.front()->cinema()->name);
 
                string body = config->kdm_email().c_str();
                boost::algorithm::replace_all (body, "$CPL_NAME", cpl_name);
                boost::algorithm::replace_all (body, "$START_TIME", name_values['b']);
                boost::algorithm::replace_all (body, "$END_TIME", name_values['e']);
-               boost::algorithm::replace_all (body, "$CINEMA_NAME", i.cinema->name);
+               boost::algorithm::replace_all (body, "$CINEMA_NAME", i.front()->cinema()->name);
 
                string screens;
-               BOOST_FOREACH (ScreenKDM const & j, i.screen_kdms) {
-                       screens += j.screen->name + ", ";
+               BOOST_FOREACH (KDMWithMetadataPtr j, i) {
+                       optional<string> screen_name = j->get('n');
+                       if (screen_name) {
+                               screens += *screen_name + ", ";
+                       }
                }
                boost::algorithm::replace_all (body, "$SCREENS", screens.substr (0, screens.length() - 2));
 
-               Emailer email (config->kdm_from(), i.cinema->emails, subject, body);
+               Emailer email (config->kdm_from(), i.front()->cinema()->emails, subject, body);
 
                BOOST_FOREACH (string i, config->kdm_cc()) {
                        email.add_cc (i);