File: bookmark_expanded_state_tracker.cc

package info (click to toggle)
chromium 139.0.7258.127-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 6,122,156 kB
  • sloc: cpp: 35,100,771; ansic: 7,163,530; javascript: 4,103,002; python: 1,436,920; asm: 946,517; xml: 746,709; pascal: 187,653; perl: 88,691; sh: 88,436; objc: 79,953; sql: 51,488; cs: 44,583; fortran: 24,137; makefile: 22,147; tcl: 15,277; php: 13,980; yacc: 8,984; ruby: 7,485; awk: 3,720; lisp: 3,096; lex: 1,327; ada: 727; jsp: 228; sed: 36
file content (122 lines) | stat: -rw-r--r-- 3,437 bytes parent folder | download | duplicates (6)
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
114
115
116
117
118
119
120
121
122
// Copyright 2014 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/bookmarks/bookmark_expanded_state_tracker.h"

#include <stdint.h>

#include <cstddef>
#include <utility>
#include <vector>

#include "base/check.h"
#include "base/check_op.h"
#include "base/strings/string_number_conversions.h"
#include "base/values.h"
#include "components/bookmarks/browser/bookmark_model.h"
#include "components/bookmarks/browser/bookmark_utils.h"
#include "components/bookmarks/common/bookmark_pref_names.h"
#include "components/prefs/pref_service.h"

BookmarkExpandedStateTracker::BookmarkExpandedStateTracker(
    PrefService* pref_service)
    : pref_service_(pref_service) {}

BookmarkExpandedStateTracker::~BookmarkExpandedStateTracker() = default;

void BookmarkExpandedStateTracker::Init(
    bookmarks::BookmarkModel* bookmark_model) {
  DCHECK(!bookmark_model_);
  bookmark_model_ = bookmark_model;
  bookmark_model_->AddObserver(this);
}

void BookmarkExpandedStateTracker::SetExpandedNodes(const Nodes& nodes) {
  UpdatePrefs(nodes);
}

BookmarkExpandedStateTracker::Nodes
BookmarkExpandedStateTracker::GetExpandedNodes() {
  Nodes nodes;
  if (!bookmark_model_ || !bookmark_model_->loaded()) {
    return nodes;
  }

  if (!pref_service_) {
    return nodes;
  }

  const base::Value::List& value =
      pref_service_->GetList(bookmarks::prefs::kBookmarkEditorExpandedNodes);

  bool changed = false;
  for (const auto& entry : value) {
    int64_t node_id;
    const bookmarks::BookmarkNode* node;
    const std::string* value_str = entry.GetIfString();
    if (value_str && base::StringToInt64(*value_str, &node_id) &&
        (node = GetBookmarkNodeByID(bookmark_model_, node_id)) != nullptr &&
        node->is_folder()) {
      nodes.insert(node);
    } else {
      changed = true;
    }
  }
  if (changed) {
    UpdatePrefs(nodes);
  }
  return nodes;
}

void BookmarkExpandedStateTracker::BookmarkModelLoaded(
    bool ids_reassigned) {
  if (ids_reassigned) {
    // If the ids change we can't trust the value in preferences and need to
    // reset it.
    SetExpandedNodes(Nodes());
  }
}

void BookmarkExpandedStateTracker::BookmarkModelChanged() {}

void BookmarkExpandedStateTracker::BookmarkModelBeingDeleted() {
  bookmark_model_->RemoveObserver(this);
  bookmark_model_ = nullptr;
}

void BookmarkExpandedStateTracker::BookmarkNodeRemoved(
    const bookmarks::BookmarkNode* parent,
    size_t old_index,
    const bookmarks::BookmarkNode* node,
    const std::set<GURL>& removed_urls,
    const base::Location& location) {
  if (!node->is_folder()) {
    return;  // Only care about folders.
  }

  // Ask for the nodes again, which removes any nodes that were deleted.
  GetExpandedNodes();
}

void BookmarkExpandedStateTracker::BookmarkAllUserNodesRemoved(
    const std::set<GURL>& removed_urls,
    const base::Location& location) {
  // Ask for the nodes again, which removes any nodes that were deleted.
  GetExpandedNodes();
}

void BookmarkExpandedStateTracker::UpdatePrefs(const Nodes& nodes) {
  if (!pref_service_) {
    return;
  }

  base::Value::List values;
  values.reserve(nodes.size());
  for (const auto* node : nodes) {
    values.Append(base::NumberToString(node->id()));
  }

  pref_service_->SetList(bookmarks::prefs::kBookmarkEditorExpandedNodes,
                         std::move(values));
}