mash: Support ShelfModel access in Chrome.

ash/shelf/shelf_controller.cc

 // Copyright 2016 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 "ash/shelf/shelf_controller.h"
 
-#include "ash/public/interfaces/shelf.mojom.h"
+#include "ash/public/cpp/config.h"
+#include "ash/public/cpp/remote_shelf_item_delegate.h"
 #include "ash/root_window_controller.h"
 #include "ash/session/session_controller.h"
 #include "ash/shelf/app_list_shelf_item_delegate.h"
 #include "ash/shelf/shelf.h"
 #include "ash/shell.h"
 #include "ash/wm_window.h"
+#include "base/auto_reset.h"
 #include "base/strings/utf_string_conversions.h"
 #include "ui/base/models/simple_menu_model.h"
 #include "ui/display/display.h"
 #include "ui/display/screen.h"
 
 namespace ash {
 
 namespace {
 
 // Returns the Shelf instance for the display with the given |display_id|.
 Shelf* GetShelfForDisplay(int64_t display_id) {
   // The controller may be null for invalid ids or for displays being removed.
   RootWindowController* root_window_controller =
       Shell::GetRootWindowControllerWithDisplayId(display_id);
   return root_window_controller ? root_window_controller->GetShelf() : nullptr;
 }
 
 }  // namespace
 
-ShelfController::ShelfController() {
+ShelfController::ShelfController() : shelf_model_observer_binding_(this) {
   // Create the app list item in the shelf.
   AppListShelfItemDelegate::CreateAppListItemAndDelegate(&model_);
+  model_.AddObserver(this);
 }
 
-ShelfController::~ShelfController() {}
+ShelfController::~ShelfController() {
+  model_.RemoveObserver(this);
+}
 
 void ShelfController::BindRequest(mojom::ShelfControllerRequest request) {
   bindings_.AddBinding(this, std::move(request));
 }
 
 void ShelfController::NotifyShelfInitialized(Shelf* shelf) {
   // Notify observers, Chrome will set alignment and auto-hide from prefs.
   int64_t display_id = shelf->GetWindow()->GetDisplayNearestWindow().id();
   observers_.ForAllPtrs([display_id](mojom::ShelfObserver* observer) {
     observer->OnShelfInitialized(display_id);
@@ skipping 17 matching lines @@
   });
 }
 
 void ShelfController::AddObserver(
     mojom::ShelfObserverAssociatedPtrInfo observer) {
   mojom::ShelfObserverAssociatedPtr observer_ptr;
   observer_ptr.Bind(std::move(observer));
   observers_.AddPtr(std::move(observer_ptr));
 }
 
+void ShelfController::LinkShelfModels(mojom::ShelfModelObserverPtr observer,
+                                      LinkShelfModelsCallback callback) {
+  // Notify Ash of all the existing local (Chrome) items and delegates.
+  std::move(callback).Run(
+      shelf_model_observer_binding_.CreateInterfacePtrAndBind(),
+      model_.items());
+  for (const ShelfItem& item : model_.items()) {
+    ShelfItemDelegate* delegate = model_.GetShelfItemDelegate(item.id);
+    mojom::ShelfItemDelegatePtr delegate_ptr;
+    if (delegate)
+      delegate_ptr = delegate->CreateInterfacePtrAndBind();
+    observer->OnShelfItemDelegateChanged(item.id, std::move(delegate_ptr));
+  }
+  model_observers_.AddPtr(std::move(observer));
+}
+
 void ShelfController::SetAlignment(ShelfAlignment alignment,
                                    int64_t display_id) {
   Shelf* shelf = GetShelfForDisplay(display_id);
   // TODO(jamescook): The session state check should not be necessary, but
   // otherwise this wrongly tries to set the alignment on a secondary display
   // during login before the ShelfLockingManager is created.
   if (shelf && Shell::Get()->session_controller()->IsActiveUserSessionStarted())
     shelf->SetAlignment(alignment);
 }
 
 void ShelfController::SetAutoHideBehavior(ShelfAutoHideBehavior auto_hide,
                                           int64_t display_id) {
   Shelf* shelf = GetShelfForDisplay(display_id);
   // TODO(jamescook): The session state check should not be necessary, but
   // otherwise this wrongly tries to set auto-hide state on a secondary display
   // during login.
   if (shelf && Shell::Get()->session_controller()->IsActiveUserSessionStarted())
     shelf->SetAutoHideBehavior(auto_hide);
 }
 
-void ShelfController::PinItem(
-    const ShelfItem& item,
-    mojom::ShelfItemDelegateAssociatedPtrInfo delegate) {
-  NOTIMPLEMENTED();
+void ShelfController::OnShelfItemAdded(int32_t index, const ShelfItem& item) {
+  DCHECK_EQ(Shell::GetAshConfig(), Config::MASH) << "Unexpected model sync";
+  DCHECK(!applying_remote_shelf_model_changes_) << "Unexpected model change";
+  DCHECK_GE(model_.item_count(), index) << "Models out of sync";
+  base::AutoReset<bool> reset(&applying_remote_shelf_model_changes_, true);
+  model_.AddAt(index, item);
 }
 
-void ShelfController::UnpinItem(const std::string& app_id) {
-  NOTIMPLEMENTED();
+void ShelfController::OnShelfItemRemoved(int32_t index, const ShelfItem& item) {
+  DCHECK_EQ(Shell::GetAshConfig(), Config::MASH) << "Unexpected model sync";
+  DCHECK(!applying_remote_shelf_model_changes_) << "Unexpected model change";
+  DCHECK_GT(model_.item_count(), index) << "Models out of sync";
+  DCHECK_EQ(model_.items()[index].id, item.id) << "Models out of sync";
+  base::AutoReset<bool> reset(&applying_remote_shelf_model_changes_, true);
+  model_.RemoveItemAt(index);
 }
 
-void ShelfController::SetItemImage(const std::string& app_id,
-                                   const SkBitmap& image) {
-  NOTIMPLEMENTED();
+void ShelfController::OnShelfItemMoved(int32_t start_index,
+                                       int32_t target_index) {
+  DCHECK_EQ(Shell::GetAshConfig(), Config::MASH) << "Unexpected model sync";
+  DCHECK(!applying_remote_shelf_model_changes_) << "Unexpected model change";
+  DCHECK_GT(model_.item_count(), start_index) << "Models out of sync";
+  DCHECK_GT(model_.item_count(), target_index) << "Models out of sync";
+  base::AutoReset<bool> reset(&applying_remote_shelf_model_changes_, true);
+  model_.Move(start_index, target_index);
+}
+
+void ShelfController::OnShelfItemChanged(int32_t index, const ShelfItem& item) {
+  DCHECK_EQ(Shell::GetAshConfig(), Config::MASH) << "Unexpected model sync";
+  DCHECK(!applying_remote_shelf_model_changes_) << "Unexpected model change";
+  DCHECK_GT(model_.item_count(), index) << "Models out of sync";
+  DCHECK_EQ(model_.items()[index].id, item.id) << "Models out of sync";
+  base::AutoReset<bool> reset(&applying_remote_shelf_model_changes_, true);
+  model_.Set(index, item);
+}
+
+void ShelfController::OnShelfItemDelegateChanged(
+    const ShelfID& id,
+    mojom::ShelfItemDelegatePtr delegate) {
+  DCHECK_EQ(Shell::GetAshConfig(), Config::MASH) << "Unexpected model sync";
+  DCHECK(!applying_remote_shelf_model_changes_) << "Unexpected model change";
+  base::AutoReset<bool> reset(&applying_remote_shelf_model_changes_, true);
+  if (delegate.is_bound())
+    model_.SetShelfItemDelegate(
+        id, base::MakeUnique<RemoteShelfItemDelegate>(id, std::move(delegate)));
+  else
+    model_.SetShelfItemDelegate(id, nullptr);
+}
+
+void ShelfController::ShelfItemAdded(int index) {
+  if (applying_remote_shelf_model_changes_)
+    return;
+
+  const ShelfItem& item = model_.items()[index];
+  model_observers_.ForAllPtrs(
+      [index, item](mojom::ShelfModelObserver* observer) {
+        observer->OnShelfItemAdded(index, item);
+      });
+}
+
+void ShelfController::ShelfItemRemoved(int index, const ShelfItem& old_item) {
+  if (applying_remote_shelf_model_changes_)
+    return;
+
+  model_observers_.ForAllPtrs(
+      [index, old_item](mojom::ShelfModelObserver* observer) {
+        observer->OnShelfItemRemoved(index, old_item);
+      });
+}
+
+void ShelfController::ShelfItemMoved(int start_index, int target_index) {
+  if (applying_remote_shelf_model_changes_)
+    return;
+
+  model_observers_.ForAllPtrs(
+      [start_index, target_index](mojom::ShelfModelObserver* observer) {
+        observer->OnShelfItemMoved(start_index, target_index);
+      });
+}
+
+void ShelfController::ShelfItemChanged(int index, const ShelfItem& old_item) {
+  if (applying_remote_shelf_model_changes_)
+    return;
+
+  const ShelfItem& item = model_.items()[index];
+  model_observers_.ForAllPtrs(
+      [index, item](mojom::ShelfModelObserver* observer) {
+        observer->OnShelfItemChanged(index, item);
+      });
+}
+
+void ShelfController::ShelfItemDelegateChanged(const ShelfID& id,
+                                               ShelfItemDelegate* delegate) {
+  if (applying_remote_shelf_model_changes_)
+    return;
+
+  model_observers_.ForAllPtrs(
+      [id, delegate](mojom::ShelfModelObserver* observer) {
+        observer->OnShelfItemDelegateChanged(
+            id, delegate ? delegate->CreateInterfacePtrAndBind()
+                         : mojom::ShelfItemDelegatePtr());
+      });
 }
 
 }  // namespace ash