127 lines
3.4 KiB
C++
127 lines
3.4 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 <cstdio>
|
|
#include <cstdint>
|
|
#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.
|
|
std::vector<char> slurp (const std::experimental::filesystem::path&);
|
|
std::vector<char> slurp (FILE *);
|
|
|
|
|
|
//-------------------------------------------------------------------------
|
|
void write (const posix::fd&, const void *restrict data, size_t bytes);
|
|
|
|
template <typename T>
|
|
void write (const posix::fd&, const T &data);
|
|
|
|
template <typename T>
|
|
void write (const posix::fd&, const T *restrict first, const T *restrict last);
|
|
|
|
//-------------------------------------------------------------------------
|
|
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 {
|
|
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:
|
|
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
|