// Copyright 2023 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "chrome/browser/ui/performance_controls/memory_saver_utils.h"

#include "base/byte_size.h"
#include "base/check_op.h"
#include "chrome/browser/performance_manager/public/user_tuning/user_performance_tuning_manager.h"
#include "components/performance_manager/public/user_tuning/prefs.h"
#include "content/public/common/url_constants.h"

namespace memory_saver {

bool IsURLSupported(GURL url) {
  return !url.SchemeIs(content::kChromeUIScheme);
}

std::optional<::mojom::LifecycleUnitDiscardReason> GetDiscardReason(
    content::WebContents* contents) {
  auto* pre_discard_resource_usage =
      performance_manager::user_tuning::UserPerformanceTuningManager::
          PreDiscardResourceUsage::FromWebContents(contents);
  return pre_discard_resource_usage == nullptr
             ? std::nullopt
             : std::make_optional<::mojom::LifecycleUnitDiscardReason>(
                   pre_discard_resource_usage->discard_reason());
}

base::ByteSize GetDiscardedMemorySavings(content::WebContents* contents) {
  const auto* const pre_discard_resource_usage =
      performance_manager::user_tuning::UserPerformanceTuningManager::
          PreDiscardResourceUsage::FromWebContents(contents);
  if (pre_discard_resource_usage == nullptr) {
    return base::ByteSize(0);
  } else {
    return pre_discard_resource_usage->memory_footprint_estimate();
  }
}

}  // namespace memory_saver