summaryrefslogtreecommitdiff
path: root/rockworkd/libpebble/watchlogendpoint.cpp
blob: 4b6ab26b6b8e3c374f668a7196bf67de44581808 (plain)
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
124
125
126
127
128
#include "watchlogendpoint.h"
#include "watchdatawriter.h"
#include "watchdatareader.h"
#include "pebble.h"
#include "ziphelper.h"

#include <QDir>

WatchLogEndpoint::WatchLogEndpoint(Pebble *pebble, WatchConnection *connection):
    QObject(pebble),
    m_pebble(pebble),
    m_connection(connection)
{
    qsrand(QDateTime::currentMSecsSinceEpoch());
    m_connection->registerEndpointHandler(WatchConnection::EndpointLogDump, this, "logMessageReceived");
}

void WatchLogEndpoint::fetchLogs(const QString &fileName)
{
    if (m_currentEpoch != 0) {
        qWarning() << "Already dumping logs. Not starting a second time";
        return;
    }

    m_currentFile.setFileName(fileName);
    if (!m_currentFile.open(QFile::WriteOnly | QFile::Truncate)) {
        qWarning() << "Cannot open log file for writing" << m_currentFile.fileName();
        emit logsFetched(false);
        return;
    }

    fetchForEpoch(m_currentEpoch);
}

void WatchLogEndpoint::fetchForEpoch(quint8 epoch)
{
    qDebug() << "Dumping logs for epoch" << epoch;
    QString line("=== Generation: %1 ===\n");
    line = line.arg(epoch);
    m_currentFile.write(line.toUtf8());
    RequestLogPacket packet(WatchLogEndpoint::LogCommandRequestLogs, epoch, qrand());
    m_connection->writeToPebble(WatchConnection::EndpointLogDump, packet.serialize());
}

void WatchLogEndpoint::logMessageReceived(const QByteArray &data)
{
    WatchDataReader reader(data);
    quint8 command = reader.read<quint8>();
    switch (command) {
    case LogCommandLogMessage: {
        LogMessage m(data.right(data.length() - 1));
        QString line("%1 %2 :%3> %4\n");
        line = line.arg(m.level()).arg(m.timestamp().toString("yyyy-MM-dd hh:mm:ss")).arg(m.line()).arg(m.message());
        m_currentFile.write(line.toUtf8());
        break;
    }
    case LogCommandLogMessageDone: {
        qDebug() << "Log for epoch" << m_currentEpoch << "fetched";
        m_currentEpoch++;
        if (m_currentEpoch == 0) {
            // Depending on the capabilities, there might not be a LogCommandNoLogMessages. Make sure we don't cycle endlessly
            qDebug() << "All 255 epocs fetched. Stopping";
            m_currentFile.close();
            emit logsFetched(true);
            return;
        }
        fetchForEpoch(m_currentEpoch);
        break;
    }
    case LogCommandNoLogMessages:
        qDebug() << "Log dumping finished";
        m_currentEpoch = 0;
        m_currentFile.close();
        emit logsFetched(true);
        break;
    default:
        qWarning() << "LogEndpoint: Unhandled command" << command;
    }
}

RequestLogPacket::RequestLogPacket(WatchLogEndpoint::LogCommand command, quint8 generation, quint32 cookie):
    m_command(command),
    m_generation(generation),
    m_cookie(cookie)
{

}

QByteArray RequestLogPacket::serialize() const
{
    QByteArray msg;
    WatchDataWriter writer(&msg);
    writer.write<quint8>(m_command);
    writer.write<quint8>(m_generation);
    writer.write<quint32>(m_cookie);
    return msg;
}

LogMessage::LogMessage(const QByteArray &data)
{
    WatchDataReader reader(data);
    m_cookie = reader.read<quint32>();
    m_timestamp = QDateTime::fromTime_t(reader.read<quint32>());
    int level = reader.read<quint8>();
    switch (level) {
    case 0:
        m_level = '*';
        break;
    case 1:
        m_level = 'E';
        break;
    case 50:
        m_level = 'W';
        break;
    case 100:
        m_level = 'I';
        break;
    case 200:
        m_level = 'D';
    case 250:
        m_level = 'V';
    }

    m_length = reader.read<quint8>();
    m_line = reader.read<quint16>();
    m_filename = reader.readFixedString(16);
    m_message = reader.readFixedString(m_length);
}