blob: 33335051b1af1a5e3bbce65b0cf093e9fa82a238 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
|
/*
* This source code is released into the public domain.
*/
module;
#include <iostream>
export module nihil.util:capture_stream;
namespace nihil {
/*
* Capture output written to a stream and redirect it to an internal string
* buffer. Call .str() to get the data written. Call .release() to stop
* capturing (or simply delete the capture_stream object).
*/
export template<typename Char, typename Traits>
struct capture_stream {
capture_stream(std::basic_ostream<Char, Traits> &stream)
: m_stream(&stream)
{
m_old_streambuf = m_stream->rdbuf();
m_stream->rdbuf(m_buffer.rdbuf());
}
~capture_stream() {
if (m_old_streambuf == nullptr)
return;
release();
}
/*
* Release this capture, returning the stream to its previous state.
*/
auto release(this capture_stream &self) -> void
{
if (self.m_old_streambuf == nullptr)
throw std::logic_error(
"release() called on empty capture_stream");
self.m_stream->rdbuf(self.m_old_streambuf);
self.m_old_streambuf = nullptr;
}
/*
* Get the data which has been written to the stream.
*/
[[nodiscard]] auto str(this capture_stream const &self)
-> std::basic_string_view<Char, Traits>
{
return self.m_buffer.view();
}
private:
std::basic_ostringstream<Char, Traits> m_buffer;
std::basic_ostream<Char, Traits> *m_stream;
std::streambuf *m_old_streambuf;
};
} // namespace nihil
|
