Update device name in chooser when device changes name (desktops)

chrome/browser/ui/bluetooth/bluetooth_chooser_controller.cc

 // Copyright 2015 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "chrome/browser/ui/bluetooth/bluetooth_chooser_controller.h"
 
+#include <algorithm>
+
 #include "base/logging.h"
 #include "base/strings/utf_string_conversions.h"
 #include "chrome/browser/net/referrer.h"
 #include "chrome/browser/profiles/profile_manager.h"
 #include "chrome/browser/ui/bluetooth/bluetooth_chooser_desktop.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/browser/ui/scoped_tabbed_browser_displayer.h"
 #include "chrome/common/url_constants.h"
 #include "chrome/grit/generated_resources.h"
 #include "content/browser/bluetooth/bluetooth_metrics.h"
@@ skipping 34 matching lines @@
 
 size_t BluetoothChooserController::NumOptions() const {
   return device_ids_.size();
 }
 
 base::string16 BluetoothChooserController::GetOption(size_t index) const {
   DCHECK_LT(index, device_ids_.size());
   const std::string& device_id = device_ids_[index];
   const auto& device_name_it = device_id_to_name_map_.find(device_id);
   DCHECK(device_name_it != device_id_to_name_map_.end());
-  const auto& it = device_name_map_.find(device_name_it->second);
-  DCHECK(it != device_name_map_.end());
+  const auto& it = device_name_counts_.find(device_name_it->second);
+  DCHECK(it != device_name_counts_.end());
   return it->second == 1
              ? device_name_it->second
              : l10n_util::GetStringFUTF16(
                    IDS_DEVICE_CHOOSER_DEVICE_NAME_WITH_ID,
                    device_name_it->second, base::UTF8ToUTF16(device_id));
 }
 
 void BluetoothChooserController::RefreshOptions() {
   if (event_handler_.is_null())
     return;
@@ skipping 90 matching lines @@
   }
 }
 
 void BluetoothChooserController::AddOrUpdateDevice(
     const std::string& device_id,
     bool should_update_name,
     const base::string16& device_name,
     bool is_gatt_connected,
     bool is_paired,
     const int8_t* rssi) {
-  auto result = device_id_to_name_map_.insert({device_id, device_name});
-  if (!result.second) {
-    // TODO(ortuno): Update device's information.
-    // https://crbug.com/634366 Update name
+  auto name_it = device_id_to_name_map_.find(device_id);
+  if (name_it != device_id_to_name_map_.end()) {
+    if (should_update_name) {
+      base::string16 previous_device_name = name_it->second;
+      name_it->second = device_name;
+
+      const auto& it = device_name_counts_.find(previous_device_name);
+      DCHECK(it != device_name_counts_.end());
+      DCHECK_GT(it->second, 0);
+
+      if (--(it->second) == 0)
+        device_name_counts_.erase(it);
+
+      ++device_name_counts_[device_name];
+    }
+
+    size_t index = std::distance(
+        device_ids_.begin(),
+        std::find(device_ids_.begin(), device_ids_.end(), device_id));
+    DCHECK_NE(device_ids_.size(), index);
     // http://crbug.com/543466 Update connection and paired status
     // http://crbug.com/629689 Update RSSI.
+    if (view())
+      view()->OnOptionUpdated(index);
     return;
   }
 
   device_ids_.push_back(device_id);
-  ++device_name_map_[device_name];
+  device_id_to_name_map_.insert({device_id, device_name});
+  ++device_name_counts_[device_name];
   if (view())
     view()->OnOptionAdded(device_ids_.size() - 1);
 }
 
 void BluetoothChooserController::RemoveDevice(const std::string& device_id) {
   const auto& name_it = device_id_to_name_map_.find(device_id);
   if (name_it == device_id_to_name_map_.end())
     return;
 
   size_t index = 0;
   for (const auto& saved_device_id : device_ids_) {
     if (saved_device_id != device_id) {
       ++index;
       continue;
     }
 
     device_ids_.erase(device_ids_.begin() + index);
 
-    const auto& it = device_name_map_.find(name_it->second);
-    DCHECK(it != device_name_map_.end());
+    const auto& it = device_name_counts_.find(name_it->second);
+    DCHECK(it != device_name_counts_.end());
     DCHECK_GT(it->second, 0);
 
     if (--(it->second) == 0)
-      device_name_map_.erase(it);
+      device_name_counts_.erase(it);
 
     device_id_to_name_map_.erase(name_it);
 
     if (view())
       view()->OnOptionRemoved(index);
     return;
   }
 }
 
 void BluetoothChooserController::ResetEventHandler() {
   event_handler_.Reset();
 }
 
 void BluetoothChooserController::ClearAllDevices() {
   device_ids_.clear();
   device_id_to_name_map_.clear();
-  device_name_map_.clear();
+  device_name_counts_.clear();
 }