Gadgetbridge/app/src/main/java/nodomain/freeyourgadget/gadgetbridge/service/devices/xiaomi/activity/impl/ManualSamplesParser.java

/*  Copyright (C) 2024 José Rebelo

    This file is part of Gadgetbridge.

    Gadgetbridge is free software: you can redistribute it and/or modify
    it under the terms of the GNU Affero General Public License as published
    by the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    Gadgetbridge is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU Affero General Public License for more details.

    You should have received a copy of the GNU Affero General Public License
    along with this program.  If not, see <https://www.gnu.org/licenses/>. */
package nodomain.freeyourgadget.gadgetbridge.service.devices.xiaomi.activity.impl;

import android.widget.Toast;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.nio.ByteBuffer;
import java.nio.ByteOrder;
import java.util.ArrayList;
import java.util.List;

import nodomain.freeyourgadget.gadgetbridge.GBApplication;
import nodomain.freeyourgadget.gadgetbridge.database.DBHandler;
import nodomain.freeyourgadget.gadgetbridge.database.DBHelper;
import nodomain.freeyourgadget.gadgetbridge.devices.xiaomi.XiaomiManualSampleProvider;
import nodomain.freeyourgadget.gadgetbridge.entities.DaoSession;
import nodomain.freeyourgadget.gadgetbridge.entities.Device;
import nodomain.freeyourgadget.gadgetbridge.entities.User;
import nodomain.freeyourgadget.gadgetbridge.entities.XiaomiManualSample;
import nodomain.freeyourgadget.gadgetbridge.impl.GBDevice;
import nodomain.freeyourgadget.gadgetbridge.service.devices.xiaomi.XiaomiSupport;
import nodomain.freeyourgadget.gadgetbridge.service.devices.xiaomi.activity.XiaomiActivityFileId;
import nodomain.freeyourgadget.gadgetbridge.service.devices.xiaomi.activity.XiaomiActivityParser;
import nodomain.freeyourgadget.gadgetbridge.util.GB;

public class ManualSamplesParser extends XiaomiActivityParser {
    private static final Logger LOG = LoggerFactory.getLogger(ManualSamplesParser.class);

    @Override
    public boolean parse(final XiaomiSupport support, final XiaomiActivityFileId fileId, final byte[] bytes) {
        if (fileId.getVersion() != 2) {
            LOG.warn("Unknown manual samples version {}", fileId.getVersion());
            return false;
        }

        final ByteBuffer buf = ByteBuffer.wrap(bytes).order(ByteOrder.LITTLE_ENDIAN);

        // Looks like there is no header, it starts right away with samples:
        // 8A90A965 12 63 <- spo2
        // ... multiple 13 00
        // C793A965 13 00
        // 9698A965 13 1C <- stress
        // E79CA965 44 5A0E0000 <- body temperature
        // 729FA965 44 590E0000 <- body temperature

        final List<XiaomiManualSample> samples = new ArrayList<>();

        while (buf.position() < buf.limit()) {
            final int timestamp = buf.getInt();
            final int type = buf.get() & 0xff;

            final int value;
            switch (type) {
                case XiaomiManualSampleProvider.TYPE_HR:
                case XiaomiManualSampleProvider.TYPE_SPO2:
                case XiaomiManualSampleProvider.TYPE_STRESS:
                    value = buf.get() & 0xff;
                    break;
                case XiaomiManualSampleProvider.TYPE_TEMPERATURE:
                    value = buf.getInt();
                    break;
                default:
                    LOG.warn("Unknown sample type {}", type);
                    // We need to abort parsing, as we don't know the sample size
                    return false;
            }

            if (value == 0) {
                continue;
            }

            LOG.debug("Got manual sample: ts={} type={} value={}", timestamp, type, value);

            final XiaomiManualSample sample = new XiaomiManualSample();
            sample.setTimestamp(timestamp * 1000L);
            sample.setType(type);
            sample.setValue(value);

            samples.add(sample);
        }

        try (DBHandler handler = GBApplication.acquireDB()) {
            final DaoSession session = handler.getDaoSession();

            final GBDevice gbDevice = support.getDevice();
            final Device device = DBHelper.getDevice(gbDevice, session);
            final User user = DBHelper.getUser(session);

            for (final XiaomiManualSample sample : samples) {
                sample.setDevice(device);
                sample.setUser(user);
            }

            final XiaomiManualSampleProvider sampleProvider = new XiaomiManualSampleProvider(gbDevice, session);
            sampleProvider.addSamples(samples);
        } catch (final Exception e) {
            GB.toast(support.getContext(), "Error saving manual samples", Toast.LENGTH_LONG, GB.ERROR);
            LOG.error("Error saving manual samples", e);
            return false;
        }

        return true;
    }
}