mythmetadatalookup.cpp

Go to the documentation of this file.

// C headers
#include <unistd.h> // close
 
// C++ headers
#include <iostream>
#include <memory>
#include <thread>
 
// Qt headers
#include <QtGlobal>
#if QT_VERSION >= QT_VERSION_CHECK(6,5,0)
#include <QtEnvironmentVariables>
#endif
#include <QCoreApplication>
#include <QEventLoop>
#ifdef Q_OS_DARWIN
#include <QProcessEnvironment>
#endif
 
// MythTV
#include "libmyth/mythcontext.h"
#include "libmythbase/exitcodes.h"
#include "libmythbase/mythappname.h"
#include "libmythbase/mythconfig.h"
#include "libmythbase/mythdb.h"
#include "libmythbase/mythlogging.h"
#include "libmythbase/mythmiscutil.h"
#include "libmythbase/mythtranslation.h"
#include "libmythbase/mythversion.h"
#include "libmythtv/jobqueue.h"
 
// MythMetadataLookup
#include "lookup.h"
#include "mythmetadatalookup_commandlineparser.h"
 
int main(int argc, char *argv[])
{
    MythMetadataLookupCommandLineParser cmdline;
    if (!cmdline.Parse(argc, argv))
    {
        cmdline.PrintHelp();
        return GENERIC_EXIT_INVALID_CMDLINE;
    }
 
    if (cmdline.toBool("showhelp"))
    {
        cmdline.PrintHelp();
        return GENERIC_EXIT_OK;
    }
 
    if (cmdline.toBool("showversion"))
    {
        MythMetadataLookupCommandLineParser::PrintVersion();
        return GENERIC_EXIT_OK;
    }
 
    QCoreApplication a(argc, argv);
    QCoreApplication::setApplicationName(MYTH_APPNAME_MYTHMETADATALOOKUP);
 
#ifdef Q_OS_DARWIN
    QString path = QCoreApplication::applicationDirPath();
    qputenv("PYTHONPATH",
           QString("%1/../Resources/lib/%2:%1/../Resources/lib/%2/site-packages:%1/../Resources/lib/%2/lib-dynload:%3")
           .arg(path)
           .arg(QFileInfo(PYTHON_EXE).fileName())
           .arg(QProcessEnvironment::systemEnvironment().value("PYTHONPATH"))
           .toUtf8().constData());
#endif
 
    int retval = cmdline.ConfigureLogging();
    if (retval != GENERIC_EXIT_OK)
        return retval;
 
    // Don't listen to console input
    close(0);
 
    MythContext context {MYTH_BINARY_VERSION};
    if (!context.Init(false))
    {
        LOG(VB_GENERAL, LOG_ERR, "Failed to init MythContext, exiting.");
        return GENERIC_EXIT_NO_MYTHCONTEXT;
    }
 
    myth_nice(19);
 
    MythTranslation::load("mythfrontend");
 
    std::unique_ptr<LookerUpper> lookup = std::make_unique<LookerUpper>();
 
    LOG(VB_GENERAL, LOG_INFO,
            "Testing grabbers and metadata sites for functionality...");
    if (!LookerUpper::AllOK())
        return GENERIC_EXIT_NOT_OK;
    LOG(VB_GENERAL, LOG_INFO,
            "All grabbers tested and working.  Continuing...");
 
    if (cmdline.toBool("jobid"))
    {
        uint chanid = 0;
        QDateTime starttime;
        int jobType = JOB_METADATA;
 
        if (!JobQueue::GetJobInfoFromID(cmdline.toInt("jobid"), jobType,
                                        chanid, starttime))
        {
            LOG(VB_GENERAL, LOG_INFO,
                        QString("No valid job found for jobid: %1")
                                    .arg(cmdline.toInt("jobid")));
            return GENERIC_EXIT_NOT_OK;
        }
 
        lookup->HandleSingleRecording(chanid, starttime,
                                      cmdline.toBool("refresh-rules"));
    }
    else if (cmdline.toBool("chanid") && cmdline.toBool("starttime"))
    {
        lookup->HandleSingleRecording(cmdline.toUInt("chanid"),
                                      cmdline.toDateTime("starttime"),
                                      cmdline.toBool("refresh-rules"));
    }
    else if (cmdline.toBool("refresh-all-rules"))
    {
        lookup->HandleAllRecordingRules();
        LookerUpper::CopyRuleInetrefsToRecordings();
    }
    else if (cmdline.toBool("refresh-all-artwork"))
    {
        lookup->HandleAllArtwork(false);
    }
    else if (cmdline.toBool("refresh-all-artwork-dangerously"))
    {
        lookup->HandleAllArtwork(true);
    }
    else
    {
        // refresh-all is default behavior if no other arguments given
        LookerUpper::CopyRuleInetrefsToRecordings();
        lookup->HandleAllRecordings(cmdline.toBool("refresh-rules"));
    }
 
    while (lookup->StillWorking())
    {
        std::this_thread::sleep_for(1s);
        qApp->processEvents();
    }
 
    LOG(VB_GENERAL, LOG_NOTICE, "MythMetadataLookup run complete.");
 
    return GENERIC_EXIT_OK;
}