chrome/browser/fuchsia/element_manager_impl.cc

// Copyright 2022 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/fuchsia/element_manager_impl.h"

#include <lib/fpromise/promise.h>

#include "base/command_line.h"
#include "base/files/file_path.h"
#include "base/functional/bind.h"
#include "base/logging.h"
#include "base/strings/string_util.h"
#include "chrome/browser/chrome_browser_main.h"
#include "chrome/browser/ui/browser_list.h"
#include "url/gurl.h"
#include "url/url_constants.h"

namespace {

bool HaveBrowser() {
  return !BrowserList::GetInstance()->empty();
}

bool IsChromeBrowserUrl(const GURL& url) {
  return url.SchemeIs("fuchsia-pkg") &&
         base::EndsWith(url.path_piece(), "/chrome") &&
         url.ref_piece() == "meta/chrome.cm";
}

}  // namespace

bool ElementManagerImpl::AnnotationKeyCompare::operator()(
    const fuchsia::element::AnnotationKey& lhs,
    const fuchsia::element::AnnotationKey& rhs) const {
  return std::tie(lhs.namespace_, lhs.value) <
         std::tie(rhs.namespace_, rhs.value);
}

ElementManagerImpl::ElementManagerImpl(
    sys::OutgoingDirectory* outgoing_directory,
    NewProposalCallback callback)
    : binding_(outgoing_directory, this),
      new_proposal_callback_(std::move(callback)) {
  DCHECK(new_proposal_callback_);
  BrowserList::AddObserver(this);
}

ElementManagerImpl::~ElementManagerImpl() {
  BrowserList::RemoveObserver(this);
}

void ElementManagerImpl::ProposeElement(
    fuchsia::element::Spec spec,
    fidl::InterfaceRequest<fuchsia::element::Controller> element_controller,
    ProposeElementCallback callback) {
  if (!spec.has_component_url()) {
    callback(fuchsia::element::Manager_ProposeElement_Result::WithErr(
        fuchsia::element::ProposeElementError::INVALID_ARGS));
    return;
  }

  base::CommandLine command_line(base::CommandLine::NO_PROGRAM);

  // component_url must either specify a web resource to open in a new tab,
  // or refer to Chrome's own component manifest.
  GURL url(spec.component_url());
  if (!url.is_valid()) {
    callback(fuchsia::element::Manager_ProposeElement_Result::WithErr(
        fuchsia::element::ProposeElementError::INVALID_ARGS));
    return;
  }

  if (url.SchemeIsHTTPOrHTTPS()) {
    command_line.AppendArg(spec.component_url());
  } else if (!IsChromeBrowserUrl(url)) {
    callback(fuchsia::element::Manager_ProposeElement_Result::WithErr(
        fuchsia::element::ProposeElementError::INVALID_ARGS));
    return;
  }

  // Store the annotations to be used for all subsequent window-creation
  // actions.
  annotations_manager_.UpdateAnnotations(
      std::move(*spec.mutable_annotations()));

  // Request that the caller act on the request, e.g. by opening a new tab.
  if (!new_proposal_callback_.Run(command_line)) {
    callback(fuchsia::element::Manager_ProposeElement_Result::WithErr(
        fuchsia::element::ProposeElementError::INVALID_ARGS));
  }

  if (element_controller) {
    controller_bindings_.AddBinding(this, std::move(element_controller));
  }

  callback(fuchsia::element::Manager_ProposeElement_Result::WithResponse({}));
}

void ElementManagerImpl::UpdateAnnotations(
    std::vector<fuchsia::element::Annotation> annotations_to_set,
    std::vector<fuchsia::element::AnnotationKey> annotations_to_delete,
    UpdateAnnotationsCallback callback) {
  if (annotations_manager_.UpdateAnnotations(
          std::move(annotations_to_set), std::move(annotations_to_delete))) {
    callback(fpromise::ok());
  } else {
    callback(fpromise::error(
        fuchsia::element::UpdateAnnotationsError::INVALID_ARGS));
  }
}

void ElementManagerImpl::GetAnnotations(GetAnnotationsCallback callback) {
  callback(fpromise::ok(annotations_manager_.GetAnnotations()));
}

void ElementManagerImpl::OnBrowserRemoved(Browser* browser) {
  // If the browser was the last, clear all active bindings to notify the shell.
  bool have_browser =
      have_browser_for_test_ ? have_browser_for_test_.Run() : HaveBrowser();
  if (!have_browser) {
    controller_bindings_.CloseAll();
  }
}