// Copyright 2022 Google LLC
//
// 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
//
//     https://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.

#ifndef THIRD_PARTY_NEARBY_SHARING_ATTACHMENT_H_
#define THIRD_PARTY_NEARBY_SHARING_ATTACHMENT_H_

#include <cstdint>

#include "absl/strings/string_view.h"
#include "proto/sharing_enums.pb.h"
#include "sharing/common/nearby_share_enums.h"

namespace nearby::sharing {

// A single attachment to be sent by / received from a ShareTarget, can be
// either a file or text.
class Attachment {
 public:
  enum class Family {
    kFile,
    kText,
    kWifiCredentials,
    kMaxValue = kWifiCredentials
  };

  Attachment(
      Family family, int64_t size, int32_t batch_id,
      location::nearby::proto::sharing::AttachmentSourceType source_type);
  Attachment(
      int64_t id, Family family, int64_t size, int32_t batch_id,
      location::nearby::proto::sharing::AttachmentSourceType source_type);
  virtual ~Attachment() = default;

  int64_t id() const { return id_; }
  Family family() const { return family_; }
  int64_t size() const { return size_; }
  void set_size(int64_t size) { size_ = size; }
  int32_t batch_id() const { return batch_id_; }
  location::nearby::proto::sharing::AttachmentSourceType source_type() const {
    return source_type_;
  }

  virtual absl::string_view GetDescription() const = 0;
  virtual ShareType GetShareType() const = 0;

 private:
  const int64_t id_;
  const Family family_;
  const int32_t batch_id_;
  const location::nearby::proto::sharing::AttachmentSourceType source_type_;
  int64_t size_;
};

}  // namespace nearby::sharing

#endif  // THIRD_PARTY_NEARBY_SHARING_ATTACHMENT_H_