//
// Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2018
//
// Distributed under the Boost Software License, Version 1.0. (See accompanying
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
//
#pragma once

#include "td/telegram/td_api.h"
#include "td/telegram/telegram_api.h"

#include "td/telegram/files/FileId.h"
#include "td/telegram/Photo.h"
#include "td/telegram/SecretInputMedia.h"

#include "td/utils/buffer.h"
#include "td/utils/common.h"

#include <unordered_map>

namespace td {
class Td;
}  // namespace td

namespace td {

class AudiosManager {
 public:
  explicit AudiosManager(Td *td);

  int32 get_audio_duration(FileId file_id) const;

  tl_object_ptr<td_api::audio> get_audio_object(FileId file_id);

  void create_audio(FileId file_id, PhotoSize thumbnail, string file_name, string mime_type, int32 duration,
                    string title, string performer, bool replace);

  tl_object_ptr<telegram_api::InputMedia> get_input_media(FileId file_id,
                                                          tl_object_ptr<telegram_api::InputFile> input_file,
                                                          tl_object_ptr<telegram_api::InputFile> input_thumbnail) const;

  SecretInputMedia get_secret_input_media(FileId audio_file_id,
                                          tl_object_ptr<telegram_api::InputEncryptedFile> input_file,
                                          const string &caption, BufferSlice thumbnail) const;

  FileId get_audio_thumbnail_file_id(FileId file_id) const;

  void delete_audio_thumbnail(FileId file_id);

  FileId dup_audio(FileId new_id, FileId old_id);

  bool merge_audios(FileId new_id, FileId old_id, bool can_delete_old);

  template <class T>
  void store_audio(FileId file_id, T &storer) const;

  template <class T>
  FileId parse_audio(T &parser);

  string get_audio_search_text(FileId file_id) const;

 private:
  class Audio {
   public:
    string file_name;
    string mime_type;
    int32 duration = 0;
    string title;
    string performer;
    PhotoSize thumbnail;

    FileId file_id;

    bool is_changed = true;
  };

  const Audio *get_audio(FileId file_id) const;

  FileId on_get_audio(unique_ptr<Audio> new_audio, bool replace);

  Td *td_;
  std::unordered_map<FileId, unique_ptr<Audio>, FileIdHash> audios_;
};

}  // namespace td
