libcruft-util/io.hpp

162 lines
4.6 KiB
C++

/*
* 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.
*
* Copyright 2010-2014 Danny Robson <danny@nerdcruft.net>
*/
#ifndef __UTIL_IO_HPP
#define __UTIL_IO_HPP
#include "platform.hpp"
#include "posix/fd.hpp"
#include <cstddef>
#include <cstdint>
#include <cstdio>
#include <experimental/filesystem>
#include <vector>
#include <streambuf>
#ifdef PLATFORM_WIN32
#include <windows.h>
#else
#define O_BINARY 0
#endif
namespace util {
//-------------------------------------------------------------------------
/// Reads an entire file into memory.
template <typename T = std::byte>
std::vector<T> slurp (const std::experimental::filesystem::path&);
template <typename T = std::byte>
std::vector<T> slurp (FILE *);
///////////////////////////////////////////////////////////////////////////
void write (const posix::fd&, const void *restrict data, size_t bytes);
//-------------------------------------------------------------------------
inline void
write (const posix::fd &dst, util::view<const char*> data)
{
write (dst, std::data (data), std::size (data));
}
//-------------------------------------------------------------------------
template <typename T>
void write (const posix::fd &_fd, const T &data)
{
return write (_fd, make_view (data));
}
///------------------------------------------------------------------------
/// writes all data from the provided view into the file-like-object
///
/// writing will continually iterate until the entire buffer has been
/// dispatched so as to avoid issues with partial writes. will block until
/// such time as the entire buffer has written.
///
/// an exception may be thrown in the event forward progress is impossible.
/// in this event the progress may not be reported to the caller. in the
/// future an exception member variable may expose the information.
template <typename DstT, typename IteratorA, typename IteratorB>
void
write (DstT &dst, util::view<IteratorA, IteratorB> src)
{
for (auto cursor = src.begin (); cursor != src.end (); ) {
cursor = dst.write (view{cursor, src.end ()});
}
}
///////////////////////////////////////////////////////////////////////////
class indenter : public std::streambuf {
protected:
std::streambuf* m_dest;
bool m_line_start;
std::string m_indent;
std::ostream* m_owner;
protected:
virtual int overflow (int ch) override;
public:
explicit indenter (std::streambuf* _dest, size_t _indent = 4);
explicit indenter (std::ostream& _dest, size_t _indent = 4);
virtual ~indenter ();
};
//-------------------------------------------------------------------------
template <typename T>
struct indented {
explicit indented (const T &_data);
const T &data;
};
//-------------------------------------------------------------------------
template <typename T>
indented<T>
make_indented (const T &_data);
//-------------------------------------------------------------------------
template <typename T>
std::ostream&
operator<< (std::ostream &os, const util::indented<T> &v);
//-------------------------------------------------------------------------
class scoped_cwd {
public:
scoped_cwd ();
~scoped_cwd ();
protected:
std::string m_original;
};
//-------------------------------------------------------------------------
class path_error : public std::runtime_error {
public:
explicit path_error (const std::experimental::filesystem::path &path);
const std::experimental::filesystem::path& path (void) const noexcept;
private:
const std::experimental::filesystem::path m_path;
};
class stream_error : public std::exception {
public:
virtual const char* what (void) const noexcept
{ return "error on C stream"; }
};
}
#ifdef PLATFORM_WIN32
#include "io_win32.hpp"
#else
#include "io_posix.hpp"
#endif
#include "io.ipp"
#endif