text: // Copyright (c) 2016 GitHub, Inc.
  // Use of this source code is governed by the MIT license that can be
  // found in the LICENSE file.
  
  #ifndef SHELL_COMMON_API_ATOM_API_KEY_WEAK_MAP_H_
  #define SHELL_COMMON_API_ATOM_API_KEY_WEAK_MAP_H_
  
  #include "gin/handle.h"
  #include "shell/common/gin_converters/std_converter.h"
  #include "shell/common/gin_helper/object_template_builder.h"
  #include "shell/common/gin_helper/wrappable.h"
  #include "shell/common/key_weak_map.h"
  
  namespace electron {
  
  namespace api {
  
  template <typename K>
  class KeyWeakMap : public gin_helper::Wrappable<KeyWeakMap<K>> {
   public:
    static gin::Handle<KeyWeakMap<K>> Create(v8::Isolate* isolate) {
      return gin::CreateHandle(isolate, new KeyWeakMap<K>(isolate));
    }
  
    static void BuildPrototype(v8::Isolate* isolate,
                               v8::Local<v8::FunctionTemplate> prototype) {
      prototype->SetClassName(gin::StringToV8(isolate, "KeyWeakMap"));
      gin_helper::ObjectTemplateBuilder(isolate, prototype->PrototypeTemplate())
          .SetMethod("set", &KeyWeakMap<K>::Set)
          .SetMethod("get", &KeyWeakMap<K>::Get)
          .SetMethod("has", &KeyWeakMap<K>::Has)
          .SetMethod("remove", &KeyWeakMap<K>::Remove);
    }
  
   protected:
    explicit KeyWeakMap(v8::Isolate* isolate) {
      gin_helper::Wrappable<KeyWeakMap<K>>::Init(isolate);
    }
    ~KeyWeakMap() override {}
  
   private:
    // API for KeyWeakMap.
    void Set(v8::Isolate* isolate, const K& key, v8::Local<v8::Object> object) {
      key_weak_map_.Set(isolate, key, object);
    }
  
    v8::Local<v8::Object> Get(v8::Isolate* isolate, const K& key) {
      return key_weak_map_.Get(isolate, key).ToLocalChecked();
    }
  
    bool Has(const K& key) { return key_weak_map_.Has(key); }
  
    void Remove(const K& key) { key_weak_map_.Remove(key); }
  
    electron::KeyWeakMap<K> key_weak_map_;
  
    DISALLOW_COPY_AND_ASSIGN(KeyWeakMap);
  };
  
  }  // namespace api
  
  }  // namespace electron
  
  #endif  // SHELL_COMMON_API_ATOM_API_KEY_WEAK_MAP_H_
  