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

#ifndef COMPONENTS_FEED_CORE_V2_IOS_SHARED_EXPERIMENTS_TRANSLATOR_H_
#define COMPONENTS_FEED_CORE_V2_IOS_SHARED_EXPERIMENTS_TRANSLATOR_H_

#include <map>
#include <memory>
#include <optional>
#include <string>
#include <vector>

#include "components/feed/core/proto/v2/wire/chrome_feed_response_metadata.pb.h"

namespace feed {

struct ExperimentGroup {
  std::string name;
  int experiment_id;

  bool operator==(const ExperimentGroup& rhs) const;
};

// A map of trial names (key) and list of groups.
// sent from the server.
typedef std::map<std::string, std::vector<ExperimentGroup>> Experiments;

std::optional<Experiments> TranslateExperiments(
    const google::protobuf::RepeatedPtrField<feedwire::Experiment>&
        wire_experiments);

}  // namespace feed

#endif  // COMPONENTS_FEED_CORE_V2_IOS_SHARED_EXPERIMENTS_TRANSLATOR_H_