Ticket #3176: mythtv-mythshutdown.diff

programs/mythshutdown/main.cpp

@@ -11 +11 @@
 #include <exitcodes.h>
 #include <mythcontext.h>
 #include <mythdbcon.h>
+#include "libmythtv/programinfo.h"
+#include "tv.h"
 
 void setGlobalSetting(const QString &key, const QString &value)
 {
@@ -103 +105 @@
     return dtDateTime;
 }
 
+bool isRecording()
+{
+    bool m_isRecording = false;
+
+    if (!gContext->IsConnectedToMaster())
+    {
+        VERBOSE(VB_IMPORTANT, "isRecording: Attempting to connect to master server...");
+        if (!gContext->ConnectToMasterServer(false))
+        {
+            VERBOSE(VB_IMPORTANT, "isRecording: Could not connect to master server!");
+            return false;
+        }
+    }
+
+    QStringList strlist;
+
+    // get list of current recordings
+    QString querytext = QString("SELECT cardid FROM capturecard WHERE parentid = 0;");
+    MSqlQuery query(MSqlQuery::InitCon());
+    query.exec(querytext);
+    QString Status = "";
+
+    if (query.isActive() && query.numRowsAffected())
+    {
+        VERBOSE(VB_IMPORTANT, "isRecording: Query active");
+        while(query.next())
+        {
+            QString status = "";
+            int cardid = query.value(0).toInt();
+            int state = kState_ChangingState;
+            QString channelName = "";
+            QString title = "";
+            QString subtitle = "";
+            QDateTime dtStart = QDateTime();
+            QDateTime dtEnd = QDateTime();
+
+            QString cmd = QString("QUERY_REMOTEENCODER %1").arg(cardid);
+
+            while (state == kState_ChangingState)
+            {
+                strlist = cmd;
+                strlist << "GET_STATE";
+                gContext->SendReceiveStringList(strlist);
+
+                state = strlist[0].toInt();
+                if (state == kState_ChangingState)
+                    usleep(500);
+            }
+            VERBOSE(VB_IMPORTANT, "isRecording: Successfully queried encoder.");
+
+            if (state == kState_RecordingOnly || state == kState_WatchingRecording)
+            {
+                VERBOSE(VB_IMPORTANT, "Recorder is recording. Returning true");
+                m_isRecording = true;
+            }
+        }
+    }
+    return m_isRecording;
+}
+
 int getStatus()
 {
     VERBOSE(VB_GENERAL, "Mythshutdown: --status");
@@ -127 +189 @@
         res += 4;
     }
 
+    if (isRecording())
+    {
+        VERBOSE(VB_IMPORTANT, "Recording in progress...");
+        res += 8;
+    }
+    
     if (getGlobalSetting("MythShutdownLock", "0") == "1")
     {
         VERBOSE(VB_IMPORTANT, "Shutdown is locked");
@@ -250 +318 @@
     return 0;
 }
 
