Issue 4128, 5138 - Add Parser and Serializer implemented in C++

compiled/subscription/Subscription.cpp

 /*
  * This file is part of Adblock Plus <https://adblockplus.org/>,
  * Copyright (C) 2006-present eyeo GmbH
  *
  * Adblock Plus is free software: you can redistribute it and/or modify
  * it under the terms of the GNU General Public License version 3 as
  * published by the Free Software Foundation.
  *
  * Adblock Plus is distributed in the hope that it will be useful,
  * but WITHOUT ANY WARRANTY; without even the implied warranty of
@@ skipping 13 matching lines @@
 #include "UserDefinedSubscription.h"
 #include "../StringMap.h"
 
 ABP_NS_USING
 
 namespace
 {
   StringMap<Subscription*> knownSubscriptions(16);
 }
 
-Subscription::Subscription(Type type, const String& id)
+Subscription::Subscription(Type type, const String& id, const KeyValues& properties)
     : mID(id), mType(type), mDisabled(false), mListed(false)
 {
   annotate_address(this, "Subscription");
+  parseProperty(properties, mTitle, u"title"_str);
+  parseProperty(properties, mDisabled, u"disabled"_str);
 }
 
 Subscription::~Subscription()
 {
   knownSubscriptions.erase(mID);
 }
 
 Filter* Subscription::FilterAt(Subscription::Filters::size_type index)
 {
   if (index >= mFilters.size())
     return nullptr;
 
   FilterPtr result(mFilters[index]);
   return result.release();
 }
 
 int Subscription::IndexOfFilter(const Filter& filter)
 {
   for (Filters::size_type i = 0; i < mFilters.size(); i++)
     if (mFilters[i] == &filter)
       return i;
   return -1;
 }
 
-OwnedString Subscription::Serialize() const
+void Subscription::AddFilter(Filter& filter)
 {
-  OwnedString result(u"[Subscription]\nurl="_str);
+  mFilters.emplace_back(&filter);
+}
+
+OwnedString Subscription::SerializeProperties() const
+{
+  if (const auto* downcastedSubscription = As<DownloadableSubscription>())
+    return downcastedSubscription->SerializeProperties();
+  else if (const auto* downcastedSubscription = As<UserDefinedSubscription>())
+    return downcastedSubscription->SerializeProperties();
+
+  return DoSerializeProperties();
+}
+
+
+OwnedString Subscription::DoSerializeProperties() const
+{
+  OwnedString result(u"url="_str);
   result.append(mID);
   result.append(u'\n');
   if (!mTitle.empty())
   {
     result.append(u"title="_str);
     result.append(mTitle);
     result.append(u'\n');
   }
   if (mDisabled)
     result.append(u"disabled=true\n"_str);
 
   return result;
 }
 
-OwnedString Subscription::SerializeFilters() const
-{
-  if (!mFilters.size())
-    return OwnedString();
-
-  OwnedString result(u"[Subscription filters]\n"_str);
-  for (const auto& filter : mFilters)
-  {
-    // TODO: Escape [ characters
-    result.append(filter->GetText());
-    result.append(u'\n');
-  }
-  return result;
-}
-
-Subscription* Subscription::FromID(const String& id)
+SubscriptionPtr Subscription::FromProperties(const String& id, const KeyValues& keyValues)
 {
   if (id.empty())
   {
     // Generate a new random ID
     std::mt19937 gen(knownSubscriptions.size());
     OwnedString randomID(u"~user~000000"_str);
     do
     {
       int number = gen();
       for (String::size_type i = randomID.length() - 6; i < randomID.length(); i++)
       {
         randomID[i] = '0' + (number % 10);
         number /= 10;
       }
     } while (knownSubscriptions.find(randomID));
-    return FromID(randomID);
+    return FromProperties(randomID, keyValues);
   }
 
   auto knownSubscription = knownSubscriptions.find(id);
   if (knownSubscription)
-  {
-    knownSubscription->second->AddRef();
-    return knownSubscription->second;
-  }
+    return SubscriptionPtr(knownSubscription->second);
 
   SubscriptionPtr subscription;
-  if (!id.empty() && id[0] == '~')
-    subscription = SubscriptionPtr(new UserDefinedSubscription(id), false);
+  if (id[0] == '~')
+    subscription.reset(new UserDefinedSubscription(id, keyValues), false);
   else
-    subscription = SubscriptionPtr(new DownloadableSubscription(id), false);
+    subscription.reset(new DownloadableSubscription(id, keyValues), false);
 
   // This is a hack: we looked up the entry using id but create it using
   // subscription->mID. This works because both are equal at this point.
   // However, id refers to a temporary buffer which will go away.
   enter_context("Adding to known subscriptions");
   knownSubscription.assign(subscription->mID, subscription.get());
   exit_context();
 
-  return subscription.release();
+  return subscription;
 }
+
+SubscriptionPtr Subscription::FromProperties(const KeyValues& properties)
+{
+  const String* id = findPropertyValue(properties, u"url"_str);
+  if (id == nullptr || id->empty())
+    return SubscriptionPtr();
+  return FromProperties(*id, properties);
+}
+
+const String* Subscription::findPropertyValue(const Subscription::KeyValues& properties, const String& propertyName)
+{
+  for (const auto& property : properties)
+    if (property.first == propertyName)
+      return &property.second;
+  return nullptr;
+}