file_transfer_progress.cpp

/*
 * MIT License
 *
 * Copyright (c) 2023 Denis Kovalchuk
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in all
 * copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 */

#include <iostream>
#include <sstream>
#include <memory>
#include <boost/timer/progress_display.hpp>
#include <ftp/client.hpp>
#include <ftp/transfer_callback.hpp>
#include <ftp/stream/istream_adapter.hpp>
#include "reply_handlers.hpp"

namespace
{

/* Display the progress in the following format:
 *
 *    0%   10   20   30   40   50   60   70   80   90   100%
 *    |----|----|----|----|----|----|----|----|----|----|
 *    **************************************
 */

class transfer_callback : public ftp::transfer_callback
{
public:
    explicit transfer_callback(std::size_t total_bytes)
        : total_bytes_(total_bytes)
    {}

private:
    void begin() override
    {
        progress_ =
            std::make_unique<boost::timer::progress_display>(static_cast<unsigned long>(total_bytes_),
                                                             std::cout);
    }

    void notify(std::size_t bytes_transferred) override
    {
        if (progress_)
        {
            *progress_ += static_cast<unsigned long>(bytes_transferred);
        }
    }

    void end() override
    {
        std::cout << std::endl;
    }

    std::size_t total_bytes_;
    std::unique_ptr<boost::timer::progress_display> progress_;
};

} // namespace

int main(int argc, char *argv[])
{
    try
    {
        ftp::client client;

        handle_reply(client.connect("localhost", 2121, "user", "password"));

        /* Prepare 32 MB for uploading. */
        std::string data(32 * 1024 * 1024, 'a');
        std::istringstream iss(data);

        transfer_callback transfer_cb(data.size());
        handle_reply(client.upload_file(ftp::istream_adapter(iss), "file", false, &transfer_cb));

        handle_reply(client.disconnect());

        return EXIT_SUCCESS;
    }
    catch (const std::exception & ex)
    {
        std::cerr << ex.what() << std::endl;
        return EXIT_FAILURE;
    }
}