// Copyright 2015 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/update_client/task_update.h"

#include <utility>

#include "base/functional/bind.h"
#include "base/location.h"
#include "base/task/sequenced_task_runner.h"
#include "components/update_client/update_client.h"
#include "components/update_client/update_engine.h"

namespace update_client {

TaskUpdate::TaskUpdate(
    scoped_refptr<UpdateEngine> update_engine,
    bool is_foreground,
    bool is_install,
    const std::vector<std::string>& ids,
    UpdateClient::CrxDataCallback crx_data_callback,
    UpdateClient::CrxStateChangeCallback crx_state_change_callback,
    Callback callback)
    : update_engine_(update_engine),
      is_foreground_(is_foreground),
      is_install_(is_install),
      ids_(ids),
      crx_data_callback_(std::move(crx_data_callback)),
      crx_state_change_callback_(crx_state_change_callback),
      callback_(std::move(callback)) {}

TaskUpdate::~TaskUpdate() = default;

void TaskUpdate::Run() {
  DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);

  if (ids_.empty()) {
    TaskComplete(Error::INVALID_ARGUMENT);
    return;
  }

  if (cancelled_) {
    TaskComplete(Error::UPDATE_CANCELED);
    return;
  }

  cancel_callback_ = update_engine_->Update(
      is_foreground_, is_install_, ids_, std::move(crx_data_callback_),
      std::move(crx_state_change_callback_),
      base::BindOnce(&TaskUpdate::TaskComplete, this));
}

void TaskUpdate::Cancel() {
  DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
  cancelled_ = true;
  cancel_callback_.Run();
}

std::vector<std::string> TaskUpdate::GetIds() const {
  DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
  return ids_;
}

void TaskUpdate::TaskComplete(Error error) {
  DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);

  base::SequencedTaskRunner::GetCurrentDefault()->PostTask(
      FROM_HERE, base::BindOnce(&TaskUpdate::RunCallback, this, error));
}

void TaskUpdate::RunCallback(Error error) {
  DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
  std::move(callback_).Run(this, error);
}

}  // namespace update_client