// 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 WEBKIT_TOOLS_TEST_SHELL_NOTIFICATION_PRESENTER_H_
#define WEBKIT_TOOLS_TEST_SHELL_NOTIFICATION_PRESENTER_H_

#include <map>
#include <set>
#include <string>

#include "third_party/WebKit/WebKit/chromium/public/WebNotificationPresenter.h"
#include "webkit/tools/test_shell/test_shell.h"

// A class that implements NotificationPresenter for the test shell.
class TestNotificationPresenter : public WebKit::WebNotificationPresenter {
 public:
  explicit TestNotificationPresenter(TestShell* shell)
      : shell_(shell) {
  }

  // Called by the LayoutTestController to simulate a user granting
  // permission.
  void grantPermission(const std::string& origin);

  // WebKit::WebNotificationPresenter interface
  virtual bool show(const WebKit::WebNotification&);
  virtual void cancel(const WebKit::WebNotification&);
  virtual void objectDestroyed(const WebKit::WebNotification&);
  virtual Permission checkPermission(const WebKit::WebURL& url);
  virtual void requestPermission(const WebKit::WebSecurityOrigin& origin,
      WebKit::WebNotificationPermissionCallback* callback);

  void Reset() {
    allowed_origins_.clear();
  }

 private:
  // Non-owned pointer.  The NotificationPresenter is owned by the test shell.
  TestShell* shell_;

  // List of allowed origins.
  std::set<std::string> allowed_origins_;

  // Map of active replacement IDs to the titles of those notifications
  std::map<std::string, std::string> replacements_;
};

#endif  // WEBKIT_TOOLS_TEST_SHELL_NOTIFICATION_PRESENTER_H_