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

#include "services/viz/public/cpp/compositing/compositor_frame_transition_directive_mojom_traits.h"

#include <utility>
#include <vector>

#include "base/time/time.h"
#include "base/unguessable_token.h"
#include "components/viz/common/quads/compositor_frame_transition_directive.h"
#include "components/viz/common/quads/compositor_render_pass.h"
#include "mojo/public/cpp/base/time_mojom_traits.h"
#include "mojo/public/cpp/base/unguessable_token_mojom_traits.h"
#include "services/viz/public/cpp/compositing/compositor_render_pass_id_mojom_traits.h"
#include "services/viz/public/cpp/compositing/view_transition_element_resource_id_mojom_traits.h"
#include "services/viz/public/mojom/compositing/compositor_frame_transition_directive.mojom-shared.h"
#include "third_party/blink/public/common/tokens/tokens_mojom_traits.h"
#include "ui/gfx/display_color_spaces.h"
#include "ui/gfx/mojom/display_color_spaces_mojom_traits.h"

namespace viz {
using NavigationId = base::UnguessableToken;
}

namespace mojo {

// static
viz::mojom::CompositorFrameTransitionDirectiveType
EnumTraits<viz::mojom::CompositorFrameTransitionDirectiveType,
           viz::CompositorFrameTransitionDirective::Type>::
    ToMojom(viz::CompositorFrameTransitionDirective::Type type) {
  switch (type) {
    case viz::CompositorFrameTransitionDirective::Type::kSave:
      return viz::mojom::CompositorFrameTransitionDirectiveType::kSave;
    case viz::CompositorFrameTransitionDirective::Type::kAnimateRenderer:
      return viz::mojom::CompositorFrameTransitionDirectiveType::
          kAnimateRenderer;
    case viz::CompositorFrameTransitionDirective::Type::kRelease:
      return viz::mojom::CompositorFrameTransitionDirectiveType::kRelease;
  }
  NOTREACHED();
}

// static
viz::CompositorFrameTransitionDirective::Type
EnumTraits<viz::mojom::CompositorFrameTransitionDirectiveType,
           viz::CompositorFrameTransitionDirective::Type>::
    FromMojom(viz::mojom::CompositorFrameTransitionDirectiveType input) {
  switch (input) {
    case viz::mojom::CompositorFrameTransitionDirectiveType::kSave:
      return viz::CompositorFrameTransitionDirective::Type::kSave;
    case viz::mojom::CompositorFrameTransitionDirectiveType::kAnimateRenderer:
      return viz::CompositorFrameTransitionDirective::Type::kAnimateRenderer;
    case viz::mojom::CompositorFrameTransitionDirectiveType::kRelease:
      return viz::CompositorFrameTransitionDirective::Type::kRelease;
  }
  NOTREACHED();
}

// static
base::expected<void, DeserializationError> StructTraits<
    viz::mojom::CompositorFrameTransitionDirectiveSharedElementDataView,
    viz::CompositorFrameTransitionDirective::SharedElement>::
    Read(viz::mojom::CompositorFrameTransitionDirectiveSharedElementDataView
             data,
         viz::CompositorFrameTransitionDirective::SharedElement* out) {
  if (!data.ReadRenderPassId(&out->render_pass_id) ||
      !data.ReadViewTransitionElementResourceId(
          &out->view_transition_element_resource_id)) {
    return base::unexpected(DeserializationError());
  }
  return base::ok();
}

// static
base::expected<void, DeserializationError>
StructTraits<viz::mojom::CompositorFrameTransitionDirectiveDataView,
             viz::CompositorFrameTransitionDirective>::
    Read(viz::mojom::CompositorFrameTransitionDirectiveDataView data,
         viz::CompositorFrameTransitionDirective* out) {
  uint32_t sequence_id = data.sequence_id();
  bool maybe_cross_frame_sink = data.maybe_cross_frame_sink();

  blink::ViewTransitionToken transition_token;
  viz::CompositorFrameTransitionDirective::Type type;
  std::vector<viz::CompositorFrameTransitionDirective::SharedElement>
      shared_elements;
  gfx::DisplayColorSpaces display_color_spaces;
  if (!data.ReadTransitionToken(&transition_token) || !data.ReadType(&type) ||
      !data.ReadSharedElements(&shared_elements) ||
      !data.ReadDisplayColorSpaces(&display_color_spaces)) {
    return base::unexpected(DeserializationError());
  }

  // The renderer should never create a directive other than save with shared
  // elements.
  if (type != viz::CompositorFrameTransitionDirective::Type::kSave &&
      !shared_elements.empty()) {
    return base::unexpected(DeserializationError());
  }

  switch (type) {
    case viz::CompositorFrameTransitionDirective::Type::kSave:
      *out = viz::CompositorFrameTransitionDirective::CreateSave(
          transition_token, maybe_cross_frame_sink, sequence_id,
          std::move(shared_elements), display_color_spaces,
          data.delay_layer_tree_view_deletion());
      break;
    case viz::CompositorFrameTransitionDirective::Type::kAnimateRenderer:
      *out = viz::CompositorFrameTransitionDirective::CreateAnimate(
          transition_token, maybe_cross_frame_sink, sequence_id,
          data.delay_layer_tree_view_deletion());
      break;
    case viz::CompositorFrameTransitionDirective::Type::kRelease:
      *out = viz::CompositorFrameTransitionDirective::CreateRelease(
          transition_token, maybe_cross_frame_sink, sequence_id,
          data.delay_layer_tree_view_deletion());
  }

  return base::ok();
}

}  // namespace mojo