tdlib-fork/td/telegram/files/FileHashUploader.h

//
// 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/files/FileLoaderActor.h"
#include "td/telegram/files/FileLocation.h"
#include "td/telegram/files/ResourceManager.h"

#include "td/utils/BufferedFd.h"
#include "td/utils/crypto.h"
#include "td/utils/port/FileFd.h"
#include "td/utils/Status.h"

namespace td {

class FileHashUploader : public FileLoaderActor {
 public:
  class Callback {
   public:
    Callback() = default;
    Callback(const Callback &) = delete;
    Callback &operator=(const Callback &) = delete;
    virtual ~Callback() = default;
    virtual void on_ok(const FullRemoteFileLocation &locatioin) = 0;
    virtual void on_error(Status status) = 0;
  };

  FileHashUploader(const FullLocalFileLocation &local, int64 size, std::unique_ptr<Callback> callback)
      : local_(local), size_(size), size_left_(size), callback_(std::move(callback)) {
  }

  void set_resource_manager(ActorShared<ResourceManager> resource_manager) override {
    resource_manager_ = std::move(resource_manager);
    send_closure(resource_manager_, &ResourceManager::update_resources, resource_state_);
  }

  void update_priority(int8 priority) override {
    send_closure(resource_manager_, &ResourceManager::update_priority, priority);
  }
  void update_resources(const ResourceState &other) override {
    if (stop_flag_) {
      return;
    }
    resource_state_.update_slave(other);
    loop();
  }

 private:
  ResourceState resource_state_;
  BufferedFd<FileFd> fd_;

  FullLocalFileLocation local_;
  int64 size_;
  int64 size_left_;
  unique_ptr<Callback> callback_;

  ActorShared<ResourceManager> resource_manager_;

  enum { CalcSha, NetRequest, WaitNetResult } state_ = CalcSha;
  bool stop_flag_ = false;
  Sha256State sha256_state_;

  void start_up() override;
  Status init();

  void loop() override;

  Status loop_impl();

  Status loop_sha();

  void on_result(NetQueryPtr net_query) override;

  Status on_result_impl(NetQueryPtr net_query);
};
}  // namespace td