X-Git-Url: https://git.carlh.net/gitweb/?a=blobdiff_plain;f=src%2Flib%2Fkdm_cli.cc;h=a6656fa0e52ba736a5190447839d05bb8bf5506a;hb=ff639b3cf30afcc097bfd21d39c8d15f466cadd6;hp=d7b81a135170c21cb387ffa9f6f8f60a3df6e5f8;hpb=8af4df634d45f50e06d3e1c32cf071d54e34b6f8;p=dcpomatic.git diff --git a/src/lib/kdm_cli.cc b/src/lib/kdm_cli.cc index d7b81a135..a6656fa0e 100644 --- a/src/lib/kdm_cli.cc +++ b/src/lib/kdm_cli.cc @@ -36,11 +36,8 @@ #include #include #include -#include -using std::cerr; -using std::cout; using std::dynamic_pointer_cast; using std::list; using std::make_shared; @@ -57,33 +54,37 @@ using namespace dcpomatic; static void -help () +help (std::function out) { - cerr << "Syntax: " << program_name << " [OPTION] \n" - " -h, --help show this help\n" - " -o, --output output file or directory\n" - " -K, --filename-format filename format for KDMs\n" - " -Z, --container-name-format filename format for ZIP containers\n" - " -f, --valid-from valid from time (in local time zone of the cinema) (e.g. \"2013-09-28 01:41:51\") or \"now\"\n" - " -t, --valid-to valid to time (in local time zone of the cinema) (e.g. \"2014-09-28 01:41:51\")\n" - " -d, --valid-duration valid duration (e.g. \"1 day\", \"4 hours\", \"2 weeks\")\n" - " -F, --formulation modified-transitional-1, multiple-modified-transitional-1, dci-any or dci-specific [default modified-transitional-1]\n" - " -p, --disable-forensic-marking-picture disable forensic marking of pictures essences\n" - " -a, --disable-forensic-marking-audio disable forensic marking of audio essences (optionally above a given channel, e.g 12)\n" - " -e, --email email KDMs to cinemas\n" - " -z, --zip ZIP each cinema's KDMs into its own file\n" - " -v, --verbose be verbose\n" - " -c, --cinema specify a cinema, either by name or email address\n" - " -S, --screen screen description\n" - " -C, --certificate file containing projector certificate\n" - " -T, --trusted-device file containing a trusted device's certificate\n" - " --list-cinemas list known cinemas from the DCP-o-matic settings\n" - " --list-dkdm-cpls list CPLs for which DCP-o-matic has DKDMs\n\n" - "CPL-ID must be the ID of a CPL that is mentioned in DCP-o-matic's DKDM list.\n\n" - "For example:\n\n" - "Create KDMs for my_great_movie to play in all of Fred's Cinema's screens for the next two weeks and zip them up.\n" - "(Fred's Cinema must have been set up in DCP-o-matic's KDM window)\n\n" - "\t" << program_name << " -c \"Fred's Cinema\" -f now -d \"2 weeks\" -z my_great_movie\n\n"; + out (String::compose("Syntax: %1 [OPTION] ", program_name)); + out (" -h, --help show this help"); + out (" -o, --output output file or directory"); + out (" -K, --filename-format filename format for KDMs"); + out (" -Z, --container-name-format filename format for ZIP containers"); + out (" -f, --valid-from valid from time (in local time zone of the cinema) (e.g. \"2013-09-28 01:41:51\") or \"now\""); + out (" -t, --valid-to valid to time (in local time zone of the cinema) (e.g. \"2014-09-28 01:41:51\")"); + out (" -d, --valid-duration valid duration (e.g. \"1 day\", \"4 hours\", \"2 weeks\")"); + out (" -F, --formulation modified-transitional-1, multiple-modified-transitional-1, dci-any or dci-specific [default modified-transitional-1]"); + out (" -p, --disable-forensic-marking-picture disable forensic marking of pictures essences"); + out (" -a, --disable-forensic-marking-audio disable forensic marking of audio essences (optionally above a given channel, e.g 12)"); + out (" -e, --email email KDMs to cinemas"); + out (" -z, --zip ZIP each cinema's KDMs into its own file"); + out (" -v, --verbose be verbose"); + out (" -c, --cinema cinema name (when using -C) or name/email (to filter cinemas)"); + out (" -S, --screen screen name (when using -C) or screen name (to filter screens when using -c)"); + out (" -C, --certificate file containing projector certificate"); + out (" -T, --trusted-device file containing a trusted device's certificate"); + out (" --list-cinemas list known cinemas from the DCP-o-matic settings"); + out (" --list-dkdm-cpls list CPLs for which DCP-o-matic has DKDMs"); + out (""); + out ("CPL-ID must be the ID of a CPL that is mentioned in DCP-o-matic's DKDM list."); + out (""); + out ("For example:"); + out (""); + out ("Create KDMs for my_great_movie to play in all of Fred's Cinema's screens for the next two weeks and zip them up."); + out ("(Fred's Cinema must have been set up in DCP-o-matic's KDM window)"); + out (""); + out (String::compose("\t%1 -c \"Fred's Cinema\" -f now -d \"2 weeks\" -z my_great_movie", program_name)); } @@ -148,7 +149,8 @@ write_files ( boost::filesystem::path output, dcp::NameFormat container_name_format, dcp::NameFormat filename_format, - bool verbose + bool verbose, + std::function out ) { if (zip) { @@ -161,7 +163,7 @@ write_files ( ); if (verbose) { - cout << "Wrote " << N << " ZIP files to " << output << "\n"; + out (String::compose("Wrote %1 ZIP files to %2", N, output)); } } else { int const N = write_files ( @@ -170,7 +172,7 @@ write_files ( ); if (verbose) { - cout << "Wrote " << N << " KDM files to " << output << "\n"; + out (String::compose("Wrote %1 KDM files to %2", N, output)); } } } @@ -213,7 +215,8 @@ from_film ( bool disable_forensic_marking_picture, optional disable_forensic_marking_audio, bool email, - bool zip + bool zip, + std::function out ) { shared_ptr film; @@ -221,7 +224,7 @@ from_film ( film = make_shared(film_dir); film->read_metadata (); if (verbose) { - cout << "Read film " << film->name () << "\n"; + out (String::compose("Read film %1", film->name())); } } catch (std::exception& e) { throw KDMCLIError (String::compose("error reading film \"%1\" (%2)", film_dir.string(), e.what())); @@ -245,9 +248,9 @@ from_film ( kdms.push_back (p); } } - write_files (kdms, zip, output, container_name_format, filename_format, verbose); + write_files (kdms, zip, output, container_name_format, filename_format, verbose, out); if (email) { - send_emails ({kdms}, container_name_format, filename_format, film->dcp_name()); + send_emails ({kdms}, container_name_format, filename_format, film->dcp_name(), {}); } } catch (FileError& e) { throw KDMCLIError (String::compose("%1 (%2)", e.what(), e.file().string())); @@ -338,7 +341,8 @@ from_dkdm ( bool disable_forensic_marking_picture, optional disable_forensic_marking_audio, bool email, - bool zip + bool zip, + std::function out ) { dcp::NameFormat::Map values; @@ -350,8 +354,11 @@ from_dkdm ( continue; } - dcp::LocalTime begin(valid_from, i->cinema->utc_offset_hour(), i->cinema->utc_offset_minute()); - dcp::LocalTime end(valid_to, i->cinema->utc_offset_hour(), i->cinema->utc_offset_minute()); + int const offset_hour = i->cinema ? i->cinema->utc_offset_hour() : 0; + int const offset_minute = i->cinema ? i->cinema->utc_offset_minute() : 0; + + dcp::LocalTime begin(valid_from, offset_hour, offset_minute); + dcp::LocalTime end(valid_to, offset_hour, offset_minute); auto const kdm = kdm_from_dkdm( dkdm, @@ -365,18 +372,18 @@ from_dkdm ( ); dcp::NameFormat::Map name_values; - name_values['c'] = i->cinema->name; + name_values['c'] = i->cinema ? i->cinema->name : ""; name_values['s'] = i->name; - name_values['f'] = dkdm.annotation_text().get_value_or(""); + name_values['f'] = kdm.content_title_text(); name_values['b'] = begin.date() + " " + begin.time_of_day(true, false); name_values['e'] = end.date() + " " + end.time_of_day(true, false); name_values['i'] = kdm.cpl_id(); - kdms.push_back (make_shared(name_values, i->cinema.get(), i->cinema->emails, kdm)); + kdms.push_back (make_shared(name_values, i->cinema.get(), i->cinema ? i->cinema->emails : list(), kdm)); } - write_files (kdms, zip, output, container_name_format, filename_format, verbose); + write_files (kdms, zip, output, container_name_format, filename_format, verbose, out); if (email) { - send_emails ({kdms}, container_name_format, filename_format, dkdm.annotation_text().get_value_or("")); + send_emails ({kdms}, container_name_format, filename_format, dkdm.annotation_text().get_value_or(""), {}); } } catch (FileError& e) { throw KDMCLIError (String::compose("%1 (%2)", e.what(), e.file().string())); @@ -386,40 +393,37 @@ from_dkdm ( static void -dump_dkdm_group (shared_ptr group, int indent) +dump_dkdm_group (shared_ptr group, int indent, std::function out) { + auto const indent_string = string(indent, ' '); + if (indent > 0) { - for (int i = 0; i < indent; ++i) { - cout << " "; - } - cout << group->name() << "\n"; + out (indent_string + group->name()); } for (auto i: group->children()) { auto g = dynamic_pointer_cast(i); if (g) { - dump_dkdm_group (g, indent + 2); + dump_dkdm_group (g, indent + 2, out); } else { - for (int j = 0; j < indent; ++j) { - cout << " "; - } auto d = dynamic_pointer_cast(i); assert(d); - cout << d->dkdm().cpl_id() << "\n"; + out (indent_string + d->dkdm().cpl_id()); } } } optional -kdm_cli (int argc, char* argv[]) +kdm_cli (int argc, char* argv[], std::function out) try { - boost::filesystem::path output = "."; + boost::filesystem::path output = boost::filesystem::current_path(); auto container_name_format = Config::instance()->kdm_container_name_format(); auto filename_format = Config::instance()->kdm_filename_format(); optional cinema_name; shared_ptr cinema; - string screen_description; + optional certificate; + optional screen; list> screens; optional dkdm; optional valid_from; @@ -436,6 +440,9 @@ try program_name = argv[0]; + /* Reset getopt() so we can call this method several times in one test process */ + optind = 1; + int option_index = 0; while (true) { static struct option long_options[] = { @@ -469,7 +476,7 @@ try switch (c) { case 'h': - help (); + help (out); exit (EXIT_SUCCESS); case 'o': output = optarg; @@ -531,19 +538,14 @@ try cinema = make_shared(optarg, list(), "", 0, 0); break; case 'S': - screen_description = optarg; + /* Similarly, this could be the name of a new (temporary) screen or the name of a screen + * to search for. + */ + screen = optarg; break; case 'C': - { - /* Make a new screen and add it to the current cinema */ - dcp::CertificateChain chain (dcp::file_to_string(optarg)); - auto screen = make_shared(screen_description, "", chain.leaf(), vector()); - if (cinema) { - cinema->add_screen (screen); - } - screens.push_back (screen); + certificate = optarg; break; - } case 'T': /* A trusted device ends up in the last screen we made */ if (!screens.empty ()) { @@ -559,16 +561,26 @@ try } } + if (certificate) { + /* Make a new screen and add it to the current cinema */ + dcp::CertificateChain chain(dcp::file_to_string(*certificate)); + auto screen_to_add = std::make_shared(screen.get_value_or(""), "", chain.leaf(), boost::none, vector()); + if (cinema) { + cinema->add_screen(screen_to_add); + } + screens.push_back(screen_to_add); + } + if (list_cinemas) { auto cinemas = Config::instance()->cinemas (); for (auto i: cinemas) { - cout << i->name << " (" << Emailer::address_list (i->emails) << ")\n"; + out (String::compose("%1 (%2)", i->name, Emailer::address_list (i->emails))); } exit (EXIT_SUCCESS); } if (list_dkdm_cpls) { - dump_dkdm_group (Config::instance()->dkdms(), 0); + dump_dkdm_group (Config::instance()->dkdms(), 0, out); exit (EXIT_SUCCESS); } @@ -581,8 +593,7 @@ try } if (optind >= argc) { - help (); - exit (EXIT_FAILURE); + throw KDMCLIError ("no film, CPL ID or DKDM specified"); } if (screens.empty()) { @@ -591,17 +602,17 @@ try } screens = find_cinema (*cinema_name)->screens (); + if (screen) { + screens.erase(std::remove_if(screens.begin(), screens.end(), [&screen](shared_ptr s) { return s->name != *screen; }), screens.end()); + } } if (duration_string) { valid_to = valid_from.get() + duration_from_string (*duration_string); } - dcpomatic_setup_path_encoding (); - dcpomatic_setup (); - if (verbose) { - cout << "Making KDMs valid from " << valid_from.get() << " to " << valid_to.get() << "\n"; + out (String::compose("Making KDMs valid from %1 to %2", boost::posix_time::to_simple_string(valid_from.get()), boost::posix_time::to_simple_string(valid_to.get()))); } string const thing = argv[optind]; @@ -619,7 +630,8 @@ try disable_forensic_marking_picture, disable_forensic_marking_audio, email, - zip + zip, + out ); } else { if (boost::filesystem::is_regular_file(thing)) { @@ -645,7 +657,8 @@ try disable_forensic_marking_picture, disable_forensic_marking_audio, email, - zip + zip, + out ); }