Switch "command-like" options --list-cinemas and --list-dkdm-cpls to actual commands.
[dcpomatic.git] / src / lib / kdm_cli.cc
index 1ad96192873eaccb3d567c0a4349737211a5e408..f63fd2a550d30275c1a3ed16e468ddf79efb12d2 100644 (file)
@@ -27,7 +27,7 @@
 #include "cinema.h"
 #include "config.h"
 #include "dkdm_wrapper.h"
-#include "emailer.h"
+#include "email.h"
 #include "exceptions.h"
 #include "film.h"
 #include "kdm_with_metadata.h"
@@ -57,13 +57,15 @@ using namespace dcpomatic;
 static void
 help (std::function<void (string)> out)
 {
-       out (String::compose("Syntax: %1 [OPTION] <FILM|CPL-ID|DKDM>", program_name));
+       out (String::compose("Syntax: %1 [OPTION] [COMMAND] <FILM|CPL-ID|DKDM>", program_name));
+       out ("Commands:");
+       out ("create          create KDMs; default if no other command is specified");
+       out ("list-cinemas    list known cinemas from DCP-o-matic settings");
+       out ("list-dkdm-cpls  list CPLs for which DCP-o-matic has DKDMs");
        out ("  -h, --help                               show this help");
        out ("  -o, --output <path>                      output file or directory");
        out ("  -K, --filename-format <format>           filename format for KDMs");
        out ("  -Z, --container-name-format <format>     filename format for ZIP containers");
-       out ("  -f, --valid-from <time>                  valid from time (e.g. \"2013-09-2 01:41:51\") or \"now\"");
-       out ("  -t, --valid-to <time>                    valid to time (in local time zone of the cinema) (e.g. \"2014-09-28 01:41:51\")");
        out ("  -f, --valid-from <time>                  valid from time (e.g. \"2013-09-28T01:41:51+04:00\", \"2018-01-01T12:00:30\") or \"now\"");
        out ("  -t, --valid-to <time>                    valid to time (e.g. \"2014-09-28T01:41:51\")");
        out ("  -d, --valid-duration <duration>          valid duration (e.g. \"1 day\", \"4 hours\", \"2 weeks\")");
@@ -75,11 +77,11 @@ help (std::function<void (string)> out)
        out ("  -v, --verbose                            be verbose");
        out ("  -c, --cinema <name|email>                cinema name (when using -C) or name/email (to filter cinemas)");
        out ("  -S, --screen <name>                      screen name (when using -C) or screen name (to filter screens when using -c)");
-       out ("  -C, --certificate <file>                 file containing projector certificate");
-       out ("  -T, --trusted-device <file>              file containing a trusted device's certificate");
+       out ("  -C, --projector-certificate <file>       file containing projector certificate");
+       out ("  -T, --trusted-device-certificate <file>  file containing a trusted device's certificate");
+       out ("      --decryption-key <file>              file containing the private key which can decrypt the given DKDM");
+       out ("                                           (DCP-o-matic's configured private key will be used otherwise)");
        out ("      --cinemas-file <file>                use the given file as a list of cinemas instead of the current configuration");
-       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 ("");
@@ -247,7 +249,6 @@ from_film (
                        }
                }
 
