components/attribution_reporting/event_trigger_data.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 "components/attribution_reporting/event_trigger_data.h"

#include <stdint.h>

#include <utility>

#include "base/check_op.h"
#include "base/types/expected.h"
#include "base/types/expected_macros.h"
#include "base/values.h"
#include "components/attribution_reporting/constants.h"
#include "components/attribution_reporting/filters.h"
#include "components/attribution_reporting/parsing_utils.h"
#include "components/attribution_reporting/trigger_registration_error.mojom.h"

namespace attribution_reporting {

namespace {

using ::attribution_reporting::mojom::TriggerRegistrationError;

}  // namespace

// static
base::expected<EventTriggerData, TriggerRegistrationError>
EventTriggerData::FromJSON(base::Value& value) {
  base::Value::Dict* dict = value.GetIfDict();
  if (!dict) {
    return base::unexpected(
        TriggerRegistrationError::kEventTriggerDataWrongType);
  }

  EventTriggerData out;

  ASSIGN_OR_RETURN(
      out.data,
      ParseUint64(*dict, kTriggerData).transform(&ValueOrZero<uint64_t>),
      [](ParseError) {
        return TriggerRegistrationError::kEventTriggerDataValueInvalid;
      });

  ASSIGN_OR_RETURN(out.priority, ParsePriority(*dict), [](ParseError) {
    return TriggerRegistrationError::kEventPriorityValueInvalid;
  });

  ASSIGN_OR_RETURN(out.dedup_key, ParseDeduplicationKey(*dict), [](ParseError) {
    return TriggerRegistrationError::kEventDedupKeyValueInvalid;
  });

  ASSIGN_OR_RETURN(out.filters, FilterPair::FromJSON(*dict));

  return out;
}

EventTriggerData::EventTriggerData() = default;

EventTriggerData::EventTriggerData(uint64_t data,
                                   int64_t priority,
                                   std::optional<uint64_t> dedup_key,
                                   FilterPair filters)
    : data(data),
      priority(priority),
      dedup_key(dedup_key),
      filters(std::move(filters)) {}

base::Value::Dict EventTriggerData::ToJson() const {
  base::Value::Dict dict;

  filters.SerializeIfNotEmpty(dict);

  SerializeUint64(dict, kTriggerData, data);
  SerializePriority(dict, priority);
  SerializeDeduplicationKey(dict, dedup_key);

  return dict;
}

// static
base::expected<EventTriggerValue, TriggerRegistrationError>
EventTriggerValue::Parse(const base::Value::Dict& dict) {
  const base::Value* v = dict.Find(kValue);
  if (!v) {
    return EventTriggerValue();
  }

  ASSIGN_OR_RETURN(uint32_t value, ParseUint32(*v), [](ParseError) {
    return TriggerRegistrationError::kEventValueInvalid;
  });

  if (value == 0) {
    return base::unexpected(TriggerRegistrationError::kEventValueInvalid);
  }

  return EventTriggerValue(value);
}

EventTriggerValue::EventTriggerValue(uint32_t value) : value_(value) {
  CHECK_GT(value_, 0u);
}

void EventTriggerValue::Serialize(base::Value::Dict& dict) const {
  dict.Set(kValue, Uint32ToJson(value_));
}

}  // namespace attribution_reporting