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

#include "extensions/browser/scoped_active_install.h"

#include "extensions/browser/active_install_data.h"
#include "extensions/buildflags/buildflags.h"

static_assert(BUILDFLAG(ENABLE_EXTENSIONS_CORE));

namespace extensions {

ScopedActiveInstall::ScopedActiveInstall(InstallTracker* tracker,
                                         const ActiveInstallData& install_data)
    : tracker_(tracker), extension_id_(install_data.extension_id) {
  Init();
  tracker_->AddActiveInstall(install_data);
}

ScopedActiveInstall::ScopedActiveInstall(InstallTracker* tracker,
                                         const std::string& extension_id)
    : tracker_(tracker), extension_id_(extension_id) {
  Init();
}

ScopedActiveInstall::~ScopedActiveInstall() {
  if (tracker_) {
    tracker_->RemoveActiveInstall(extension_id_);
  }
}

void ScopedActiveInstall::CancelDeregister() {
  tracker_observation_.Reset();
  tracker_ = nullptr;
}

void ScopedActiveInstall::Init() {
  DCHECK(!extension_id_.empty());
  DCHECK(tracker_);
  tracker_observation_.Observe(tracker_.get());
}

void ScopedActiveInstall::OnShutdown() {
  CancelDeregister();
}

}  // namespace extensions