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

#import "ios/chrome/browser/intelligence/proto_wrappers/page_context_utils.h"

#import "components/autofill/ios/browser/autofill_util.h"
#import "ios/web/public/content_type_util.h"
#import "ios/web/public/web_state.h"
#import "url/gurl.h"

bool CanExtractPageContextForWebState(web::WebState* web_state,
                                      bool pdf_enabled) {
  if (!web_state) {
    return false;
  }

  const GURL& url = web_state->GetVisibleURL();
  const std::string mime_type = web_state->GetContentsMimeType();

  bool mime_type_ok = web::IsContentTypeHtml(mime_type) ||
                      web::IsContentTypeImage(mime_type) ||
                      (web::IsContentTypePdf(mime_type) && pdf_enabled);

  return url.SchemeIsHTTPOrHTTPS() && mime_type_ok;
}

std::optional<autofill::LocalFrameToken> DeserializeFrameIdAsLocalFrameToken(
    const std::string& serialized_id) {
  std::optional<base::UnguessableToken> token =
      autofill::DeserializeJavaScriptFrameId(serialized_id);
  return token ? std::make_optional(autofill::LocalFrameToken(*token))
               : std::nullopt;
}

std::optional<autofill::RemoteFrameToken> DeserializeFrameIdAsRemoteFrameToken(
    const std::string& serialized_id) {
  std::optional<base::UnguessableToken> token =
      autofill::DeserializeJavaScriptFrameId(serialized_id);
  return token ? std::make_optional(autofill::RemoteFrameToken(*token))
               : std::nullopt;
}