jellyfin-media-player/src/utils/CachedRegexMatcher.cpp

//
// Created by Tobias Hieta on 20/08/15.
//

#include <QDebug>
#include "CachedRegexMatcher.h"

/////////////////////////////////////////////////////////////////////////////////////////
bool CachedRegexMatcher::addMatcher(const QString& pattern, const QVariant& result)
{
  QRegExp matcher(pattern);
  if (!matcher.isValid())
  {
    qWarning() << "Could not compile pattern:" << pattern;
    return false;
  }

  // Remove older mapping if it exists.
  if (!m_allowMultiplePatterns)
  {
    auto newEnd = std::remove_if(m_matcherList.begin(),m_matcherList.end(), [pattern](auto mp)
    {
      return mp.first.pattern() == pattern;
    });
    m_matcherList.erase(newEnd, m_matcherList.end());
  }

  m_matcherList.push_back(qMakePair(matcher, result));
  return true;
}

/////////////////////////////////////////////////////////////////////////////////////////
QVariantList CachedRegexMatcher::match(const QString& input)
{
  // first we check if this match has already happened before
  if (m_matcherCache.contains(input))
    return m_matcherCache.value(input);

  QVariantList matches;

  // otherwise try to iterate our list and find a match
  for(const MatcherValuePair& matcher : m_matcherList)
  {
    QRegExp re(matcher.first);

    if (re.indexIn(input) != -1)
    {
      // found match
      QVariant returnValue = matcher.second;

      if (re.captureCount() > 0 && matcher.second.type() == QVariant::String)
      {
        QString value(matcher.second.toString());

        for (int i = 0; i < re.captureCount(); i ++)
        {
          QString argFmt = QString("%%1").arg(i + 1);
          if (value.contains(argFmt))
            value = value.arg(re.cap(i + 1));
        }
        returnValue = QVariant(value);
      }

      matches << returnValue;
    }
  }

  qDebug() << "No match for:" << input;

  if (!matches.isEmpty())
  {
    m_matcherCache.insert(input, matches);
    return matches;
  }

  return QVariantList();
}

/////////////////////////////////////////////////////////////////////////////////////////
void CachedRegexMatcher::clear()
{
  m_matcherCache.clear();
  m_matcherList.clear();
}