src/chrome/browser/custom_home_pages_table_model.h

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #ifndef CHROME_BROWSER_CUSTOM_HOME_PAGES_TABLE_MODEL_H_
   6 #define CHROME_BROWSER_CUSTOM_HOME_PAGES_TABLE_MODEL_H_
   7
   8 #include <string>
   9 #include <vector>
  10
  11 #include "base/compiler_specific.h"
  12 #include "base/task/cancelable_task_tracker.h"
  13 #include "components/history/core/browser/history_types.h"
  14 #include "ui/base/models/table_model.h"
  15
  16 class GURL;
  17 class Profile;
  18 class HistoryService;
  19
  20 namespace history {
  21 class URLRow;
  22 }
  23
  24 namespace ui {
  25 class TableModelObserver;
  26 }
  27
  28 // CustomHomePagesTableModel is the model for the TableView showing the list
  29 // of pages the user wants opened on startup.
  30
  31 class CustomHomePagesTableModel : public ui::TableModel {
  32  public:
  33   explicit CustomHomePagesTableModel(Profile* profile);
  34   ~CustomHomePagesTableModel() override;
  35
  36   // Sets the set of urls that this model contains.
  37   void SetURLs(const std::vector<GURL>& urls);
  38
  39   // Collect all entries indexed by |index_list|, and moves them to be right
  40   // before the element addressed by |insert_before|. Used by Drag&Drop.
  41   // Expects |index_list| to be ordered ascending.
  42   void MoveURLs(int insert_before, const std::vector<int>& index_list);
  43
  44   // Adds an entry at the specified index.
  45   void Add(int index, const GURL& url);
  46
  47   // Removes the entry at the specified index.
  48   void Remove(int index);
  49
  50   // Clears any entries and fills the list with pages currently opened in the
  51   // browser.
  52   void SetToCurrentlyOpenPages();
  53
  54   // Returns the set of urls this model contains.
  55   std::vector<GURL> GetURLs();
  56
  57   // TableModel overrides:
  58   int RowCount() override;
  59   base::string16 GetText(int row, int column_id) override;
  60   base::string16 GetTooltip(int row) override;
  61   void SetObserver(ui::TableModelObserver* observer) override;
  62
  63  private:
  64   // Each item in the model is represented as an Entry. Entry stores the URL
  65   // and title of the page.
  66   struct Entry;
  67
  68   // Loads the title for the specified entry.
  69   void LoadTitle(Entry* entry);
  70
  71   // Callback from history service. Updates the title of the Entry whose
  72   // |url| matches |entry_url| and notifies the observer of the change.
  73   void OnGotTitle(const GURL& entry_url,
  74                   bool found_url,
  75                   const history::URLRow& row,
  76                   const history::VisitVector& visits);
  77
  78   // Returns the URL for a particular row, formatted for display to the user.
  79   base::string16 FormattedURL(int row) const;
  80
  81   // Set of entries we're showing.
  82   std::vector<Entry> entries_;
  83
  84   // Profile used to load titles.
  85   Profile* profile_;
  86
  87   ui::TableModelObserver* observer_;
  88
  89   // Used in loading titles.
  90   base::CancelableTaskTracker task_tracker_;
  91
  92   DISALLOW_COPY_AND_ASSIGN(CustomHomePagesTableModel);
  93 };
  94
  95 #endif  // CHROME_BROWSER_CUSTOM_HOME_PAGES_TABLE_MODEL_H_