copying_source.hpp

Go to the documentation of this file.

/*
 * Copyright 2015, alex at staticlibs.net
 *
 * 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.
 */

/* 
 * File:   copying_source.hpp
 * Author: alex
 *
 * Created on October 9, 2015, 12:24 PM
 */

#ifndef STATICLIB_IO_COPYING_SOURCE_HPP
#define STATICLIB_IO_COPYING_SOURCE_HPP

#include <ios>
#include <utility>

#include "staticlib/config.hpp"

#include "staticlib/io/operations.hpp"
#include "staticlib/io/reference_sink.hpp"
#include "staticlib/io/reference_source.hpp"
#include "staticlib/io/span.hpp"

namespace staticlib {
namespace io {

template<typename Source, typename Sink>
class copying_source {
    Source src;
    Sink sink;

public:
    copying_source(Source&& src, Sink&& sink) :
    src(std::move(src)),
    sink(std::move(sink)) { }

    copying_source(const copying_source&) = delete;

    copying_source& operator=(const copying_source&) = delete;

    copying_source(copying_source&& other) STATICLIB_NOEXCEPT :
    src(std::move(other.src)),
    sink(std::move(other.sink))  { }

    copying_source& operator=(copying_source&& other) STATICLIB_NOEXCEPT {
        src = std::move(other.src);
        sink = std::move(other.sink);
        return *this;
    }

    std::streamsize read(span<char> span) {
        std::streamsize res = src.read(span);
        if (std::char_traits<char>::eof() != res) {
            write_all(sink, {span.data(), res});
        }
        return res;
    }

    std::streamsize flush() {
        return sink.flush();
    }

    Source& get_source() {
        return src;
    }

    Sink& get_sink() {
        return sink;
    }
    
};

template<typename Source, typename Sink,
        class = typename std::enable_if<!std::is_lvalue_reference<Source>::value>::type,
        class = typename std::enable_if<!std::is_lvalue_reference<Sink>::value>::type>
copying_source<Source, Sink> make_copying_source(Source&& source, Sink&& sink) {
    return copying_source<Source, Sink>(std::move(source), std::move(sink));
}

template<typename Source, typename Sink>
copying_source<reference_source<Source>, reference_sink<Sink>> 
make_copying_source(Source& source, Sink& sink) {
    return copying_source<reference_source<Source>, reference_sink<Sink>>(
            make_reference_source(source), make_reference_sink(sink));
}

} // namespace
}

#endif /* STATICLIB_IO_COPYING_SOURCE_HPP */