+int setScheduledWakeupTime()
+{
+    typedef struct
+    {
+        QString channel, title, subtitle;
+        QDateTime startTime, endTime;
+    } ProgramDetail;
+
+    if (!gContext->IsConnectedToMaster())
+    {
+        VERBOSE(VB_IMPORTANT, "setScheduledWakeupTime: Attempting to connect to master server...");
+        if (!gContext->ConnectToMasterServer(false))
+        {
+            VERBOSE(VB_IMPORTANT, "setScheduledWakeupTime: Could not connect to master server!");
+            return false;
+        }
+    }
+    
+    QDateTime m_nextRecordingStart = QDateTime();
+
+    ProgramList *progList = new ProgramList(true);
+    ProgramInfo *progInfo;
+
+    if (progList->FromScheduler())
+    {
+        if (progList->count() > 0)
+        {
+            VERBOSE(VB_IMPORTANT, "setScheduledWakeupTime: At least one scheduled recording found.");
+            // find the earliest scheduled recording
+            for (progInfo = progList->first(); progInfo; progInfo = progList->next())
+            {
+                if (progInfo->recstatus == rsWillRecord)
+                {
+                    if (m_nextRecordingStart.isNull() ||
+                            m_nextRecordingStart > progInfo->recstartts)
+                    {
+                        m_nextRecordingStart = progInfo->recstartts;
+                    }
+                }
+            }
+
+            // save the details of the earliest recording(s)
+            for (progInfo = progList->first(); progInfo; progInfo = progList->next())
+            {
+                if (progInfo->recstatus == rsWillRecord)
+                {
+                    if (progInfo->recstartts == m_nextRecordingStart)
+                    {
+                        ProgramDetail *prog = new ProgramDetail;
+                        prog->channel = progInfo->channame;
+                        prog->title = progInfo->title;
+                        prog->subtitle = progInfo->subtitle;
+                        prog->startTime = progInfo->recstartts;
+                        prog->endTime = progInfo->recendts;
+                    }
+                }
+            }
+        }
+    }
+
+    delete progList;
+
+    // set the wakeup time for the next scheduled recording
+    if (!m_nextRecordingStart.isNull())
+    {
+        int m_preRollSeconds = gContext->GetNumSetting("RecordPreRoll");
+        QDateTime restarttime = m_nextRecordingStart.addSecs((-1) * m_preRollSeconds);
+
+        int add = gContext->GetNumSetting("StartupSecsBeforeRecording", 240);
+        if (add)
+            restarttime = restarttime.addSecs((-1) * add);
+
+        QString wakeup_timeformat = gContext->GetSetting("WakeupTimeFormat",
+                                                            "yyyy-MM-ddThh:mm");
+        setWakeupTime(restarttime.toString(wakeup_timeformat));
+    
+        return true;
+    }
+    return false;
+}
+
 int shutdown()
 {
     VERBOSE(VB_GENERAL, "Mythshutdown: --shutdown");
@@ -542 +691 @@
 void showUsage()
 {
   cout << "Usage of mythshutdown\n"; 
-  cout << "-w/--setwakeup time (sets the wakeup time. time=yyyy-MM-ddThh:mm:ss.\n";
+  cout << "-w/--setwakeup time (sets the wakeup time. time=yyyy-MM-ddThh:mm:ss)\n";
+  cout << "-t/--setscheduledwakeup (sets the wakeup time to the next scheduled recording)\n";
   cout << "                     doesn't write it into nvram)\n";
   cout << "-q/--shutdown       (set nvram-wakeup time and shutdown)\n";
+  cout << "-x/--safeshutdown       (equal to -c -t -q.  check shutdown possible, set\n";
+  cout <<"                                 scheduled wakeup and shutdown)\n";
   cout << "-p/--startup        (check startup. check will return 0 if automatic\n";
   cout << "                                                      1 for manually)\n";
   cout << "-c/--check          (check shutdown possible returns 0 ok to shutdown\n";
@@ -556 +708 @@
   cout << "         1 - Transcoding\n";
   cout << "         2 - Commercial Flagging\n";
   cout << "         4 - Grabbing EPG data\n";
-  cout << "         8 - Not used\n";
+  cout << "         8 - Recording\n";
   cout << "        16 - Locked\n";
   cout << "        32 - Not used\n";
   cout << "        64 - In a daily wakeup/shutdown period\n";
@@ -577 +729 @@
     if (!gContext->Init(false))
     {
         cout << "mythshutdown: Could not initialize myth context. "
-                "Exiting." << endl;;
+                "Exiting." << endl;
         return FRONTEND_EXIT_NO_MYTHCONTEXT;
     }
 
@@ -590 +742 @@
     bool bSetWakeupTime = false;
     QString sWakeupTime = "";
     bool bShowUsage = false;
+    bool bSetScheduledWakeupTime = false;
+    bool bCheckAndShutdown = false;
 
     //  Check command line arguments
     for (int argpos = 1; argpos < a.argc(); ++argpos)
@@ -663 +817 @@
         {
             bShowUsage = true;
         }
+        else if (!strcmp(a.argv()[argpos],"-t") ||
+            !strcmp(a.argv()[argpos],"--setscheduledwakeup"))
+        {
+            bSetScheduledWakeupTime = true;
+        }
+        else if (!strcmp(a.argv()[argpos],"-x") ||
+            !strcmp(a.argv()[argpos],"--safeshutdown"))
+        {
+            bCheckAndShutdown = true;
+        }
+
         else
         {
             cout << "Invalid argument: " << a.argv()[argpos] << endl;
@@ -681 +846 @@
         res = unlockShutdown();
     else if (bCheckOKShutdown)
         res = checkOKShutdown();
+    else if (bSetScheduledWakeupTime)
+        res = setScheduledWakeupTime();
     else if (bStartup)
         res = startup();
     else if (bShutdown)
@@ -689 +856 @@
         res = getStatus();
     else if (bSetWakeupTime)
         res = setWakeupTime(sWakeupTime);
+    else if (bCheckAndShutdown)
+    {
+        cout << "into checkAndShutdown" << endl;
+        res = checkOKShutdown();
+        if (res == 0)     // Nothing to stop a shutdown (eg. recording in progress).
+        { 
+             res = setScheduledWakeupTime();
+             res = shutdown();
+        }
+    }
     else 
         showUsage();