File: ai_crx_component.cc

package info (click to toggle)
chromium 140.0.7339.185-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 6,193,740 kB
  • sloc: cpp: 35,093,945; ansic: 7,161,670; javascript: 4,199,694; python: 1,441,797; asm: 949,904; xml: 747,515; pascal: 187,748; perl: 88,691; sh: 88,248; objc: 79,953; sql: 52,714; cs: 44,599; fortran: 24,137; makefile: 22,114; tcl: 15,277; php: 13,980; yacc: 9,000; ruby: 7,485; awk: 3,720; lisp: 3,096; lex: 1,327; ada: 727; jsp: 228; sed: 36
file content (113 lines) | stat: -rw-r--r-- 4,016 bytes parent folder | download | duplicates (5)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
// Copyright 2025 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "chrome/browser/ai/ai_crx_component.h"

#include <memory>

#include "components/update_client/crx_update_item.h"

namespace on_device_ai {

namespace {

bool IsDownloadEvent(const component_updater::CrxUpdateItem& item) {
  // See class comment: components/update_client/component.h
  switch (item.state) {
    case update_client::ComponentState::kDownloading:
    case update_client::ComponentState::kUpdating:
    case update_client::ComponentState::kUpToDate:
    case update_client::ComponentState::kDownloadingDiff:
    case update_client::ComponentState::kUpdatingDiff:
      return item.downloaded_bytes >= 0 && item.total_bytes >= 0;
    case update_client::ComponentState::kNew:
    case update_client::ComponentState::kChecking:
    case update_client::ComponentState::kCanUpdate:
    case update_client::ComponentState::kUpdated:
    case update_client::ComponentState::kUpdateError:
    case update_client::ComponentState::kRun:
    case update_client::ComponentState::kLastStatus:
      return false;
  }
}

bool IsAlreadyInstalled(const component_updater::CrxUpdateItem& item) {
  // See class comment: components/update_client/component.h
  switch (item.state) {
    case update_client::ComponentState::kUpdated:
    case update_client::ComponentState::kUpToDate:
      return true;
    case update_client::ComponentState::kNew:
    case update_client::ComponentState::kChecking:
    case update_client::ComponentState::kCanUpdate:
    case update_client::ComponentState::kDownloading:
    case update_client::ComponentState::kUpdating:
    case update_client::ComponentState::kUpdateError:
    case update_client::ComponentState::kRun:
    case update_client::ComponentState::kDownloadingDiff:
    case update_client::ComponentState::kUpdatingDiff:
    case update_client::ComponentState::kLastStatus:
      return false;
  }
}

}  // namespace

// static
base::flat_set<std::unique_ptr<AIModelDownloadProgressManager::Component>>
AICrxComponent::FromComponentIds(
    component_updater::ComponentUpdateService* component_update_service,
    base::flat_set<std::string> component_ids) {
  base::flat_set<std::unique_ptr<Component>> components;
  components.reserve(component_ids.size());

  for (std::string component_id : component_ids) {
    components.emplace(std::make_unique<AICrxComponent>(
        component_update_service, std::move(component_id)));
  }

  return components;
}

AICrxComponent::AICrxComponent(
    component_updater::ComponentUpdateService* component_update_service,
    std::string component_id)
    : component_id_(std::move(component_id)) {
  component_updater::CrxUpdateItem item;
  bool success =
      component_update_service->GetComponentDetails(component_id_, &item);

  // When `success` returns false, it means the component hasn't
  // been registered yet. `GetComponentDetails` doesn't fill out `item` in this
  // case, and we can just treat the component as if it had a state of `kNew`.
  if (success && IsAlreadyInstalled(item)) {
    // We just need to set the downloaded bytes and total bytes equal to each
    // other to indicate to the `AIModelDownloadProgressManager` that we're
    // installed.
    //
    // We don't set it to `item`'s `downloaded_bytes` and `total_bytes` because
    // they may be less than zero which `AIModelDownloadProgressManager` doesn't
    // allow.
    SetDownloadedBytes(0);
    SetTotalBytes(0);
    return;
  }

  // Watch for progress updates.
  component_updater_observation_.Observe(component_update_service);
}

AICrxComponent::~AICrxComponent() = default;

// component_updater::ServiceObserver:
void AICrxComponent::OnEvent(const component_updater::CrxUpdateItem& item) {
  if (!IsDownloadEvent(item) || item.id != component_id_) {
    return;
  }

  SetDownloadedBytes(item.downloaded_bytes);
  SetTotalBytes(item.total_bytes);
}

}  // namespace on_device_ai