/*
 * Copyright (C) 2021 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#pragma once

#include <cstdint>
#include <memory>
#include <optional>
#include <string>

#include <wmediumd/api.h>

#include "common/libs/fs/shared_fd.h"

namespace cuttlefish {

enum class WmediumdMessageType : uint32_t {
  kInvalid = 0,
  kAck = 1,
  kRegister = 2,
  kUnregister = 3,
  kNetlink = 4,
  kSetControl = 5,
  kTxStart = 6,
  kGetStations = 7,
  kSetSnr = 8,
  kReloadConfig = 9,
  kReloadCurrentConfig = 10,
  kStartPcap = 11,
  kStopPcap = 12,
  kStationsList = 13,
};

class WmediumdMessage {
 public:
  virtual ~WmediumdMessage() {}

  std::string Serialize(void) const;

  virtual WmediumdMessageType Type() const = 0;

 private:
  virtual void SerializeBody(std::string&) const {};
};

class WmediumdMessageSetControl : public WmediumdMessage {
 public:
  WmediumdMessageSetControl(uint32_t flags) : flags_(flags) {}

  WmediumdMessageType Type() const override {
    return WmediumdMessageType::kSetControl;
  }

 private:
  void SerializeBody(std::string& out) const override;
  uint32_t flags_;
};

class WmediumdMessageSetSnr : public WmediumdMessage {
 public:
  WmediumdMessageSetSnr(const std::string& node1, const std::string& node2,
                        uint8_t snr);

  WmediumdMessageType Type() const override {
    return WmediumdMessageType::kSetSnr;
  }

 private:
  void SerializeBody(std::string& out) const override;

  uint8_t node1_mac_[6];
  uint8_t node2_mac_[6];
  uint8_t snr_;
};

class WmediumdMessageReloadConfig : public WmediumdMessage {
 public:
  WmediumdMessageReloadConfig(const std::string& configPath)
      : config_path_(configPath) {}

  WmediumdMessageType Type() const override {
    return WmediumdMessageType::kReloadConfig;
  }

 private:
  void SerializeBody(std::string& out) const override;

  std::string config_path_;
};

class WmediumdMessageReloadCurrentConfig : public WmediumdMessage {
 public:
  WmediumdMessageReloadCurrentConfig() = default;

  WmediumdMessageType Type() const override {
    return WmediumdMessageType::kReloadCurrentConfig;
  }
};

class WmediumdMessageStartPcap : public WmediumdMessage {
 public:
  WmediumdMessageStartPcap(const std::string& pcapPath)
      : pcap_path_(pcapPath) {}

  WmediumdMessageType Type() const override {
    return WmediumdMessageType::kStartPcap;
  }

 private:
  void SerializeBody(std::string& out) const override;

  std::string pcap_path_;
};

class WmediumdMessageStopPcap : public WmediumdMessage {
 public:
  WmediumdMessageStopPcap() = default;

  WmediumdMessageType Type() const override {
    return WmediumdMessageType::kStopPcap;
  }
};

class WmediumdMessageGetStations : public WmediumdMessage {
 public:
  WmediumdMessageGetStations() = default;

  WmediumdMessageType Type() const override {
    return WmediumdMessageType::kGetStations;
  }
};

class WmediumdMessageReply : public WmediumdMessage {
 public:
  WmediumdMessageReply() = default;
  WmediumdMessageReply(WmediumdMessageType type, const std::string& data)
      : type_(type), data_(data) {}

  WmediumdMessageType Type() const override { return type_; }

  size_t Size() const { return data_.size(); }
  const char* Data() const { return data_.data(); }

 private:
  WmediumdMessageType type_;
  std::string data_;
};

class WmediumdMessageStationsList : public WmediumdMessage {
 public:
  WmediumdMessageStationsList() = default;
  static std::optional<WmediumdMessageStationsList> Parse(
      const WmediumdMessageReply& reply);

  WmediumdMessageType Type() const override {
    return WmediumdMessageType::kStationsList;
  }

  const std::vector<wmediumd_station_info>& GetStations() const {
    return station_list_;
  }

 private:
  std::vector<wmediumd_station_info> station_list_;
};

}  // namespace cuttlefish