C++程序  |  83行  |  2.84 KB

// Copyright (c) 2011 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_NOTIFICATIONS_NOTIFICATIONS_PREFS_CACHE_H_
#define CHROME_BROWSER_NOTIFICATIONS_NOTIFICATIONS_PREFS_CACHE_H_
#pragma once

#include <set>
#include <vector>

#include "base/memory/ref_counted.h"
#include "chrome/common/content_settings.h"
#include "googleurl/src/gurl.h"

class ListValue;

// Class which caches notification preferences.
// Construction occurs on the UI thread when the contents
// of the profile preferences are initialized.  Once is_initialized() is set,
// access can only be done from the IO thread.
class NotificationsPrefsCache
    : public base::RefCountedThreadSafe<NotificationsPrefsCache> {
 public:
  NotificationsPrefsCache();

  // Once is_initialized() is set, all accesses must happen on the IO thread.
  // Before that, all accesses need to happen on the UI thread.
  void set_is_initialized(bool val) { is_initialized_ = val; }
  bool is_initialized() { return is_initialized_; }

  // Checks to see if a given origin has permission to create desktop
  // notifications.  Returns a constant from WebNotificationPresenter
  // class.
  int HasPermission(const GURL& origin);

  // Updates the cache with a new origin allowed or denied.
  void CacheAllowedOrigin(const GURL& origin);
  void CacheDeniedOrigin(const GURL& origin);

  // Set the cache to the supplied values.  This clears the current
  // contents of the cache.
  void SetCacheAllowedOrigins(const std::vector<GURL>& allowed);
  void SetCacheDeniedOrigins(const std::vector<GURL>& denied);
  void SetCacheDefaultContentSetting(ContentSetting setting);

  static void ListValueToGurlVector(const ListValue& origin_list,
                                    std::vector<GURL>* origin_vector);

  // Exposed for testing.
  ContentSetting CachedDefaultContentSetting() {
    return default_content_setting_;
  }

 private:
  friend class base::RefCountedThreadSafe<NotificationsPrefsCache>;

  virtual ~NotificationsPrefsCache();

  // Helper functions which read preferences.
  bool IsOriginAllowed(const GURL& origin);
  bool IsOriginDenied(const GURL& origin);

  // Helper that ensures we are running on the expected thread.
  void CheckThreadAccess();

  // Storage of the actual preferences.
  std::set<GURL> allowed_origins_;
  std::set<GURL> denied_origins_;

  // The default setting, used for origins that are neither in
  // |allowed_origins_| nor |denied_origins_|.
  ContentSetting default_content_setting_;

  // Set to true once the initial cached settings have been completely read.
  // Once this is done, the class can no longer be accessed on the UI thread.
  bool is_initialized_;

  DISALLOW_COPY_AND_ASSIGN(NotificationsPrefsCache);
};

#endif  // CHROME_BROWSER_NOTIFICATIONS_NOTIFICATIONS_PREFS_CACHE_H_