/*
 * Copyright 2024 Google LLC
 *
 * 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.
 */

#include "health_sync_endpoint.h"

#include "services/common/comm_session/session.h"
#include "services/common/system_task.h"
#include "services/normal/data_logging/data_logging_service.h"
#include "system/logging.h"
#include "util/attributes.h"

#define HEALTH_SYNC_ENDPOINT_ID 911
#define ACK 0x1
#define NACK 0x2

typedef enum HealthSyncEndpointCmd {
  HealthSyncEndpointCmd_Sync = 0x1,
  HealthSyncEndpointCmd_Ack = 0x11,
} HealthSyncEndpointCmd;

typedef struct PACKED HealthSyncEndpointSyncMsg {
  HealthSyncEndpointCmd cmd : 8;
  uint32_t seconds_since_sync;
} HealthSyncEndpointSyncMsg;

typedef struct PACKED HealthSyncEndpointAckMsg {
  HealthSyncEndpointCmd cmd : 8;
  uint8_t ack_nack;
} HealthSyncEndpointAckMsg;

static void prv_send_ack_nack(bool ok) {
  const HealthSyncEndpointAckMsg msg = {
    .cmd = HealthSyncEndpointCmd_Ack,
    .ack_nack = ok ? ACK : NACK,
  };

  comm_session_send_data(comm_session_get_system_session(),
                         HEALTH_SYNC_ENDPOINT_ID,
                         (uint8_t*)&msg,
                         sizeof(HealthSyncEndpointAckMsg),
                         COMM_SESSION_DEFAULT_TIMEOUT);
}

#if CAPABILITY_HAS_HEALTH_TRACKING
#include "services/normal/activity/activity_algorithm.h"

static void prv_sync_health_system_task_cb(void *unused) {
  if (activity_tracking_on()) {
    // tell the activity service to pool the minutes it's got so far
    activity_algorithm_send_minutes();
  }

  // send all data logging data
  dls_send_all_sessions();
  // ACK
  prv_send_ack_nack(true /*ok*/);
}

static void prv_handle_sync(const uint8_t *msg, size_t len) {
  if (len < sizeof(HealthSyncEndpointSyncMsg)) {
    PBL_LOG(LOG_LEVEL_ERROR, "Invalid SYNC msg received, length: %u", len);
    return;
  }

  PBL_LOG(LOG_LEVEL_DEBUG, "Received health SYNC request");

  system_task_add_callback(prv_sync_health_system_task_cb, NULL);
}

#endif

void health_sync_protocol_msg_callback(CommSession *session, const uint8_t *msg, size_t len) {
#if CAPABILITY_HAS_HEALTH_TRACKING
  if (len < 1) {
    PBL_LOG(LOG_LEVEL_ERROR, "Invalid message received, length: %u", len);
  }

  HealthSyncEndpointCmd cmd = *msg;
  switch (cmd) {
    case HealthSyncEndpointCmd_Sync:
      prv_handle_sync(msg, len);
      break;

    default:
      PBL_LOG(LOG_LEVEL_WARNING, "Unexpected command received, 0x%x", cmd);
      return;
  }
#else
  prv_send_ack_nack(false /*ok*/);
#endif
}