Relanding: Add js api for hosted/packaged apps to request notification authorization

chrome/renderer/weak_v8_function_map.cc

+// 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.
+
+#include "base/logging.h"
+#include "chrome/renderer/weak_v8_function_map.h"
+
+// Extra data to be passed to MakeWeak/RemoveFromMap to know which entry
+// to remove from which map.
+struct WeakV8FunctionMapData {
+  base::WeakPtr<WeakV8FunctionMap> map;
+  int key;
+};
+
+// Disposes of a callback function and its corresponding entry in the callback
+// map, if that callback map is still alive.
+static void RemoveFromMap(v8::Persistent<v8::Value> context,
+                                  void* data) {
+  WeakV8FunctionMapData* map_data =
+      static_cast<WeakV8FunctionMapData*>(data);
+  if (map_data->map)
+    map_data->map->Remove(map_data->key);
+  delete map_data;
+  context.Dispose();
+}
+
+WeakV8FunctionMap::WeakV8FunctionMap()
+    : ALLOW_THIS_IN_INITIALIZER_LIST(weak_ptr_factory_(this)) {}
+
+WeakV8FunctionMap::~WeakV8FunctionMap() {}
+
+
+void WeakV8FunctionMap::Add(int key,
+                            v8::Local<v8::Function> callback_function) {
+  WeakV8FunctionMapData* map_data = new WeakV8FunctionMapData();
+  map_data->key = key;
+  map_data->map = weak_ptr_factory_.GetWeakPtr();
+  v8::Persistent<v8::Function> wrapper =
+      v8::Persistent<v8::Function>::New(callback_function);
+  map_[key] = wrapper;
+  wrapper.MakeWeak(map_data, RemoveFromMap);
+}
+
+v8::Persistent<v8::Function> WeakV8FunctionMap::Remove(int key) {
+  WeakV8FunctionMap::Map::iterator i = map_.find(key);
+  if (i == map_.end())
+    return v8::Persistent<v8::Function>();
+  v8::Persistent<v8::Function> callback = i->second;
+  map_.erase(i);
+  return callback;
+}