//
// Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2022
//
// Distributed under the Boost Software License, Version 1.0. (See accompanying
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
//
#include "td/telegram/Premium.h"

#include "td/telegram/ConfigShared.h"
#include "td/telegram/Global.h"

#include "td/utils/SliceBuilder.h"

namespace td {

const vector<Slice> &get_premium_limit_keys() {
  static const vector<Slice> limit_keys{"channels_limit",
                                        "saved_gifs_limit",
                                        "stickers_faved_limit",
                                        "dialog_filters_limit",
                                        "dialog_filters_chats_limit",
                                        "dialogs_pinned_limit",
                                        "dialogs_folder_pinned_limit",
                                        "channels_public_limit",
                                        "caption_length_limit"};
  return limit_keys;
}

static string get_premium_source(const td_api::PremiumLimitType *limit_type) {
  if (limit_type == nullptr) {
    return string();
  }

  auto key = [&]() {
    switch (limit_type->get_id()) {
      case td_api::premiumLimitTypeSupergroupCount::ID:
        return "channels";
      case td_api::premiumLimitTypeSavedAnimationCount::ID:
        return "saved_gifs";
      case td_api::premiumLimitTypeFavoriteStickerCount::ID:
        return "stickers_faved";
      case td_api::premiumLimitTypeChatFilterCount::ID:
        return "dialog_filters";
      case td_api::premiumLimitTypeChatFilterChosenChatCount::ID:
        return "dialog_filters_chats";
      case td_api::premiumLimitTypePinnedChatCount::ID:
        return "dialogs_pinned";
      case td_api::premiumLimitTypePinnedArchivedChatCount::ID:
        return "dialogs_folder_pinned";
      case td_api::premiumLimitTypeCreatedPublicChatCount::ID:
        return "channels_public";
      case td_api::premiumLimitTypeCaptionLength::ID:
        return "caption_length";
      default:
        UNREACHABLE();
        return "";
    }
  }();
  return PSTRING() << "double_limits__" << key;
}

static string get_premium_source(const td_api::PremiumFeature *feature) {
  if (feature == nullptr) {
    return string();
  }

  switch (feature->get_id()) {
    case td_api::premiumFeatureIncreasedLimits::ID:
      return "double_limits";
    case td_api::premiumFeatureIncreasedFileSize::ID:
      return "more_upload";
    case td_api::premiumFeatureImprovedDownloadSpeed::ID:
      return "faster_download";
    case td_api::premiumFeatureVoiceRecognition::ID:
      return "voice_to_text";
    case td_api::premiumFeatureDisabledAds::ID:
      return "no_ads";
    case td_api::premiumFeatureUniqueReactions::ID:
      return "unique_reactions";
    case td_api::premiumFeatureUniqueStickers::ID:
      return "premium_stickers";
    case td_api::premiumFeatureAdvancedChatManagement::ID:
      return "advanced_chat_management";
    case td_api::premiumFeatureProfileBadge::ID:
      return "profile_badge";
    case td_api::premiumFeatureAnimatedProfilePhoto::ID:
      return "animated_userpics";
    default:
      UNREACHABLE();
  }
  return string();
}

static string get_premium_source(const td_api::object_ptr<td_api::PremiumSource> &source) {
  if (source == nullptr) {
    return string();
  }
  switch (source->get_id()) {
    case td_api::premiumSourceLimitExceeded::ID: {
      auto *limit_type = static_cast<const td_api::premiumSourceLimitExceeded *>(source.get())->limit_type_.get();
      return get_premium_source(limit_type);
    }
    case td_api::premiumSourceFeature::ID: {
      auto *feature = static_cast<const td_api::premiumSourceFeature *>(source.get())->feature_.get();
      return get_premium_source(feature);
    }
    case td_api::premiumSourceLink::ID: {
      auto &referrer = static_cast<const td_api::premiumSourceLink *>(source.get())->referrer_;
      if (referrer.empty()) {
        return "deeplink";
      }
      return PSTRING() << "deeplink_" << referrer;
    }
    case td_api::premiumSourceSettings::ID:
      return "settings";
    default:
      UNREACHABLE();
      return string();
  }
}

void get_premium_features(const td_api::object_ptr<td_api::PremiumSource> &source,
                          Promise<td_api::object_ptr<td_api::premiumFeatures>> &&promise) {
  auto premium_features =
      full_split(G()->shared_config().get_option_string(
                     "premium_features",
                     "double_limits,more_upload,faster_download,voice_to_text,no_ads,unique_reactions,premium_stickers,"
                     "advanced_chat_management,profile_badge,animated_userpics"),
                 ',');
  vector<td_api::object_ptr<td_api::PremiumFeature>> features;
  for (const auto &premium_feature : premium_features) {
    auto feature = [&]() -> td_api::object_ptr<td_api::PremiumFeature> {
      if (premium_feature == "double_limits") {
        return td_api::make_object<td_api::premiumFeatureIncreasedLimits>();
      }
      if (premium_feature == "more_upload") {
        return td_api::make_object<td_api::premiumFeatureIncreasedFileSize>();
      }
      if (premium_feature == "faster_download") {
        return td_api::make_object<td_api::premiumFeatureImprovedDownloadSpeed>();
      }
      if (premium_feature == "voice_to_text") {
        return td_api::make_object<td_api::premiumFeatureVoiceRecognition>();
      }
      if (premium_feature == "no_ads") {
        return td_api::make_object<td_api::premiumFeatureDisabledAds>();
      }
      if (premium_feature == "unique_reactions") {
        return td_api::make_object<td_api::premiumFeatureUniqueReactions>();
      }
      if (premium_feature == "premium_stickers") {
        return td_api::make_object<td_api::premiumFeatureUniqueStickers>();
      }
      if (premium_feature == "advanced_chat_management") {
        return td_api::make_object<td_api::premiumFeatureAdvancedChatManagement>();
      }
      if (premium_feature == "profile_badge") {
        return td_api::make_object<td_api::premiumFeatureProfileBadge>();
      }
      if (premium_feature == "animated_userpics") {
        return td_api::make_object<td_api::premiumFeatureAnimatedProfilePhoto>();
      }
      return nullptr;
    }();
    if (feature != nullptr) {
      features.push_back(std::move(feature));
    }
  }

  auto &limit_keys = get_premium_limit_keys();
  vector<td_api::object_ptr<td_api::premiumLimit>> limits;
  for (auto key : limit_keys) {
    int32 default_limit = static_cast<int32>(G()->shared_config().get_option_integer(PSLICE() << key << "_default"));
    int32 premium_limit = static_cast<int32>(G()->shared_config().get_option_integer(PSLICE() << key << "_premium"));
    if (default_limit > 0 && premium_limit > 0) {
      auto type = [&]() -> td_api::object_ptr<td_api::PremiumLimitType> {
        if (key == "channels_limit") {
          return td_api::make_object<td_api::premiumLimitTypeSupergroupCount>();
        }
        if (key == "saved_gifs_limit") {
          return td_api::make_object<td_api::premiumLimitTypeSavedAnimationCount>();
        }
        if (key == "stickers_faved_limit") {
          return td_api::make_object<td_api::premiumLimitTypeFavoriteStickerCount>();
        }
        if (key == "dialog_filters_limit") {
          return td_api::make_object<td_api::premiumLimitTypeChatFilterCount>();
        }
        if (key == "dialog_filters_chats_limit") {
          return td_api::make_object<td_api::premiumLimitTypeChatFilterChosenChatCount>();
        }
        if (key == "dialogs_pinned_limit") {
          return td_api::make_object<td_api::premiumLimitTypePinnedChatCount>();
        }
        if (key == "dialogs_folder_pinned_limit") {
          return td_api::make_object<td_api::premiumLimitTypePinnedArchivedChatCount>();
        }
        if (key == "channels_public_limit") {
          return td_api::make_object<td_api::premiumLimitTypeCreatedPublicChatCount>();
        }
        if (key == "caption_length_limit") {
          return td_api::make_object<td_api::premiumLimitTypeCaptionLength>();
        }
        UNREACHABLE();
        return nullptr;
      }();
      limits.push_back(td_api::make_object<td_api::premiumLimit>(std::move(type), default_limit, premium_limit));
    }
  }

  auto source_str = get_premium_source(source);
  // TODO use source_str

  promise.set_value(td_api::make_object<td_api::premiumFeatures>(std::move(features), std::move(limits)));
}

}  // namespace td