X-Git-Url: https://git.carlh.net/gitweb/?a=blobdiff_plain;f=src%2Ftools%2Fdcpomatic_kdm.cc;h=3f156f4b9ece9fbdbe131cfdc69ed2c131472966;hb=bab59a964fe7fe6dabf1fb3df95ed5d7ec8df88a;hp=6257d60af3f636e2a25eaf3464eebafa993c5ee3;hpb=cb990adba9c57e5107ef2aa9716cf0a26c1df83d;p=dcpomatic.git diff --git a/src/tools/dcpomatic_kdm.cc b/src/tools/dcpomatic_kdm.cc index 6257d60af..5393a7fd1 100644 --- a/src/tools/dcpomatic_kdm.cc +++ b/src/tools/dcpomatic_kdm.cc @@ -1,302 +1,921 @@ /* - Copyright (C) 2013-2014 Carl Hetherington + Copyright (C) 2015-2021 Carl Hetherington - This program is free software; you can redistribute it and/or modify + This file is part of DCP-o-matic. + + DCP-o-matic is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation; either version 2 of the License, or (at your option) any later version. - This program is distributed in the hope that it will be useful, + DCP-o-matic is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details. You should have received a copy of the GNU General Public License - along with this program; if not, write to the Free Software - Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. + along with DCP-o-matic. If not, see . */ -#include -#include -#include "lib/film.h" + +#include "wx/about_dialog.h" +#include "wx/dcpomatic_button.h" +#include "wx/editable_list.h" +#include "wx/file_dialog.h" +#include "wx/file_picker_ctrl.h" +#include "wx/full_config_dialog.h" +#include "wx/job_view_dialog.h" +#include "wx/kdm_output_panel.h" +#include "wx/kdm_timing_panel.h" +#include "wx/nag_dialog.h" +#include "wx/new_dkdm_folder_dialog.h" +#include "wx/report_problem_dialog.h" +#include "wx/screens_panel.h" +#include "wx/static_text.h" +#include "wx/wx_signal_manager.h" +#include "wx/wx_util.h" #include "lib/cinema.h" -#include "lib/kdm.h" +#include "lib/collator.h" +#include "lib/compose.hpp" +#include "lib/constants.h" #include "lib/config.h" +#include "lib/cross.h" +#include "lib/dcpomatic_log.h" +#include "lib/dkdm_wrapper.h" #include "lib/exceptions.h" -#include "lib/safe_stringstream.h" - -using std::string; -using std::cout; -using std::cerr; +#include "lib/file_log.h" +#include "lib/job_manager.h" +#include "lib/kdm_util.h" +#include "lib/kdm_with_metadata.h" +#include "lib/screen.h" +#include "lib/send_kdm_email_job.h" +#include +#include +#include +#include +LIBDCP_DISABLE_WARNINGS +#include +#include +#include +#include +#include +#include +#include +LIBDCP_ENABLE_WARNINGS +#ifdef __WXOSX__ +#include +#endif +#include +#include + +#ifdef check +#undef check +#endif + + +using std::dynamic_pointer_cast; +using std::exception; using std::list; +using std::make_shared; +using std::map; +using std::pair; +using std::shared_ptr; +using std::string; +using std::unordered_set; using std::vector; -using boost::shared_ptr; +using boost::bind; +using boost::optional; +using boost::ref; +#if BOOST_VERSION >= 106100 +using namespace boost::placeholders; +#endif +using namespace dcpomatic; -static string program_name; -static void -help () -{ - cerr << "Syntax: " << program_name << " [OPTION] []\n" - " -h, --help show this help\n" - " -o, --output output file or directory\n" - " -f, --valid-from valid from time (in local time zone) (e.g. \"2013-09-28 01:41:51\") or \"now\"\n" - " -t, --valid-to valid to time (in local time zone) (e.g. \"2014-09-28 01:41:51\")\n" - " -d, --valid-duration valid duration (e.g. \"1 day\", \"4 hours\", \"2 weeks\")\n" - " --formulation modified-transitional-1, dci-any or dci-specific [default modified-transitional-1]\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" - " --cinemas list known cinemas from the DCP-o-matic settings\n" - " --certificate file containing projector certificate\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" - "\tdcpomatic_kdm -c \"Fred's Cinema\" -f now -d \"2 weeks\" -z my_great_movie\n\n"; -} - -static void -error (string m) -{ - cerr << program_name << ": " << m << "\n"; - exit (EXIT_FAILURE); -} +enum { + ID_help_report_a_problem = 1, +}; + -static boost::posix_time::ptime -time_from_string (string t) +class DOMFrame : public wxFrame { - if (t == "now") { - return boost::posix_time::second_clock::local_time (); +public: + explicit DOMFrame (wxString const & title) + : wxFrame (nullptr, -1, title) + , _config_dialog (nullptr) + , _job_view (nullptr) + { +#if defined(DCPOMATIC_WINDOWS) + if (Config::instance()->win32_console ()) { + AllocConsole(); + + HANDLE handle_out = GetStdHandle(STD_OUTPUT_HANDLE); + int hCrt = _open_osfhandle((intptr_t) handle_out, _O_TEXT); + FILE* hf_out = _fdopen(hCrt, "w"); + setvbuf(hf_out, NULL, _IONBF, 1); + *stdout = *hf_out; + + HANDLE handle_in = GetStdHandle(STD_INPUT_HANDLE); + hCrt = _open_osfhandle((intptr_t) handle_in, _O_TEXT); + FILE* hf_in = _fdopen(hCrt, "r"); + setvbuf(hf_in, NULL, _IONBF, 128); + *stdin = *hf_in; + + std::cout << "DCP-o-matic KDM creator is starting." << "\n"; + } +#endif + + auto bar = new wxMenuBar; + setup_menu (bar); + SetMenuBar (bar); + + Bind (wxEVT_MENU, boost::bind (&DOMFrame::file_exit, this), wxID_EXIT); + Bind (wxEVT_MENU, boost::bind (&DOMFrame::edit_preferences, this), wxID_PREFERENCES); + Bind (wxEVT_MENU, boost::bind (&DOMFrame::help_about, this), wxID_ABOUT); + Bind (wxEVT_MENU, boost::bind (&DOMFrame::help_report_a_problem, this), ID_help_report_a_problem); + + /* Use a panel as the only child of the Frame so that we avoid + the dark-grey background on Windows. + */ + auto overall_panel = new wxPanel (this, wxID_ANY); + auto main_sizer = new wxBoxSizer (wxHORIZONTAL); + + auto horizontal = new wxBoxSizer (wxHORIZONTAL); + auto left = new wxBoxSizer (wxVERTICAL); + auto right = new wxBoxSizer (wxVERTICAL); + + horizontal->Add (left, 1, wxEXPAND | wxRIGHT, DCPOMATIC_SIZER_X_GAP * 2); + horizontal->Add (right, 1, wxEXPAND); + + wxFont subheading_font (*wxNORMAL_FONT); + subheading_font.SetWeight (wxFONTWEIGHT_BOLD); + + auto h = new StaticText (overall_panel, _("Screens")); + h->SetFont (subheading_font); + left->Add (h, 0, wxBOTTOM, DCPOMATIC_SIZER_Y_GAP); + _screens = new ScreensPanel (overall_panel); + left->Add (_screens, 1, wxEXPAND | wxBOTTOM, DCPOMATIC_SIZER_Y_GAP); + + /// TRANSLATORS: translate the word "Timing" here; do not include the "KDM|" prefix + h = new StaticText (overall_panel, S_("KDM|Timing")); + h->SetFont (subheading_font); + right->Add (h); + _timing = new KDMTimingPanel (overall_panel); + right->Add (_timing, 0, wxALL, DCPOMATIC_SIZER_Y_GAP); + + h = new StaticText (overall_panel, _("DKDM")); + h->SetFont (subheading_font); + right->Add(h, 0, wxTOP, DCPOMATIC_SUBHEADING_TOP_PAD); + + _dkdm_search = new wxSearchCtrl(overall_panel, wxID_ANY, wxEmptyString, wxDefaultPosition, wxSize(200, search_ctrl_height())); +#ifndef __WXGTK3__ + /* The cancel button seems to be strangely broken in GTK3; clicking on it twice sometimes works */ + _dkdm_search->ShowCancelButton (true); +#endif + + right->Add(_dkdm_search, 0, wxTOP | wxBOTTOM, DCPOMATIC_SIZER_Y_GAP); + + class DKDMDropTarget : public wxFileDropTarget + { + public: + DKDMDropTarget(DOMFrame* frame) + : _frame(frame) + {} + + bool OnDropFiles(wxCoord, wxCoord, wxArrayString const& filenames) override + { + for (size_t i = 0; i < filenames.GetCount(); ++i) { + _frame->add_dkdm(boost::filesystem::path(wx_to_std(filenames[0]))); + } + + return true; + } + + private: + DOMFrame* _frame; + }; + + auto dkdm_sizer = new wxBoxSizer (wxHORIZONTAL); + _dkdm = new wxTreeCtrl ( + overall_panel, wxID_ANY, wxDefaultPosition, wxDefaultSize, wxTR_HIDE_ROOT | wxTR_HAS_BUTTONS | wxTR_LINES_AT_ROOT + ); + _dkdm->SetDropTarget(new DKDMDropTarget(this)); + dkdm_sizer->Add(_dkdm, 1, wxEXPAND | wxBOTTOM, DCPOMATIC_SIZER_Y_GAP); + auto dkdm_buttons = new wxBoxSizer(wxVERTICAL); + _add_dkdm = new Button (overall_panel, _("Add...")); + dkdm_buttons->Add (_add_dkdm, 0, wxALL | wxEXPAND, DCPOMATIC_BUTTON_STACK_GAP); + _add_dkdm_folder = new Button (overall_panel, _("Add folder...")); + dkdm_buttons->Add (_add_dkdm_folder, 0, wxALL | wxEXPAND, DCPOMATIC_BUTTON_STACK_GAP); + _remove_dkdm = new Button (overall_panel, _("Remove")); + dkdm_buttons->Add (_remove_dkdm, 0, wxALL | wxEXPAND, DCPOMATIC_BUTTON_STACK_GAP); + _export_dkdm = new Button (overall_panel, _("Export...")); + dkdm_buttons->Add (_export_dkdm, 0, wxALL | wxEXPAND, DCPOMATIC_BUTTON_STACK_GAP); + dkdm_sizer->Add (dkdm_buttons, 0, wxEXPAND | wxALL, DCPOMATIC_SIZER_GAP); + right->Add (dkdm_sizer, 1, wxEXPAND | wxBOTTOM, DCPOMATIC_SIZER_Y_GAP); + + update_dkdm_view(); + + h = new StaticText (overall_panel, _("Output")); + h->SetFont (subheading_font); + right->Add(h, 0, wxTOP, DCPOMATIC_SUBHEADING_TOP_PAD); + _output = new KDMOutputPanel (overall_panel); + right->Add (_output, 0, wxALL, DCPOMATIC_SIZER_Y_GAP); + + _create = new Button (overall_panel, _("Create KDMs")); + right->Add (_create, 0, wxALL, DCPOMATIC_SIZER_GAP); + + main_sizer->Add (horizontal, 1, wxALL | wxEXPAND, DCPOMATIC_DIALOG_BORDER); + overall_panel->SetSizer (main_sizer); + + /* Instantly save any config changes when using a DCP-o-matic GUI */ + Config::instance()->Changed.connect (boost::bind (&Config::write, Config::instance ())); + + _screens->ScreensChanged.connect (boost::bind (&DOMFrame::setup_sensitivity, this)); + _create->Bind (wxEVT_BUTTON, bind (&DOMFrame::create_kdms, this)); + _dkdm->Bind(wxEVT_TREE_SEL_CHANGED, boost::bind(&DOMFrame::dkdm_selection_changed, this)); + _dkdm->Bind (wxEVT_TREE_BEGIN_DRAG, boost::bind (&DOMFrame::dkdm_begin_drag, this, _1)); + _dkdm->Bind (wxEVT_TREE_END_DRAG, boost::bind (&DOMFrame::dkdm_end_drag, this, _1)); + _dkdm->Bind(wxEVT_TREE_ITEM_EXPANDED, boost::bind(&DOMFrame::dkdm_expanded, this, _1)); + _dkdm->Bind(wxEVT_TREE_ITEM_COLLAPSED, boost::bind(&DOMFrame::dkdm_collapsed, this, _1)); + _add_dkdm->Bind (wxEVT_BUTTON, bind (&DOMFrame::add_dkdm_clicked, this)); + _add_dkdm_folder->Bind (wxEVT_BUTTON, bind (&DOMFrame::add_dkdm_folder_clicked, this)); + _remove_dkdm->Bind (wxEVT_BUTTON, bind (&DOMFrame::remove_dkdm_clicked, this)); + _export_dkdm->Bind (wxEVT_BUTTON, bind (&DOMFrame::export_dkdm_clicked, this)); + _dkdm_search->Bind(wxEVT_TEXT, boost::bind(&DOMFrame::dkdm_search_changed, this)); + + setup_sensitivity (); + + dcpomatic_log = make_shared(State::write_path("kdm.log")); } - return boost::posix_time::time_from_string (t); -} +private: + void file_exit () + { + /* false here allows the close handler to veto the close request */ + Close (false); + } -static boost::posix_time::time_duration -duration_from_string (string d) -{ - SafeStringStream s (d); - int N; - string unit; - s >> N >> unit; + void edit_preferences () + { + if (!_config_dialog) { + _config_dialog = create_full_config_dialog (); + } + _config_dialog->Show (this); + } - if (N == 0) { - cerr << "Could not understand duration \"" << d << "\"\n"; - exit (EXIT_FAILURE); + void help_about () + { + AboutDialog dialog(this); + dialog.ShowModal(); } - if (unit == "year" || unit == "years") { - return boost::posix_time::time_duration (N * 24 * 365, 0, 0, 0); - } else if (unit == "week" || unit == "weeks") { - return boost::posix_time::time_duration (N * 24 * 7, 0, 0, 0); - } else if (unit == "day" || unit == "days") { - return boost::posix_time::time_duration (N * 24, 0, 0, 0); - } else if (unit == "hour" || unit == "hours") { - return boost::posix_time::time_duration (N, 0, 0, 0); + void help_report_a_problem () + { + ReportProblemDialog dialog(this, shared_ptr()); + if (dialog.ShowModal() == wxID_OK) { + dialog.report(); + } } - cerr << "Could not understand duration \"" << d << "\"\n"; - exit (EXIT_FAILURE); -} + void setup_menu (wxMenuBar* m) + { + auto file = new wxMenu; + +#ifdef __WXOSX__ + file->Append (wxID_EXIT, _("&Exit")); +#else + file->Append (wxID_EXIT, _("&Quit")); +#endif + +#ifdef __WXOSX__ + file->Append (wxID_PREFERENCES, _("&Preferences...\tCtrl-P")); +#else + wxMenu* edit = new wxMenu; + edit->Append (wxID_PREFERENCES, _("&Preferences...\tCtrl-P")); +#endif + + wxMenu* help = new wxMenu; +#ifdef __WXOSX__ + help->Append (wxID_ABOUT, _("About DCP-o-matic")); +#else + help->Append (wxID_ABOUT, _("About")); +#endif + help->Append (ID_help_report_a_problem, _("Report a problem...")); + + m->Append (file, _("&File")); +#ifndef __WXOSX__ + m->Append (edit, _("&Edit")); +#endif + m->Append (help, _("&Help")); + } -int main (int argc, char* argv[]) -{ - boost::filesystem::path output; - boost::optional valid_from; - boost::optional valid_to; - string certificate_file; - bool zip = false; - string cinema_name; - bool cinemas = false; - string duration_string; - bool verbose = false; - dcp::Formulation formulation = dcp::MODIFIED_TRANSITIONAL_1; - - program_name = argv[0]; - - int option_index = 0; - while (true) { - static struct option long_options[] = { - { "help", no_argument, 0, 'h'}, - { "output", required_argument, 0, 'o'}, - { "valid-from", required_argument, 0, 'f'}, - { "valid-to", required_argument, 0, 't'}, - { "certificate", required_argument, 0, 'A' }, - { "cinema", required_argument, 0, 'c' }, - { "cinemas", no_argument, 0, 'B' }, - { "zip", no_argument, 0, 'z' }, - { "duration", required_argument, 0, 'd' }, - { "verbose", no_argument, 0, 'v' }, - { "formulation", required_argument, 0, 'C' }, - { 0, 0, 0, 0 } - }; + bool confirm_overwrite (boost::filesystem::path path) + { + if (boost::filesystem::is_directory(path)) { + return confirm_dialog ( + this, + wxString::Format(_("Folder %s already exists. Do you want to overwrite it?"), std_to_wx(path.string()).data()) + ); + } else { + return confirm_dialog ( + this, + wxString::Format(_("File %s already exists. Do you want to overwrite it?"), std_to_wx(path.string()).data()) + ); + } + } + + /** @id if not nullptr this is filled in with the wxTreeItemId of the selection */ + shared_ptr selected_dkdm (wxTreeItemId* id = nullptr) const + { + wxArrayTreeItemIds selections; + _dkdm->GetSelections (selections); + if (selections.GetCount() != 1) { + if (id) { + *id = 0; + } + return {}; + } + + if (id) { + *id = selections[0]; + } - int c = getopt_long (argc, argv, "ho:f:t:c:A:Bzd:vC:", long_options, &option_index); - - if (c == -1) { - break; - } - - switch (c) { - case 'h': - help (); - exit (EXIT_SUCCESS); - case 'o': - output = optarg; - break; - case 'f': - valid_from = time_from_string (optarg); - break; - case 't': - valid_to = time_from_string (optarg); - break; - case 'A': - certificate_file = optarg; - break; - case 'c': - cinema_name = optarg; - break; - case 'B': - cinemas = true; - break; - case 'z': - zip = true; - break; - case 'd': - duration_string = optarg; - break; - case 'v': - verbose = true; - break; - case 'C': - if (string (optarg) == "modified-transitional-1") { - formulation = dcp::MODIFIED_TRANSITIONAL_1; - } else if (string (optarg) == "dci-any") { - formulation = dcp::DCI_ANY; - } else if (string (optarg) == "dci-specific") { - formulation = dcp::DCI_SPECIFIC; + auto i = _dkdm_id.find (selections[0]); + if (i == _dkdm_id.end()) { + return {}; + } + + return i->second; + } + + void create_kdms () + { + try { + auto dkdm_base = selected_dkdm (); + if (!dkdm_base) { + return; + } + + list kdms; + string title; + + auto dkdm = std::dynamic_pointer_cast(dkdm_base); + if (!dkdm) { + return; + } + + /* Decrypt the DKDM */ + dcp::DecryptedKDM decrypted (dkdm->dkdm(), Config::instance()->decryption_chain()->key().get()); + title = decrypted.content_title_text (); + + /* This is the signer for our new KDMs */ + auto signer = Config::instance()->signer_chain (); + if (!signer->valid ()) { + throw InvalidSignerError (); + } + + vector period_checks; + + std::function make_kdm = [decrypted, title](dcp::LocalTime begin, dcp::LocalTime end) { + /* Make an empty KDM */ + dcp::DecryptedKDM kdm ( + begin, + end, + decrypted.annotation_text().get_value_or(""), + title, + dcp::LocalTime().as_string() + ); + + /* Add keys from the DKDM */ + for (auto const& j: decrypted.keys()) { + kdm.add_key(j); + } + + return kdm; + }; + + for (auto i: _screens->screens()) { + + auto kdm = kdm_for_screen( + make_kdm, + i, + _timing->from(), + _timing->until(), + _output->formulation(), + !_output->forensic_mark_video(), + _output->forensic_mark_audio() ? boost::optional() : 0, + period_checks + ); + + if (kdm) { + kdms.push_back(kdm); + } + } + + if (kdms.empty()) { + return; + } + + if (find(period_checks.begin(), period_checks.end(), KDMCertificatePeriod::KDM_OUTSIDE_CERTIFICATE) != period_checks.end()) { + error_dialog( + this, + _("Some KDMs would have validity periods which are completely outside the recipient certificate periods. Such KDMs are very unlikely to work, so will not be created.") + ); + return; + } + + if (find(period_checks.begin(), period_checks.end(), KDMCertificatePeriod::KDM_OVERLAPS_CERTIFICATE) != period_checks.end()) { + message_dialog( + this, + _("For some of these KDMs the recipient certificate's validity period will not cover the whole of the KDM validity period. This might cause problems with the KDMs.") + ); + } + + auto result = _output->make ( + kdms, title, bind (&DOMFrame::confirm_overwrite, this, _1) + ); + + if (result.first) { + JobManager::instance()->add (result.first); + if (_job_view) { + _job_view->Destroy (); + _job_view = 0; + } + _job_view = new JobViewDialog (this, _("Send KDM emails"), result.first); + _job_view->ShowModal (); + } + + if (result.second > 0) { + /* XXX: proper plural form support in wxWidgets? */ + wxString s = result.second == 1 ? _("%d KDM written to %s") : _("%d KDMs written to %s"); + message_dialog ( + this, + wxString::Format (s, result.second, std_to_wx(_output->directory().string()).data()) + ); + } + } catch (dcp::BadKDMDateError& e) { + if (e.starts_too_early()) { + error_dialog(this, _("The KDM start period is before (or close to) the start of the signing certificate's validity period. Use a later start time for this KDM.")); } else { - error ("unrecognised KDM formulation " + string (optarg)); + error_dialog(this, _("The KDM end period is after (or close to) the end of the signing certificates' validity period. Either use an earlier end time for this KDM or re-create your signing certificates in the DCP-o-matic preferences window.")); } + return; + } catch (dcp::NotEncryptedError& e) { + error_dialog (this, _("CPL's content is not encrypted.")); + } catch (exception& e) { + error_dialog (this, std_to_wx(e.what())); + } catch (...) { + error_dialog (this, _("An unknown exception occurred.")); } } - if (cinemas) { - list > cinemas = Config::instance()->cinemas (); - for (list >::const_iterator i = cinemas.begin(); i != cinemas.end(); ++i) { - cout << (*i)->name << " (" << (*i)->email << ")\n"; + void setup_sensitivity () + { + _screens->setup_sensitivity (); + _output->setup_sensitivity (); + wxArrayTreeItemIds sel; + _dkdm->GetSelections (sel); + auto group = dynamic_pointer_cast(selected_dkdm()); + auto dkdm = dynamic_pointer_cast(selected_dkdm()); + _create->Enable (!_screens->screens().empty() && sel.GetCount() > 0 && dkdm); + _remove_dkdm->Enable (sel.GetCount() > 0 && (!group || group->name() != "root")); + _export_dkdm->Enable (sel.GetCount() > 0 && dkdm); + } + + void dkdm_selection_changed() + { + _selected_dkdm = selected_dkdm(); + setup_sensitivity(); + } + + void dkdm_expanded(wxTreeEvent& ev) + { + if (_ignore_expand) { + return; + } + + auto iter = _dkdm_id.find(ev.GetItem()); + if (iter != _dkdm_id.end()) { + _expanded_dkdm_groups.insert(iter->second); } - exit (EXIT_SUCCESS); } - if (duration_string.empty() && !valid_to) { - error ("you must specify a --valid-duration or --valid-to"); + void dkdm_collapsed(wxTreeEvent& ev) + { + auto iter = _dkdm_id.find(ev.GetItem()); + if (iter != _dkdm_id.end()) { + _expanded_dkdm_groups.erase(iter->second); + } } - if (!valid_from) { - error ("you must specify --valid-from"); - exit (EXIT_FAILURE); + void dkdm_begin_drag (wxTreeEvent& ev) + { + ev.Allow (); } - if (optind >= argc) { - help (); - exit (EXIT_FAILURE); + void dkdm_end_drag (wxTreeEvent& ev) + { + auto from = _dkdm_id.find (_dkdm->GetSelection ()); + auto to = _dkdm_id.find (ev.GetItem ()); + if (from == _dkdm_id.end() || to == _dkdm_id.end() || from->first == to->first) { + return; + } + + auto group = dynamic_pointer_cast (to->second); + if (!group) { + group = to->second->parent(); + } + + DCPOMATIC_ASSERT (group); + DCPOMATIC_ASSERT (from->second->parent ()); + + from->second->parent()->remove (from->second); + add_dkdm(from->second, group, dynamic_pointer_cast(to->second)); + + update_dkdm_view(); } - if (cinema_name.empty() && certificate_file.empty()) { - error ("you must specify either a cinema, a screen or a certificate file"); + void add_dkdm_clicked () + { + FileDialog dialog(this, _("Select DKDM file"), wxT("XML files|*.xml|All files|*.*"), wxFD_MULTIPLE, "AddDKDMPath"); + if (!dialog.show()) { + return; + } + + for (auto path: dialog.paths()) { + add_dkdm(path); + } } - if (!duration_string.empty ()) { - valid_to = valid_from.get() + duration_from_string (duration_string); + void add_dkdm(boost::filesystem::path path) + { + auto chain = Config::instance()->decryption_chain(); + DCPOMATIC_ASSERT (chain->key()); + + try { + dcp::EncryptedKDM ekdm(dcp::file_to_string(path, MAX_KDM_SIZE)); + /* Decrypt the DKDM to make sure that we can */ + dcp::DecryptedKDM dkdm(ekdm, chain->key().get()); + + auto new_dkdm = make_shared(ekdm); + auto group = dynamic_pointer_cast (selected_dkdm()); + if (!group) { + group = Config::instance()->dkdms (); + } + add_dkdm(new_dkdm, group); + } catch (dcp::KDMFormatError& e) { + error_dialog ( + this, + _("Could not read file as a KDM. Perhaps it is badly formatted, or not a KDM at all."), + std_to_wx(e.what()) + ); + return; + } catch (dcp::KDMDecryptionError &) { + error_dialog ( + this, + _("Could not decrypt the DKDM. Perhaps it was not created with the correct certificate.") + ); + } catch (dcp::MiscError& e) { + error_dialog ( + this, + _("Could not read file as a KDM. It is much too large. Make sure you are loading a DKDM (XML) file."), + std_to_wx(e.what()) + ); + } + + update_dkdm_view(); } - string const film_dir = argv[optind]; - - dcpomatic_setup (); + void add_dkdm_folder_clicked () + { + NewDKDMFolderDialog dialog(this); + if (dialog.ShowModal() != wxID_OK) { + return; + } - shared_ptr film; - try { - film.reset (new Film (film_dir)); - film->read_metadata (); - if (verbose) { - cout << "Read film " << film->name () << "\n"; + auto new_dkdm = make_shared(wx_to_std(dialog.get())); + auto parent = dynamic_pointer_cast(selected_dkdm()); + if (!parent) { + parent = Config::instance()->dkdms (); } - } catch (std::exception& e) { - cerr << program_name << ": error reading film `" << film_dir << "' (" << e.what() << ")\n"; - exit (EXIT_FAILURE); + add_dkdm(new_dkdm, parent); + update_dkdm_view(); } - if (verbose) { - cout << "Making KDMs valid from " << valid_from.get() << " to " << valid_to.get() << "\n"; + void update_dkdm_view() + { + _dkdm->DeleteAllItems(); + _dkdm_id.clear(); + add_dkdm_to_view(Config::instance()->dkdms()); + if (_selected_dkdm) { + auto selection_in_id_map = std::find_if(_dkdm_id.begin(), _dkdm_id.end(), [this](pair> const& entry) { + return entry.second == _selected_dkdm; + }); + if (selection_in_id_map != _dkdm_id.end()) { + _dkdm->SelectItem(selection_in_id_map->first); + } + } + } + + /** @return true if this thing or any of its children match a search string */ + bool matches(shared_ptr base, string const& search) + { + if (search.empty()) { + return true; + } + + auto name = base->name(); + transform(name.begin(), name.end(), name.begin(), ::tolower); + if (name.find(search) != string::npos) { + return true; + } + + auto group = dynamic_pointer_cast(base); + if (!group) { + return false; + } + + auto const children = group->children(); + return std::any_of(children.begin(), children.end(), [this, search](shared_ptr child) { + return matches(child, search); + }); } - /* XXX: allow specification of this */ - vector cpls = film->cpls (); - if (cpls.empty ()) { - error ("no CPLs found in film"); - } else if (cpls.size() > 1) { - error ("more than one CPL found in film"); + /** Add DKDMs to the view that match the current search */ + void add_dkdm_to_view(shared_ptr base) + { + auto search = wx_to_std(_dkdm_search->GetValue()); + transform(search.begin(), search.end(), search.begin(), ::tolower); + + optional group_to_expand; + + if (!base->parent()) { + /* This is the root group */ + _dkdm_id[_dkdm->AddRoot("root")] = base; + } else { + /* Add base to the view */ + wxTreeItemId added; + auto parent_id = dkdm_to_id(base->parent()); + added = _dkdm->AppendItem(parent_id, std_to_wx(base->name())); + /* Expand the group (later) if it matches the search or it was manually expanded */ + if (!search.empty() || _expanded_dkdm_groups.find(base) != _expanded_dkdm_groups.end()) { + group_to_expand = added; + } + _dkdm_id[added] = base; + } + + /* Add children */ + auto group = dynamic_pointer_cast(base); + if (group) { + auto children = group->children(); + children.sort( + [this](shared_ptr a, shared_ptr b) { return _collator.compare(a->name(), b->name()) < 0; } + ); + + for (auto i: children) { + if (matches(i, search)) { + add_dkdm_to_view(i); + } + } + } + + if (group_to_expand) { + _ignore_expand = true; + _dkdm->Expand(*group_to_expand); + _ignore_expand = false; + } } - boost::filesystem::path cpl = cpls.front().cpl_file; + /** @param group Group to add dkdm to */ + void add_dkdm(shared_ptr dkdm, shared_ptr group, shared_ptr previous = shared_ptr()) + { + group->add (dkdm, previous); + /* We're messing with a Config-owned object here, so tell it that something has changed. + This isn't nice. + */ + Config::instance()->changed (); + } - if (cinema_name.empty ()) { + wxTreeItemId dkdm_to_id (shared_ptr dkdm) + { + for (auto const& i: _dkdm_id) { + if (i.second == dkdm) { + return i.first; + } + } + DCPOMATIC_ASSERT (false); + } - if (output.empty ()) { - error ("you must specify --output"); + void remove_dkdm_clicked () + { + auto removed = selected_dkdm (); + if (!removed) { + return; } - - dcp::Certificate certificate (dcp::file_to_string (certificate_file)); - dcp::EncryptedKDM kdm = film->make_kdm (certificate, cpl, valid_from.get(), valid_to.get(), formulation); - kdm.as_xml (output); - if (verbose) { - cout << "Generated KDM " << output << " for certificate.\n"; + + if (NagDialog::maybe_nag ( + this, Config::NAG_DELETE_DKDM, + _("You are about to remove a DKDM. This will make it impossible to decrypt the DCP that the DKDM was made for, and it cannot be undone. " + "Are you sure?"), + true)) { + return; } - } else { - list > cinemas = Config::instance()->cinemas (); - list >::const_iterator i = cinemas.begin(); - while (i != cinemas.end() && (*i)->name != cinema_name && (*i)->email != cinema_name) { - ++i; + _dkdm->Delete (dkdm_to_id (removed)); + auto dkdms = Config::instance()->dkdms (); + dkdms->remove (removed); + Config::instance()->changed (); + } + + void export_dkdm_clicked () + { + auto removed = selected_dkdm (); + if (!removed) { + return; } - if (i == cinemas.end ()) { - cerr << program_name << ": could not find cinema \"" << cinema_name << "\"\n"; - exit (EXIT_FAILURE); + auto dkdm = dynamic_pointer_cast(removed); + if (!dkdm) { + return; } - if (output.empty ()) { - output = "."; + wxFileDialog dialog( + this, _("Select DKDM File"), wxEmptyString, wxEmptyString, wxT("XML files (*.xml)|*.xml"), + wxFD_SAVE | wxFD_OVERWRITE_PROMPT + ); + + if (dialog.ShowModal() == wxID_OK) { + dkdm->dkdm().as_xml(wx_to_std(dialog.GetPath())); } + } + + void dkdm_search_changed() + { + update_dkdm_view(); + } + + wxPreferencesEditor* _config_dialog; + ScreensPanel* _screens; + KDMTimingPanel* _timing; + wxTreeCtrl* _dkdm; + wxSearchCtrl* _dkdm_search; + typedef std::map> DKDMMap; + DKDMMap _dkdm_id; + /* Keep a separate track of the selected DKDM so that when a search happens, and some things + * get removed from the view, we can restore the selection when they are re-added. + */ + shared_ptr _selected_dkdm; + /* Keep expanded groups for the same reason */ + unordered_set> _expanded_dkdm_groups; + /* true if we are "artificially" expanding a group because it contains something found + * in a search. + */ + bool _ignore_expand = false; + wxButton* _add_dkdm; + wxButton* _add_dkdm_folder; + wxButton* _remove_dkdm; + wxButton* _export_dkdm; + wxButton* _create; + KDMOutputPanel* _output; + JobViewDialog* _job_view; + Collator _collator; +}; + + +/** @class App + * @brief The magic App class for wxWidgets. + */ +class App : public wxApp +{ +public: + App () + : wxApp () + , _frame (nullptr) + {} + +private: + + bool OnInit () override + { + wxSplashScreen* splash; try { - if (zip) { - write_kdm_zip_files ( - film, (*i)->screens(), cpl, dcp::LocalTime (valid_from.get()), dcp::LocalTime (valid_to.get()), formulation, output - ); - - if (verbose) { - cout << "Wrote ZIP files to " << output << "\n"; - } - } else { - write_kdm_files ( - film, (*i)->screens(), cpl, dcp::LocalTime (valid_from.get()), dcp::LocalTime (valid_to.get()), formulation, output - ); - - if (verbose) { - cout << "Wrote KDM files to " << output << "\n"; - } + wxInitAllImageHandlers (); + + Config::FailedToLoad.connect(boost::bind(&App::config_failed_to_load, this, _1)); + Config::Warning.connect (boost::bind (&App::config_warning, this, _1)); + + splash = maybe_show_splash (); + + SetAppName (_("DCP-o-matic KDM Creator")); + + if (!wxApp::OnInit()) { + return false; + } + +#ifdef DCPOMATIC_LINUX + unsetenv ("UBUNTU_MENUPROXY"); +#endif + +#ifdef DCPOMATIC_OSX + make_foreground_application (); +#endif + + dcpomatic_setup_path_encoding (); + + /* Enable i18n; this will create a Config object + to look for a force-configured language. This Config + object will be wrong, however, because dcpomatic_setup + hasn't yet been called and there aren't any filters etc. + set up yet. + */ + dcpomatic_setup_i18n (); + + /* Set things up, including filters etc. + which will now be internationalised correctly. + */ + dcpomatic_setup (); + + /* Force the configuration to be re-loaded correctly next + time it is needed. + */ + Config::drop (); + + _frame = new DOMFrame (_("DCP-o-matic KDM Creator")); + SetTopWindow (_frame); + _frame->Maximize (); + if (splash) { + splash->Destroy (); + splash = nullptr; } + _frame->Show (); + + signal_manager = new wxSignalManager (this); + Bind (wxEVT_IDLE, boost::bind (&App::idle, this)); + } + catch (exception& e) + { + if (splash) { + splash->Destroy (); + } + error_dialog (0, _("DCP-o-matic could not start"), std_to_wx(e.what())); + } + + return true; + } + + /* An unhandled exception has occurred inside the main event loop */ + bool OnExceptionInMainLoop () override + { + try { + throw; } catch (FileError& e) { - cerr << argv[0] << ": " << e.what() << " (" << e.file().string() << ")\n"; - exit (EXIT_FAILURE); - } catch (KDMError& e) { - cerr << argv[0] << ": " << e.what() << "\n"; - exit (EXIT_FAILURE); + error_dialog ( + 0, + wxString::Format ( + _("An exception occurred: %s (%s)\n\n") + REPORT_PROBLEM, + std_to_wx (e.what()), + std_to_wx (e.file().string().c_str ()) + ) + ); + } catch (exception& e) { + error_dialog ( + nullptr, + wxString::Format ( + _("An exception occurred: %s.\n\n") + " " + REPORT_PROBLEM, + std_to_wx(e.what()) + ) + ); + } catch (...) { + error_dialog (0, _("An unknown exception occurred.") + " " + REPORT_PROBLEM); } + + /* This will terminate the program */ + return false; } - return 0; -} + void OnUnhandledException () override + { + error_dialog (nullptr, _("An unknown exception occurred.") + " " + REPORT_PROBLEM); + } + + void idle () + { + signal_manager->ui_idle (); + } + + void config_failed_to_load(Config::LoadFailure what) + { + report_config_load_failure(_frame, what); + } + + void config_warning (string m) + { + message_dialog (_frame, std_to_wx(m)); + } + + DOMFrame* _frame; +}; + +IMPLEMENT_APP (App)