// Copyright 2012 Google Inc. All Rights Reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package com.google.collide.clientlibs.invalidation;

import com.google.collide.shared.invalidations.InvalidationObjectId;
import com.google.collide.shared.util.ListenerRegistrar.RemoverManager;


/**
 * Handles registration of Tango objects and unregisters them when the object is cleaned up.
 */
public class ScopedInvalidationRegistrar implements InvalidationRegistrar {
  private final InvalidationRegistrar registrar;
  private final RemoverManager removerManager = new RemoverManager();

  public ScopedInvalidationRegistrar(InvalidationRegistrar registrar) {
    this.registrar = registrar;
  }

  @Override
  public RemovableHandle register(
      InvalidationObjectId<?> tangoObjectId, InvalidationRegistrar.Listener eventListener) {
    RemovableHandle handle = registrar.register(tangoObjectId, eventListener);
    removerManager.track(handle);
    return handle;
  }

  public void cleanup() {
    removerManager.remove();
  }

  @Override
  public void unregister(InvalidationObjectId<?> objectId) {
    registrar.unregister(objectId);
  }
}
