-
Notifications
You must be signed in to change notification settings - Fork 2
/
Outputter.cpp
123 lines (108 loc) · 3.48 KB
/
Outputter.cpp
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
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
#include "Outputter.hpp"
#include "td/utils/Slice-decl.h"
#include "td/utils/port/Clocks.h"
#include "td/utils/format.h"
#include "td/utils/format.h"
#include "ChatWindow.hpp"
#include "windows/unicode.h"
namespace td {
namespace format {
template <typename T>
struct ZeroPaddedInt {
T value;
td::uint32 pad;
};
StringBuilder &operator<<(StringBuilder &sb, ZeroPaddedInt<td::uint32> v) {
char buf[64];
auto l = sprintf(buf, "%0*u", v.pad, v.value);
return sb << td::CSlice(buf, buf + l);
}
} // namespace format
} // namespace td
namespace tdcurses {
Outputter &Outputter::operator<<(Date date) {
time_t t = date.date;
struct tm d;
localtime_r(&t, &d);
*this << "[";
auto now = td::Clocks::system();
if (now - date.date > 60 * 86400) {
*this << (d.tm_year + 1900) << "/";
}
if (now - date.date > 36000) {
*this << td::format::ZeroPaddedInt<td::uint32>{static_cast<td::uint32>(d.tm_mon + 1), 2} << "/";
*this << td::format::ZeroPaddedInt<td::uint32>{static_cast<td::uint32>(d.tm_mday), 2} << " ";
}
*this << td::format::ZeroPaddedInt<td::uint32>{static_cast<td::uint32>(d.tm_hour), 2} << ":"
<< td::format::ZeroPaddedInt<td::uint32>{static_cast<td::uint32>(d.tm_min), 2} << ":"
<< td::format::ZeroPaddedInt<td::uint32>{static_cast<td::uint32>(d.tm_sec), 2};
return *this << "]";
}
void Outputter::add_markup(td::int32 attr, size_t f, size_t l, td::int32 val) {
markup_.emplace_back(f, l, attr, val);
}
void Outputter::set_attr_ex(td::int32 attr, td::int32 attr_val) {
auto val = (td::int32)attr;
if (attr_val == -1) {
CHECK(args_[val].size() > 0);
add_markup(attr, args_[val].back().first, sb_.as_cslice().size(), args_[val].back().second);
args_[val].pop_back();
if (args_[val].size() > 0) {
args_[val].back().first = sb_.as_cslice().size();
}
} else {
if (args_[val].size() > 0) {
add_markup(attr, args_[val].back().first, sb_.as_cslice().size(), args_[val].back().second);
}
args_[val].emplace_back(sb_.as_cslice().size(), attr_val);
}
}
void Outputter::set_attr(td::int32 attr, ChangeBool mode) {
td::int32 attr_val = 0;
auto val = (td::int32)attr;
switch (mode) {
case ChangeBool::Disable:
attr_val = 0;
break;
case ChangeBool::Enable:
attr_val = 1;
break;
case ChangeBool::Revert:
attr_val = -1;
break;
case ChangeBool::Invert:
attr_val = args_[val].size() > 0 ? !args_[val].back().second : 1;
break;
}
set_attr_ex(attr, attr_val);
}
Outputter &Outputter::operator<<(FgColor color) {
set_attr_ex(TickitPenAttr::TICKIT_PEN_FG, (td::int32)color.color);
return *this;
}
Outputter &Outputter::operator<<(BgColor color) {
set_attr_ex(TickitPenAttr::TICKIT_PEN_BG, (td::int32)color.color);
return *this;
}
std::vector<windows::MarkupElement> Outputter::markup() {
auto res = markup_;
for (size_t i = 0; i < args_.size(); i++) {
if (args_[i].size() > 0) {
res.emplace_back(args_[i].back().first, sb_.as_cslice().size(), (td::int32)i, args_[i].back().second);
}
}
return res;
}
const td::td_api::message *Outputter::get_message(td::int64 chat_id, td::int64 message_id) {
if (!cur_chat_) {
return nullptr;
}
return cur_chat_->get_message_as_message(chat_id, message_id);
}
Outputter &Outputter::operator<<(const RightPad &x) {
td::int32 code = RIGHT_ALIGN_BLOCK_START + (td::int32)x.data.size();
char buf[6];
utf8_code_to_str(code, buf);
return *this << td::CSlice((char *)buf) << x.data;
}
} // namespace tdcurses