pdf/test/pdf_ink_test_helpers.cc

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

#include "pdf/test/pdf_ink_test_helpers.h"

#include <string>
#include <utility>

#include "base/values.h"

namespace chrome_pdf {

base::Value::Dict CreateSetAnnotationBrushMessageForTesting(
    const std::string& type,
    double size,
    const TestAnnotationBrushMessageParams* params) {
  base::Value::Dict message;
  message.Set("type", "setAnnotationBrush");

  base::Value::Dict data;
  data.Set("type", type);
  data.Set("size", size);
  if (params) {
    base::Value::Dict color;
    color.Set("r", params->color_r);
    color.Set("g", params->color_g);
    color.Set("b", params->color_b);
    data.Set("color", std::move(color));
  }
  message.Set("data", std::move(data));
  return message;
}

base::Value::Dict CreateSetAnnotationModeMessageForTesting(bool enable) {
  base::Value::Dict message;
  message.Set("type", "setAnnotationMode");
  message.Set("enable", enable);
  return message;
}

}  // namespace chrome_pdf