Upstream Chrome on iOS source code [9/11].

ios/chrome/browser/ui/toolbar/toolbar_model_impl_ios.mm

Index: ios/chrome/browser/ui/toolbar/toolbar_model_impl_ios.mm
diff --git a/ios/chrome/browser/ui/toolbar/toolbar_model_impl_ios.mm b/ios/chrome/browser/ui/toolbar/toolbar_model_impl_ios.mm
new file mode 100644
index 0000000000000000000000000000000000000000..af65aef02d7bdb81a92b30995dcce8cabdb2bb30
--- /dev/null
+++ b/ios/chrome/browser/ui/toolbar/toolbar_model_impl_ios.mm
@@ -0,0 +1,84 @@
+// Copyright 2012 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 "ios/chrome/browser/ui/toolbar/toolbar_model_impl_ios.h"
+
+#include "components/bookmarks/browser/bookmark_model.h"
+#include "components/toolbar/toolbar_model_impl.h"
+#include "ios/chrome/browser/bookmarks/bookmark_model_factory.h"
+#include "ios/chrome/browser/browser_state/chrome_browser_state.h"
+#include "ios/chrome/browser/chrome_url_constants.h"
+#import "ios/chrome/browser/tabs/tab.h"
+#include "ios/chrome/browser/ui/toolbar/toolbar_model_delegate_ios.h"
+#import "ios/web/public/web_state/web_state.h"
+
+namespace {
+const size_t kMaxURLDisplayChars = 32 * 1024;
+
+bookmarks::BookmarkModel* GetBookmarkModelForTab(Tab* tab) {
+  web::WebState* web_state = [tab webState];
+  if (!web_state)
+    return nullptr;
+  web::BrowserState* browser_state = web_state->GetBrowserState();
+  if (!browser_state)
+    return nullptr;
+  return ios::BookmarkModelFactory::GetForBrowserState(
+      ios::ChromeBrowserState::FromBrowserState(browser_state));
+}
+}  // namespace
+
+ToolbarModelImplIOS::ToolbarModelImplIOS(ToolbarModelDelegateIOS* delegate) {
+  delegate_ = delegate;
+  toolbar_model_.reset(new ToolbarModelImpl(delegate, kMaxURLDisplayChars));
+}
+
+ToolbarModelImplIOS::~ToolbarModelImplIOS() {}
+
+ToolbarModel* ToolbarModelImplIOS::GetToolbarModel() {
+  return toolbar_model_.get();
+}
+
+bool ToolbarModelImplIOS::IsLoading() {
+  // Please note, ToolbarModel's notion of isLoading is slightly different from
+  // WebState's IsLoading().
+  web::WebState* web_state = delegate_->GetCurrentTab().webState;
+  return web_state && web_state->IsLoading() && !IsCurrentTabNativePage();
+}
+
+CGFloat ToolbarModelImplIOS::GetLoadProgressFraction() {
+  web::WebState* webState = delegate_->GetCurrentTab().webState;
+  return webState ? webState->GetLoadingProgress() : 0.0;
+}
+
+bool ToolbarModelImplIOS::CanGoBack() {
+  return delegate_->GetCurrentTab().canGoBack;
+}
+
+bool ToolbarModelImplIOS::CanGoForward() {
+  return delegate_->GetCurrentTab().canGoForward;
+}
+
+bool ToolbarModelImplIOS::IsCurrentTabNativePage() {
+  Tab* currentTab = delegate_->GetCurrentTab();
+  return currentTab && currentTab.url.SchemeIs(kChromeUIScheme);
+}
+
+bool ToolbarModelImplIOS::IsCurrentTabBookmarked() {
+  Tab* currentTab = delegate_->GetCurrentTab();
+  bookmarks::BookmarkModel* bookmarkModel = GetBookmarkModelForTab(currentTab);
+  return currentTab && bookmarkModel &&
+         bookmarkModel->IsBookmarked(currentTab.url);
+}
+
+bool ToolbarModelImplIOS::IsCurrentTabBookmarkedByUser() {
+  Tab* currentTab = delegate_->GetCurrentTab();
+  bookmarks::BookmarkModel* bookmarkModel = GetBookmarkModelForTab(currentTab);
+  return currentTab && bookmarkModel &&
+         bookmarkModel->GetMostRecentlyAddedUserNodeForURL(currentTab.url);
+}
+
+bool ToolbarModelImplIOS::ShouldDisplayHintText() {
+  Tab* currentTab = delegate_->GetCurrentTab();
+  return [currentTab.webController wantsLocationBarHintText];
+}