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

#ifndef PATCHPANEL_PROTO_UTILS_H_
#define PATCHPANEL_PROTO_UTILS_H_

#include <map>
#include <memory>
#include <string>

#include <chromeos/net-base/ipv4_address.h>
#include <chromeos/net-base/ipv6_address.h>
#include <chromeos/net-base/network_config.h>
#include <patchpanel/proto_bindings/patchpanel_service.pb.h>

#include "patchpanel/arc_service.h"
#include "patchpanel/connmark_updater.h"
#include "patchpanel/crostini_service.h"
#include "patchpanel/downstream_network_info.h"

namespace patchpanel {

// Fills a protobuf TerminaVmStartupResponse object with the given
// |termina_device| Device.
void FillTerminaAllocationProto(
    const CrostiniService::CrostiniDevice& termina_device,
    TerminaVmStartupResponse* output);

// Fills a protobuf ParallelsVmStartupResponse object with the given
// |parallels_device| Device.
void FillParallelsAllocationProto(
    const CrostiniService::CrostiniDevice& parallels_device,
    ParallelsVmStartupResponse* output);

// Fills a protobuf BruschettaVmStartupResponse object with the given
// |Bruschetta_device| Device.
void FillBruschettaAllocationProto(
    const CrostiniService::CrostiniDevice& Bruschetta_device,
    BruschettaVmStartupResponse* output);

// Fills a protobuf BorealisVmStartupResponse object with the given
// |borealis_device| device.
void FillBorealisAllocationProto(
    const CrostiniService::CrostiniDevice& borealis_device,
    BorealisVmStartupResponse* output);

// Fills a protobuf IPv4Subnet object with the IPv4CIDR.
void FillSubnetProto(const net_base::IPv4CIDR& cidr, IPv4Subnet* output);
void FillSubnetProto(const Subnet& virtual_subnet, IPv4Subnet* output);

void FillArcDeviceDnsProxyProto(
    const ArcService::ArcDevice& arc_device,
    NetworkDevice* output,
    const std::map<std::string, net_base::IPv4Address>& ipv4_addrs,
    const std::map<std::string, net_base::IPv6Address>& ipv6_addrs);

void FillDownstreamNetworkProto(
    const DownstreamNetworkInfo& downstream_network_info,
    DownstreamNetwork* output);

void FillNetworkClientInfoProto(const DownstreamClientInfo& network_client_info,
                                NetworkClientInfo* output);

net_base::NetworkConfig DeserializeNetworkConfig(
    const patchpanel::NetworkConfig& in);

std::optional<ConnmarkUpdater::Conntrack5Tuple> GetConntrack5Tuple(
    const SocketConnectionEvent& msg);

}  // namespace patchpanel
#endif  // PATCHPANEL_PROTO_UTILS_H_