1 files changed, 69 insertions, 0 deletions
diff --git a/chrome/browser/geolocation/geolocation_exceptions_table_model.h b/chrome/browser/geolocation/geolocation_exceptions_table_model.h
new file mode 100644
index 0000000..6328166
--- /dev/null
+++ b/chrome/browser/geolocation/geolocation_exceptions_table_model.h
@@ -0,0 +1,69 @@
+// Copyright (c) 2010 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.
+
+#ifndef CHROME_BROWSER_GEOLOCATION_GEOLOCATION_EXCEPTIONS_TABLE_MODEL_H_
+#define CHROME_BROWSER_GEOLOCATION_GEOLOCATION_EXCEPTIONS_TABLE_MODEL_H_
+
+#include <set>
+#include <vector>
+
+#include "app/table_model.h"
+#include "chrome/browser/geolocation/geolocation_content_settings_map.h"
+#include "chrome/common/content_settings.h"
+#include "chrome/common/content_settings_types.h"
+
+class GeolocationExceptionsTableModel : public TableModel {
+ public:
+  typedef std::set<size_t> Rows;
+
+  explicit GeolocationExceptionsTableModel(
+      GeolocationContentSettingsMap* map);
+
+  // Return whether the given set of rows can be removed.  A parent with setting
+  // of CONTENT_SETTING_DEFAULT can't be removed unless all its children are
+  // also being removed.
+  bool CanRemoveExceptions(const Rows& rows) const;
+
+  // Removes the exceptions at the specified indexes.  If an exception is a
+  // parent, and it has children, the row in model will be updated to have
+  // CONTENT_SETTING_DEFAULT.  If it is the only child of a
+  // CONTENT_SETTING_DEFAULT parent, the parent will be removed from the model
+  // too.
+  void RemoveExceptions(const Rows& rows);
+
+  // Removes all the exceptions from both the map and model.
+  void RemoveAll();
+
+  // TableModel overrides:
+  virtual int RowCount();
+  virtual std::wstring GetText(int row, int column_id);
+  virtual void SetObserver(TableModelObserver* observer);
+  virtual int CompareValues(int row1, int row2, int column_id);
+
+ private:
+  struct Entry {
+    Entry(const GURL& origin,
+          const GURL& embedding_origin,
+          ContentSetting setting);
+
+    GURL origin;
+    GURL embedding_origin;
+    ContentSetting setting;
+  };
+
+  void AddEntriesForOrigin(
+      const GURL& origin,
+      const GeolocationContentSettingsMap::OneOriginSettings& settings);
+
+  GeolocationContentSettingsMap* map_;
+
+  typedef std::vector<Entry> EntriesVector;
+  EntriesVector entries_;
+
+  TableModelObserver* observer_;
+
+  DISALLOW_COPY_AND_ASSIGN(GeolocationExceptionsTableModel);
+};
+
+#endif  // CHROME_BROWSER_GEOLOCATION_GEOLOCATION_EXCEPTIONS_TABLE_MODEL_H_