/*
 * Copyright (C) 2023 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.
 */

package com.android.server.healthconnect.fitness.aggregation;

import android.annotation.IntDef;
import android.annotation.Nullable;

import com.android.server.healthconnect.storage.utils.SqlJoin;

import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.util.ArrayList;
import java.util.List;
import java.util.Objects;

/**
 * Collection of parameters of {@link AggregateRecordRequest}.
 *
 * @hide
 */
public final class AggregateParams {
    private final String mTableName;

    /** Column used for time filtering. Start time for interval records. */
    private String mTimeColumnName;

    private final List<String> mColumnsToFetch;
    private SqlJoin mJoin;

    // Additional column used for time filtering. End time for interval records,
    // null for other records.
    @Nullable private String mExtraTimeColumnName = null;

    private String mTimeOffsetColumnName;

    private PriorityAggregationExtraParams mPriorityAggregationExtraParams;

    public AggregateParams(String tableName, List<String> columnsToFetch) {
        this(tableName, columnsToFetch, PriorityAggregationExtraParams.VALUE_TYPE_DOUBLE);
    }

    @SuppressWarnings("NullAway.Init") // TODO(b/317029272): fix this suppression
    public AggregateParams(
            String tableName,
            List<String> columnsToFetch,
            @PriorityAggregationExtraParams.ValueColumnType int priorityColumnDataType) {
        mTableName = tableName;
        mColumnsToFetch = new ArrayList<>();
        mColumnsToFetch.addAll(columnsToFetch);

        // TODO(b/277776749): remove dependency on columns orders
        mPriorityAggregationExtraParams =
                new PriorityAggregationExtraParams(columnsToFetch.get(0), priorityColumnDataType);
    }

    public SqlJoin getJoin() {
        return mJoin;
    }

    public String getTableName() {
        return mTableName;
    }

    public String getTimeColumnName() {
        return mTimeColumnName;
    }

    /**
     * Additional column used for time filtering. End time for interval records, null for other
     * records.
     */
    @Nullable
    public String getExtraTimeColumnName() {
        return mExtraTimeColumnName;
    }

    public List<String> getColumnsToFetch() {
        return mColumnsToFetch;
    }

    public String getTimeOffsetColumnName() {
        return mTimeOffsetColumnName;
    }

    /** Sets join type. */
    public AggregateParams setJoin(SqlJoin join) {
        mJoin = join;
        return this;
    }

    public AggregateParams setTimeColumnName(String columnName) {
        mTimeColumnName = columnName;
        return this;
    }

    /** Appends additional columns to fetch. */
    public AggregateParams appendAdditionalColumns(List<String> additionColumns) {
        mColumnsToFetch.addAll(additionColumns);
        return this;
    }

    /** Sets params for priority aggregation. */
    public AggregateParams setPriorityAggregationExtraParams(
            PriorityAggregationExtraParams extraParams) {
        mPriorityAggregationExtraParams = extraParams;
        return this;
    }

    /** Returns params for priority aggregation. */
    public PriorityAggregationExtraParams getPriorityAggregationExtraParams() {
        return mPriorityAggregationExtraParams;
    }

    public AggregateParams setExtraTimeColumn(String extraTimeColumn) {
        mExtraTimeColumnName = extraTimeColumn;
        return this;
    }

    public AggregateParams setOffsetColumnToFetch(String mainTimeColumnOffset) {
        Objects.requireNonNull(mainTimeColumnOffset);
        mTimeOffsetColumnName = mainTimeColumnOffset;
        return this;
    }

    /** Collections of parameters of priority AggregationRequest. */
    public static class PriorityAggregationExtraParams {

        public static final int VALUE_TYPE_LONG = 0;
        public static final int VALUE_TYPE_DOUBLE = 1;

        /** @hide */
        @IntDef({
            VALUE_TYPE_LONG,
            VALUE_TYPE_DOUBLE,
        })
        @Retention(RetentionPolicy.SOURCE)
        public @interface ValueColumnType {}

        @ValueColumnType private int mColumnToAggregateType;

        @Nullable private String mColumnToAggregateName;
        @Nullable private String mExcludeIntervalEndColumnName;
        @Nullable private String mExcludeIntervalStartColumnName;

        public PriorityAggregationExtraParams(
                String excludeIntervalStartColumnName, String excludeIntervalEndColumnName) {
            mExcludeIntervalStartColumnName =
                    Objects.requireNonNull(excludeIntervalStartColumnName);
            mExcludeIntervalEndColumnName = Objects.requireNonNull(excludeIntervalEndColumnName);
        }

        public PriorityAggregationExtraParams(
                String columnToAggregateName, @ValueColumnType int dataType) {
            mColumnToAggregateName = columnToAggregateName;
            mColumnToAggregateType = dataType;
        }

        @Nullable
        public String getExcludeIntervalStartColumnName() {
            return mExcludeIntervalStartColumnName;
        }

        @Nullable
        public String getExcludeIntervalEndColumnName() {
            return mExcludeIntervalEndColumnName;
        }

        @ValueColumnType
        public int getColumnToAggregateType() {
            return mColumnToAggregateType;
        }

        @Nullable
        public String getColumnToAggregateName() {
            return mColumnToAggregateName;
        }
    }
}