streambuf_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:   streambuf_source.hpp
 * Author: alex
 *
 * Created on October 9, 2015, 12:32 PM
 */

#ifndef STATICLIB_IO_STREAMBUF_SOURCE_HPP
#define STATICLIB_IO_STREAMBUF_SOURCE_HPP

#include <ios>
#include <streambuf>

#include "staticlib/config.hpp"

#include "staticlib/io/span.hpp"

namespace staticlib {
namespace io {

class streambuf_source {
    std::streambuf* streambuf;

public:
    explicit streambuf_source(std::streambuf* streambuf) :
    streambuf(streambuf) { }

    streambuf_source(const streambuf_source&) = delete;

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

    streambuf_source(streambuf_source&& other) STATICLIB_NOEXCEPT :
    streambuf(other.streambuf) {
        other.streambuf = nullptr;
    }

    streambuf_source& operator=(streambuf_source&& other) STATICLIB_NOEXCEPT {
        streambuf = other.streambuf;
        other.streambuf = nullptr;
        return *this;
    }

    std::streamsize read(span<char> span) {
        std::streamsize res = streambuf->sgetn(span.data(), span.size_signed());
        if (res > 0) {
            return res;
        } else if (0 == res && std::char_traits<char>::eof() == streambuf->sbumpc()) {
            return std::char_traits<char>::eof();
        } else {
            streambuf->sungetc();
            return res;
        }
    }
    
};

} // namespace
}

#endif /* STATICLIB_IO_STREAMBUF_SOURCE_HPP */