// Copyright 2016 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "components/offline_pages/core/background/mark_attempt_completed_task.h"

#include <utility>

#include "base/functional/bind.h"
#include "base/time/time.h"

namespace offline_pages {

MarkAttemptCompletedTask::MarkAttemptCompletedTask(
    RequestQueueStore* store,
    int64_t request_id,
    FailState fail_state,
    RequestQueueStore::UpdateCallback callback)
    : UpdateRequestTask(store, request_id, std::move(callback)),
      fail_state_(fail_state) {}

MarkAttemptCompletedTask::~MarkAttemptCompletedTask() = default;

void MarkAttemptCompletedTask::UpdateRequestImpl(
    UpdateRequestsResult read_result) {
  if (!ValidateReadResult(read_result)) {
    CompleteWithResult(std::move(read_result));
    return;
  }

  // It is perfectly fine to reuse the read_result.updated_items collection, as
  // it is owned by this callback and will be destroyed when out of scope.
  read_result.updated_items[0].MarkAttemptCompleted(fail_state_);
  store()->UpdateRequests(
      read_result.updated_items,
      base::BindOnce(&MarkAttemptCompletedTask::CompleteWithResult,
                     GetWeakPtr()));
}

}  // namespace offline_pages