-
                if (find_if(
                        period_checks.begin(),
                        period_checks.end(),
@@ -450,15 +451,15 @@ try
        auto filename_format = Config::instance()->kdm_filename_format();
        optional<string> cinema_name;
        shared_ptr<Cinema> cinema;
-       optional<boost::filesystem::path> certificate;
+       optional<boost::filesystem::path> projector_certificate;
+       optional<boost::filesystem::path> decryption_key;
        optional<string> screen;
        vector<shared_ptr<Screen>> screens;
        optional<dcp::EncryptedKDM> dkdm;
        optional<dcp::LocalTime> valid_from;
        optional<dcp::LocalTime> valid_to;
        bool zip = false;
-       bool list_cinemas = false;
-       bool list_dkdm_cpls = false;
+       string command = "create";
        optional<string> duration_string;
        bool verbose = false;
        dcp::Formulation formulation = dcp::Formulation::MODIFIED_TRANSITIONAL_1;
@@ -490,15 +491,14 @@ try
                        { "verbose", no_argument, 0, 'v' },
                        { "cinema", required_argument, 0, 'c' },
                        { "screen", required_argument, 0, 'S' },
-                       { "certificate", required_argument, 0, 'C' },
-                       { "trusted-device", required_argument, 0, 'T' },
-                       { "list-cinemas", no_argument, 0, 'B' },
-                       { "list-dkdm-cpls", no_argument, 0, 'D' },
+                       { "projector-certificate", required_argument, 0, 'C' },
+                       { "trusted-device-certificate", required_argument, 0, 'T' },
+                       { "decryption-key", required_argument, 0, 'G' },
                        { "cinemas-file", required_argument, 0, 'E' },
                        { 0, 0, 0, 0 }
                };
 
-               int c = getopt_long (argc, argv, "ho:K:Z:f:t:d:F:pae::zvc:S:C:T:BDE:", long_options, &option_index);
+               int c = getopt_long (argc, argv, "ho:K:Z:f:t:d:F:pae::zvc:S:C:T:E:G:", long_options, &option_index);
 
                if (c == -1) {
                        break;
@@ -574,7 +574,7 @@ try
                        screen = optarg;
                        break;
                case 'C':
-                       certificate = optarg;
+                       projector_certificate = optarg;
                        break;
                case 'T':
                        /* A trusted device ends up in the last screen we made */
@@ -582,11 +582,8 @@ try
                                screens.back()->trusted_devices.push_back(TrustedDevice(dcp::Certificate(dcp::file_to_string(optarg))));
                        }
                        break;
-               case 'B':
-                       list_cinemas = true;
-                       break;
-               case 'D':
-                       list_dkdm_cpls = true;
+               case 'G':
+                       decryption_key = optarg;
                        break;
                case 'E':
                        cinemas_file = optarg;
@@ -594,13 +591,33 @@ try
                }
        }
 
+       vector<string> commands = {
+               "create",
+               "list-cinemas",
+               "list-dkdm-cpls"
+       };
+
+       if (optind < argc - 1) {
+               /* Command with some KDM / CPL / whever specified afterwards */
+               command = argv[optind++];
+       } else if (optind < argc) {
+               /* Look for a valid command, hoping that it's not the name of the KDM / CPL / whatever */
+               if (std::find(commands.begin(), commands.end(), argv[optind]) != commands.end()) {
+                       command = argv[optind];
+               }
+       }
+
+       if (std::find(commands.begin(), commands.end(), command) == commands.end()) {
+               throw KDMCLIError(String::compose("Unrecognised command %1", command));
+       }
+
        if (cinemas_file) {
                Config::instance()->set_cinemas_file(*cinemas_file);
        }
 
-       if (certificate) {
+       if (projector_certificate) {
                /* Make a new screen and add it to the current cinema */
-               dcp::CertificateChain chain(dcp::file_to_string(*certificate));
+               dcp::CertificateChain chain(dcp::file_to_string(*projector_certificate));
                auto screen_to_add = std::make_shared<Screen>(screen.get_value_or(""), "", chain.leaf(), boost::none, vector<TrustedDevice>());
                if (cinema) {
                        cinema->add_screen(screen_to_add);
@@ -608,15 +625,15 @@ try
                screens.push_back(screen_to_add);
        }
 
-       if (list_cinemas) {
+       if (command == "list-cinemas") {
                auto cinemas = Config::instance()->cinemas ();
                for (auto i: cinemas) {
-                       out (String::compose("%1 (%2)", i->name, Emailer::address_list (i->emails)));
+                       out (String::compose("%1 (%2)", i->name, Email::address_list(i->emails)));
                }
                return {};
        }
 
-       if (list_dkdm_cpls) {
+       if (command == "list-dkdm-cpls") {
                dump_dkdm_group (Config::instance()->dkdms(), 0, out);
                return {};
        }
@@ -682,9 +699,11 @@ try
                        throw KDMCLIError ("could not find film or CPL ID corresponding to " + thing);
                }
 
+               string const key = decryption_key ? dcp::file_to_string(*decryption_key) : Config::instance()->decryption_chain()->key().get();
+
                from_dkdm (
                        screens,
-                       dcp::DecryptedKDM (*dkdm, Config::instance()->decryption_chain()->key().get()),
+                       dcp::DecryptedKDM(*dkdm, key),
                        verbose,
                        output,
                        container_name_format,