1 files changed, 190 insertions, 0 deletions
diff --git a/xbmc/pvr/PVRChannelNumberInputHandler.cpp b/xbmc/pvr/PVRChannelNumberInputHandler.cpp
new file mode 100644
index 0000000..a57c774
--- /dev/null
+++ b/xbmc/pvr/PVRChannelNumberInputHandler.cpp
@@ -0,0 +1,190 @@
+/*
+ *  Copyright (C) 2012-2018 Team Kodi
+ *  This file is part of Kodi - https://kodi.tv
+ *
+ *  SPDX-License-Identifier: GPL-2.0-or-later
+ *  See LICENSES/README.md for more information.
+ */
+
+#include "PVRChannelNumberInputHandler.h"
+
+#include "ServiceBroker.h"
+#include "settings/AdvancedSettings.h"
+#include "settings/SettingsComponent.h"
+#include "utils/StringUtils.h"
+
+#include <algorithm>
+#include <cstdlib>
+#include <mutex>
+#include <string>
+
+using namespace PVR;
+using namespace std::chrono_literals;
+
+CPVRChannelNumberInputHandler::CPVRChannelNumberInputHandler()
+  : CPVRChannelNumberInputHandler(CServiceBroker::GetSettingsComponent()
+                                      ->GetAdvancedSettings()
+                                      ->m_iPVRNumericChannelSwitchTimeout,
+                                  CHANNEL_NUMBER_INPUT_MAX_DIGITS)
+{
+}
+
+CPVRChannelNumberInputHandler::CPVRChannelNumberInputHandler(
+    int iDelay, int iMaxDigits /* = CHANNEL_NUMBER_INPUT_MAX_DIGITS */)
+  : m_iDelay(iDelay), m_iMaxDigits(iMaxDigits), m_timer(this)
+{
+}
+
+void CPVRChannelNumberInputHandler::OnTimeout()
+{
+  if (m_inputBuffer.empty())
+  {
+    std::unique_lock<CCriticalSection> lock(m_mutex);
+    SetLabel("");
+  }
+  else
+  {
+    // call the overridden worker method
+    OnInputDone();
+
+    std::unique_lock<CCriticalSection> lock(m_mutex);
+
+    // erase input buffer immediately , but...
+    m_inputBuffer.erase();
+
+    // ... display the label for another .5 secs if we stopped the timer before regular timeout.
+    if (m_timer.IsRunning())
+      SetLabel("");
+    else
+      m_timer.Start(500ms);
+  }
+}
+
+void CPVRChannelNumberInputHandler::ExecuteAction()
+{
+  m_timer.Stop(true /* wait until worker thread ended */);
+  OnTimeout();
+}
+
+bool CPVRChannelNumberInputHandler::CheckInputAndExecuteAction()
+{
+  const CPVRChannelNumber channelNumber = GetChannelNumber();
+  if (channelNumber.IsValid())
+  {
+    // we have a valid channel number; execute the associated action now.
+    ExecuteAction();
+    return true;
+  }
+  return false;
+}
+
+void CPVRChannelNumberInputHandler::AppendChannelNumberCharacter(char cCharacter)
+{
+  if (cCharacter != CPVRChannelNumber::SEPARATOR && (cCharacter < '0' || cCharacter > '9'))
+    return;
+
+  std::unique_lock<CCriticalSection> lock(m_mutex);
+
+  if (cCharacter == CPVRChannelNumber::SEPARATOR)
+  {
+    // no leading separator
+    if (m_inputBuffer.empty())
+      return;
+
+    // max one separator
+    if (m_inputBuffer.find(CPVRChannelNumber::SEPARATOR) != std::string::npos)
+      return;
+  }
+
+  if (m_inputBuffer.size() == static_cast<size_t>(m_iMaxDigits))
+  {
+    m_inputBuffer.erase(m_inputBuffer.begin());
+    SetLabel(m_inputBuffer);
+  }
+  else if (m_inputBuffer.empty())
+  {
+    m_sortedChannelNumbers.clear();
+    GetChannelNumbers(m_sortedChannelNumbers);
+
+    std::sort(m_sortedChannelNumbers.begin(), m_sortedChannelNumbers.end());
+  }
+
+  m_inputBuffer.append(&cCharacter, 1);
+  SetLabel(m_inputBuffer);
+
+  for (auto it = m_sortedChannelNumbers.begin(); it != m_sortedChannelNumbers.end();)
+  {
+    const std::string channel = *it;
+    ++it;
+
+    if (StringUtils::StartsWith(channel, m_inputBuffer))
+    {
+      if (it != m_sortedChannelNumbers.end() && StringUtils::StartsWith(*it, m_inputBuffer))
+      {
+        // there are alternative numbers; wait for more input
+        break;
+      }
+
+      // no alternatives; complete the number and fire immediately
+      m_inputBuffer = channel;
+      SetLabel(m_inputBuffer);
+      ExecuteAction();
+      return;
+    }
+  }
+
+  if (!m_timer.IsRunning())
+    m_timer.Start(std::chrono::milliseconds(m_iDelay));
+  else
+    m_timer.Restart();
+}
+
+CPVRChannelNumber CPVRChannelNumberInputHandler::GetChannelNumber() const
+{
+  int iChannelNumber = 0;
+  int iSubChannelNumber = 0;
+
+  std::unique_lock<CCriticalSection> lock(m_mutex);
+
+  size_t pos = m_inputBuffer.find(CPVRChannelNumber::SEPARATOR);
+  if (pos != std::string::npos)
+  {
+    // main + sub
+    if (pos != 0)
+    {
+      iChannelNumber = std::atoi(m_inputBuffer.substr(0, pos).c_str());
+      if (pos != m_inputBuffer.size() - 1)
+        iSubChannelNumber = std::atoi(m_inputBuffer.substr(pos + 1).c_str());
+    }
+  }
+  else
+  {
+    // only main
+    iChannelNumber = std::atoi(m_inputBuffer.c_str());
+  }
+
+  return CPVRChannelNumber(iChannelNumber, iSubChannelNumber);
+}
+
+bool CPVRChannelNumberInputHandler::HasChannelNumber() const
+{
+  return !m_inputBuffer.empty();
+}
+
+std::string CPVRChannelNumberInputHandler::GetChannelNumberLabel() const
+{
+  std::unique_lock<CCriticalSection> lock(m_mutex);
+  return m_label;
+}
+
+void CPVRChannelNumberInputHandler::SetLabel(const std::string& label)
+{
+  std::unique_lock<CCriticalSection> lock(m_mutex);
+  if (label != m_label)
+  {
+    m_label = label;
+
+    // inform subscribers
+    m_events.Publish(PVRChannelNumberInputChangedEvent(m_label));
+  }
+}