acecode/src/tui/compact_notice_row.cpp at master · tmoonlight/acecode · GitHub

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
#include "tui/compact_notice_row.hpp"

#include "session/compact_notice.hpp"

namespace acecode { namespace tui {

bool append_compact_notice_row(std::vector<TuiState::Message>& rows,
                               const ChatMessage& message) {
    const auto notice = decode_compact_notice(message);
    if (!notice.has_value()) return false;

    TuiState::Message* row = nullptr;
    if (!rows.empty() &&
        rows.back().role == "compact_notice" &&
        rows.back().compact_notice_id == notice->id) {
        row = &rows.back();
        if (!message.content.empty()) {
            if (!row->content.empty()) row->content += "\n\n";
            row->content += message.content;
        }
    } else {
        TuiState::Message compact_row;
        compact_row.role = "compact_notice";
        compact_row.content = message.content;
        compact_row.compact_notice_id = notice->id;
        rows.push_back(std::move(compact_row));
        row = &rows.back();
    }

    row->compact_notice_complete =
        row->compact_notice_complete || notice->complete;
    row->expanded = !row->compact_notice_complete;
    return true;
}

bool compact_notice_row_is_expanded(const TuiState::Message& row,
                                    bool transcript_expanded) {
    return row.role == "compact_notice" &&
        (!row.compact_notice_complete || row.expanded || transcript_expanded);
}

bool toggle_completed_compact_notice_row(TuiState::Message& row) {
    if (row.role != "compact_notice" || !row.compact_notice_complete) {
        return false;
    }
    row.expanded = !row.expanded;
    return true;
}

}} // namespace acecode::tui