librepods/linux/dbusadaptor.hpp at 939f7bacdcb9ce7cb9e9feaf7489fbc22ce960ed · kavishdevar/librepods · 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
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
#pragma once

#include <QObject>
#include <QDBusAbstractAdaptor>
#include <QDBusConnection>
#include <QDBusMessage>
#include "battery.hpp"
#include "deviceinfo.hpp"

class BatteryDBusAdaptor : public QDBusAbstractAdaptor
{
    Q_OBJECT
    Q_CLASSINFO("D-Bus Interface", "me.kavishdevar.librepods.Battery")

    // Battery levels (0-100)
    Q_PROPERTY(int LeftLevel READ leftLevel NOTIFY BatteryChanged)
    Q_PROPERTY(int RightLevel READ rightLevel NOTIFY BatteryChanged)
    Q_PROPERTY(int CaseLevel READ caseLevel NOTIFY BatteryChanged)
    Q_PROPERTY(int HeadsetLevel READ headsetLevel NOTIFY BatteryChanged)

    // Charging status
    Q_PROPERTY(bool LeftCharging READ leftCharging NOTIFY BatteryChanged)
    Q_PROPERTY(bool RightCharging READ rightCharging NOTIFY BatteryChanged)
    Q_PROPERTY(bool CaseCharging READ caseCharging NOTIFY BatteryChanged)
    Q_PROPERTY(bool HeadsetCharging READ headsetCharging NOTIFY BatteryChanged)

    // Availability (connected/detected)
    Q_PROPERTY(bool LeftAvailable READ leftAvailable NOTIFY BatteryChanged)
    Q_PROPERTY(bool RightAvailable READ rightAvailable NOTIFY BatteryChanged)
    Q_PROPERTY(bool CaseAvailable READ caseAvailable NOTIFY BatteryChanged)
    Q_PROPERTY(bool HeadsetAvailable READ headsetAvailable NOTIFY BatteryChanged)

    // Device info
    Q_PROPERTY(QString DeviceName READ deviceName NOTIFY DeviceChanged)
    Q_PROPERTY(bool Connected READ connected NOTIFY DeviceChanged)

public:
    BatteryDBusAdaptor(Battery *battery, DeviceInfo *deviceInfo, QObject *parent)
        : QDBusAbstractAdaptor(parent), m_battery(battery), m_deviceInfo(deviceInfo)
    {
        setAutoRelaySignals(true);

        // Connect battery signals to our relay
        connect(m_battery, &Battery::batteryStatusChanged, this, [this]() {
            emit BatteryChanged();
        });

        connect(m_deviceInfo, &DeviceInfo::batteryStatusChanged, this, [this]() {
            emit BatteryChanged();
        });

        connect(m_deviceInfo, &DeviceInfo::deviceNameChanged, this, [this]() {
            emit DeviceChanged();
        });
    }

    // Battery levels
    int leftLevel() const { return m_battery->getLeftPodLevel(); }
    int rightLevel() const { return m_battery->getRightPodLevel(); }
    int caseLevel() const { return m_battery->getCaseLevel(); }
    int headsetLevel() const { return m_battery->getHeadsetLevel(); }

    // Charging status
    bool leftCharging() const { return m_battery->isLeftPodCharging(); }
    bool rightCharging() const { return m_battery->isRightPodCharging(); }
    bool caseCharging() const { return m_battery->isCaseCharging(); }
    bool headsetCharging() const { return m_battery->isHeadsetCharging(); }

    // Availability
    bool leftAvailable() const { return m_battery->isLeftPodAvailable(); }
    bool rightAvailable() const { return m_battery->isRightPodAvailable(); }
    bool caseAvailable() const { return m_battery->isCaseAvailable(); }
    bool headsetAvailable() const { return m_battery->isHeadsetAvailable(); }

    // Device info - connected if device name is set and any battery is available
    QString deviceName() const { return m_deviceInfo->deviceName(); }
    bool connected() const {
        return !m_deviceInfo->deviceName().isEmpty() &&
               (leftAvailable() || rightAvailable() || headsetAvailable());
    }

public slots:
    // Method to get all battery info at once (useful for waybar)
    QVariantMap GetBatteryInfo()
    {
        QVariantMap info;
        info["left_level"] = leftLevel();
        info["left_charging"] = leftCharging();
        info["left_available"] = leftAvailable();
        info["right_level"] = rightLevel();
        info["right_charging"] = rightCharging();
        info["right_available"] = rightAvailable();
        info["case_level"] = caseLevel();
        info["case_charging"] = caseCharging();
        info["case_available"] = caseAvailable();
        info["headset_level"] = headsetLevel();
        info["headset_charging"] = headsetCharging();
        info["headset_available"] = headsetAvailable();
        info["device_name"] = deviceName();
        info["connected"] = connected();
        return info;
    }

signals:
    void BatteryChanged();
    void DeviceChanged();

private:
    Battery *m_battery;
    DeviceInfo *m_deviceInfo;
};