ProtoToAggregateDataRow.kt

/*
 * Copyright (C) 2022 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
@file:RestrictTo(RestrictTo.Scope.LIBRARY)

package androidx.health.connect.client.impl.converters.aggregate

import androidx.annotation.RestrictTo
import androidx.health.connect.client.aggregate.AggregationResult
import androidx.health.connect.client.aggregate.AggregationResultGroupedByDuration
import androidx.health.connect.client.aggregate.AggregationResultGroupedByPeriod
import androidx.health.connect.client.records.metadata.DataOrigin
import androidx.health.platform.client.proto.DataProto
import java.time.Instant
import java.time.LocalDateTime
import java.time.ZoneOffset

// ZoneOffset.ofTotalSeconds() has been banned but safe here for serialization.
@SuppressWarnings("GoodTime")
fun DataProto.AggregateDataRow.toAggregateDataRowGroupByDuration():
    AggregationResultGroupedByDuration {
    require(hasStartTimeEpochMs()) { "start time must be set" }
    require(hasEndTimeEpochMs()) { "end time must be set" }

    return AggregationResultGroupedByDuration(
        result = retrieveAggregateDataRow(),
        startTime = Instant.ofEpochMilli(startTimeEpochMs),
        endTime = Instant.ofEpochMilli(endTimeEpochMs),
        zoneOffset = ZoneOffset.ofTotalSeconds(zoneOffsetSeconds)
    )
}

fun DataProto.AggregateDataRow.toAggregateDataRowGroupByPeriod(): AggregationResultGroupedByPeriod {
    require(hasStartLocalDateTime()) { "start time must be set" }
    require(hasEndLocalDateTime()) { "end time must be set" }

    return AggregationResultGroupedByPeriod(
        result = retrieveAggregateDataRow(),
        startTime = LocalDateTime.parse(startLocalDateTime),
        endTime = LocalDateTime.parse(endLocalDateTime),
    )
}

fun DataProto.AggregateDataRow.retrieveAggregateDataRow() =
    AggregationResult(
        longValues = longValuesMap,
        doubleValues = doubleValuesMap,
        dataOrigins = dataOriginsList.mapTo(HashSet()) { DataOrigin(it.applicationId) }
    )